Hu Chunming / FFNvDecoder

  #include "AbstractDecoder.h"

  #include<iostream>
  #include<vector>
  #include<map>
  

  #include <mutex>
  

  using namespace std;
  

  struct MgrDecConfig
  {

  	DECODER_TYPE dec_type;	// 解码器类型

      FFDecConfig cfg;    // 解码器配置
      string name{""};    // 解码器名称

  	long snap_time_interval; // 定时抓拍时间间隔

  };
  

  // #define USE_NVDEC

  // #define USE_DVPP

  /**
   * 解码器管理类，单例类

   * 谨防死锁

   **/

  class FFNvDecoderManager {

  public:

      /**************************************************
  	* 接口：getInstance
  	* 功能：获取解码器管理者实例
  	* 参数：无
  	* 返回：成功返回 解码器管理者实例， 失败返回 nullptr
  	* 备注：调用其他接口前，需要先调用该接口获取管理者实例
  	**************************************************/

      static FFNvDecoderManager* getInstance(){
  		static FFNvDecoderManager* singleton = nullptr;
  		if (singleton == nullptr){
  			singleton = new FFNvDecoderManager();
  		}
  		return singleton;
  	}
  
      ~FFNvDecoderManager()
      {
          closeAllDecoder();
      }
  

      /**************************************************
  	* 接口：createDecoder
  	* 功能：根据配置信息创建解码器
  	* 参数：MgrDecConfig& config 解码器配置信息
  	* 返回：成功返回解码器， 失败返回 nullptr
  	* 备注：
  	**************************************************/

      AbstractDecoder* createDecoder(MgrDecConfig config);

      /**************************************************

  	* 接口：setPostDecArg

  	* 功能：设置解码数据回调接口的用户自定义参数
  	* 参数：string name 解码器名称
      *       const void * userPtr  用户自定义的要传到解码数据回调接口的数据
  	* 返回：设置成功返回true，失败返回false
  	* 备注：
  	**************************************************/

      bool setPostDecArg(const string name, const void * userPtr);
  
  	/**************************************************
  	* 接口：setFinishedDecArg
  	* 功能：设置解码结束回调接口的用户自定义参数
  	* 参数：string name 解码器名称
      *       const void * userPtr  用户自定义的要传到解码数据回调接口的数据
  	* 返回：设置成功返回true，失败返回false
  	* 备注：
  	**************************************************/
      bool setFinishedDecArg(const string name, const void * userPtr);

  
      /**************************************************
  	* 接口：getDecoderByName
  	* 功能：根据解码器名称返回解码器对象指针
  	* 参数：const string name 解码器名称
  	* 返回：成功返回对应的解码器对象的指针，失败返回nullptr
  	* 备注：
  	**************************************************/

      AbstractDecoder* getDecoderByName(const string name);

      /**************************************************
  	* 接口：startDecode
  	* 功能：启动解码
  	* 参数：FFNvDecoder* 解码器指针
  	* 返回：void
  	* 备注：
  	**************************************************/

      bool startDecode(AbstractDecoder*);

  
      /**************************************************

  	* 接口：startAllDecode

  	* 功能：启动全部解码
  	* 参数：void
  	* 返回：void
  	* 备注：
  	**************************************************/

      void startAllDecode();

      /**************************************************
  	* 接口：startDecodeByName

  	* 功能：启动名称对应的解码器

  	* 参数：string name 解码器名称
  	* 返回：成功返回true，失败返回false
  	* 备注：
  	**************************************************/
      bool startDecodeByName(const string name);
  
      /**************************************************
  	* 接口：closeDecoderByName
  	* 功能：关闭解码器名称对应的解码

  	* 参数：const string name 解码器名称

  	* 返回：成功返回true，失败返回false
  	* 备注：
  	**************************************************/
      bool closeDecoderByName(const string name);
  
      /**************************************************
  	* 接口：closeAllDecoder
  	* 功能：关闭全部解码器
  	* 参数：void
  	* 返回：void
  	* 备注：
  	**************************************************/

      void closeAllDecoder();

  	/**************************************************
  	* 接口：closeAllDecoderByGpuid
  	* 功能：关闭某张显卡撒花姑娘的全部解码器
  	* 参数：const string gpuid gpu的id
  	* 返回：void
  	* 备注：
  	**************************************************/
      void closeAllDecoderByGpuid(const string gpuid);
  

      /**************************************************
  	* 接口：pauseDecoder
  	* 功能：暂停指定名称的解码器
  	* 参数：const string name 解码器名称
  	* 返回：成功返回true，失败返回false
  	* 备注：
  	**************************************************/
      bool pauseDecoder(const string name);
  
      /**************************************************
  	* 接口：pauseDecoder
  	* 功能：恢复指定名称的解码器
  	* 参数：const string name 解码器名称
  	* 返回：成功返回true，失败返回false
  	* 备注：
  	**************************************************/
      bool resumeDecoder(const string name);
  
      /**************************************************
  	* 接口：isSurport
  	* 功能：是否支持指定配置的解码
  	* 参数：FFDecConfig& cfg 解码器配置
  	* 返回：支持返回true，不支持返回false
  	* 备注：
  	**************************************************/

      bool isSurport(MgrDecConfig& config);

      /**************************************************
  	* 接口：isRunning
  	* 功能：根据解码器名称判断解码器是否正在运行
  	* 参数：const string name 解码器名称
  	* 返回：正在运行返回true，否则返回false
  	* 备注：
  	**************************************************/
      bool isRunning(const string name);
  

  	/**************************************************
  	* 接口：isFinished
  	* 功能：根据解码器名称判断解码器是否已经结束
  	* 参数：const string name 解码器名称
  	* 返回：正在运行返回true，否则返回false
  	* 备注：
  	**************************************************/
      bool isFinished(const string name);
  
  	/**************************************************
  	* 接口：isPausing
  	* 功能：根据解码器名称判断解码器是否暂停
  	* 参数：const string name 解码器名称
  	* 返回：正在运行返回true，否则返回false
  	* 备注：
  	**************************************************/
      bool isPausing(const string name);
  

      /**************************************************
  	* 接口：count
  	* 功能：获取正在运行的解码器数量
  	* 参数：void
  	* 返回：正在运行的解码器数量
  	* 备注：
  	**************************************************/

      int count();
  

  	/**************************************************
  	* 接口：setDecKeyframe
  	* 功能：设置是否只解码关键帧。默认全解
  	* 参数：const string name 解码器名称
  	*		bool bKeyframe   是否只解码关键帧。true，只解码关键帧；false,普通的全解码

  	* 返回：bool 成功返回true,失败返回false

  	* 备注：
  	**************************************************/
  	bool setDecKeyframe(const string name, bool bKeyframe);
  

  	/**************************************************
  	* 接口：getResolution
  	* 功能：获取视频分辨率
  	* 参数：const string name 解码器名称
  	*		int &width   从 width 返回视频宽度
  	*		int &height	 从 height 返回视频高度
  	* 返回：bool 成功获取返回true,失败返回false
  	* 备注：
  	**************************************************/
  	bool getResolution(const string name, int &width, int &height);
  
  	/**************************************************
  	* 接口：getAllDecodeName
  	* 功能：获取全部解码器名称
  	* 参数：void
  	* 返回：vector<string> 返回全部解码器名称
  	* 备注：
  	**************************************************/
  	vector<string> getAllDecodeName();
  
  	/**************************************************
  	* 接口：getCachedQueueLength
  	* 功能：获取解码缓冲队列当前长度
  	* 参数：const string name 解码器名称
  	* 返回：int 解码缓冲队列当前长度
  	* 备注：
  	**************************************************/
  	int getCachedQueueLength(const string name);
  
  	/**************************************************

  	* 接口：releaseFFImgInfo
  	* 功能：释放视频快照信息
  	* 参数：FFImgInfo* info 视频快照信息
  	* 返回：void
  	* 备注：
  	**************************************************/
  	void releaseFFImgInfo(FFImgInfo* info);
  

  	FFImgInfo* snapshot_in_task(const string name);
  
  	vector<FFImgInfo*> timing_snapshot_all();
  

  private:
      FFNvDecoderManager(){}

      
      void closeAllFinishedDecoder();

  
  private:

      map<string, AbstractDecoder*> decoderMap;

      mutex m_mutex;

  };