From 535695f0b97527a1496d31c1e522a34469f8e0bc Mon Sep 17 00:00:00 2001
From: Ivan Grokhotkov <ivan@espressif.com>
Date: Fri, 22 Sep 2017 23:06:52 +0800
Subject: [PATCH] freertos: add frequency switching hooks to ISR and idle task

---
 components/esp32/freertos_hooks.c    | 25 +++++++++++++++----------
 components/freertos/xtensa_vectors.S |  4 ++++
 2 files changed, 19 insertions(+), 10 deletions(-)

diff --git a/components/esp32/freertos_hooks.c b/components/esp32/freertos_hooks.c
index ed88be883..abe88a97e 100644
--- a/components/esp32/freertos_hooks.c
+++ b/components/esp32/freertos_hooks.c
@@ -20,6 +20,10 @@
 #include "esp_attr.h"
 #include "esp_freertos_hooks.h"
 
+#include "sdkconfig.h"
+#include "esp_pm.h"
+#include "pm_impl.h"
+
 //We use just a static array here because it's not expected many components will need
 //an idle or tick hook.
 #define MAX_HOOKS 8
@@ -41,20 +45,21 @@ void IRAM_ATTR esp_vApplicationTickHook()
 
 void esp_vApplicationIdleHook() 
 {
-    bool doWait=true;
-    bool r;
-    int n;
+    bool can_go_idle=true;
     int core = xPortGetCoreID();
-    for (n=0; n<MAX_HOOKS; n++) {
-        if (idle_cb[core][n]!=NULL) {
-            r=idle_cb[core][n]();
-            if (!r) doWait=false;
+    for (int n = 0; n < MAX_HOOKS; n++) {
+        if (idle_cb[core][n] != NULL && !idle_cb[core][n]()) {
+            can_go_idle = false;
         }
     }
-    if (doWait) {
-        //Wait for whatever interrupt comes next... this should save some power.
-        asm("waiti 0");
+    if (!can_go_idle) {
+        return;
     }
+
+#ifdef CONFIG_PM_ENABLE
+    esp_pm_impl_idle_hook();
+#endif
+    asm("waiti 0");
 }
 
 esp_err_t esp_register_freertos_idle_hook_for_cpu(esp_freertos_idle_cb_t new_idle_cb, UBaseType_t cpuid)
diff --git a/components/freertos/xtensa_vectors.S b/components/freertos/xtensa_vectors.S
index c64f6d440..46750081e 100644
--- a/components/freertos/xtensa_vectors.S
+++ b/components/freertos/xtensa_vectors.S
@@ -206,6 +206,10 @@ SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
     #endif
     #endif
 
+    #ifdef CONFIG_PM_ENABLE
+    call4 esp_pm_impl_isr_hook
+    #endif
+
     #ifdef XT_INTEXC_HOOKS
     /* Call interrupt hook if present to (pre)handle interrupts. */
     movi    a4, _xt_intexc_hooks