esp32/Makefile Updated to add esp32/machine_rtc.c

MrSurly · MrSurly · commit 57d4906f2e47 · 2017-10-26T09:22:18.000-07:00
esp32/modmachine.c Added RTC object
esp32/modmachine.h Added RTC object

Just enough code for this to work on ESP32:

    import machine
    rtc = machine.RTC() # Need to implement this first!
    rtc.irq(trigger=rtc.ALARM0, wake=machine.DEEPSLEEP) # syntactic sugar to match, probably a NOP
    rtc.alarm(rtc.ALARM0, 10000) # calls esp_deep_sleep_enable_timer_wakeup() in IDF
    machine.deepsleep() # calls esp_deep_sleep_start() in IDF
diff --git a/ports/esp32/Makefile b/ports/esp32/Makefile
@@ -143,6 +143,7 @@ SRC_C = \
 	machine_hw_spi.c \
 	machine_wdt.c \
 	mpthreadport.c \
+	machine_rtc.c \
 	$(SRC_MOD)
 
 EXTMOD_SRC_C = $(addprefix extmod/,\
diff --git a/ports/esp32/modmachine.c b/ports/esp32/modmachine.c
@@ -46,6 +46,10 @@
 
 #if MICROPY_PY_MACHINE
 
+// amount of time to sleep
+extern uint64_t machine_rtc_expiry;
+#define MACHINE_WAKE_DEEPSLEEP (0x04)
+
 STATIC mp_obj_t machine_freq(size_t n_args, const mp_obj_t *args) {
     if (n_args == 0) {
         // get
@@ -64,6 +68,16 @@ STATIC mp_obj_t machine_freq(size_t n_args, const mp_obj_t *args) {
 }
 STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(machine_freq_obj, 0, 1, machine_freq);
 
+STATIC mp_obj_t machine_deepsleep(void) {
+    if (machine_rtc_expiry != 0) {
+        esp_deep_sleep_enable_timer_wakeup(machine_rtc_expiry);
+    }
+    esp_deep_sleep_start();
+    return mp_const_none;
+}
+
+STATIC MP_DEFINE_CONST_FUN_OBJ_0(machine_deepsleep_obj, machine_deepsleep);
+
 STATIC mp_obj_t machine_reset(void) {
     esp_restart();
     return mp_const_none;
@@ -106,6 +120,7 @@ STATIC const mp_rom_map_elem_t machine_module_globals_table[] = {
     { MP_ROM_QSTR(MP_QSTR_freq), MP_ROM_PTR(&machine_freq_obj) },
     { MP_ROM_QSTR(MP_QSTR_reset), MP_ROM_PTR(&machine_reset_obj) },
     { MP_ROM_QSTR(MP_QSTR_unique_id), MP_ROM_PTR(&machine_unique_id_obj) },
+    { MP_ROM_QSTR(MP_QSTR_deepsleep), MP_ROM_PTR(&machine_deepsleep_obj) },
     { MP_ROM_QSTR(MP_QSTR_idle), MP_ROM_PTR(&machine_idle_obj) },
 
     { MP_ROM_QSTR(MP_QSTR_disable_irq), MP_ROM_PTR(&machine_disable_irq_obj) },
@@ -115,13 +130,17 @@ STATIC const mp_rom_map_elem_t machine_module_globals_table[] = {
 
     { MP_ROM_QSTR(MP_QSTR_Timer), MP_ROM_PTR(&machine_timer_type) },
     { MP_ROM_QSTR(MP_QSTR_WDT), MP_ROM_PTR(&machine_wdt_type) },
+    // wake abilities
+    { MP_ROM_QSTR(MP_QSTR_DEEPSLEEP), MP_ROM_INT(MACHINE_WAKE_DEEPSLEEP) },
+
     { MP_ROM_QSTR(MP_QSTR_Pin), MP_ROM_PTR(&machine_pin_type) },
     { MP_ROM_QSTR(MP_QSTR_Signal), MP_ROM_PTR(&machine_signal_type) },
     { MP_ROM_QSTR(MP_QSTR_TouchPad), MP_ROM_PTR(&machine_touchpad_type) },
     { MP_ROM_QSTR(MP_QSTR_ADC), MP_ROM_PTR(&machine_adc_type) },
     { MP_ROM_QSTR(MP_QSTR_DAC), MP_ROM_PTR(&machine_dac_type) },
     { MP_ROM_QSTR(MP_QSTR_I2C), MP_ROM_PTR(&machine_i2c_type) },
     { MP_ROM_QSTR(MP_QSTR_PWM), MP_ROM_PTR(&machine_pwm_type) },
+    { MP_ROM_QSTR(MP_QSTR_RTC), MP_ROM_PTR(&machine_rtc_type) },
     { MP_ROM_QSTR(MP_QSTR_SPI), MP_ROM_PTR(&mp_machine_soft_spi_type) },
     { MP_ROM_QSTR(MP_QSTR_UART), MP_ROM_PTR(&machine_uart_type) },
 };
diff --git a/ports/esp32/modmachine.h b/ports/esp32/modmachine.h
@@ -15,5 +15,6 @@ extern const mp_obj_type_t machine_uart_type;
 
 void machine_pins_init(void);
 void machine_pins_deinit(void);
+extern const mp_obj_type_t machine_rtc_type;
 
 #endif // MICROPY_INCLUDED_ESP32_MODMACHINE_H
