Hu Chunming / vpt_ascend_arm

  #include "FFRecoder.h"

  #include "FFRecoder2.h"

  
  #include "../../ai_platform/common_header.h"
  #include "depend_headers.h"
  
  #include <list>
  #include <queue>
  #include <map>
  #include <string>
  #include <thread>
  #include <mutex>
  
  using namespace std;
  
  struct RecodeParam {
      AVRational time_base;
      RecoderInfo recoderInfo;
      AVCodecContext* avctx;
  };
  

  typedef std::function<bool(const char *msg)> mq_callback_t;
  

  class FFRecoderTaskManager {
  public:
      FFRecoderTaskManager();
  	virtual ~FFRecoderTaskManager();
  
      void cache_pkt(AVPacket* pkt, long long frame_nb);

      bool init(AVStream* stream, AVCodecContext* avctx);

      bool init2(int w, int h, int fps, int bit_rate);

      void create_recode_task(RecoderInfo& recoderInfo);

      void close();
  
      void set_mq_callback(mq_callback_t cb);
  

  public:

      void recode_thread();

      void recode_thread2();

  
  private:

      void save_intask_recoderinfo(RecoderInfo info);

      list<DataPacket*>::iterator getStartIterator(unsigned long long frame_nb);

      
  private:
      std::queue<int> m_key_frame_interval;
      std::list<unsigned long long> m_keyframe_nb_list;
      std::list<DataPacket*> m_pkt_list;
      mutex m_pkt_list_mtx; 

  
      std::list<RecoderInfo> m_recoderinfo_list;
      mutex m_recoderinfo_list_mtx;
  
      bool m_bExit{false};
  

      AVRational m_time_base;
      AVCodecContext* m_avctx;

      AVStream* m_inStream;

      thread* m_recoder_thread{nullptr};
  
      mq_callback_t mq_publish_func;

  
      // FFRecoder2
      int m_width;
      int m_height;
      int m_fps;
      int m_bit_rate;

  };