Add menu option for encoding timelapse with ffmpeg

Author: felipemanga

data/settings.ini

@@ -19,6 +19,8 @@ new-image-palette = %appdata/palettes/default.png
 mouse-wheel-invert-x = false
 mouse-wheel-invert-y = false
 
+#ffmpeg-path = C:/ffmpeg.exe
+
 #OpenGLMajor = 3
 #OpenGLMinor = 0
 #OpenGLProfile = es

data/skins/default/gui/MainWindow.xml

@@ -80,7 +80,7 @@
  </clickout>
 
  <clickout id="recmenu" class="menu">
- <image class="menucontainer" width="170px" x="100">
+ <image class="menucontainer" width="200px" x="100">
  <menubutton
  controller="recorderstatus"
  click="togglerec"
@@ -91,6 +91,11 @@
  click="resetrecsequence"
  label="Reset Frame (#${value})"
  forward="labelspan.text=text"/>
+ <menubutton
+ controller="recorderencode"
+ click="recorderencode"
+ label="Encode MP4"
+ forward="labelspan.text=text"/>
  </image>
  </clickout>
 

src/cmd/Recording.cpp

@@ -3,6 +3,7 @@
 // Read LICENSE.txt for more information.
 
 #include <cmd/Command.hpp>
+#include <common/Config.hpp>
 #include <common/Messages.hpp>
 #include <common/PubSub.hpp>
 #include <doc/Cell.hpp>
@@ -12,11 +13,16 @@
 #include <gui/Controller.hpp>
 #include <gui/Node.hpp>
 #include <log/Log.hpp>
+#include <optional>
+#include <task/TaskManager.hpp>
 
 namespace msg {
 struct RecorderStart {};
 struct RecorderStop {};
 struct RecorderUpdateSequence {U32 number;};
+struct RecorderEncodingStart {};
+struct RecorderEncodingDone {};
+struct RecorderEncodingFail {};
 }
 
 U32 initSequence() {
@@ -41,6 +47,47 @@ U32& sequence() {
  return num;
 }
 
+class Encoder {
+public:
+ TaskHandle handle;
+
+ bool done() {
+ return handle.task->isDone();
+ }
+
+ void run() {
+ String cmd;
+ PubSub<>::pub(msg::RecorderEncodingStart{});
+
+ auto userdata = std::filesystem::path{inject<FileSystem>{}->find("%userdata", "dir")->getUID()};
+ cmd += "cd " + join(split(userdata.string(), "\\"), "\\\\") + " && ";
+
+ cmd += "\"";
+ if (auto ffmpeg = inject<Config>{}->properties->get<String>("ffmpeg-path"); ffmpeg.empty()) {
+ cmd += "ffmpeg";
+ } else {
+ cmd += join(split(ffmpeg, "\\"), "\\\\");
+ }
+ cmd += "\" -y ";
+ cmd += "-start_number 1 -i \"recording_%d.png\" -c:v libx264 -r 30 -pix_fmt yuv420p out.mp4";
+
+ logI(cmd);
+
+ handle = inject<TaskManager>{}->add(
+ [=]() -> int {
+ return system(cmd.c_str());
+ },
+ [=](int code){
+ if (code == 0) {
+ PubSub<>::pub(msg::RecorderEncodingDone{});
+ } else {
+ PubSub<>::pub(msg::RecorderEncodingFail{});
+ }
+ });
+ }
+};
+static std::optional<Encoder> encoder;
+
 class Recorder {
  PubSub<msg::ModifyDocument,
  msg::Tick> pub{this};
@@ -118,6 +165,43 @@ class RecorderSequence : public ui::Controller {
 };
 static ui::Controller::Shared<RecorderSequence> recseq{"recordersequence"};
 
+class RecorderEncode : public ui::Controller {
+public:
+ PubSub<msg::RecorderEncodingStart,
+ msg::RecorderEncodingDone,
+ msg::RecorderEncodingFail,
+ msg::RecorderUpdateSequence> pub{this};
+ Property<String> label{this, "label", "${value}"};
+
+ void attach() override {
+ if (encoder) {
+ if (encoder->done())
+ on(msg::RecorderEncodingDone{});
+ else
+ on(msg::RecorderEncodingStart{});
+ } else {
+ on(msg::RecorderUpdateSequence{});
+ }
+ }
+
+ void on(const msg::RecorderUpdateSequence&) {
+ node()->set("text", "Encode MP4");
+ }
+
+ void on(const msg::RecorderEncodingDone&) {
+ node()->set("text", "Encoding Done");
+ }
+
+ void on(const msg::RecorderEncodingStart&) {
+ node()->set("text", "Encoding MP4");
+ }
+
+ void on(const msg::RecorderEncodingFail&) {
+ node()->set("text", "Encoding Error");
+ }
+};
+static ui::Controller::Shared<RecorderEncode> recenc{"recorderencode"};
+
 class ToggleRecording : public Command {
 public:
  void run() override {
@@ -138,3 +222,16 @@ class ResetRecordingSequence : public Command {
  }
 };
 static Command::Shared<ResetRecordingSequence> cmd2{"resetrecsequence"};
+
+class RecorderEncodeCmd : public Command {
+public:
+ void run() override {
+ if (encoder && !encoder->done()) {
+ return;
+ }
+ encoder = std::nullopt;
+ encoder.emplace();
+ encoder->run();
+ }
+};
+static Command::Shared<RecorderEncodeCmd> cmd3{"recorderencode"};