Add a console command to dump a snapshot

3 files changed, 31 insertions, 0 deletions

diff --git a/src/tangara/ui/ui_events.hpp b/src/tangara/ui/ui_events.hpp
index 05fd4483..0f371769 100644
--- a/src/tangara/ui/ui_events.hpp
+++ b/src/tangara/ui/ui_events.hpp
@@ -30,6 +30,10 @@ struct OnLuaError : tinyfsm::Event {
 
 struct DumpLuaStack : tinyfsm::Event {};
 
+struct Screenshot : tinyfsm::Event {
+  std::string filename;
+};
+
 namespace internal {
 
 struct InitDisplay : tinyfsm::Event {
diff --git a/src/tangara/ui/ui_fsm.cpp b/src/tangara/ui/ui_fsm.cpp
index 476732db..4f93fe61 100644
--- a/src/tangara/ui/ui_fsm.cpp
+++ b/src/tangara/ui/ui_fsm.cpp
@@ -13,6 +13,7 @@
 #include <variant>
 
 #include "FreeRTOSConfig.h"
+#include "draw/lv_draw_buf.h"
 #include "drivers/bluetooth.hpp"
 #include "lvgl.h"
 
@@ -26,6 +27,9 @@
 #include "freertos/projdefs.h"
 #include "lua.hpp"
 #include "luavgl.h"
+#include "misc/lv_color.h"
+#include "misc/lv_utils.h"
+#include "others/snapshot/lv_snapshot.h"
 #include "tick/lv_tick.h"
 #include "tinyfsm.hpp"
 
@@ -363,6 +367,28 @@ int UiState::PopScreen() {
   return sScreens.size();
 }
 
+void UiState::react(const Screenshot& ev) {
+  if (!sCurrentScreen) {
+    return;
+  }
+  ESP_LOGI(kTag, "taking snapshot");
+  lv_draw_buf_t* buf =
+      lv_snapshot_take(sCurrentScreen->root(), LV_COLOR_FORMAT_RGB888);
+  if (!buf) {
+    ESP_LOGW(kTag, "snapshot failed");
+    return;
+  }
+  ESP_LOGI(kTag, "writing to file");
+  std::string fullpath = "//sdcard/" + ev.filename;
+  auto res = lv_draw_buf_save_to_file(buf, fullpath.c_str());
+  lv_draw_buf_destroy(buf);
+  if (res == LV_RESULT_OK) {
+    ESP_LOGI(kTag, "write okay!");
+  } else {
+    ESP_LOGE(kTag, "write failed!");
+  }
+}
+
 void UiState::react(const system_fsm::KeyLockChanged& ev) {
   sDisplay->SetDisplayOn(!ev.locking);
   sInput->lock(ev.locking);
diff --git a/src/tangara/ui/ui_fsm.hpp b/src/tangara/ui/ui_fsm.hpp
index 72688fa0..cef9a13a 100644
--- a/src/tangara/ui/ui_fsm.hpp
+++ b/src/tangara/ui/ui_fsm.hpp
@@ -53,6 +53,7 @@ class UiState : public tinyfsm::Fsm<UiState> {
   /* Fallback event handler. Does nothing. */
   void react(const tinyfsm::Event& ev) {}
 
+  void react(const Screenshot&);
   virtual void react(const OnLuaError&) {}
   virtual void react(const DumpLuaStack&) {}
   virtual void react(const internal::BackPressed&) {}