first commit

nodes/record/vp_record_node.h

@@ -0,0 +1,72 @@
+#pragma once
+
+#include <list>
+#include <deque>
+#include <map>
+
+#include "../vp_node.h"
+#include "../../objects/vp_image_record_control_meta.h"
+#include "../../objects/vp_video_record_control_meta.h"
+
+#include "vp_image_record_task.h"
+#include "vp_video_record_task.h"
+#include "vp_record_status_hookable.h"
+
+namespace vp_nodes {
+    // video/image recording node, save it to local disk.
+    // it is a middle node but works asynchronously, so recording would not block the pipeline.
+    // note record node could work on multi channels at the same time.
+    class vp_record_node: public vp_node, public vp_record_status_hookable
+    {
+    private:
+        /* config data */
+        // video save directory
+        std::string video_save_dir;
+        // image save directory
+        std::string image_save_dir;
+        // pre record time for video (seconds)
+        int pre_record_video_duration;
+        // record time for video (seconds), not including pre_record_video_duration
+        int record_video_duration;
+        // auto create sub directory by date and channel or not, such as `./video_save_dir/2022-10-8/1/**.mp4`
+        bool auto_sub_dir;
+        // width and height
+        vp_objects::vp_size resolution_w_h = {};
+        // bitrate for video record
+        int bitrate = 1024;
+        // record osd frame or not
+        bool osd = false;
+
+        // fps for current video
+        // int fps = 0;
+        std::map<int, int> all_fps;
+        
+        /* record task list */
+        // std::list<std::shared_ptr<vp_nodes::vp_record_task>> record_tasks;
+        std::map<int, std::list<std::shared_ptr<vp_nodes::vp_record_task>>> all_record_tasks;
+
+        /* pre-record for video */
+        // std::deque<std::shared_ptr<vp_objects::vp_frame_meta>> pre_records;
+        std::map<int, std::deque<std::shared_ptr<vp_objects::vp_frame_meta>>> all_pre_records;
+
+        // new record task
+        void auto_new_record_task(std::shared_ptr<vp_objects::vp_control_meta>& meta);
+
+    protected:
+        // re-implementation
+        virtual std::shared_ptr<vp_objects::vp_meta> handle_frame_meta(std::shared_ptr<vp_objects::vp_frame_meta> meta) override; 
+        // re-implementation
+        virtual std::shared_ptr<vp_objects::vp_meta> handle_control_meta(std::shared_ptr<vp_objects::vp_control_meta> meta) override;
+    public:
+        vp_record_node(std::string node_name, 
+                        std::string video_save_dir, 
+                        std::string image_save_dir,
+                        vp_objects::vp_size resolution_w_h = {}, 
+                        bool osd = false,
+                        int pre_record_video_duration = 5, 
+                        int record_video_duration = 20,
+                        bool auto_sub_dir = true,
+                        int bitrate = 1024);
+        ~vp_record_node();
+    };
+}