Hu Chunming / FFNvDecoder

  #include<string>
  #include <pthread.h>
  

  #include <mutex>
  

  extern "C"
  {
  	#include <libavcodec/avcodec.h> 
  	#include <libavdevice/avdevice.h> 
  	#include <libavformat/avformat.h> 
  	#include <libavfilter/avfilter.h> 
  	#include <libavutil/avutil.h> 
      #include <libavutil/pixdesc.h> 
  	#include <libswscale/swscale.h>
      #include <libavutil/imgutils.h>
  }
  
  #include "common_header.h"
  
  #include "../interface/AbstractDecoder.h"
  

  using namespace std;
  

  class FFNvDecoder : public AbstractDecoder {

  public:
      FFNvDecoder();
      ~FFNvDecoder();

      bool init(FFDecConfig& cfg);

      void close();

      bool start();

      void pause();
      void resume();
  

      void setDecKeyframe(bool bKeyframe);
  

      bool isRunning();

      bool isFinished();

      bool isPausing();

      bool getResolution( int &width, int &height );
  

      bool isSurport(FFDecConfig& cfg);
  

      int getCachedQueueLength();
  

      float fps();

      DECODER_TYPE getDecoderType(){ return DECODER_TYPE_FFMPEG; }

      FFImgInfo* snapshot();
  

      void setName(string nm){
          m_dec_name = nm;
      }
  
      string getName(){
          return m_dec_name;
      }
  
      void setPostDecArg(const void* postDecArg);
      void setFinishedDecArg(const void* finishedDecArg);
  

  public:
      AVPixelFormat getHwPixFmt();
  
  private:
      void decode_thread();
      void post_decode_thread();

      bool init(const char* uri, const char* gpuid, bool force_tcp);

      void decode_finished();

      DeviceRgbMemory* convert2bgr(AVFrame * gpuFrame);
  

  private:

      string m_dec_name;
      FFDecConfig m_cfg;
  

      AVStream* stream;
      AVCodecContext *avctx;
      int stream_index;
      AVFormatContext *fmt_ctx;
      AVPixelFormat hw_pix_fmt;
  
      pthread_t m_decode_thread;
      pthread_t m_post_decode_thread;
      
      bool m_bRunning;

      bool m_bFinished;

  
      bool m_bPause;

  
      bool m_bReal; // 是否实时流

      float m_fps;

  
      queue<AVFrame*> mFrameQueue;
      mutex m_queue_mutex;
      mutex m_snapshot_mutex;

      long m_index{0};
  
      bool m_dec_keyframe;
  
      const void * m_postDecArg;
      POST_DECODE_CALLBACK post_decoded_cbk;  // 解码数据回调接口
  
      const void * m_finishedDecArg;
      DECODE_FINISHED_CALLBACK decode_finished_cbk;

  };