Hu Chunming / mvpt_dq_ascend_arm

  #include "FFRecoder.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 FFRecoderTask {

  public:

      FFRecoderTask();
  	virtual ~FFRecoderTask();

      void cache_pkt(AVPacket* pkt, long long frame_nb);

      bool init(int w, int h, int fps, int bit_rate, string dec_name);

      void create_recode_task(RecoderInfo& recoderInfo);
  
      void close();
  
      void set_mq_callback(mq_callback_t cb);
  
  public:
      void recode_thread();
      void pkt_cache_thread();

      std::string GetName(){return m_dec_name;}

  
  private:
      void save_intask_recoderinfo(RecoderInfo info);
      list<DataPacket*>::iterator getStartIterator(unsigned long long frame_nb);
      
  private:

      std::string m_dec_name;

      std::queue<int> m_key_frame_interval;
      std::list<unsigned long long> m_keyframe_nb_list;
      std::list<DataPacket*> m_pkt_list_short;    // 临时缓存
      mutex m_pkt_list_short_mtx; 
      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};
      bool m_bExitRecoderThread{false};
  
      AVRational m_time_base;
      AVCodecContext* m_avctx;
      AVStream* m_inStream;
  

      thread* m_recoder_thread{nullptr};

      thread* m_cache_thread{nullptr};
  
      mq_callback_t mq_publish_func;
  
      // FFRecoder
      int m_width;
      int m_height;
      int m_fps;
      int m_bit_rate;
  };