Hu Chunming / vpt_ascend_arm

  #ifndef __DVPP_STREAM_DECODER_H__
  #define __DVPP_STREAM_DECODER_H__
  
  #include "depend_headers.h"
  #include "dvpp_headers.h"
  
  #include "VpcUtils.h"
  

  #include "FFRecoderTaskManager.h"
  

  using namespace std;
  
  class DvppStreamDecoder
  {
  public:
      DvppStreamDecoder(/* args */);
      ~DvppStreamDecoder();
  

      bool Init(FFDecConfig cfg);

      int SendData(int videoType, char* data, int len, int isKey, uint64_t pts);
  
      void Close();
  
      DvppDataMemory* GetFrame();

      void doRecode(RecoderInfo& recoderInfo);
  
      void set_mq_callback(mq_callback_t cb);
  

  public:
      void doProcessReport();
      void doVdppVdecCallBack(acldvppStreamDesc *input, acldvppPicDesc *output, void *pUserData);
  
  private:
      bool sendVdecEos(aclvdecChannelDesc *vdecChannelDesc);

      int sendPkt(aclvdecChannelDesc *vdecChannelDesc, AVPacket* pkt, unsigned long long frame_nb, int vdec_out_size);
  
      int parse_stream_info(int videoType, AVPacket* pkt);

      void release_dvpp();
      void release_channeldesc();
      int getVdecType(int videoType);
  

  private:
      string m_dec_name {""};
      int m_deviceId {-1};
  
      int m_dvpp_channel {-1};
      aclrtContext m_context{nullptr};
  
      aclvdecChannelDesc *vdecChannelDesc {nullptr};

      pthread_t report_thread;
      bool m_bExitReportThd{false};

  
      int m_vdec_out_size {-1};

      int m_fps {-1};
      int frameW {-1};
      int frameH {-1};

      int out_frame_width{0};
  	int out_frame_height{0};

  
      VpcUtils m_vpcUtils;
  
      unsigned long m_frame_nb {0};

      std::atomic<int> m_DvppCacheCounter{0};
  
      queue<DvppDataMemory*> m_decoded_data_queue;
      mutex m_decoded_data_queue_mtx;
  
      bool m_bExit {false};
  
      const void * m_finishedDecArg {nullptr};
      DECODE_FINISHED_CALLBACK decode_finished_cbk {nullptr};
  
      AVBSFContext * h264bsfc{nullptr};

  
      uint64_t m_in_count {0};
      uint64_t m_out_count {0};
  
      bool m_bResize {false};

  
      FFRecoderTaskManager m_recoderManager;

  };
  
  
  
  
  #endif      // __DVPP_STREAM_DECODER_H__