Blame view

src/FFNvDecoderManager.cpp 4.03 KB
aac5773f   hucm   功能基本完成,接口待打磨
1
2
3
4
5
  #include "FFNvDecoderManager.h"
  #include<iostream>
  
  using namespace std;
  
7319ea36   Hu Chunming   多显卡设置
6
7
  FFNvDecoder* FFNvDecoderManager::createDecoder(MgrDecConfig& config){
      int num = decoderMap.count(config.name);
aac5773f   hucm   功能基本完成,接口待打磨
8
9
10
11
12
13
14
15
16
17
18
19
      if (num > 0)
      {
          cout << "已存在name所标记的解码器" << endl;
          return nullptr;
      }
      
      FFNvDecoder* dec = new FFNvDecoder();
      if (dec == nullptr)
      {
          return nullptr;
      }
      
e96e6489   Hu Chunming   优化代码;添加isRunning函数
20
      bool bRet= dec->init(config.cfg);
aac5773f   hucm   功能基本完成,接口待打磨
21
22
      if (bRet)
      {
7319ea36   Hu Chunming   多显卡设置
23
24
25
          dec->setName(config.name) ;
          dec->post_decoded_cbk = config.cfg.post_decoded_cbk;
          decoderMap[config.name] = dec;
aac5773f   hucm   功能基本完成,接口待打磨
26
27
28
29
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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
          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;
      }
  
      auto dec = decoderMap.find(name);
      if (dec != decoderMap.end())
      {
          dec->second->close();
          delete dec->second;
          decoderMap.erase(dec);
          return true;
      }
      
  
      cout << "没有找到name为" << name << "的解码器!" << endl;
      return false;
  }
  
  void FFNvDecoderManager::closeAllDecoder()
  {
      for(auto iter = decoderMap.begin(); iter != decoderMap.end(); iter++){
          iter->second->close();
      }
  }
  
  int FFNvDecoderManager::count()
  {
      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函数
175
  bool FFNvDecoderManager::isSurport(FFDecConfig& cfg)
aac5773f   hucm   功能基本完成,接口待打磨
176
177
  {
      FFNvDecoder dec;
e96e6489   Hu Chunming   优化代码;添加isRunning函数
178
      bool bRet = dec.init(cfg);
aac5773f   hucm   功能基本完成,接口待打磨
179
180
      dec.close();
      return bRet;
e96e6489   Hu Chunming   优化代码;添加isRunning函数
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
  }
  
  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   功能基本完成,接口待打磨
198
  }