保存触发id之后的视频作为触发视频

zhangmeng

2020-07-24 f93ee1a42e8c47e472332287b7350b66a6b0fa11

 csrc/buz/recorder.hpp

@@ -13,19 +13,17 @@

#include "../common/callback.hpp"

#include "../common.hpp"

struct AVPacket;

namespace ffwrapper{
    class FormatIn;
    class FormatOut;

    class CodedData;
}

namespace cffmpeg_wrap{
    namespace buz{
        struct avpacket{
            std::shared_ptr<ffwrapper::CodedData> data;
            int64_t id;
        };

        class Recorder{
            public:
@@ -33,44 +31,65 @@
                ~Recorder();

            public: 
                int Run(const char* output, const int mind, const int maxd);
                int CachePacket(const avpacket &pkt);
                int Run(const char* output, const int mind, const int maxd, const bool audio);
                int PushPacket(std::list<CPacket> &lst);
                int StartWritePacket(std::list<CPacket> &lst, const int64_t &id, const int start, const int end);
                int FireRecorder(const int64_t &id);

                void SetCallback(FUNC_REC_INFO cb){
                    func_rec_info_ = cb;
                }

                const bool ErrorOcurred(){return error_occured_;}
                const std::string& RecID()const{return id_;}
            private:
                void run_thread();

                int init_writer();
                void start_writer();
                int write_correctly(const avpacket &pkt);
                int init_writer(const bool audio);
                int write_correctly(const CPacket &pkt);
                void end_writer();

                void maybe_dump_gop();

                int init_write_h264(const bool audio);
                int write_h264(const CPacket &pkt);
                int end_write_h264();

                int init_write_hevc(const bool audio);
                int write_hevc(const CPacket &pkt);
                int end_write_hevc();
                int mux_hevc(FILE *fp, const char *outfile);
            private: 
                ffwrapper::FormatIn     *in_;
                ffwrapper::FormatOut    *out_;

                int     maxduration;
                int     minduration;
                int     end_frame;
                int     cur_frame;

                std::list<avpacket>     list_pkt_;

                std::list<CPacket>      list_pkt_;
                std::atomic_bool        stop_recorder_;
                std::mutex              mutex_pkt_;
               std::condition_variable cv_;

                std::string             dir_;
                std::string             id_;

                int64_t                 id_frame_;
                int                     file_frame_index_;
                int                     id_frame_in_file_;

                std::string             file_path_;
                FUNC_REC_INFO           func_rec_info_;
                FILE                    *fp_;
                bool                    audio_;

                std::unique_ptr<std::thread> thrd_;


                int     end_frame_;
                int     v_cur_frame_;
                int     a_cur_frame_;
                
                int64_t last_rec_id_;

                int     maxduration;
                int     minduration;

                bool                    error_occured_;
        };
    }
}