adafruit · tannewt · Apr 22, 2025 · Apr 14, 2025 · Apr 20, 2025 · Apr 20, 2025
diff --git a/ports/espressif/common-hal/wifi/Radio.c b/ports/espressif/common-hal/wifi/Radio.c
@@ -139,23 +139,44 @@ void common_hal_wifi_radio_set_tx_power(wifi_radio_obj_t *self, const mp_float_t
     esp_wifi_set_max_tx_power(tx_power * 4.0f);
 }
 
-mp_int_t common_hal_wifi_radio_get_listen_interval(wifi_radio_obj_t *self) {
-    wifi_config_t *config = &self->sta_config;
-    return config->sta.listen_interval;
+wifi_power_management_t common_hal_wifi_radio_get_power_management(wifi_radio_obj_t *self) {
+    wifi_ps_type_t ps;
+    esp_err_t ret = esp_wifi_get_ps(&ps);
+    if (ret == ESP_OK) {
+        switch (ps) {
+            case WIFI_PS_MIN_MODEM:
+                return POWER_MANAGEMENT_MIN;
+            case WIFI_PS_MAX_MODEM:
+                return POWER_MANAGEMENT_MAX;
+            case WIFI_PS_NONE:
+                return POWER_MANAGEMENT_NONE;
+        }
+    }
+    return POWER_MANAGEMENT_MIN;
 }
 
-void common_hal_wifi_radio_set_listen_interval(wifi_radio_obj_t *self, const mp_int_t listen_interval) {
-    wifi_config_t *config = &self->sta_config;
-    config->sta.listen_interval = listen_interval;
-    if (listen_interval == 1) {
-        esp_wifi_set_ps(WIFI_PS_MIN_MODEM);
-    } else if (listen_interval > 1) {
-        esp_wifi_set_ps(WIFI_PS_MAX_MODEM);
-    } else {
-        esp_wifi_set_ps(WIFI_PS_NONE);
-    }
 
-    esp_wifi_set_config(ESP_IF_WIFI_STA, config);
+void common_hal_wifi_radio_set_power_management(wifi_radio_obj_t *self, wifi_power_management_t power_management) {
+    switch (power_management) {
+        case POWER_MANAGEMENT_MIN:
+            esp_wifi_set_ps(WIFI_PS_MIN_MODEM);
+            break;
+        case POWER_MANAGEMENT_MAX: {
+            // listen_interval is only used in this case.
+            wifi_config_t *config = &self->sta_config;
+            // This is a typical value seen in various examples.
+            config->sta.listen_interval = 3;
+            esp_wifi_set_ps(WIFI_PS_MAX_MODEM);
+            esp_wifi_set_config(ESP_IF_WIFI_STA, config);
+        }
+        break;
+        case POWER_MANAGEMENT_NONE:
+            esp_wifi_set_ps(WIFI_PS_NONE);
+            break;
+        case POWER_MANAGEMENT_UNKNOWN:
+            // This should be prevented in shared-bindings.
+            break;
+    }
 }
 
 mp_obj_t common_hal_wifi_radio_get_mac_address_ap(wifi_radio_obj_t *self) {

diff --git a/ports/espressif/mpconfigport.mk b/ports/espressif/mpconfigport.mk
@@ -40,9 +40,6 @@ CIRCUITPY_FULL_BUILD ?= 1
 # If SSL is enabled, it's mbedtls
 CIRCUITPY_SSL_MBEDTLS = 1
 
-# Wifi Power Save
-CIRCUITPY_WIFI_RADIO_SETTABLE_LISTEN_INTERVAL = 1
-
 # Never use our copy of MBEDTLS
 CIRCUITPY_HASHLIB_MBEDTLS_ONLY = 0
 

diff --git a/ports/raspberrypi/bindings/cyw43/__init__.c b/ports/raspberrypi/bindings/cyw43/__init__.c
@@ -16,7 +16,7 @@
 
 #include "lib/cyw43-driver/src/cyw43.h"
 
-static int power_management_value = PM_DISABLED;
+static uint32_t power_management_value = CONST_CYW43_DEFAULT_PM;
 
 void cyw43_enter_deep_sleep(void) {
 #define WL_REG_ON 23
@@ -43,14 +43,23 @@ MP_DEFINE_CONST_OBJ_TYPE(
     print, shared_bindings_microcontroller_pin_print
     );
 
+uint32_t cyw43_get_power_management_value() {
+    return power_management_value;
+}
+
+void cyw43_set_power_management_value(uint32_t value) {
+    power_management_value = value;
+    bindings_cyw43_wifi_enforce_pm();
+}
+
 //| PM_STANDARD: int
-//| """The standard power management mode"""
+//| """The default power management mode; same as PM_PERFORMANCE"""
 //| PM_AGGRESSIVE: int
 //| """Aggressive power management mode for optimal power usage at the cost of performance"""
 //| PM_PERFORMANCE: int
 //| """Performance power management mode where more power is used to increase performance"""
 //| PM_DISABLED: int
-//| """Disable power management and always use highest power mode. CircuitPython sets this value at reset time, because it provides the best connectivity reliability."""
+//| """Disable power management and always use highest power mode."""
 //|
 //|
 //| def set_power_management(value: int) -> None:
@@ -85,8 +94,7 @@ MP_DEFINE_CONST_OBJ_TYPE(
 //|
 static mp_obj_t cyw43_set_power_management(const mp_obj_t value_in) {
     mp_int_t value = mp_obj_get_int(value_in);
-    power_management_value = value;
-    bindings_cyw43_wifi_enforce_pm();
+    cyw43_set_power_management_value(value);
     return mp_const_none;
 }
 static MP_DEFINE_CONST_FUN_OBJ_1(cyw43_set_power_management_obj, cyw43_set_power_management);
@@ -126,10 +134,10 @@ static const mp_rom_map_elem_t cyw43_module_globals_table[] = {
     { MP_ROM_QSTR(MP_QSTR_CywPin), MP_ROM_PTR(&cyw43_pin_type) },
     { MP_ROM_QSTR(MP_QSTR_set_power_management), &cyw43_set_power_management_obj },
     { MP_ROM_QSTR(MP_QSTR_get_power_management), &cyw43_get_power_management_obj },
-    { MP_ROM_QSTR(MP_QSTR_PM_STANDARD), MP_ROM_INT(PM_STANDARD) },
-    { MP_ROM_QSTR(MP_QSTR_PM_AGGRESSIVE), MP_ROM_INT(PM_AGGRESSIVE) },
-    { MP_ROM_QSTR(MP_QSTR_PM_PERFORMANCE), MP_ROM_INT(PM_PERFORMANCE) },
-    { MP_ROM_QSTR(MP_QSTR_PM_DISABLED), MP_ROM_INT(PM_DISABLED) },
+    { MP_ROM_QSTR(MP_QSTR_PM_STANDARD), MP_ROM_INT(CONST_CYW43_DEFAULT_PM) },
+    { MP_ROM_QSTR(MP_QSTR_PM_AGGRESSIVE), MP_ROM_INT(CONST_CYW43_AGGRESSIVE_PM) },
+    { MP_ROM_QSTR(MP_QSTR_PM_PERFORMANCE), MP_ROM_INT(CONST_CYW43_PERFORMANCE_PM) },
+    { MP_ROM_QSTR(MP_QSTR_PM_DISABLED), MP_ROM_INT(CONST_CYW43_NONE_PM) },
 };
 
 static MP_DEFINE_CONST_DICT(cyw43_module_globals, cyw43_module_globals_table);

diff --git a/ports/raspberrypi/bindings/cyw43/__init__.h b/ports/raspberrypi/bindings/cyw43/__init__.h
@@ -16,21 +16,25 @@ const mcu_pin_obj_t *validate_obj_is_free_pin_including_cyw43(mp_obj_t obj, qstr
 const mcu_pin_obj_t *validate_obj_is_free_pin_or_gpio29(mp_obj_t obj, qstr arg_name);
 const mcu_pin_obj_t *validate_obj_is_pin_including_cyw43(mp_obj_t obj, qstr arg_name);
 
-#define CONSTANT_CYW43_PM_VALUE(pm_mode, pm2_sleep_ret_ms, li_beacon_period, li_dtim_period, li_assoc) \
+// This is equivalent to the code in cyw43.h, except that the values are computed at compile time.
+// A `CONST_` prefix has been added to the computation function (expressed as a macro) and the values.
+
+#define CONST_cyw43_pm_value(pm_mode, pm2_sleep_ret_ms, li_beacon_period, li_dtim_period, li_assoc) \
     (li_assoc << 20 | /* listen interval sent to ap */ \
         li_dtim_period << 16 | \
         li_beacon_period << 12 | \
         (pm2_sleep_ret_ms / 10) << 4 | /* cyw43_ll_wifi_pm multiplies this by 10 */ \
         pm_mode /* CYW43_PM2_POWERSAVE_MODE etc */)
 
-// CYW43_DEFAULT_PM (except a compile-time constant)
-#define PM_STANDARD CONSTANT_CYW43_PM_VALUE(CYW43_PM2_POWERSAVE_MODE, 200, 1, 1, 10)
-// CYW43_AGGRESSIVE_PM (except a compile-time constant)
-#define PM_AGGRESSIVE CONSTANT_CYW43_PM_VALUE(CYW43_PM2_POWERSAVE_MODE, 2000, 1, 1, 10)
-// CYW43_PERFORMANCE_PM (except a compile-time constant)
-#define PM_PERFORMANCE CONSTANT_CYW43_PM_VALUE(CYW43_PM2_POWERSAVE_MODE, 20, 1, 1, 1)
-// The 0xa11140 magic value
-#define PM_DISABLED CONSTANT_CYW43_PM_VALUE(CYW43_NO_POWERSAVE_MODE, 200, 1, 1, 10)
+#define CONST_CYW43_DEFAULT_PM (CONST_CYW43_PERFORMANCE_PM)
+
+#define CONST_CYW43_NONE_PM (CONST_cyw43_pm_value(CYW43_NO_POWERSAVE_MODE, 10, 0, 0, 0))
+
+#define CONST_CYW43_AGGRESSIVE_PM (CONST_cyw43_pm_value(CYW43_PM1_POWERSAVE_MODE, 10, 0, 0, 0))
+
+#define CONST_CYW43_PERFORMANCE_PM (CONST_cyw43_pm_value(CYW43_PM2_POWERSAVE_MODE, 200, 1, 1, 10))
 
+extern uint32_t cyw43_get_power_management_value(void);
+extern void cyw43_set_power_management_value(uint32_t value);
 extern void bindings_cyw43_wifi_enforce_pm(void);
 void cyw43_enter_deep_sleep(void);
diff --git a/ports/raspberrypi/common-hal/wifi/Radio.c b/ports/raspberrypi/common-hal/wifi/Radio.c
@@ -5,6 +5,7 @@
 // SPDX-License-Identifier: MIT
 
 #include "supervisor/port.h"
+#include "shared-bindings/wifi/PowerManagement.h"
 #include "shared-bindings/wifi/Radio.h"
 #include "shared-bindings/wifi/Network.h"
 
@@ -107,6 +108,41 @@ void common_hal_wifi_radio_set_tx_power(wifi_radio_obj_t *self, const mp_float_t
     cyw43_ioctl(&cyw43_state, CYW43_IOCTL_SET_VAR, 9 + 4, buf, CYW43_ITF_AP);
 }
 
+wifi_power_management_t common_hal_wifi_radio_get_power_management(wifi_radio_obj_t *self) {
+    uint32_t pm_value = cyw43_get_power_management_value();
+
+    switch (pm_value) {
+        case CONST_CYW43_PERFORMANCE_PM:
+            return POWER_MANAGEMENT_MIN;
+        case CONST_CYW43_AGGRESSIVE_PM:
+            return POWER_MANAGEMENT_MAX;
+        case CONST_CYW43_NONE_PM:
+            return POWER_MANAGEMENT_NONE;
+        default:
+            return POWER_MANAGEMENT_UNKNOWN;
+    }
+}
+
+
+void common_hal_wifi_radio_set_power_management(wifi_radio_obj_t *self, wifi_power_management_t power_management) {
+    uint32_t pm_setting = CONST_CYW43_DEFAULT_PM;
+    switch (power_management) {
+        case POWER_MANAGEMENT_MIN:
+            pm_setting = CONST_CYW43_PERFORMANCE_PM;
+            break;
+        case POWER_MANAGEMENT_MAX:
+            pm_setting = CONST_CYW43_AGGRESSIVE_PM;
+            break;
+        case POWER_MANAGEMENT_NONE:
+            pm_setting = CONST_CYW43_NONE_PM;
+            break;
+        default:
+            // Should not get here.
+            break;
+    }
+    cyw43_set_power_management_value(pm_setting);
+}
+
 mp_obj_t common_hal_wifi_radio_get_mac_address_ap(wifi_radio_obj_t *self) {
     return common_hal_wifi_radio_get_mac_address(self);
 }

diff --git a/ports/zephyr-cp/common-hal/wifi/Radio.c b/ports/zephyr-cp/common-hal/wifi/Radio.c
@@ -137,10 +137,44 @@ void common_hal_wifi_radio_set_tx_power(wifi_radio_obj_t *self, const mp_float_t
     // esp_wifi_set_max_tx_power(tx_power * 4.0f);
 }
 
-mp_int_t common_hal_wifi_radio_get_listen_interval(wifi_radio_obj_t *self) {
-    // wifi_config_t *config = &self->sta_config;
-    // return config->sta.listen_interval;
-    return 0;
+wifi_power_management_t common_hal_wifi_radio_get_power_management(wifi_radio_obj_t *self) {
+    // wifi_ps_type_t ps;
+    // esp_err_t ret = esp_wifi_get_ps(&ps);
+    // if (ret == ESP_OK) {
+    //     switch (ps) {
+    //         case WIFI_PS_MIN_MODEM:
+    //             return POWER_MANAGEMENT_MIN;
+    //         case WIFI_PS_MAX_MODEM:
+    //             return POWER_MANAGEMENT_MAX;
+    //         case WIFI_PS_NONE:
+    //             return POWER_MANAGEMENT_NONE;
+    //     }
+    // }
+    return POWER_MANAGEMENT_UNKNOWN;
+}
+
+
+void common_hal_wifi_radio_set_power_management(wifi_radio_obj_t *self, wifi_power_management_t power_management) {
+    // switch (power_management) {
+    //     case POWER_MANAGEMENT_MIN:
+    //         esp_wifi_set_ps(WIFI_PS_MIN_MODEM);
+    //         break;
+    //     case POWER_MANAGEMENT_MAX: {
+    //         // listen_interval is only used in this case.
+    //         wifi_config_t *config = &self->sta_config;
+    //         // This is a typical value seen in various examples.
+    //         config->sta.listen_interval = 3;
+    //         esp_wifi_set_ps(WIFI_PS_MAX_MODEM);
+    //         esp_wifi_set_config(ESP_IF_WIFI_STA, config);
+    //     }
+    //     break;
+    //     case POWER_MANAGEMENT_NONE:
+    //         esp_wifi_set_ps(WIFI_PS_NONE);
+    //         break;
+    //     case POWER_MANAGEMENT_UNKNOWN:
+    //         // This should be prevented in shared-bindings.
+    //         break;
+    // }
 }
 
 void common_hal_wifi_radio_set_listen_interval(wifi_radio_obj_t *self, const mp_int_t listen_interval) {

diff --git a/py/circuitpy_defns.mk b/py/circuitpy_defns.mk
@@ -620,6 +620,7 @@ $(filter $(SRC_PATTERNS), \
 	supervisor/StatusBar.c \
 	wifi/AuthMode.c \
 	wifi/Packet.c \
+	wifi/PowerManagement.c \
 )
 
 ifeq ($(CIRCUITPY_BLEIO_HCI),1)

diff --git a/py/circuitpy_mpconfig.mk b/py/circuitpy_mpconfig.mk
@@ -707,9 +707,6 @@ CFLAGS += -DCIRCUITPY_WEB_WORKFLOW=$(CIRCUITPY_WEB_WORKFLOW)
 CIRCUITPY_WIFI_RADIO_SETTABLE_MAC_ADDRESS?= 1
 CFLAGS += -DCIRCUITPY_WIFI_RADIO_SETTABLE_MAC_ADDRESS=$(CIRCUITPY_WIFI_RADIO_SETTABLE_MAC_ADDRESS)
 
-CIRCUITPY_WIFI_RADIO_SETTABLE_LISTEN_INTERVAL?= 0
-CFLAGS += -DCIRCUITPY_WIFI_RADIO_SETTABLE_LISTEN_INTERVAL=$(CIRCUITPY_WIFI_RADIO_SETTABLE_LISTEN_INTERVAL)
-
 # tinyusb port tailored configuration
 CIRCUITPY_TUSB_MEM_ALIGN ?= 4
 CFLAGS += -DCIRCUITPY_TUSB_MEM_ALIGN=$(CIRCUITPY_TUSB_MEM_ALIGN)

diff --git a/shared-bindings/wifi/PowerManagement.c b/shared-bindings/wifi/PowerManagement.c
@@ -0,0 +1,50 @@
+// This file is part of the CircuitPython project: https://circuitpython.org
+//
+// SPDX-FileCopyrightText: Copyright (c) 2025 Dan Halbert for Adafruit Industries
+//
+// SPDX-License-Identifier: MIT
+
+#include "py/enum.h"
+
+#include "shared-bindings/wifi/PowerManagement.h"
+
+//| class PowerManagement:
+//|     """Power-saving options for wifi
+//|
+//|     .. note:: On boards using the CYW43 radio module, the choices below correspond to the
+//|       power management values defined in the `cyw43` module:
+//|       `PowerManagement.MIN` is the same as `cyw43.PM_PERFORMANCE`, `PowerManagement.MAX`
+//|       is the same as `cyw43.PM_AGGRESSIVE`, and `PowerManagement.NONE` is the same as
+//|       `cyw43.PM_DISABLED`. If a custom value was set with `cyw43.set_power_management()`
+//|       not corresponding to one of these three values, then `PowerManagement.UNKNOWN` will be returned.
+//|       """
+//|
+//|     MIN: PowerManagement
+//|     """Minimum power management (default). The WiFi station wakes up to receive a beacon every DTIM period.
+//|        The DTIM period is set by the access point."""
+//|     MAX: PowerManagement
+//|     """Maximum power management, at the expense of some performance. The WiFi station wakes up less often than `MIN`."""
+//|     NONE: PowerManagement
+//|     """No power management: the WiFi station does not sleep."""
+//|     UNKNOWN: PowerManagement
+//|     """Power management setting cannot be determined."""
+//|
+
+// In order of the enum type.
+MAKE_ENUM_VALUE(wifi_power_management_type, power_management, NONE, POWER_MANAGEMENT_NONE);
+MAKE_ENUM_VALUE(wifi_power_management_type, power_management, MIN, POWER_MANAGEMENT_MIN);
+MAKE_ENUM_VALUE(wifi_power_management_type, power_management, MAX, POWER_MANAGEMENT_MAX);
+MAKE_ENUM_VALUE(wifi_power_management_type, power_management, UNKNOWN, POWER_MANAGEMENT_UNKNOWN);
+
+MAKE_ENUM_MAP(wifi_power_management) {
+    MAKE_ENUM_MAP_ENTRY(power_management, NONE),
+    MAKE_ENUM_MAP_ENTRY(power_management, MIN),
+    MAKE_ENUM_MAP_ENTRY(power_management, MAX),
+    MAKE_ENUM_MAP_ENTRY(power_management, UNKNOWN),
+};
+
+static MP_DEFINE_CONST_DICT(wifi_power_management_locals_dict, wifi_power_management_locals_table);
+
+MAKE_PRINTER(wifi, wifi_power_management);
+
+MAKE_ENUM_TYPE(wifi, PowerManagement, wifi_power_management);
diff --git a/shared-bindings/wifi/PowerManagement.h b/shared-bindings/wifi/PowerManagement.h
@@ -0,0 +1,19 @@
+// This file is part of the CircuitPython project: https://circuitpython.org
+//
+// SPDX-FileCopyrightText: Copyright (c) 2025 Dan Halbert for Adafruit Industries
+//
+// SPDX-License-Identifier: MIT
+
+#pragma once
+
+#include "py/enum.h"
+
+typedef enum {
+    POWER_MANAGEMENT_NONE = 0,
+    POWER_MANAGEMENT_MIN = 1,
+    POWER_MANAGEMENT_MAX = 2,
+    // Value can't be determined.
+    POWER_MANAGEMENT_UNKNOWN = 3,
+} wifi_power_management_t;
+
+extern const mp_obj_type_t wifi_power_management_type;