ports/esp32/modesp32.c - lite/micropython - Linaro Git Browser

 /*
  * This file is part of the MicroPython project, http://micropython.org/
  *
  * The MIT License (MIT)
  *
  * Copyright (c) 2017 "Eric Poulsen" <eric@zyxod.com>
  *
  * Permission is hereby granted, free of charge, to any person obtaining a copy
  * of this software and associated documentation files (the "Software"), to deal
  * in the Software without restriction, including without limitation the rights
  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  * copies of the Software, and to permit persons to whom the Software is
  * furnished to do so, subject to the following conditions:
  *
  * The above copyright notice and this permission notice shall be included in
  * all copies or substantial portions of the Software.
  *
  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  * THE SOFTWARE.
  */

 #include <stdio.h>
 #include <string.h>

 #include <time.h>
 #include <sys/time.h>
 #include "driver/gpio.h"
 #include "esp_heap_caps.h"
 #include "multi_heap.h"

 #include "py/nlr.h"
 #include "py/obj.h"
 #include "py/runtime.h"
 #include "py/mphal.h"
 #include "shared/timeutils/timeutils.h"
 #include "modmachine.h"
 #include "machine_rtc.h"
 #include "modesp32.h"

 // These private includes are needed for idf_heap_info.
 #define MULTI_HEAP_FREERTOS
 #include "../multi_heap_platform.h"
 #include "../heap_private.h"

 #if SOC_TOUCH_SENSOR_SUPPORTED
 static mp_obj_t esp32_wake_on_touch(const mp_obj_t wake) {

     #if SOC_PM_SUPPORT_EXT0_WAKEUP
     if (machine_rtc_config.ext0_pin != -1) {
         mp_raise_ValueError(MP_ERROR_TEXT("no resources"));
     }
     #endif

     machine_rtc_config.wake_on_touch = mp_obj_is_true(wake);
     return mp_const_none;
 }
 static MP_DEFINE_CONST_FUN_OBJ_1(esp32_wake_on_touch_obj, esp32_wake_on_touch);
 #endif

 #if SOC_PM_SUPPORT_EXT0_WAKEUP
 static mp_obj_t esp32_wake_on_ext0(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args) {

     #if SOC_TOUCH_SENSOR_SUPPORTED
     if (machine_rtc_config.wake_on_touch) {
         mp_raise_ValueError(MP_ERROR_TEXT("no resources"));
     }
     #endif

     enum {ARG_pin, ARG_level};
     const mp_arg_t allowed_args[] = {
         { MP_QSTR_pin,  MP_ARG_OBJ, {.u_obj = mp_obj_new_int(machine_rtc_config.ext0_pin)} },
         { MP_QSTR_level,  MP_ARG_BOOL, {.u_bool = machine_rtc_config.ext0_level} },
     };
     mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
     mp_arg_parse_all(n_args, pos_args, kw_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);

     if (args[ARG_pin].u_obj == mp_const_none) {
         machine_rtc_config.ext0_pin = -1; // "None"
     } else {
         gpio_num_t pin_id = machine_pin_get_id(args[ARG_pin].u_obj);
         if (pin_id != machine_rtc_config.ext0_pin) {
             if (!RTC_IS_VALID_EXT_PIN(pin_id)) {
                 mp_raise_ValueError(MP_ERROR_TEXT("invalid pin"));
             }
             machine_rtc_config.ext0_pin = pin_id;
         }
     }

     machine_rtc_config.ext0_level = args[ARG_level].u_bool;
     machine_rtc_config.ext0_wake_types = MACHINE_WAKE_SLEEP | MACHINE_WAKE_DEEPSLEEP;

     return mp_const_none;
 }
 static MP_DEFINE_CONST_FUN_OBJ_KW(esp32_wake_on_ext0_obj, 0, esp32_wake_on_ext0);
 #endif

 #if SOC_PM_SUPPORT_EXT1_WAKEUP
 static mp_obj_t esp32_wake_on_ext1(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args) {
     enum {ARG_pins, ARG_level};
     const mp_arg_t allowed_args[] = {
         { MP_QSTR_pins, MP_ARG_OBJ, {.u_obj = mp_const_none} },
         { MP_QSTR_level, MP_ARG_BOOL, {.u_bool = machine_rtc_config.ext1_level} },
     };
     mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
     mp_arg_parse_all(n_args, pos_args, kw_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);
     uint64_t ext1_pins = machine_rtc_config.ext1_pins;


     // Check that all pins are allowed
     if (args[ARG_pins].u_obj != mp_const_none) {
         size_t len = 0;
         mp_obj_t *elem;
         mp_obj_get_array(args[ARG_pins].u_obj, &len, &elem);
         ext1_pins = 0;

         for (int i = 0; i < len; i++) {

             gpio_num_t pin_id = machine_pin_get_id(elem[i]);
             if (!RTC_IS_VALID_EXT_PIN(pin_id)) {
                 mp_raise_ValueError(MP_ERROR_TEXT("invalid pin"));
                 break;
             }
             ext1_pins |= (1ll << pin_id);
         }
     }

     machine_rtc_config.ext1_level = args[ARG_level].u_bool;
     machine_rtc_config.ext1_pins = ext1_pins;

     return mp_const_none;
 }
 static MP_DEFINE_CONST_FUN_OBJ_KW(esp32_wake_on_ext1_obj, 0, esp32_wake_on_ext1);
 #endif

 #if SOC_ULP_SUPPORTED
 static mp_obj_t esp32_wake_on_ulp(const mp_obj_t wake) {
     #if SOC_PM_SUPPORT_EXT0_WAKEUP
     if (machine_rtc_config.ext0_pin != -1) {
         mp_raise_ValueError(MP_ERROR_TEXT("no resources"));
     }
     #endif
     machine_rtc_config.wake_on_ulp = mp_obj_is_true(wake);
     return mp_const_none;
 }
 static MP_DEFINE_CONST_FUN_OBJ_1(esp32_wake_on_ulp_obj, esp32_wake_on_ulp);
 #endif

 #if !SOC_GPIO_SUPPORT_HOLD_SINGLE_IO_IN_DSLP
 static mp_obj_t esp32_gpio_deep_sleep_hold(const mp_obj_t enable) {
     if (mp_obj_is_true(enable)) {
         gpio_deep_sleep_hold_en();
     } else {
         gpio_deep_sleep_hold_dis();
     }
     return mp_const_none;
 }
 static MP_DEFINE_CONST_FUN_OBJ_1(esp32_gpio_deep_sleep_hold_obj, esp32_gpio_deep_sleep_hold);
 #endif

 #if CONFIG_IDF_TARGET_ESP32

 #include "soc/sens_reg.h"

 static mp_obj_t esp32_raw_temperature(void) {
     SET_PERI_REG_BITS(SENS_SAR_MEAS_WAIT2_REG, SENS_FORCE_XPD_SAR, 3, SENS_FORCE_XPD_SAR_S);
     SET_PERI_REG_BITS(SENS_SAR_TSENS_CTRL_REG, SENS_TSENS_CLK_DIV, 10, SENS_TSENS_CLK_DIV_S);
     CLEAR_PERI_REG_MASK(SENS_SAR_TSENS_CTRL_REG, SENS_TSENS_POWER_UP);
     CLEAR_PERI_REG_MASK(SENS_SAR_TSENS_CTRL_REG, SENS_TSENS_DUMP_OUT);
     SET_PERI_REG_MASK(SENS_SAR_TSENS_CTRL_REG, SENS_TSENS_POWER_UP_FORCE);
     SET_PERI_REG_MASK(SENS_SAR_TSENS_CTRL_REG, SENS_TSENS_POWER_UP);
     esp_rom_delay_us(100);
     SET_PERI_REG_MASK(SENS_SAR_TSENS_CTRL_REG, SENS_TSENS_DUMP_OUT);
     esp_rom_delay_us(5);
     int res = GET_PERI_REG_BITS2(SENS_SAR_SLAVE_ADDR3_REG, SENS_TSENS_OUT, SENS_TSENS_OUT_S);

     return mp_obj_new_int(res);
 }
 static MP_DEFINE_CONST_FUN_OBJ_0(esp32_raw_temperature_obj, esp32_raw_temperature);

 #else

 // IDF 5 exposes new internal temperature interface, and the ESP32C3/S2/S3
 // now have calibrated temperature settings in 5 discrete ranges.
 #include "driver/temperature_sensor.h"

 static mp_obj_t esp32_mcu_temperature(void) {
     static temperature_sensor_handle_t temp_sensor = NULL;
     float tvalue;
     if (temp_sensor == NULL) {
         temperature_sensor_config_t temp_sensor_config = TEMPERATURE_SENSOR_CONFIG_DEFAULT(-10, 80);
         ESP_ERROR_CHECK(temperature_sensor_install(&temp_sensor_config, &temp_sensor));
     }
     ESP_ERROR_CHECK(temperature_sensor_enable(temp_sensor));
     ESP_ERROR_CHECK(temperature_sensor_get_celsius(temp_sensor, &tvalue));
     ESP_ERROR_CHECK(temperature_sensor_disable(temp_sensor));
     return mp_obj_new_int((int)(tvalue + 0.5));
 }
 static MP_DEFINE_CONST_FUN_OBJ_0(esp32_mcu_temperature_obj, esp32_mcu_temperature);

 #endif

 static mp_obj_t esp32_idf_heap_info(const mp_obj_t cap_in) {
     mp_int_t cap = mp_obj_get_int(cap_in);
     multi_heap_info_t info;
     heap_t *heap;
     mp_obj_t heap_list = mp_obj_new_list(0, 0);
     SLIST_FOREACH(heap, &registered_heaps, next) {
         if (heap_caps_match(heap, cap)) {
             multi_heap_get_info(heap->heap, &info);
             mp_obj_t data[] = {
                 MP_OBJ_NEW_SMALL_INT(heap->end - heap->start), // total heap size
                 MP_OBJ_NEW_SMALL_INT(info.total_free_bytes),   // total free bytes
                 MP_OBJ_NEW_SMALL_INT(info.largest_free_block), // largest free contiguous
                 MP_OBJ_NEW_SMALL_INT(info.minimum_free_bytes), // minimum free seen
             };
             mp_obj_t this_heap = mp_obj_new_tuple(4, data);
             mp_obj_list_append(heap_list, this_heap);
         }
     }
     return heap_list;
 }
 static MP_DEFINE_CONST_FUN_OBJ_1(esp32_idf_heap_info_obj, esp32_idf_heap_info);

 #if CONFIG_FREERTOS_USE_TRACE_FACILITY
 static mp_obj_t esp32_idf_task_info(void) {
     const size_t task_count_max = uxTaskGetNumberOfTasks();
     TaskStatus_t *task_array = m_new(TaskStatus_t, task_count_max);
     uint32_t total_time;
     const size_t task_count = uxTaskGetSystemState(task_array, task_count_max, &total_time);

     mp_obj_list_t *task_list = MP_OBJ_TO_PTR(mp_obj_new_list(task_count, NULL));
     for (size_t i = 0; i < task_count; i++) {
         mp_obj_t task_data[] = {
             mp_obj_new_int_from_uint((mp_uint_t)task_array[i].xHandle),
             mp_obj_new_str(task_array[i].pcTaskName, strlen(task_array[i].pcTaskName)),
             MP_OBJ_NEW_SMALL_INT(task_array[i].eCurrentState),
             MP_OBJ_NEW_SMALL_INT(task_array[i].uxCurrentPriority),
             #if CONFIG_FREERTOS_GENERATE_RUN_TIME_STATS
             mp_obj_new_int_from_uint(task_array[i].ulRunTimeCounter),
             #else
             mp_const_none,
             #endif
             mp_obj_new_int_from_uint(task_array[i].usStackHighWaterMark),
             #if CONFIG_FREERTOS_VTASKLIST_INCLUDE_COREID
             MP_OBJ_NEW_SMALL_INT(task_array[i].xCoreID),
             #else
             mp_const_none,
             #endif
         };
         task_list->items[i] = mp_obj_new_tuple(7, task_data);
     }

     m_del(TaskStatus_t, task_array, task_count_max);
     mp_obj_t task_stats[] = {
         #if CONFIG_FREERTOS_GENERATE_RUN_TIME_STATS
         MP_OBJ_NEW_SMALL_INT(total_time),
         #else
         mp_const_none,
         #endif
         task_list
     };
     return mp_obj_new_tuple(2, task_stats);
 }
 static MP_DEFINE_CONST_FUN_OBJ_0(esp32_idf_task_info_obj, esp32_idf_task_info);
 #endif

 static const mp_rom_map_elem_t esp32_module_globals_table[] = {
     { MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_esp32) },

     #if SOC_TOUCH_SENSOR_SUPPORTED
     { MP_ROM_QSTR(MP_QSTR_wake_on_touch), MP_ROM_PTR(&esp32_wake_on_touch_obj) },
     #endif
     #if SOC_PM_SUPPORT_EXT0_WAKEUP
     { MP_ROM_QSTR(MP_QSTR_wake_on_ext0), MP_ROM_PTR(&esp32_wake_on_ext0_obj) },
     #endif
     #if SOC_PM_SUPPORT_EXT1_WAKEUP
     { MP_ROM_QSTR(MP_QSTR_wake_on_ext1), MP_ROM_PTR(&esp32_wake_on_ext1_obj) },
     #endif
     #if SOC_ULP_SUPPORTED
     { MP_ROM_QSTR(MP_QSTR_wake_on_ulp), MP_ROM_PTR(&esp32_wake_on_ulp_obj) },
     #endif
     #if !SOC_GPIO_SUPPORT_HOLD_SINGLE_IO_IN_DSLP
     { MP_ROM_QSTR(MP_QSTR_gpio_deep_sleep_hold), MP_ROM_PTR(&esp32_gpio_deep_sleep_hold_obj) },
     #endif
     #if CONFIG_IDF_TARGET_ESP32
     { MP_ROM_QSTR(MP_QSTR_raw_temperature), MP_ROM_PTR(&esp32_raw_temperature_obj) },
     #else
     { MP_ROM_QSTR(MP_QSTR_mcu_temperature), MP_ROM_PTR(&esp32_mcu_temperature_obj) },
     #endif
     { MP_ROM_QSTR(MP_QSTR_idf_heap_info), MP_ROM_PTR(&esp32_idf_heap_info_obj) },
     #if CONFIG_FREERTOS_USE_TRACE_FACILITY
     { MP_ROM_QSTR(MP_QSTR_idf_task_info), MP_ROM_PTR(&esp32_idf_task_info_obj) },
     #endif

     { MP_ROM_QSTR(MP_QSTR_NVS), MP_ROM_PTR(&esp32_nvs_type) },
     { MP_ROM_QSTR(MP_QSTR_Partition), MP_ROM_PTR(&esp32_partition_type) },
     { MP_ROM_QSTR(MP_QSTR_RMT), MP_ROM_PTR(&esp32_rmt_type) },
     #if CONFIG_IDF_TARGET_ESP32 || CONFIG_IDF_TARGET_ESP32S2 || CONFIG_IDF_TARGET_ESP32S3
     { MP_ROM_QSTR(MP_QSTR_ULP), MP_ROM_PTR(&esp32_ulp_type) },
     #endif

     { MP_ROM_QSTR(MP_QSTR_WAKEUP_ALL_LOW), MP_ROM_FALSE },
     { MP_ROM_QSTR(MP_QSTR_WAKEUP_ANY_HIGH), MP_ROM_TRUE },

     { MP_ROM_QSTR(MP_QSTR_HEAP_DATA), MP_ROM_INT(MALLOC_CAP_8BIT) },
     { MP_ROM_QSTR(MP_QSTR_HEAP_EXEC), MP_ROM_INT(MALLOC_CAP_EXEC) },
 };

 static MP_DEFINE_CONST_DICT(esp32_module_globals, esp32_module_globals_table);

 const mp_obj_module_t esp32_module = {
     .base = { &mp_type_module },
     .globals = (mp_obj_dict_t *)&esp32_module_globals,
 };

 MP_REGISTER_MODULE(MP_QSTR_esp32, esp32_module);
	/*
	* This file is part of the MicroPython project, http://micropython.org/
	*
	* The MIT License (MIT)
	*
	* Copyright (c) 2017 "Eric Poulsen" <eric@zyxod.com>
	*
	* Permission is hereby granted, free of charge, to any person obtaining a copy
	* of this software and associated documentation files (the "Software"), to deal
	* in the Software without restriction, including without limitation the rights
	* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
	* copies of the Software, and to permit persons to whom the Software is
	* furnished to do so, subject to the following conditions:
	*
	* The above copyright notice and this permission notice shall be included in
	* all copies or substantial portions of the Software.
	*
	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
	* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
	* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
	* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
	* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
	* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
	* THE SOFTWARE.
	*/

	#include <stdio.h>
	#include <string.h>

	#include <time.h>
	#include <sys/time.h>
	#include "driver/gpio.h"
	#include "esp_heap_caps.h"
	#include "multi_heap.h"

	#include "py/nlr.h"
	#include "py/obj.h"
	#include "py/runtime.h"
	#include "py/mphal.h"
	#include "shared/timeutils/timeutils.h"
	#include "modmachine.h"
	#include "machine_rtc.h"
	#include "modesp32.h"

	// These private includes are needed for idf_heap_info.
	#define MULTI_HEAP_FREERTOS
	#include "../multi_heap_platform.h"
	#include "../heap_private.h"

	#if SOC_TOUCH_SENSOR_SUPPORTED
	static mp_obj_t esp32_wake_on_touch(const mp_obj_t wake) {

	#if SOC_PM_SUPPORT_EXT0_WAKEUP
	if (machine_rtc_config.ext0_pin != -1) {
	mp_raise_ValueError(MP_ERROR_TEXT("no resources"));
	}
	#endif

	machine_rtc_config.wake_on_touch = mp_obj_is_true(wake);
	return mp_const_none;
	}
	static MP_DEFINE_CONST_FUN_OBJ_1(esp32_wake_on_touch_obj, esp32_wake_on_touch);
	#endif

	#if SOC_PM_SUPPORT_EXT0_WAKEUP
	static mp_obj_t esp32_wake_on_ext0(size_t n_args, const mp_obj_t pos_args, mp_map_t kw_args) {

	#if SOC_TOUCH_SENSOR_SUPPORTED
	if (machine_rtc_config.wake_on_touch) {
	mp_raise_ValueError(MP_ERROR_TEXT("no resources"));
	}
	#endif

	enum {ARG_pin, ARG_level};
	const mp_arg_t allowed_args[] = {
	{ MP_QSTR_pin, MP_ARG_OBJ, {.u_obj = mp_obj_new_int(machine_rtc_config.ext0_pin)} },
	{ MP_QSTR_level, MP_ARG_BOOL, {.u_bool = machine_rtc_config.ext0_level} },
	};
	mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
	mp_arg_parse_all(n_args, pos_args, kw_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);

	if (args[ARG_pin].u_obj == mp_const_none) {
	machine_rtc_config.ext0_pin = -1; // "None"
	} else {
	gpio_num_t pin_id = machine_pin_get_id(args[ARG_pin].u_obj);
	if (pin_id != machine_rtc_config.ext0_pin) {
	if (!RTC_IS_VALID_EXT_PIN(pin_id)) {
	mp_raise_ValueError(MP_ERROR_TEXT("invalid pin"));
	}
	machine_rtc_config.ext0_pin = pin_id;
	}
	}

	machine_rtc_config.ext0_level = args[ARG_level].u_bool;
	machine_rtc_config.ext0_wake_types = MACHINE_WAKE_SLEEP \| MACHINE_WAKE_DEEPSLEEP;

	return mp_const_none;
	}
	static MP_DEFINE_CONST_FUN_OBJ_KW(esp32_wake_on_ext0_obj, 0, esp32_wake_on_ext0);
	#endif

	#if SOC_PM_SUPPORT_EXT1_WAKEUP
	static mp_obj_t esp32_wake_on_ext1(size_t n_args, const mp_obj_t pos_args, mp_map_t kw_args) {
	enum {ARG_pins, ARG_level};
	const mp_arg_t allowed_args[] = {
	{ MP_QSTR_pins, MP_ARG_OBJ, {.u_obj = mp_const_none} },
	{ MP_QSTR_level, MP_ARG_BOOL, {.u_bool = machine_rtc_config.ext1_level} },
	};
	mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
	mp_arg_parse_all(n_args, pos_args, kw_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);
	uint64_t ext1_pins = machine_rtc_config.ext1_pins;


	// Check that all pins are allowed
	if (args[ARG_pins].u_obj != mp_const_none) {
	size_t len = 0;
	mp_obj_t *elem;
	mp_obj_get_array(args[ARG_pins].u_obj, &len, &elem);
	ext1_pins = 0;

	for (int i = 0; i < len; i++) {

	gpio_num_t pin_id = machine_pin_get_id(elem[i]);
	if (!RTC_IS_VALID_EXT_PIN(pin_id)) {
	mp_raise_ValueError(MP_ERROR_TEXT("invalid pin"));
	break;
	}
	ext1_pins \|= (1ll << pin_id);
	}
	}

	machine_rtc_config.ext1_level = args[ARG_level].u_bool;
	machine_rtc_config.ext1_pins = ext1_pins;

	return mp_const_none;
	}
	static MP_DEFINE_CONST_FUN_OBJ_KW(esp32_wake_on_ext1_obj, 0, esp32_wake_on_ext1);
	#endif

	#if SOC_ULP_SUPPORTED
	static mp_obj_t esp32_wake_on_ulp(const mp_obj_t wake) {
	#if SOC_PM_SUPPORT_EXT0_WAKEUP
	if (machine_rtc_config.ext0_pin != -1) {
	mp_raise_ValueError(MP_ERROR_TEXT("no resources"));
	}
	#endif
	machine_rtc_config.wake_on_ulp = mp_obj_is_true(wake);
	return mp_const_none;
	}
	static MP_DEFINE_CONST_FUN_OBJ_1(esp32_wake_on_ulp_obj, esp32_wake_on_ulp);
	#endif

	#if !SOC_GPIO_SUPPORT_HOLD_SINGLE_IO_IN_DSLP
	static mp_obj_t esp32_gpio_deep_sleep_hold(const mp_obj_t enable) {
	if (mp_obj_is_true(enable)) {
	gpio_deep_sleep_hold_en();
	} else {
	gpio_deep_sleep_hold_dis();
	}
	return mp_const_none;
	}
	static MP_DEFINE_CONST_FUN_OBJ_1(esp32_gpio_deep_sleep_hold_obj, esp32_gpio_deep_sleep_hold);
	#endif

	#if CONFIG_IDF_TARGET_ESP32

	#include "soc/sens_reg.h"

	static mp_obj_t esp32_raw_temperature(void) {
	SET_PERI_REG_BITS(SENS_SAR_MEAS_WAIT2_REG, SENS_FORCE_XPD_SAR, 3, SENS_FORCE_XPD_SAR_S);
	SET_PERI_REG_BITS(SENS_SAR_TSENS_CTRL_REG, SENS_TSENS_CLK_DIV, 10, SENS_TSENS_CLK_DIV_S);
	CLEAR_PERI_REG_MASK(SENS_SAR_TSENS_CTRL_REG, SENS_TSENS_POWER_UP);
	CLEAR_PERI_REG_MASK(SENS_SAR_TSENS_CTRL_REG, SENS_TSENS_DUMP_OUT);
	SET_PERI_REG_MASK(SENS_SAR_TSENS_CTRL_REG, SENS_TSENS_POWER_UP_FORCE);
	SET_PERI_REG_MASK(SENS_SAR_TSENS_CTRL_REG, SENS_TSENS_POWER_UP);
	esp_rom_delay_us(100);
	SET_PERI_REG_MASK(SENS_SAR_TSENS_CTRL_REG, SENS_TSENS_DUMP_OUT);
	esp_rom_delay_us(5);
	int res = GET_PERI_REG_BITS2(SENS_SAR_SLAVE_ADDR3_REG, SENS_TSENS_OUT, SENS_TSENS_OUT_S);

	return mp_obj_new_int(res);
	}
	static MP_DEFINE_CONST_FUN_OBJ_0(esp32_raw_temperature_obj, esp32_raw_temperature);

	#else

	// IDF 5 exposes new internal temperature interface, and the ESP32C3/S2/S3
	// now have calibrated temperature settings in 5 discrete ranges.
	#include "driver/temperature_sensor.h"

	static mp_obj_t esp32_mcu_temperature(void) {
	static temperature_sensor_handle_t temp_sensor = NULL;
	float tvalue;
	if (temp_sensor == NULL) {
	temperature_sensor_config_t temp_sensor_config = TEMPERATURE_SENSOR_CONFIG_DEFAULT(-10, 80);
	ESP_ERROR_CHECK(temperature_sensor_install(&temp_sensor_config, &temp_sensor));
	}
	ESP_ERROR_CHECK(temperature_sensor_enable(temp_sensor));
	ESP_ERROR_CHECK(temperature_sensor_get_celsius(temp_sensor, &tvalue));
	ESP_ERROR_CHECK(temperature_sensor_disable(temp_sensor));
	return mp_obj_new_int((int)(tvalue + 0.5));
	}
	static MP_DEFINE_CONST_FUN_OBJ_0(esp32_mcu_temperature_obj, esp32_mcu_temperature);

	#endif

	static mp_obj_t esp32_idf_heap_info(const mp_obj_t cap_in) {
	mp_int_t cap = mp_obj_get_int(cap_in);
	multi_heap_info_t info;
	heap_t *heap;
	mp_obj_t heap_list = mp_obj_new_list(0, 0);
	SLIST_FOREACH(heap, &registered_heaps, next) {
	if (heap_caps_match(heap, cap)) {
	multi_heap_get_info(heap->heap, &info);
	mp_obj_t data[] = {
	MP_OBJ_NEW_SMALL_INT(heap->end - heap->start), // total heap size
	MP_OBJ_NEW_SMALL_INT(info.total_free_bytes), // total free bytes
	MP_OBJ_NEW_SMALL_INT(info.largest_free_block), // largest free contiguous
	MP_OBJ_NEW_SMALL_INT(info.minimum_free_bytes), // minimum free seen
	};
	mp_obj_t this_heap = mp_obj_new_tuple(4, data);
	mp_obj_list_append(heap_list, this_heap);
	}
	}
	return heap_list;
	}
	static MP_DEFINE_CONST_FUN_OBJ_1(esp32_idf_heap_info_obj, esp32_idf_heap_info);

	#if CONFIG_FREERTOS_USE_TRACE_FACILITY
	static mp_obj_t esp32_idf_task_info(void) {
	const size_t task_count_max = uxTaskGetNumberOfTasks();
	TaskStatus_t *task_array = m_new(TaskStatus_t, task_count_max);
	uint32_t total_time;
	const size_t task_count = uxTaskGetSystemState(task_array, task_count_max, &total_time);

	mp_obj_list_t *task_list = MP_OBJ_TO_PTR(mp_obj_new_list(task_count, NULL));
	for (size_t i = 0; i < task_count; i++) {
	mp_obj_t task_data[] = {
	mp_obj_new_int_from_uint((mp_uint_t)task_array[i].xHandle),
	mp_obj_new_str(task_array[i].pcTaskName, strlen(task_array[i].pcTaskName)),
	MP_OBJ_NEW_SMALL_INT(task_array[i].eCurrentState),
	MP_OBJ_NEW_SMALL_INT(task_array[i].uxCurrentPriority),
	#if CONFIG_FREERTOS_GENERATE_RUN_TIME_STATS
	mp_obj_new_int_from_uint(task_array[i].ulRunTimeCounter),
	#else
	mp_const_none,
	#endif
	mp_obj_new_int_from_uint(task_array[i].usStackHighWaterMark),
	#if CONFIG_FREERTOS_VTASKLIST_INCLUDE_COREID
	MP_OBJ_NEW_SMALL_INT(task_array[i].xCoreID),
	#else
	mp_const_none,
	#endif
	};
	task_list->items[i] = mp_obj_new_tuple(7, task_data);
	}

	m_del(TaskStatus_t, task_array, task_count_max);
	mp_obj_t task_stats[] = {
	#if CONFIG_FREERTOS_GENERATE_RUN_TIME_STATS
	MP_OBJ_NEW_SMALL_INT(total_time),
	#else
	mp_const_none,
	#endif
	task_list
	};
	return mp_obj_new_tuple(2, task_stats);
	}
	static MP_DEFINE_CONST_FUN_OBJ_0(esp32_idf_task_info_obj, esp32_idf_task_info);
	#endif

	static const mp_rom_map_elem_t esp32_module_globals_table[] = {
	{ MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_esp32) },

	#if SOC_TOUCH_SENSOR_SUPPORTED
	{ MP_ROM_QSTR(MP_QSTR_wake_on_touch), MP_ROM_PTR(&esp32_wake_on_touch_obj) },
	#endif
	#if SOC_PM_SUPPORT_EXT0_WAKEUP
	{ MP_ROM_QSTR(MP_QSTR_wake_on_ext0), MP_ROM_PTR(&esp32_wake_on_ext0_obj) },
	#endif
	#if SOC_PM_SUPPORT_EXT1_WAKEUP
	{ MP_ROM_QSTR(MP_QSTR_wake_on_ext1), MP_ROM_PTR(&esp32_wake_on_ext1_obj) },
	#endif
	#if SOC_ULP_SUPPORTED
	{ MP_ROM_QSTR(MP_QSTR_wake_on_ulp), MP_ROM_PTR(&esp32_wake_on_ulp_obj) },
	#endif
	#if !SOC_GPIO_SUPPORT_HOLD_SINGLE_IO_IN_DSLP
	{ MP_ROM_QSTR(MP_QSTR_gpio_deep_sleep_hold), MP_ROM_PTR(&esp32_gpio_deep_sleep_hold_obj) },
	#endif
	#if CONFIG_IDF_TARGET_ESP32
	{ MP_ROM_QSTR(MP_QSTR_raw_temperature), MP_ROM_PTR(&esp32_raw_temperature_obj) },
	#else
	{ MP_ROM_QSTR(MP_QSTR_mcu_temperature), MP_ROM_PTR(&esp32_mcu_temperature_obj) },
	#endif
	{ MP_ROM_QSTR(MP_QSTR_idf_heap_info), MP_ROM_PTR(&esp32_idf_heap_info_obj) },
	#if CONFIG_FREERTOS_USE_TRACE_FACILITY
	{ MP_ROM_QSTR(MP_QSTR_idf_task_info), MP_ROM_PTR(&esp32_idf_task_info_obj) },
	#endif

	{ MP_ROM_QSTR(MP_QSTR_NVS), MP_ROM_PTR(&esp32_nvs_type) },
	{ MP_ROM_QSTR(MP_QSTR_Partition), MP_ROM_PTR(&esp32_partition_type) },
	{ MP_ROM_QSTR(MP_QSTR_RMT), MP_ROM_PTR(&esp32_rmt_type) },
	#if CONFIG_IDF_TARGET_ESP32 \|\| CONFIG_IDF_TARGET_ESP32S2 \|\| CONFIG_IDF_TARGET_ESP32S3
	{ MP_ROM_QSTR(MP_QSTR_ULP), MP_ROM_PTR(&esp32_ulp_type) },
	#endif

	{ MP_ROM_QSTR(MP_QSTR_WAKEUP_ALL_LOW), MP_ROM_FALSE },
	{ MP_ROM_QSTR(MP_QSTR_WAKEUP_ANY_HIGH), MP_ROM_TRUE },

	{ MP_ROM_QSTR(MP_QSTR_HEAP_DATA), MP_ROM_INT(MALLOC_CAP_8BIT) },
	{ MP_ROM_QSTR(MP_QSTR_HEAP_EXEC), MP_ROM_INT(MALLOC_CAP_EXEC) },
	};

	static MP_DEFINE_CONST_DICT(esp32_module_globals, esp32_module_globals_table);

	const mp_obj_module_t esp32_module = {
	.base = { &mp_type_module },
	.globals = (mp_obj_dict_t *)&esp32_module_globals,
	};

	MP_REGISTER_MODULE(MP_QSTR_esp32, esp32_module);