esp-hal 1.1.0

use strum::FromRepr;

use crate::{
    peripherals::{APB_CTRL, EXTMEM, LPWR, SPI0, SPI1, SYSTEM},
    soc::regi2c,
};

pub(crate) fn init() {
    let rtc_cntl = LPWR::regs();

    regi2c::I2C_DIG_REG_XPD_DIG_REG.write_field(0);
    regi2c::I2C_DIG_REG_XPD_RTC_REG.write_field(0);

    unsafe {
        rtc_cntl.timer1().modify(|_, w| {
            w.pll_buf_wait().bits(20u8);
            w.ck8m_wait().bits(20u8)
        });

        rtc_cntl.timer5().modify(|_, w| w.min_slp_val().bits(2u8));
    }

    calibrate_ocode();

    set_rtc_dig_dbias();

    clock_control_init();

    power_control_init();

    unsafe {
        rtc_cntl.int_ena().write(|w| w.bits(0));
        rtc_cntl.int_clr().write(|w| w.bits(u32::MAX));
    }

    regi2c::I2C_ULP_IR_FORCE_XPD_CK.write_field(0);
}

fn calibrate_ocode() {}

fn set_rtc_dig_dbias() {}

/// Perform clock control related initialization
fn clock_control_init() {
    let extmem = EXTMEM::regs();
    let spi_mem_0 = SPI0::regs();
    let spi_mem_1 = SPI1::regs();

    // Clear CMMU clock force on
    extmem
        .cache_mmu_power_ctrl()
        .modify(|_, w| w.cache_mmu_mem_force_on().clear_bit());

    // Clear tag clock force on
    extmem
        .icache_tag_power_ctrl()
        .modify(|_, w| w.icache_tag_mem_force_on().clear_bit());

    // Clear register clock force on
    spi_mem_0.clock_gate().modify(|_, w| w.clk_en().clear_bit());
    spi_mem_1.clock_gate().modify(|_, w| w.clk_en().clear_bit());
}

/// Perform power control related initialization
fn power_control_init() {
    let rtc_cntl = LPWR::regs();
    let system = SYSTEM::regs();

    rtc_cntl
        .clk_conf()
        .modify(|_, w| w.ck8m_force_pu().clear_bit());

    // Cancel XTAL force PU if no need to force power up
    // Cannot cancel XTAL force PU if PLL is force power on
    rtc_cntl
        .options0()
        .modify(|_, w| w.xtl_force_pu().clear_bit());

    // Force PD APLL
    rtc_cntl.ana_conf().modify(|_, w| {
        w.plla_force_pu().clear_bit();
        w.plla_force_pd().set_bit();
        // Open SAR_I2C protect function to avoid SAR_I2C
        // Reset when rtc_ldo is low.
        w.i2c_reset_por_force_pd().clear_bit()
    });

    // Cancel BBPLL force PU if setting no force power up
    rtc_cntl.options0().modify(|_, w| {
        w.bbpll_force_pu().clear_bit();
        w.bbpll_i2c_force_pu().clear_bit();
        w.bb_i2c_force_pu().clear_bit()
    });

    rtc_cntl
        .rtc_cntl()
        .modify(|_, w| w.regulator_force_pu().clear_bit());

    // If this mask is enabled, all soc memories cannot enter power down mode.
    // We should control soc memory power down mode from RTC,
    // so we will not touch this register any more.
    system
        .mem_pd_mask()
        .modify(|_, w| w.lslp_mem_pd_mask().clear_bit());

    rtc_sleep_pu();

    rtc_cntl
        .dig_pwc()
        .modify(|_, w| w.dg_wrap_force_pu().clear_bit());

    rtc_cntl
        .dig_iso()
        .modify(|_, w| w.dg_wrap_force_noiso().clear_bit());

    // Cancel digital PADS force no iso
    system
        .cpu_per_conf()
        .modify(|_, w| w.cpu_wait_mode_force_on().clear_bit());

    // If SYSTEM_CPU_WAIT_MODE_FORCE_ON == 0,
    // the CPU clock will be closed when CPU enter WAITI mode.
    rtc_cntl.dig_iso().modify(|_, w| {
        w.dg_pad_force_unhold().clear_bit();
        w.dg_pad_force_noiso().clear_bit()
    });
}

/// Configure whether certain peripherals are powered down in deep sleep
fn rtc_sleep_pu() {
    LPWR::regs()
        .dig_pwc()
        .modify(|_, w| w.lslp_mem_force_pu().clear_bit());

    APB_CTRL::regs().front_end_mem_pd().modify(|_, w| {
        w.dc_mem_force_pu().clear_bit();
        w.pbus_mem_force_pu().clear_bit();
        w.agc_mem_force_pu().clear_bit()
    });
    APB_CTRL::regs().mem_power_up().modify(|_, w| unsafe {
        w.sram_power_up().bits(0u8);
        w.rom_power_up().bits(0u8)
    });
}

// Terminology:
//
// CPU Reset:    Reset CPU core only, once reset done, CPU will execute from
//               reset vector
// Core Reset:   Reset the whole digital system except RTC sub-system
// System Reset: Reset the whole digital system, including RTC sub-system
// Chip Reset:   Reset the whole chip, including the analog part

#[derive(Debug, Clone, Copy, PartialEq, Eq, FromRepr)]
/// SOC Reset Reason.
pub enum SocResetReason {
    /// Power on reset
    ChipPowerOn   = 0x01,
    /// Software resets the digital core by RTC_CNTL_SW_SYS_RST
    CoreSw        = 0x03,
    /// Deep sleep reset the digital core
    CoreDeepSleep = 0x05,
    /// Main watch dog 0 resets digital core
    CoreMwdt0     = 0x07,
    /// RTC watch dog resets digital core
    CoreRtcWdt    = 0x09,
    /// Main watch dog 0 resets CPU 0
    Cpu0Mwdt0     = 0x0B,
    /// Software resets CPU 0 by RTC_CNTL_SW_PROCPU_RST
    Cpu0Sw        = 0x0C,
    /// RTC watch dog resets CPU 0
    Cpu0RtcWdt    = 0x0D,
    /// VDD voltage is not stable and resets the digital core
    SysBrownOut   = 0x0F,
    /// RTC watch dog resets digital core and rtc module
    SysRtcWdt     = 0x10,
    /// Super watch dog resets the digital core and rtc module
    SysSuperWdt   = 0x12,
    /// Glitch on clock resets the digital core and rtc module
    SysClkGlitch  = 0x13,
    /// eFuse CRC error resets the digital core
    CoreEfuseCrc  = 0x14,
    /// JTAG resets the CPU 0
    Cpu0Jtag      = 0x18,
}