Merge branch 'dev-cmhu' into 'master'

Dev cmhu See merge request !2

Merge branch 'dev-cmhu' into 'master'
Dev cmhu See merge request !2
Hu Chunming
2 parents 58d98527 48330793
Showing 9 changed files with 296 additions and 202 deletions
src/FFNvDecoder.cpp
src/FFNvDecoder.h
src/FFNvDecoderManager.cpp
src/FFNvDecoderManager.h
src/NV12ToRGB.cu
src/NvJpegEncoder.cpp
src/NvJpegEncoder.h
src/cuda_kernels.h
src/main.cpp
@@ -38,6 +38,11 @@ FFNvDecoder::FFNvDecoder()
 	m_bPause = false;
 	m_bReal = true;
+
+	m_decode_thread = 0;
+	m_post_decode_thread = 0;
+
+	m_bFinished = false;
 }
 FFNvDecoder::~FFNvDecoder()
@@ -45,9 +50,11 @@ FFNvDecoder::~FFNvDecoder()
 }
-bool FFNvDecoder::init(const string& path)
+bool FFNvDecoder::init(FFDecConfig& cfg)
 {
-	fstream infile(path);
+	m_cfg = cfg;
+
+	fstream infile(cfg.uri);
 	if (infile.is_open()){
 		m_bReal = false;
 		infile.close();
@@ -55,24 +62,25 @@ bool FFNvDecoder::init(const string&amp; path)
 		m_bReal = true;
 	}
-	// 查找对应的硬件解码设备
-	const char* device_name = "cuda";
-	AVHWDeviceType hw_device_type = av_hwdevice_find_type_by_name(device_name);
-	if (hw_device_type == AV_HWDEVICE_TYPE_NONE) {
-		while ((hw_device_type = av_hwdevice_iterate_types(hw_device_type)) != AV_HWDEVICE_TYPE_NONE)
-			cout << av_hwdevice_get_type_name(hw_device_type);
-		return false;
-	}
+	return init(cfg.uri.c_str(), cfg.gpuid.c_str(),cfg.force_tcp);
+}
+
+bool FFNvDecoder::init(const char* uri, const char* gpuid, bool force_tcp)
+{
+	av_register_all();
+	avformat_network_init();
 	// 打开输入视频文件
 	AVDictionary *options = nullptr;
-	av_dict_set( &options, "bufsize", "1024000", 0 );
-	av_dict_set( &options, "rtsp_transport", "tcp", 0 );
-	av_dict_set( &options, "listen_timeout", "30", 0 ); // 单位为s
+	av_dict_set( &options, "bufsize", "655360", 0 );
+	av_dict_set( &options, "rtsp_transport", force_tcp ? "tcp" : "udp", 0 );
+	// av_dict_set( &options, "listen_timeout", "30", 0 ); // 单位为s
+	av_dict_set( &options, "stimeout", "30000000", 0 ); // 单位为 百万分之一秒
+	
 	fmt_ctx = avformat_alloc_context();
-	const char* input_file = path.c_str();
+	const char* input_file = uri;
 	if (avformat_open_input(&fmt_ctx, input_file, nullptr, &options) != 0) {
-		cout << "Cannot open input file" << input_file;
+		cout << "Cannot open input file: " << input_file;
 		return false;
 	}
@@ -90,24 +98,9 @@ bool FFNvDecoder::init(const string&amp; path)
 		return false;
 	}
-	// 判断硬件解码设备是否兼容视频解码器
-	for (int i = 0;; i++) {
-		const AVCodecHWConfig *config = avcodec_get_hw_config(decoder, i);
-		if (!config) {
-			cout << "Decoder does not support device hw_device_type"
-				<< decoder->name << av_hwdevice_get_type_name(hw_device_type);
-			return false;
-		}
-		// 得到硬件解码像素格式
-		if (config->methods & AV_CODEC_HW_CONFIG_METHOD_HW_DEVICE_CTX &&
-			config->device_type == hw_device_type) {
-			hw_pix_fmt = config->pix_fmt;
-			break;
-		}
-	}
-
-	// 得到解码器管理器
-	if (!(avctx = avcodec_alloc_context3(decoder)))
+	string cuvid_dec_name = string(decoder->name) + "_cuvid";
+	AVCodec *vcodec = avcodec_find_decoder_by_name(cuvid_dec_name.c_str());
+	if (!(avctx = avcodec_alloc_context3(vcodec)))
 		return (bool)AVERROR(ENOMEM);
 	// 得到视频流对象
@@ -119,16 +112,12 @@ bool FFNvDecoder::init(const string&amp; path)
 	// 设置解码器管理器的像素格式回调函数
 	avctx->get_format = get_hw_format;
-	// 初始化硬件解码器
-	AVBufferRef *hw_device_ctx = nullptr;
-	if (av_hwdevice_ctx_create(&hw_device_ctx, hw_device_type, nullptr, nullptr, 0) < 0) {
-		cout << "Failed to create specified HW device.";
-		return false;
-	}
-	avctx->hw_device_ctx = av_buffer_ref(hw_device_ctx);
+	hw_pix_fmt = AV_PIX_FMT_CUDA;
 	// 打开解码器流
-	if (avcodec_open2(avctx, decoder, nullptr) < 0) {
+	AVDictionary *op = nullptr;
+	av_dict_set( &op, "gpu", gpuid, 0 );
+	if (avcodec_open2(avctx, vcodec, &op) < 0) {
 		cout << "Failed to open codec for stream" << stream_index;
 		return false;
 	}
@@ -148,6 +137,13 @@ void FFNvDecoder::start(){
             return (void*)0;
         }
     ,this);
+}
+
+void FFNvDecoder::decode_thread()
+{
+	AVPacket* pkt ;
+	pkt = av_packet_alloc();
+	av_init_packet( pkt );
 	pthread_create(&m_post_decode_thread,0,
         [](void* arg)
@@ -157,13 +153,6 @@ void FFNvDecoder::start(){
             return (void*)0;
         }
     ,this);
-}
-
-void FFNvDecoder::decode_thread()
-{
-	AVPacket* pkt ;
-	pkt = av_packet_alloc();
-	av_init_packet( pkt );
 	while (m_bRunning)
 	{
@@ -210,7 +199,7 @@ void FFNvDecoder::decode_thread()
 			result = avcodec_send_packet(avctx, pkt);
 			if (result < 0)
 			{
-				cout << "Failed to send pkt" << result << endl;
+				cout << "Failed to send pkt:" << result << endl;
 				continue;
 			}
@@ -222,24 +211,41 @@ void FFNvDecoder::decode_thread()
 			}
 			mFrameQueue.addTail();
-
-			// if (gpuFrame->format == hw_pix_fmt) 
-			// {
-			// 	/* retrieve data from GPU to CPU */
-			// 	if (av_hwframe_transfer_data(pong_frame, gpuFrame, 0) < 0) 
-			// 	{
-			// 		cout << "Failed to transfer the data to system memory";
-			// 		return;
-			// 	}
-			// }
 		}
 		av_packet_unref(pkt);
 	}
 	m_bRunning = false;
+
+	if (m_post_decode_thread != 0)
+	{
+		pthread_join(m_post_decode_thread,0);
+	}
+
+	decode_finished();
+
 	cout << "decode thread exited." << endl;
 }
+void FFNvDecoder::decode_finished()
+{
+	if (avctx)
+	{
+		if (avctx->hw_device_ctx)
+		{
+			av_buffer_unref(&avctx->hw_device_ctx);
+		}
+		avcodec_free_context(&avctx);
+	}
+	
+	if (fmt_ctx)
+	{
+		avformat_close_input(&fmt_ctx);
+	}
+
+	m_bFinished = true;
+}
+
 void FFNvDecoder::post_decode_thread()
 {
 	 while (m_bRunning)
@@ -262,21 +268,8 @@ void FFNvDecoder::post_decode_thread()
 void FFNvDecoder::close()
 {
 	m_bRunning=false;
-    pthread_join(m_decode_thread,0);
-	pthread_join(m_post_decode_thread,0);
-
-	if (avctx)
-	{
-		if (avctx->hw_device_ctx)
-		{
-			av_buffer_unref(&avctx->hw_device_ctx);
-		}
-		avcodec_free_context(&avctx);
-	}
-	
-	if (fmt_ctx)
-	{
-		avformat_close_input(&fmt_ctx);
+	if(m_decode_thread != 0){
+		pthread_join(m_decode_thread,0);
 	}
 }
@@ -298,6 +291,11 @@ bool FFNvDecoder::isRunning()
 	return m_bRunning;
 }
+bool FFNvDecoder::isFinished()
+{
+	return m_bFinished;
+}
+
 bool FFNvDecoder::getResolution( int &width, int &height )
 {
 	if (avctx != nullptr)
@@ -20,7 +20,7 @@ using namespace std;
 * 接口：DXDECODER_CALLBACK
 * 功能：解码数据回调接口
 * 参数：const dx_void * userPtr 用户自定义数据
-*       AVFrame * gpuFrame 解码结果帧数据
+*       AVFrame * gpuFrame 解码结果帧数据，在设置的对应的gpu上，要十分注意这一点，尤其是多线程情况
 * 返回：无
 * 备注：当解码库数据源为实时流时(RTSP/GB28181)，本接
 *       口内不可进行阻塞/耗时操作。当解码库数据源为
@@ -29,17 +29,26 @@ using namespace std;
 **************************************************/
 typedef void(*POST_DECODE_CALLBACK)(const void * userPtr, AVFrame * gpuFrame);
+struct FFDecConfig
+{
+    string uri;
+    POST_DECODE_CALLBACK post_decoded_cbk;
+    string gpuid;
+    bool force_tcp{true};
+};
+
 class FFNvDecoder{
 public:
     FFNvDecoder();
     ~FFNvDecoder();
-    bool init(const string& path);
+    bool init(FFDecConfig& cfg);
     void close();
     void start();
     void pause();
     void resume();
     bool isRunning();
+    bool isFinished();
     bool getResolution( int &width, int &height );
     void setName(string nm);
@@ -51,10 +60,13 @@ public:
 private:
     void decode_thread();
     void post_decode_thread();
+    bool init(const char* uri, const char* gpuid, bool force_tcp);
+    void decode_finished();
 public:
     POST_DECODE_CALLBACK post_decoded_cbk;
     const void * m_userPtr;
+    FFDecConfig m_cfg;
 private:
     AVStream* stream;
@@ -67,6 +79,7 @@ private:
     pthread_t m_post_decode_thread;
     bool m_bRunning;
+    bool m_bFinished;
     string name;
     bool m_bPause;
@@ -3,8 +3,12 @@
 using namespace std;
-FFNvDecoder* FFNvDecoderManager::createDecoder(const char * uri, POST_DECODE_CALLBACK post_decoded_cbk, string name){
-    int num = decoderMap.count(name);
+
+FFNvDecoder* FFNvDecoderManager::createDecoder(MgrDecConfig& config){
+
+    closeAllFinishedDecoder();
+
+    int num = decoderMap.count(config.name);
     if (num > 0)
     {
         cout << "已存在name所标记的解码器" << endl;
@@ -17,12 +21,12 @@ FFNvDecoder* FFNvDecoderManager::createDecoder(const char * uri, POST_DECODE_CAL
         return nullptr;
     }
-    bool bRet= dec->init(uri);
+    bool bRet= dec->init(config.cfg);
     if (bRet)
     {
-        dec->setName(name) ;
-        dec->post_decoded_cbk = post_decoded_cbk;
-        decoderMap[name] = dec;
+        dec->setName(config.name) ;
+        dec->post_decoded_cbk = config.cfg.post_decoded_cbk;
+        decoderMap[config.name] = dec;
         return dec;
     }
@@ -126,11 +130,26 @@ void FFNvDecoderManager::closeAllDecoder()
 {
     for(auto iter = decoderMap.begin(); iter != decoderMap.end(); iter++){
         iter->second->close();
+        delete iter->second;
+    }
+    decoderMap.clear();
+}
+
+void FFNvDecoderManager::closeAllFinishedDecoder()
+{
+     for(auto iter = decoderMap.begin(); iter != decoderMap.end(); iter++){
+        if (iter->second->isFinished())
+        {
+            delete iter->second;
+            decoderMap.erase(iter);
+        }
     }
 }
 int FFNvDecoderManager::count()
 {
+    closeAllFinishedDecoder();
+
     return decoderMap.size();
 }
@@ -172,10 +191,27 @@ bool FFNvDecoderManager::resumeDecoder(string name)
     return false;
 }
-bool FFNvDecoderManager::isSurport(const char* uri)
+bool FFNvDecoderManager::isSurport(FFDecConfig& cfg)
 {
     FFNvDecoder dec;
-    bool bRet = dec.init(uri);
+    bool bRet = dec.init(cfg);
     dec.close();
     return bRet;
+}
+
+bool FFNvDecoderManager::isRunning(string name){
+     if (name.empty())
+    {
+        cout << "name 为空!"<< endl;
+        return false;
+    }
+
+    auto dec = decoderMap.find(name);
+    if (dec != decoderMap.end())
+    {
+        return dec->second->isRunning();
+    }
+    
+    cout << "没有找到name为" << name << "的解码器!" << endl;
+    return false;
 }
 \ No newline at end of file
@@ -5,6 +5,12 @@
 using namespace std;
+struct MgrDecConfig
+{
+    FFDecConfig cfg;
+    string name{""};
+};
+
 class FFNvDecoderManager{
 public:
     static FFNvDecoderManager* getInstance(){
@@ -20,7 +26,7 @@ public:
         closeAllDecoder();
     }
-    FFNvDecoder* createDecoder(const char * uri, POST_DECODE_CALLBACK post_decoded_cbk, string name = "");
+    FFNvDecoder* createDecoder(MgrDecConfig& config);
     void setUserPtr(string name, const void * userPtr);
     FFNvDecoder* getDecoderByName(const string name);
@@ -31,11 +37,16 @@ public:
     bool closeDecoderByName(string name);
     void closeAllDecoder();
+    void closeAllFinishedDecoder();
+
+    bool removeDecoderByName(string name);
     bool pauseDecoder(string name);
     bool resumeDecoder(string name);
-    bool isSurport(const char* uri);
+    bool isSurport(FFDecConfig& cfg);
+
+    bool isRunning(string name);
     int count();
@@ -301,26 +301,6 @@ namespace cuda_common
 			fprintf(stderr, "cudaMemcpyToSymbol failed: %s\n", cudaGetErrorString(cudaStatus));
 		}
-		//showHueCSC();
-
-		return cudaStatus;
-	}
-
-	cudaError_t showHueCSC3()
-	{
- /*
-		cudaError_t cudaStatus;
-
-		float data[9];
-		cudaStatus = cudaMemcpyFromSymbol(data, constHueColorSpaceMat2, sizeof(data));
-		printf("hueCSC  ");
-		for (int i = 0; i < 9; i++)
-		{
-			printf("%f  ,", data[i]);
-		}
-		printf("\n");
-*/
-		cudaError_t cudaStatus;
 		return cudaStatus;
 	}
 #include "NvJpegEncoder.h"
-#include <nvjpeg.h>
-
 #include <fstream>
 #include <vector>
 #include <iostream>
+#define CHECK_NVJPEG(S) do {nvjpegStatus_t  status; \
+        status = S; \
+        if (status != NVJPEG_STATUS_SUCCESS ) std::cout << __LINE__ <<" CHECK_NVJPEG - status = " << status << std::endl; \
+        } while (false)
+
 int saveJpeg(const char * filepath, unsigned char* d_srcBGR, int width, int height)
 {
-	nvjpegHandle_t nvjpeg_handle;
-	nvjpegEncoderState_t encoder_state;
-	nvjpegEncoderParams_t encoder_params;
-
-	cudaEvent_t ev_start, ev_end;
-	cudaEventCreate(&ev_start);
-	cudaEventCreate(&ev_end);
-	
-
-	nvjpegImage_t input;
-	nvjpegInputFormat_t input_format = NVJPEG_INPUT_BGRI;
-	int image_width = width;
-	int image_height = height;
-
-	// int channel_size = image_width * image_height;
-	// for (int i = 0; i < 3; i++)
-	// {
-	// 	input.pitch[i] = image_width;
-	// 	(cudaMalloc((void**)&(input.channel[i]), channel_size));
-	// 	(cudaMemset(input.channel[i], 50 * 40 * i, channel_size));
-	// }
+    nvjpegHandle_t nvjpeg_handle;
+    nvjpegEncoderState_t encoder_state;
+    nvjpegEncoderParams_t encoder_params;
+
+    cudaEvent_t ev_start, ev_end;
+    cudaEventCreate(&ev_start);
+    cudaEventCreate(&ev_end);
+
+    nvjpegImage_t input;
+    nvjpegInputFormat_t input_format = NVJPEG_INPUT_BGRI;
+    int image_width = width;
+    int image_height = height;
+
+    // int channel_size = image_width * image_height;
+    // for (int i = 0; i < 3; i++)
+    // {
+    //     input.pitch[i] = image_width;
+    //     (cudaMalloc((void**)&(input.channel[i]), channel_size));
+    //     (cudaMemset(input.channel[i], 50 * 40 * i, channel_size));
+    // }
     input.channel[0] = d_srcBGR;
     input.pitch[0] = image_width * 3;
-	nvjpegBackend_t backend = NVJPEG_BACKEND_DEFAULT;
-
-	nvjpegCreate(backend, nullptr, &nvjpeg_handle);
-	
-	nvjpegEncoderParamsCreate(nvjpeg_handle, &encoder_params, NULL);
-	nvjpegEncoderStateCreate(nvjpeg_handle, &encoder_state, NULL);
-
-	// set params
-	nvjpegEncoderParamsSetEncoding(encoder_params, nvjpegJpegEncoding_t::NVJPEG_ENCODING_PROGRESSIVE_DCT_HUFFMAN, NULL);
-	nvjpegEncoderParamsSetOptimizedHuffman(encoder_params, 1, NULL);
-	nvjpegEncoderParamsSetQuality(encoder_params, 70, NULL);
-	nvjpegEncoderParamsSetSamplingFactors(encoder_params, nvjpegChromaSubsampling_t::NVJPEG_CSS_420, NULL);
-
-	cudaEventRecord(ev_start);
-	nvjpegEncodeImage(nvjpeg_handle, encoder_state, encoder_params, &input, input_format, image_width, image_height, NULL);
-	cudaEventRecord(ev_end);
-
-	std::vector<unsigned char> obuffer;
-	size_t length;
-	nvjpegEncodeRetrieveBitstream(
-		nvjpeg_handle,
-		encoder_state,
-		NULL,
-		&length,
-		NULL);
-
-	obuffer.resize(length);
-	nvjpegEncodeRetrieveBitstream(
-		nvjpeg_handle,
-		encoder_state,
-		obuffer.data(),
-		&length,
-		NULL);
-
-	std::ofstream outputFile(filepath, std::ios::out | std::ios::binary);
-	outputFile.write(reinterpret_cast<const char *>(obuffer.data()), static_cast<int>(length));
-
-	cudaEventSynchronize(ev_end);
-
-	float ms;
-	cudaEventElapsedTime(&ms, ev_start, ev_end);
-	std::cout << "time spend " << ms << " ms" << std::endl;
-	
-	return 0;
+    nvjpegBackend_t backend = NVJPEG_BACKEND_DEFAULT;
+
+    CHECK_NVJPEG(nvjpegCreate(backend, nullptr, &nvjpeg_handle));
+    
+    CHECK_NVJPEG(nvjpegEncoderParamsCreate(nvjpeg_handle, &encoder_params, NULL));
+    CHECK_NVJPEG(nvjpegEncoderStateCreate(nvjpeg_handle, &encoder_state, NULL));
+
+    // set params
+    CHECK_NVJPEG(nvjpegEncoderParamsSetEncoding(encoder_params, nvjpegJpegEncoding_t::NVJPEG_ENCODING_PROGRESSIVE_DCT_HUFFMAN, NULL));
+    CHECK_NVJPEG(nvjpegEncoderParamsSetOptimizedHuffman(encoder_params, 1, NULL));
+    CHECK_NVJPEG(nvjpegEncoderParamsSetQuality(encoder_params, 70, NULL));
+    CHECK_NVJPEG(nvjpegEncoderParamsSetSamplingFactors(encoder_params, nvjpegChromaSubsampling_t::NVJPEG_CSS_420, NULL));
+
+    cudaEventRecord(ev_start);
+    CHECK_NVJPEG(nvjpegEncodeImage(nvjpeg_handle, encoder_state, encoder_params, &input, input_format, image_width, image_height, NULL));
+    cudaEventRecord(ev_end);
+
+    std::vector<unsigned char> obuffer;
+    size_t length;
+    CHECK_NVJPEG(nvjpegEncodeRetrieveBitstream(
+        nvjpeg_handle,
+        encoder_state,
+        NULL,
+        &length,
+        NULL));
+
+    obuffer.resize(length);
+    CHECK_NVJPEG(nvjpegEncodeRetrieveBitstream(
+        nvjpeg_handle,
+        encoder_state,
+        obuffer.data(),
+        &length,
+        NULL));
+
+    cudaEventSynchronize(ev_end);
+
+    // 用完销毁，避免显存泄露
+    nvjpegEncoderParamsDestroy(encoder_params);
+    nvjpegEncoderStateDestroy(encoder_state);
+    nvjpegDestroy(nvjpeg_handle);
+
+    float ms;
+    cudaEventElapsedTime(&ms, ev_start, ev_end);
+    std::cout << "time spend " << ms << " ms" << std::endl;
+
+    std::ofstream outputFile(filepath, std::ios::out | std::ios::binary);
+    outputFile.write(reinterpret_cast<const char *>(obuffer.data()), static_cast<int>(length));
+    outputFile.close();
+    
+    return 0;
 }
 \ No newline at end of file
-
+#include <nvjpeg.h>
 int saveJpeg(const char * filepath, unsigned char* d_srcBGR, int width, int height);
 \ No newline at end of file
@@ -19,7 +19,6 @@ typedef enum
 namespace cuda_common
 {
 	cudaError_t setColorSpace2(e_ColorSpace CSC, float hue);
-	cudaError_t showHueCSC3();
 	cudaError_t NV12ToRGBnot(CUdeviceptr d_srcNV12, size_t nSourcePitch, unsigned char* d_dstRGB, int width, int height);
 	cudaError_t CUDAToBGR(CUdeviceptr dataY, CUdeviceptr dataUV, size_t pitchY, size_t pitchUV, unsigned char* d_dstRGB, int width, int height);
@@ -7,54 +7,103 @@
 unsigned char *pHwRgb = nullptr;
-
+/**
+ * 注意： gpuFrame 在解码器设置的显卡上，后续操作要十分注意这一点，尤其是多线程情况
+ * */
 void postDecoded(const void * userPtr, AVFrame * gpuFrame){
     FFNvDecoder* decoder = (FFNvDecoder*)userPtr;
     if (decoder!= nullptr)
     {
-        cout << decoder->getName() << endl;
-        const char* gpu_pixfmt = av_get_pix_fmt_name((AVPixelFormat)gpuFrame->format);
-        cout << "pixfmt: " << gpu_pixfmt << endl;
+        cout << "decode name: " << decoder->getName() << endl;
+        // const char* gpu_pixfmt = av_get_pix_fmt_name((AVPixelFormat)gpuFrame->format);
+        // cout << "pixfmt: " << gpu_pixfmt << endl;
         cout << "keyframe: " << gpuFrame->key_frame << " width: " << gpuFrame->width << " height: "<< gpuFrame->height << endl;
-        cout << "decode successed ✿✿ヽ(°▽°)ノ✿ " << endl;
+        // cout << "decode successed ✿✿ヽ(°▽°)ノ✿ " << endl;
         if (gpuFrame->format == AV_PIX_FMT_CUDA)
-        {
-             cudaError_t cudaStatus;
-            if(pHwRgb == nullptr){
-                cudaStatus = cudaMalloc((void **)&pHwRgb, 3 * gpuFrame->width * gpuFrame->height * sizeof(unsigned char));
-            }
-            cudaStatus = cuda_common::CUDAToBGR((CUdeviceptr)gpuFrame->data[0],(CUdeviceptr)gpuFrame->data[1], gpuFrame->linesize[0], gpuFrame->linesize[1], pHwRgb, gpuFrame->width, gpuFrame->height);
-            cudaDeviceSynchronize();
-            if (cudaStatus != cudaSuccess) {
-                cout << "CUDAToBGR failed !!!" << endl;
-                return;
-            }
-            saveJpeg("/home/cmhu/FFNvDecoder/a.jpg", pHwRgb, gpuFrame->width, gpuFrame->height);  // 验证 CUDAToRGB 
+        {   
+            cout << "gpuid = " << atoi(decoder->m_cfg.gpuid.c_str()) << endl;
+            // cudaSetDevice(atoi(decoder->m_cfg.gpuid.c_str()));
+            // cudaError_t cudaStatus;
+            // if(pHwRgb == nullptr){
+            //     cuda_common::setColorSpace2( ITU709, 0 );
+            //     cudaStatus = cudaMalloc((void **)&pHwRgb, 3 * gpuFrame->width * gpuFrame->height * sizeof(unsigned char));
+            // }
+            // cudaStatus = cuda_common::CUDAToBGR((CUdeviceptr)gpuFrame->data[0],(CUdeviceptr)gpuFrame->data[1], gpuFrame->linesize[0], gpuFrame->linesize[1], pHwRgb, gpuFrame->width, gpuFrame->height);
+            // cudaDeviceSynchronize();
+            // if (cudaStatus != cudaSuccess) {
+            //     cout << "CUDAToBGR failed !!!" << endl;
+            //     return;
+            // }
+
+            // string path = "/home/cmhu/FFNvDecoder/" + decoder->m_cfg.gpuid + ".jpg";
+            // saveJpeg(path.c_str(), pHwRgb, gpuFrame->width, gpuFrame->height);  // 验证 CUDAToRGB 
         }
     }
 }
-int main(){
+void createDecode(int index){
+    FFNvDecoderManager* pDecManager = FFNvDecoderManager::getInstance();
+    MgrDecConfig config;
+    config.name = "dec" + to_string(index);
+    config.cfg.uri = "rtsp://176.10.0.4:8554/stream";
+    config.cfg.post_decoded_cbk = postDecoded;
+    config.cfg.force_tcp = true;
+    config.cfg.gpuid = "1";
+    FFNvDecoder* decoder = pDecManager->createDecoder(config);
+    if (!decoder)
+    {
+        return ;
+    }
+    pDecManager->setUserPtr(config.name, decoder);
+    pDecManager->startDecodeByName(config.name);
+}
-    cuda_common::setColorSpace2( ITU709, 0 );
+int main(){
     FFNvDecoderManager* pDecManager = FFNvDecoderManager::getInstance();
-    // FFNvDecoder* decoder = pDecManager->createDecoder("rtmp://192.168.10.56:1935/objecteye/1",postDecoded,"dec1");
-    // if (!decoder)
+
+    // for (size_t i = 0; i < 20; i++)
     // {
-    //     return 1;
+    //     createDecode(i);
     // }
-    // pDecManager->setUserPtr("dec1", decoder);
-    // pDecManager->startDecodeByName("dec1");
-    FFNvDecoder* dec2 = pDecManager->createDecoder("/home/cmhu/data/duan1.avi",postDecoded,"dec2");
+    MgrDecConfig config;
+    config.name = "dec2";
+    config.cfg.uri = "/home/cmhu/data/test.mp4";
+     config.cfg.post_decoded_cbk = postDecoded;
+    config.cfg.force_tcp = true;
+    config.cfg.gpuid = "2";
+    FFNvDecoder* dec2 = pDecManager->createDecoder(config);
     if (!dec2)
     {
         return 1;
     }
-    pDecManager->setUserPtr("dec2", dec2);
-    pDecManager->startDecodeByName("dec2");
+    pDecManager->setUserPtr(config.name, dec2);
+    pDecManager->startDecodeByName(config.name);
+
+    
+    // config.name = "dec0";
+    // config.cfg.uri = "rtmp://192.168.10.56:1935/objecteye/1";
+    // config.cfg.gpuid = "0";
+    // FFNvDecoder* dec0 = pDecManager->createDecoder(config);
+    // if (!dec0)
+    // {
+    //     return 1;
+    // }
+    // pDecManager->setUserPtr(config.name, dec0);
+    // pDecManager->startDecodeByName(config.name);
+
+    // config.name = "dec01";
+    // config.cfg.uri = "rtmp://192.168.10.56:1935/objecteye/1";
+    // config.cfg.gpuid = "0";
+    // FFNvDecoder* dec01 = pDecManager->createDecoder(config);
+    // if (!dec01)
+    // {
+    //     return 1;
+    // }
+    // pDecManager->setUserPtr(config.name, dec01);
+    // pDecManager->startDecodeByName(config.name);
     // while (getchar() != 'q');