Blame view

src/FFNvDecoderManager.cpp 4.74 KB
aac5773f   hucm   功能基本完成,接口待打磨
1
2
3
4
5
  #include "FFNvDecoderManager.h"
  #include<iostream>
  
  using namespace std;
  
48330793   Hu Chunming   修正解码线程自然结束时解码器内存没...
6
  
7319ea36   Hu Chunming   多显卡设置
7
  FFNvDecoder* FFNvDecoderManager::createDecoder(MgrDecConfig& config){
48330793   Hu Chunming   修正解码线程自然结束时解码器内存没...
8
9
10
  
      closeAllFinishedDecoder();
  
7319ea36   Hu Chunming   多显卡设置
11
      int num = decoderMap.count(config.name);
aac5773f   hucm   功能基本完成,接口待打磨
12
13
14
15
16
17
18
19
20
21
22
23
      if (num > 0)
      {
          cout << "已存在name所标记的解码器" << endl;
          return nullptr;
      }
      
      FFNvDecoder* dec = new FFNvDecoder();
      if (dec == nullptr)
      {
          return nullptr;
      }
      
e96e6489   Hu Chunming   优化代码;添加isRunning函数
24
      bool bRet= dec->init(config.cfg);
aac5773f   hucm   功能基本完成,接口待打磨
25
26
      if (bRet)
      {
7319ea36   Hu Chunming   多显卡设置
27
28
29
          dec->setName(config.name) ;
          dec->post_decoded_cbk = config.cfg.post_decoded_cbk;
          decoderMap[config.name] = dec;
aac5773f   hucm   功能基本完成,接口待打磨
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
          return dec;
      }
      
      // 创建失败,关闭解码器
      dec->close();
      delete dec;
  
      return nullptr;
  }
  
  void FFNvDecoderManager::setUserPtr(string name, const void * userPtr)
  {
      if (name.empty())
      {
          cout << "name 为空!"<< endl;
          return ;
      }
  
      auto dec = decoderMap.find(name);
      if (dec != decoderMap.end())
      {
          dec->second->m_userPtr = userPtr;
          return;
      }
  
      cout << "没有找到name为" << name << "的解码器!" << endl;
      return ;
  }
  
  FFNvDecoder* FFNvDecoderManager::getDecoderByName(const string name)
  {
      if (name.empty())
      {
          cout << "name 为空!"<< endl;
          return nullptr;
      }
      
      auto dec = decoderMap.find(name);
      if (dec != decoderMap.end())
      {
          return dec->second;
      }
  
      cout << "没有找到name为" << name << "的解码器!" << endl;
      return nullptr;
  }
  
  void FFNvDecoderManager::startDecode(FFNvDecoder* dec){
      if (dec != nullptr && !dec->isRunning())
      {
          dec->start();
      }
  }
  
  bool FFNvDecoderManager::startDecodeByName(string name){
       if (name.empty())
      {
          cout << "name 为空!"<< endl;
          return false;
      }
  
      auto dec = decoderMap.find(name);
      if (dec != decoderMap.end())
      {
          dec->second->start();
          return true;
      }
  
      cout << "没有找到name为" << name << "的解码器!" << endl;
      return false;
  }
  
  void FFNvDecoderManager::startAllDecode(){
      for(auto iter = decoderMap.begin(); iter != decoderMap.end(); iter++){
          iter->second->start();
      }
  }
  
  bool FFNvDecoderManager::closeDecoderByName(string name){
      if (name.empty())
      {
          cout << "name 为空!"<< endl;
          return false;
      }
  
f40cc409   Hu Chunming   优化显存占用。当前在3080显卡上...
115
      m_mutex_erase.lock();
aac5773f   hucm   功能基本完成,接口待打磨
116
117
118
119
120
      auto dec = decoderMap.find(name);
      if (dec != decoderMap.end())
      {
          dec->second->close();
          delete dec->second;
f40cc409   Hu Chunming   优化显存占用。当前在3080显卡上...
121
          dec->second = nullptr;
aac5773f   hucm   功能基本完成,接口待打磨
122
          decoderMap.erase(dec);
f40cc409   Hu Chunming   优化显存占用。当前在3080显卡上...
123
124
  
          m_mutex_erase.unlock();
aac5773f   hucm   功能基本完成,接口待打磨
125
126
127
          return true;
      }
      
f40cc409   Hu Chunming   优化显存占用。当前在3080显卡上...
128
      m_mutex_erase.unlock();
aac5773f   hucm   功能基本完成,接口待打磨
129
130
131
132
133
134
      cout << "没有找到name为" << name << "的解码器!" << endl;
      return false;
  }
  
  void FFNvDecoderManager::closeAllDecoder()
  {
f40cc409   Hu Chunming   优化显存占用。当前在3080显卡上...
135
      m_mutex_erase.lock();
aac5773f   hucm   功能基本完成,接口待打磨
136
137
      for(auto iter = decoderMap.begin(); iter != decoderMap.end(); iter++){
          iter->second->close();
48330793   Hu Chunming   修正解码线程自然结束时解码器内存没...
138
          delete iter->second;
f40cc409   Hu Chunming   优化显存占用。当前在3080显卡上...
139
          iter->second = nullptr;
48330793   Hu Chunming   修正解码线程自然结束时解码器内存没...
140
141
      }
      decoderMap.clear();
f40cc409   Hu Chunming   优化显存占用。当前在3080显卡上...
142
      m_mutex_erase.unlock();
48330793   Hu Chunming   修正解码线程自然结束时解码器内存没...
143
144
145
146
  }
  
  void FFNvDecoderManager::closeAllFinishedDecoder()
  {
f40cc409   Hu Chunming   优化显存占用。当前在3080显卡上...
147
148
      m_mutex_erase.lock();
       for(auto iter = decoderMap.begin(); iter != decoderMap.end(); ){
48330793   Hu Chunming   修正解码线程自然结束时解码器内存没...
149
150
151
          if (iter->second->isFinished())
          {
              delete iter->second;
f40cc409   Hu Chunming   优化显存占用。当前在3080显卡上...
152
153
154
155
156
157
              iter->second = nullptr;
              iter = decoderMap.erase(iter);
          }
          else
          {
              iter++ ;
48330793   Hu Chunming   修正解码线程自然结束时解码器内存没...
158
          }
aac5773f   hucm   功能基本完成,接口待打磨
159
      }
f40cc409   Hu Chunming   优化显存占用。当前在3080显卡上...
160
      m_mutex_erase.unlock();
aac5773f   hucm   功能基本完成,接口待打磨
161
162
163
164
  }
  
  int FFNvDecoderManager::count()
  {
48330793   Hu Chunming   修正解码线程自然结束时解码器内存没...
165
166
      closeAllFinishedDecoder();
  
aac5773f   hucm   功能基本完成,接口待打磨
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
      return decoderMap.size();
  }
  
  bool FFNvDecoderManager::pauseDecoder(string name)
  {
      if (name.empty())
      {
          cout << "name 为空!"<< endl;
          return false;
      }
  
      auto dec = decoderMap.find(name);
      if (dec != decoderMap.end())
      {
          dec->second->pause();
          return true;
      }
      
      cout << "没有找到name为" << name << "的解码器!" << endl;
      return false;
  }
  
  bool FFNvDecoderManager::resumeDecoder(string name)
  {
      if (name.empty())
      {
          cout << "name 为空!"<< endl;
          return false;
      }
  
      auto dec = decoderMap.find(name);
      if (dec != decoderMap.end())
      {
          dec->second->resume();
          return true;
      }
      
      cout << "没有找到name为" << name << "的解码器!" << endl;
      return false;
  }
  
e96e6489   Hu Chunming   优化代码;添加isRunning函数
208
  bool FFNvDecoderManager::isSurport(FFDecConfig& cfg)
aac5773f   hucm   功能基本完成,接口待打磨
209
210
  {
      FFNvDecoder dec;
e96e6489   Hu Chunming   优化代码;添加isRunning函数
211
      bool bRet = dec.init(cfg);
aac5773f   hucm   功能基本完成,接口待打磨
212
213
      dec.close();
      return bRet;
e96e6489   Hu Chunming   优化代码;添加isRunning函数
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
  }
  
  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;
aac5773f   hucm   功能基本完成,接口待打磨
231
  }