Hu Chunming / FFNvDecoder

Blame view

src/main.cpp 3.65 KB
Edit Raw Normal View History
aac5773f   hucm   功能基本完成，接口待打磨 
1
2
3
 
  #include "FFNvDecoderManager.h"
  #include <iostream>
e41a52bb   Hu Chunming   1.优化数据读取线程；2. 添加A... 
4
5
6
7
8
9
 
  #include "cuda_kernels.h"
  
  #include "NvJpegEncoder.h"
  
  unsigned char *pHwRgb = nullptr;
0b43216c   Hu Chunming   添加重要注释 
10
11
12
 
  /**
   * 注意： gpuFrame 在解码器设置的显卡上，后续操作要十分注意这一点，尤其是多线程情况
   * */
aac5773f   hucm   功能基本完成，接口待打磨 
13
14
15
16
17
18
19
20
21
 
  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 << "keyframe: " << gpuFrame->key_frame << " width: " << gpuFrame->width << " height: "<< gpuFrame->height << endl;
          cout << "decode successed ✿✿ヽ(°▽°)ノ✿ " << endl;
e41a52bb   Hu Chunming   1.优化数据读取线程；2. 添加A... 
22
23
 
  
          if (gpuFrame->format == AV_PIX_FMT_CUDA)
7319ea36   Hu Chunming   多显卡设置 
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
 
          {   
              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
e41a52bb   Hu Chunming   1.优化数据读取线程；2. 添加A... 
41
 
          }
aac5773f   hucm   功能基本完成，接口待打磨 
42
43
44
45
 
      }
  }
  
  int main(){
e41a52bb   Hu Chunming   1.优化数据读取线程；2. 添加A... 
46
aac5773f   hucm   功能基本完成，接口待打磨 
47
 
      FFNvDecoderManager* pDecManager = FFNvDecoderManager::getInstance();
7319ea36   Hu Chunming   多显卡设置 
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
 
      MgrDecConfig config;
      config.name = "dec1";
      config.cfg.uri = "rtmp://192.168.10.56:1935/objecteye/1";
      config.cfg.post_decoded_cbk = postDecoded;
      config.cfg.force_tcp = true;
      config.cfg.gpuid = "1";
      FFNvDecoder* decoder = pDecManager->createDecoder(config);
      if (!decoder)
      {
          return 1;
      }
      pDecManager->setUserPtr(config.name, decoder);
      pDecManager->startDecodeByName(config.name);
  
      config.name = "dec2";
      config.cfg.uri = "rtmp://192.168.10.56:1935/objecteye/1";
      config.cfg.gpuid = "2";
      FFNvDecoder* dec2 = pDecManager->createDecoder(config);
aac5773f   hucm   功能基本完成，接口待打磨 
66
67
68
69
 
      if (!dec2)
      {
          return 1;
      }
7319ea36   Hu Chunming   多显卡设置 
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
 
      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);
aac5773f   hucm   功能基本完成，接口待打磨 
95
e41a52bb   Hu Chunming   1.优化数据读取线程；2. 添加A... 
96
 
      // while (getchar() != 'q');
aac5773f   hucm   功能基本完成，接口待打磨 
97
e41a52bb   Hu Chunming   1.优化数据读取线程；2. 添加A... 
98
99
100
 
      // // pDecManager->closeDecoderByName("dec1");
      // // pDecManager->pauseDecoder("dec1");
      // pDecManager->pauseDecoder("dec2");
aac5773f   hucm   功能基本完成，接口待打磨 
101
e41a52bb   Hu Chunming   1.优化数据读取线程；2. 添加A... 
102
 
      // while (getchar() != 'q');
aac5773f   hucm   功能基本完成，接口待打磨 
103
e41a52bb   Hu Chunming   1.优化数据读取线程；2. 添加A... 
104
105
 
      // // pDecManager->resumeDecoder("dec1");
      // pDecManager->resumeDecoder("dec2");
aac5773f   hucm   功能基本完成，接口待打磨 
106
107
108
109
110
 
  
      while (getchar() != 'q');
  
      pDecManager->closeAllDecoder();
  }