av_decoder.cpp 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784
  1. #include "av_decoder.h"
  2. #include <QDebug>
  3. #include <QThread>
  4. #include "threadpool.h"
  5. using AVTool::Decoder;
  6. Decoder::Decoder()
  7. : m_fmtCtx(nullptr)
  8. , m_maxFrameQueueSize(6)
  9. , m_maxPacketQueueSize(12)
  10. , m_audioIndex(-1)
  11. , m_videoIndex(-1)
  12. , m_exit(0)
  13. , m_duration(0)
  14. {
  15. if (!init())
  16. qDebug() << "Decoder init failed!\n";
  17. setInitVal();
  18. }
  19. Decoder::~Decoder()
  20. {
  21. exit();
  22. }
  23. AVTool::MediaInfo* Decoder::detectMediaInfo(const QString& url)
  24. {
  25. int ret = 0;
  26. int duration = 0;
  27. //解封装初始化
  28. AVFormatContext* fmtCtx = avformat_alloc_context();
  29. //用于获取流时长
  30. AVDictionary* formatOpts = nullptr;
  31. av_dict_set(&formatOpts, "probesize", "32", 0);
  32. av_dict_set(&formatOpts, "analyzeduration", "0", 0);
  33. av_dict_set(&formatOpts, "rtsp_transport", "tcp", 0);
  34. av_dict_set(&formatOpts, "fflags", "nobuffer", 0);
  35. av_dict_set(&formatOpts, "max_delay", "0", 0);
  36. ret = avformat_open_input(&fmtCtx, url.toUtf8().constData(), nullptr, &formatOpts);
  37. if (ret < 0) {
  38. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  39. qDebug() << "avformat_open_input error:" << m_errBuf;
  40. av_dict_free(&formatOpts);
  41. //打开失败释放分配的AVFormatContext内存
  42. avformat_free_context(fmtCtx);
  43. return Q_NULLPTR;
  44. }
  45. // 低延迟:关闭内部缓冲
  46. if (fmtCtx) {
  47. fmtCtx->flags |= AVFMT_FLAG_NOBUFFER;
  48. fmtCtx->max_delay = 0;
  49. }
  50. ret = avformat_find_stream_info(fmtCtx, nullptr);
  51. if (ret < 0) {
  52. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  53. qDebug() << "avformat_find_stream_info error:" << m_errBuf;
  54. av_dict_free(&formatOpts);
  55. return Q_NULLPTR;
  56. }
  57. //记录流时长
  58. AVRational q = {1, AV_TIME_BASE};
  59. duration = (uint32_t) (fmtCtx->duration * av_q2d(q));
  60. av_dict_free(&formatOpts);
  61. int videoIndex = av_find_best_stream(fmtCtx, AVMEDIA_TYPE_VIDEO, -1, -1, nullptr, 0);
  62. if (videoIndex < 0) {
  63. // 仅音频:返回时长和占位图
  64. AVTool::MediaInfo* info = new AVTool::MediaInfo;
  65. info->duration = duration;
  66. info->tipImg = QImage(300, 300, QImage::Format_RGB888);
  67. info->tipImg.fill(Qt::black);
  68. avformat_close_input(&fmtCtx);
  69. return info;
  70. }
  71. //视频解码初始化
  72. AVCodecParameters* videoCodecPar = fmtCtx->streams[videoIndex]->codecpar;
  73. if (!videoCodecPar) {
  74. qDebug() << "videocodecpar is nullptr!";
  75. return Q_NULLPTR;
  76. }
  77. AVCodecContext* codecCtx = avcodec_alloc_context3(nullptr);
  78. ret = avcodec_parameters_to_context(codecCtx, videoCodecPar);
  79. if (ret < 0) {
  80. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  81. qDebug() << "error info_avcodec_parameters_to_context:" << m_errBuf;
  82. return Q_NULLPTR;
  83. }
  84. const AVCodec* videoCodec = avcodec_find_decoder(codecCtx->codec_id);
  85. if (!videoCodec) {
  86. qDebug() << "avcodec_find_decoder failed!";
  87. return Q_NULLPTR;
  88. }
  89. codecCtx->codec_id = videoCodec->id;
  90. ret = avcodec_open2(codecCtx, videoCodec, nullptr);
  91. if (ret < 0) {
  92. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  93. qDebug() << "error info_avcodec_open2:" << m_errBuf;
  94. return Q_NULLPTR;
  95. }
  96. AVPacket* pkt = av_packet_alloc();
  97. AVFrame* frame = av_frame_alloc();
  98. bool flag = false;
  99. while (1) {
  100. ret = av_read_frame(fmtCtx, pkt);
  101. if (ret != 0) {
  102. return Q_NULLPTR;
  103. }
  104. if (pkt->stream_index == videoIndex) {
  105. ret = avcodec_send_packet(codecCtx, pkt);
  106. av_packet_unref(pkt);
  107. if (ret < 0 || ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
  108. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  109. qDebug() << "avcodec_send_packet error:" << m_errBuf;
  110. continue;
  111. }
  112. while (1) {
  113. ret = avcodec_receive_frame(codecCtx, frame);
  114. if (ret == 0) {
  115. flag = true;
  116. break;
  117. } else if (ret == AVERROR(EAGAIN)) {
  118. break;
  119. } else {
  120. return Q_NULLPTR;
  121. }
  122. }
  123. if (flag)
  124. break;
  125. } else {
  126. av_packet_unref(pkt);
  127. }
  128. }
  129. int imageWidth = videoCodecPar->width;
  130. int imageHeight = videoCodecPar->height;
  131. enum AVPixelFormat dstPixFmt = AV_PIX_FMT_RGB24;
  132. int swsFlags = SWS_BICUBIC;
  133. uint8_t* pixels[4];
  134. int pitch[4];
  135. //分配存储转换后帧数据的buffer内存
  136. int bufSize = av_image_get_buffer_size(dstPixFmt, imageWidth, imageHeight, 1);
  137. uint8_t* buffer = (uint8_t*) av_malloc(bufSize * sizeof(uint8_t));
  138. av_image_fill_arrays(pixels, pitch, buffer, dstPixFmt, imageWidth, imageHeight, 1);
  139. SwsContext* swsCtx = sws_getCachedContext(nullptr,
  140. frame->width,
  141. frame->height,
  142. (enum AVPixelFormat) frame->format,
  143. imageWidth,
  144. imageHeight,
  145. dstPixFmt,
  146. swsFlags,
  147. nullptr,
  148. nullptr,
  149. nullptr);
  150. if (swsCtx)
  151. sws_scale(swsCtx, frame->data, frame->linesize, 0, frame->height, pixels, pitch);
  152. av_frame_unref(frame);
  153. AVTool::MediaInfo* info = new AVTool::MediaInfo;
  154. info->duration = duration;
  155. info->tipImg = QImage(pixels[0], imageWidth, imageHeight, pitch[0], QImage::Format_RGB888);
  156. av_packet_free(&pkt);
  157. av_frame_free(&frame);
  158. avformat_close_input(&fmtCtx);
  159. avcodec_free_context(&codecCtx);
  160. return info;
  161. }
  162. bool Decoder::init()
  163. {
  164. if (!ThreadPool::init()) {
  165. qDebug() << "threadpool init failed!\n";
  166. return false;
  167. }
  168. m_audioPacketQueue.pktVec.resize(m_maxPacketQueueSize);
  169. m_videoPacketQueue.pktVec.resize(m_maxPacketQueueSize);
  170. m_audioFrameQueue.frameVec.resize(m_maxFrameQueueSize);
  171. m_videoFrameQueue.frameVec.resize(m_maxFrameQueueSize);
  172. m_audioPktDecoder.codecCtx = nullptr;
  173. m_videoPktDecoder.codecCtx = nullptr;
  174. return true;
  175. }
  176. void Decoder::setInitVal()
  177. {
  178. m_audioPacketQueue.size = 0;
  179. m_audioPacketQueue.pushIndex = 0;
  180. m_audioPacketQueue.readIndex = 0;
  181. m_audioPacketQueue.serial = 0;
  182. m_videoPacketQueue.size = 0;
  183. m_videoPacketQueue.pushIndex = 0;
  184. m_videoPacketQueue.readIndex = 0;
  185. m_videoPacketQueue.serial = 0;
  186. m_audioFrameQueue.size = 0;
  187. m_audioFrameQueue.readIndex = 0;
  188. m_audioFrameQueue.pushIndex = 0;
  189. m_audioFrameQueue.shown = 0;
  190. m_videoFrameQueue.size = 0;
  191. m_videoFrameQueue.readIndex = 0;
  192. m_videoFrameQueue.pushIndex = 0;
  193. m_videoFrameQueue.shown = 0;
  194. m_exit = 0;
  195. m_isSeek = 0;
  196. m_audSeek = 0;
  197. m_vidSeek = 0;
  198. m_audioPktDecoder.serial = 0;
  199. m_videoPktDecoder.serial = 0;
  200. }
  201. int Decoder::decode(const QString& url)
  202. {
  203. int ret = 0;
  204. //解封装初始化
  205. m_fmtCtx = avformat_alloc_context();
  206. //用于获取流时长
  207. AVDictionary* formatOpts = nullptr;
  208. av_dict_set(&formatOpts, "probesize", "32", 0);
  209. av_dict_set(&formatOpts, "analyzeduration", "0", 0);
  210. av_dict_set(&formatOpts, "rtsp_transport", "tcp", 0);
  211. av_dict_set(&formatOpts, "fflags", "nobuffer", 0);
  212. av_dict_set(&formatOpts, "max_delay", "0", 0);
  213. ret = avformat_open_input(&m_fmtCtx, url.toUtf8().constData(), nullptr, &formatOpts);
  214. if (ret < 0) {
  215. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  216. qDebug() << "avformat_open_input error:" << m_errBuf;
  217. av_dict_free(&formatOpts);
  218. //打开失败释放分配的AVFormatContext内存
  219. avformat_free_context(m_fmtCtx);
  220. return 0;
  221. }
  222. // 低延迟:关闭内部缓冲
  223. if (m_fmtCtx) {
  224. m_fmtCtx->flags |= AVFMT_FLAG_NOBUFFER;
  225. m_fmtCtx->max_delay = 0;
  226. }
  227. ret = avformat_find_stream_info(m_fmtCtx, nullptr);
  228. if (ret < 0) {
  229. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  230. qDebug() << "avformat_find_stream_info error:" << m_errBuf;
  231. av_dict_free(&formatOpts);
  232. return 0;
  233. }
  234. //记录流时长
  235. AVRational q = {1, AV_TIME_BASE};
  236. m_duration = (uint32_t) (m_fmtCtx->duration * av_q2d(q));
  237. av_dict_free(&formatOpts);
  238. // 查找音频/视频流,允许缺失
  239. m_audioIndex = av_find_best_stream(m_fmtCtx, AVMEDIA_TYPE_AUDIO, -1, -1, nullptr, 0);
  240. if (m_audioIndex < 0) {
  241. qDebug() << "no audio stream!";
  242. }
  243. m_videoIndex = av_find_best_stream(m_fmtCtx, AVMEDIA_TYPE_VIDEO, -1, -1, nullptr, 0);
  244. if (m_videoIndex < 0) {
  245. qDebug() << "no video stream!";
  246. }
  247. if (m_audioIndex < 0 && m_videoIndex < 0) {
  248. qDebug() << "no audio and video stream!";
  249. return 0;
  250. }
  251. // 音频解码初始化(可选)
  252. if (m_audioIndex >= 0) {
  253. AVCodecParameters* audioCodecPar = m_fmtCtx->streams[m_audioIndex]->codecpar;
  254. if (!audioCodecPar) {
  255. qDebug() << "audio par is nullptr!";
  256. return 0;
  257. }
  258. m_audioPktDecoder.codecCtx = avcodec_alloc_context3(nullptr);
  259. ret = avcodec_parameters_to_context(m_audioPktDecoder.codecCtx, audioCodecPar);
  260. if (ret < 0) {
  261. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  262. qDebug() << "error info_avcodec_parameters_to_context:" << m_errBuf;
  263. return 0;
  264. }
  265. const AVCodec* audioCodec = avcodec_find_decoder(m_audioPktDecoder.codecCtx->codec_id);
  266. if (!audioCodec) {
  267. qDebug() << "avcodec_find_decoder failed!";
  268. return 0;
  269. }
  270. m_audioPktDecoder.codecCtx->codec_id = audioCodec->id;
  271. // 低延迟解码设置(音频)
  272. m_audioPktDecoder.codecCtx->flags |= AV_CODEC_FLAG_LOW_DELAY;
  273. m_audioPktDecoder.codecCtx->flags2 |= AV_CODEC_FLAG2_FAST;
  274. m_audioPktDecoder.codecCtx->thread_type = FF_THREAD_SLICE;
  275. m_audioPktDecoder.codecCtx->thread_count = 1;
  276. ret = avcodec_open2(m_audioPktDecoder.codecCtx, audioCodec, nullptr);
  277. if (ret < 0) {
  278. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  279. qDebug() << "error info_avcodec_open2:" << m_errBuf;
  280. return 0;
  281. }
  282. }
  283. // 视频解码初始化(可选)
  284. if (m_videoIndex >= 0) {
  285. AVCodecParameters* videoCodecPar = m_fmtCtx->streams[m_videoIndex]->codecpar;
  286. if (!videoCodecPar) {
  287. qDebug() << "videocodecpar is nullptr!";
  288. return 0;
  289. }
  290. m_videoPktDecoder.codecCtx = avcodec_alloc_context3(nullptr);
  291. ret = avcodec_parameters_to_context(m_videoPktDecoder.codecCtx, videoCodecPar);
  292. if (ret < 0) {
  293. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  294. qDebug() << "error info_avcodec_parameters_to_context:" << m_errBuf;
  295. return 0;
  296. }
  297. const AVCodec* videoCodec = avcodec_find_decoder(m_videoPktDecoder.codecCtx->codec_id);
  298. if (!videoCodec) {
  299. qDebug() << "avcodec_find_decoder failed!";
  300. return 0;
  301. }
  302. m_videoPktDecoder.codecCtx->codec_id = videoCodec->id;
  303. // 低延迟解码设置(视频)
  304. m_videoPktDecoder.codecCtx->flags |= AV_CODEC_FLAG_LOW_DELAY;
  305. m_videoPktDecoder.codecCtx->flags2 |= AV_CODEC_FLAG2_FAST;
  306. m_videoPktDecoder.codecCtx->thread_type = FF_THREAD_SLICE;
  307. m_videoPktDecoder.codecCtx->thread_count = 1;
  308. ret = avcodec_open2(m_videoPktDecoder.codecCtx, videoCodec, nullptr);
  309. if (ret < 0) {
  310. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  311. qDebug() << "error info_avcodec_open2:" << m_errBuf;
  312. return 0;
  313. }
  314. //记录视频帧率
  315. m_vidFrameRate = av_guess_frame_rate(m_fmtCtx, m_fmtCtx->streams[m_videoIndex], nullptr);
  316. }
  317. // 启动解复用/解码线程
  318. setInitVal();
  319. ThreadPool::addTask(std::bind(&Decoder::demux, this, std::placeholders::_1),
  320. std::make_shared<int>(1));
  321. if (m_audioIndex >= 0) {
  322. ThreadPool::addTask(std::bind(&Decoder::audioDecode, this, std::placeholders::_1),
  323. std::make_shared<int>(2));
  324. }
  325. if (m_videoIndex >= 0) {
  326. ThreadPool::addTask(std::bind(&Decoder::videoDecode, this, std::placeholders::_1),
  327. std::make_shared<int>(3));
  328. }
  329. return 1;
  330. }
  331. void Decoder::exit()
  332. {
  333. m_exit = 1;
  334. QThread::msleep(200);
  335. clearQueueCache();
  336. if (m_fmtCtx) {
  337. avformat_close_input(&m_fmtCtx);
  338. m_fmtCtx = nullptr;
  339. }
  340. if (m_audioPktDecoder.codecCtx) {
  341. avcodec_free_context(&m_audioPktDecoder.codecCtx);
  342. m_audioPktDecoder.codecCtx = nullptr;
  343. }
  344. if (m_videoPktDecoder.codecCtx) {
  345. avcodec_free_context(&m_videoPktDecoder.codecCtx);
  346. m_videoPktDecoder.codecCtx = nullptr;
  347. }
  348. }
  349. void Decoder::clearQueueCache()
  350. {
  351. std::lock_guard<std::mutex> lockAP(m_audioPacketQueue.mutex);
  352. std::lock_guard<std::mutex> lockVP(m_videoPacketQueue.mutex);
  353. while (m_audioPacketQueue.size) {
  354. av_packet_unref(&m_audioPacketQueue.pktVec[m_audioPacketQueue.readIndex].pkt);
  355. m_audioPacketQueue.readIndex = (m_audioPacketQueue.readIndex + 1) % m_maxPacketQueueSize;
  356. m_audioPacketQueue.size--;
  357. }
  358. while (m_videoPacketQueue.size) {
  359. av_packet_unref(&m_videoPacketQueue.pktVec[m_videoPacketQueue.readIndex].pkt);
  360. m_videoPacketQueue.readIndex = (m_videoPacketQueue.readIndex + 1) % m_maxPacketQueueSize;
  361. m_videoPacketQueue.size--;
  362. }
  363. std::lock_guard<std::mutex> lockAF(m_audioFrameQueue.mutex);
  364. std::lock_guard<std::mutex> lockVF(m_videoFrameQueue.mutex);
  365. while (m_audioFrameQueue.size) {
  366. av_frame_unref(&m_audioFrameQueue.frameVec[m_audioFrameQueue.readIndex].frame);
  367. m_audioFrameQueue.readIndex = (m_audioFrameQueue.readIndex + 1) % m_maxFrameQueueSize;
  368. m_audioFrameQueue.size--;
  369. }
  370. while (m_videoFrameQueue.size) {
  371. av_frame_unref(&m_videoFrameQueue.frameVec[m_videoFrameQueue.readIndex].frame);
  372. m_videoFrameQueue.readIndex = (m_videoFrameQueue.readIndex + 1) % m_maxFrameQueueSize;
  373. m_videoFrameQueue.size--;
  374. }
  375. }
  376. void Decoder::packetQueueFlush(PacketQueue* queue)
  377. {
  378. std::lock_guard<std::mutex> lockAP(queue->mutex);
  379. while (queue->size) {
  380. av_packet_unref(&queue->pktVec[queue->readIndex].pkt);
  381. queue->readIndex = (queue->readIndex + 1) % m_maxPacketQueueSize;
  382. queue->size--;
  383. }
  384. queue->serial++;
  385. }
  386. void Decoder::seekTo(int32_t target, int32_t seekRel)
  387. {
  388. //上次跳转未完成不处理跳转请求
  389. if (m_isSeek == 1)
  390. return;
  391. if (target < 0)
  392. target = 0;
  393. m_seekTarget = target;
  394. m_seekRel = seekRel;
  395. m_isSeek = 1;
  396. }
  397. void Decoder::demux(std::shared_ptr<void> par)
  398. {
  399. int ret = -1;
  400. AVPacket* pkt = av_packet_alloc();
  401. while (1) {
  402. if (m_exit) {
  403. break;
  404. }
  405. if (m_audioPacketQueue.size >= m_maxPacketQueueSize
  406. || m_videoPacketQueue.size >= m_maxPacketQueueSize) {
  407. std::this_thread::sleep_for(std::chrono::milliseconds(50));
  408. continue;
  409. }
  410. if (m_isSeek) {
  411. //AVRational sq={1,AV_TIME_BASE};
  412. //int64_t seekMin = m_seekRel > 0 ? m_seekTarget-m_seekRel+2 : INT64_MIN;
  413. //int64_t seekMax = m_seekRel < 0 ? m_seekTarget-m_seekRel-2 : INT64_MAX;
  414. //qDebug()<<"seekMin:"<<seekMin<<" seekMax:"<<seekMax<<" seekTarget:"<<m_seekTarget<<endl;
  415. //ret=avformat_seek_file(m_fmtCtx,m_audioIndex,seekMin,m_seekTarget,seekMax,AVSEEK_FLAG_BACKWARD);
  416. int64_t seekTarget = m_seekTarget * AV_TIME_BASE;
  417. ret = av_seek_frame(m_fmtCtx, -1, seekTarget, AVSEEK_FLAG_BACKWARD);
  418. if (ret < 0) {
  419. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  420. qDebug() << "avformat_seek_file error:" << m_errBuf;
  421. } else {
  422. packetQueueFlush(&m_audioPacketQueue);
  423. packetQueueFlush(&m_videoPacketQueue);
  424. m_audSeek = 1;
  425. m_vidSeek = 1;
  426. }
  427. m_isSeek = 0;
  428. }
  429. ret = av_read_frame(m_fmtCtx, pkt);
  430. if (ret != 0) {
  431. // 仅清理内部引用,避免与循环末尾的 av_packet_free 重复释放
  432. av_packet_unref(pkt);
  433. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  434. qDebug() << "av_read_frame error:" << m_errBuf;
  435. break;
  436. }
  437. if (pkt->stream_index == m_audioIndex) {
  438. //插入音频包队列
  439. //qDebug()<<pkt->pts*av_q2d(m_fmtCtx->streams[m_audioIndex]->time_base)<<endl;
  440. pushPacket(&m_audioPacketQueue, pkt);
  441. } else if (pkt->stream_index == m_videoIndex) {
  442. //插入视频包队列
  443. pushPacket(&m_videoPacketQueue, pkt);
  444. //av_packet_unref(pkt);
  445. } else {
  446. av_packet_unref(pkt);
  447. }
  448. }
  449. av_packet_free(&pkt);
  450. //是读到文件末尾退出的才清空,强制退出不重复此操作
  451. if (!m_exit) {
  452. // 根据实际存在的流等待对应帧队列清空
  453. if (m_audioIndex >= 0) {
  454. while (m_audioFrameQueue.size)
  455. QThread::msleep(50);
  456. }
  457. if (m_videoIndex >= 0) {
  458. while (m_videoFrameQueue.size)
  459. QThread::msleep(50);
  460. }
  461. exit();
  462. }
  463. //qDebug() << "demuxthread exit";
  464. }
  465. void Decoder::audioDecode(std::shared_ptr<void> par)
  466. {
  467. AVPacket* pkt = av_packet_alloc();
  468. AVFrame* frame = av_frame_alloc();
  469. while (1) {
  470. if (m_exit) {
  471. break;
  472. }
  473. //音频帧队列长度控制
  474. if (m_audioFrameQueue.size >= m_maxFrameQueueSize) {
  475. std::this_thread::sleep_for(std::chrono::milliseconds(10));
  476. continue;
  477. }
  478. //从音频包队列取音频包
  479. int ret = getPacket(&m_audioPacketQueue, pkt, &m_audioPktDecoder);
  480. if (ret) {
  481. ret = avcodec_send_packet(m_audioPktDecoder.codecCtx, pkt);
  482. av_packet_unref(pkt);
  483. if (ret < 0) {
  484. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  485. qDebug() << "avcodec_send_packet error:" << m_errBuf;
  486. continue;
  487. }
  488. while (1) {
  489. ret = avcodec_receive_frame(m_audioPktDecoder.codecCtx, frame);
  490. if (ret == 0) {
  491. if (m_audSeek) {
  492. int pts = (int) frame->pts
  493. * av_q2d(m_fmtCtx->streams[m_audioIndex]->time_base);
  494. //qDebug()<<"audFrame pts:"<<pts<<endl;
  495. if (pts < m_seekTarget) {
  496. av_frame_unref(frame);
  497. continue;
  498. } else {
  499. m_audSeek = 0;
  500. }
  501. }
  502. //添加到待播放音频帧队列
  503. pushAFrame(frame);
  504. } else {
  505. break;
  506. }
  507. }
  508. } else {
  509. //qDebug() << "audio packetQueue is empty for decoding!";
  510. std::this_thread::sleep_for(std::chrono::milliseconds(20));
  511. }
  512. }
  513. av_packet_free(&pkt);
  514. av_frame_free(&frame);
  515. //qDebug() << "audioDecode exit";
  516. }
  517. void Decoder::videoDecode(std::shared_ptr<void> par)
  518. {
  519. AVPacket* pkt = av_packet_alloc();
  520. AVFrame* frame = av_frame_alloc();
  521. while (1) {
  522. if (m_exit) {
  523. break;
  524. }
  525. if (m_videoFrameQueue.size >= m_maxFrameQueueSize) { //视频帧队列长度控制
  526. std::this_thread::sleep_for(std::chrono::milliseconds(10));
  527. continue;
  528. }
  529. int ret = getPacket(&m_videoPacketQueue, pkt, &m_videoPktDecoder);
  530. if (ret) {
  531. ret = avcodec_send_packet(m_videoPktDecoder.codecCtx, pkt);
  532. av_packet_unref(pkt);
  533. if (ret < 0 || ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
  534. av_strerror(ret, m_errBuf, sizeof(m_errBuf));
  535. qDebug() << "avcodec_send_packet error:" << m_errBuf;
  536. continue;
  537. }
  538. while (1) {
  539. ret = avcodec_receive_frame(m_videoPktDecoder.codecCtx, frame);
  540. if (ret == 0) {
  541. if (m_vidSeek) {
  542. int pts = (int) frame->pts
  543. * av_q2d(m_fmtCtx->streams[m_videoIndex]->time_base);
  544. if (pts < m_seekTarget) {
  545. av_frame_unref(frame);
  546. continue;
  547. } else {
  548. m_vidSeek = 0;
  549. }
  550. }
  551. //AVRational vidRational=av_guess_frame_rate(m_fmtCtx,
  552. // m_fmtCtx->streams[m_videoIndex],frame);
  553. //添加到待播放视频帧队列
  554. pushVFrame(frame);
  555. } else {
  556. break;
  557. }
  558. }
  559. } else {
  560. //qDebug() << "video packetQueue is empty for decoding!";
  561. std::this_thread::sleep_for(std::chrono::milliseconds(20));
  562. }
  563. }
  564. av_packet_free(&pkt);
  565. av_frame_free(&frame);
  566. //qDebug() << "videoDecode exit";
  567. }
  568. int Decoder::getPacket(PacketQueue* queue, AVPacket* pkt, PktDecoder* decoder)
  569. {
  570. std::unique_lock<std::mutex> lock(queue->mutex);
  571. while (!queue->size) {
  572. bool ret = queue->cond.wait_for(lock, std::chrono::milliseconds(100), [&]() {
  573. return queue->size && !m_exit;
  574. });
  575. if (!ret)
  576. return 0;
  577. }
  578. if (queue->serial != decoder->serial) {
  579. //序列号不连续的帧证明发生了跳转操作则直接丢弃
  580. //并清空解码器缓存
  581. avcodec_flush_buffers(decoder->codecCtx);
  582. decoder->serial = queue->pktVec[queue->readIndex].serial;
  583. return 0;
  584. }
  585. av_packet_move_ref(pkt, &queue->pktVec[queue->readIndex].pkt);
  586. decoder->serial = queue->pktVec[queue->readIndex].serial;
  587. queue->readIndex = (queue->readIndex + 1) % m_maxPacketQueueSize;
  588. queue->size--;
  589. return 1;
  590. }
  591. void Decoder::pushPacket(PacketQueue* queue, AVPacket* pkt)
  592. {
  593. std::lock_guard<std::mutex> lock(queue->mutex);
  594. av_packet_move_ref(&queue->pktVec[queue->pushIndex].pkt, pkt);
  595. queue->pktVec[queue->pushIndex].serial = queue->serial;
  596. if (queue->serial == 1)
  597. qDebug() << "";
  598. queue->pushIndex = (queue->pushIndex + 1) % m_maxPacketQueueSize;
  599. queue->size++;
  600. }
  601. void Decoder::pushAFrame(AVFrame* frame)
  602. {
  603. std::lock_guard<std::mutex> lock(m_audioFrameQueue.mutex);
  604. av_frame_move_ref(&m_audioFrameQueue.frameVec[m_audioFrameQueue.pushIndex].frame, frame);
  605. m_audioFrameQueue.frameVec[m_audioFrameQueue.pushIndex].serial = m_audioPktDecoder.serial;
  606. m_audioFrameQueue.pushIndex = (m_audioFrameQueue.pushIndex + 1) % m_maxFrameQueueSize;
  607. m_audioFrameQueue.size++;
  608. }
  609. int Decoder::getAFrame(AVFrame* frame)
  610. {
  611. if (!frame)
  612. return 0;
  613. std::unique_lock<std::mutex> lock(m_audioFrameQueue.mutex);
  614. while (!m_audioFrameQueue.size) {
  615. bool ret = m_audioFrameQueue.cond.wait_for(lock, std::chrono::milliseconds(100), [&]() {
  616. return !m_exit && m_audioFrameQueue.size;
  617. });
  618. if (!ret)
  619. return 0;
  620. }
  621. if (m_audioFrameQueue.frameVec[m_audioFrameQueue.readIndex].serial
  622. != m_audioPacketQueue.serial) {
  623. av_frame_unref(&m_audioFrameQueue.frameVec[m_audioFrameQueue.readIndex].frame);
  624. m_audioFrameQueue.readIndex = (m_audioFrameQueue.readIndex + 1) % m_maxFrameQueueSize;
  625. m_audioFrameQueue.size--;
  626. return 0;
  627. }
  628. av_frame_move_ref(frame, &m_audioFrameQueue.frameVec[m_audioFrameQueue.readIndex].frame);
  629. m_audioFrameQueue.readIndex = (m_audioFrameQueue.readIndex + 1) % m_maxFrameQueueSize;
  630. m_audioFrameQueue.size--;
  631. return 1;
  632. }
  633. void Decoder::pushVFrame(AVFrame* frame)
  634. {
  635. std::lock_guard<std::mutex> lock(m_videoFrameQueue.mutex);
  636. m_videoFrameQueue.frameVec[m_videoFrameQueue.pushIndex].serial = m_videoPktDecoder.serial;
  637. m_videoFrameQueue.frameVec[m_videoFrameQueue.pushIndex].duration
  638. = (m_vidFrameRate.den && m_vidFrameRate.den)
  639. ? av_q2d(AVRational{m_vidFrameRate.den, m_vidFrameRate.num})
  640. : 0.00;
  641. m_videoFrameQueue.frameVec[m_videoFrameQueue.pushIndex].pts
  642. = frame->pts * av_q2d(m_fmtCtx->streams[m_videoIndex]->time_base);
  643. av_frame_move_ref(&m_videoFrameQueue.frameVec[m_videoFrameQueue.pushIndex].frame, frame);
  644. m_videoFrameQueue.pushIndex = (m_videoFrameQueue.pushIndex + 1) % m_maxFrameQueueSize;
  645. m_videoFrameQueue.size++;
  646. //qDebug()<<"RemainingVFrame:"<<m_videoFrameQueue.size-m_videoFrameQueue.shown;
  647. }
  648. Decoder::MyFrame* Decoder::peekLastVFrame()
  649. {
  650. Decoder::MyFrame* frame = &m_videoFrameQueue.frameVec[m_videoFrameQueue.readIndex];
  651. return frame;
  652. }
  653. Decoder::MyFrame* Decoder::peekVFrame()
  654. {
  655. while (!m_videoFrameQueue.size) {
  656. std::unique_lock<std::mutex> lock(m_videoFrameQueue.mutex);
  657. bool ret = m_videoFrameQueue.cond.wait_for(lock, std::chrono::milliseconds(100), [&]() {
  658. return !m_exit && m_videoFrameQueue.size;
  659. });
  660. if (!ret)
  661. return nullptr;
  662. }
  663. int index = (m_videoFrameQueue.readIndex + m_videoFrameQueue.shown) % m_maxFrameQueueSize;
  664. Decoder::MyFrame* frame = &m_videoFrameQueue.frameVec[index];
  665. return frame;
  666. }
  667. Decoder::MyFrame* Decoder::peekNextVFrame()
  668. {
  669. while (m_videoFrameQueue.size < 2) {
  670. std::unique_lock<std::mutex> lock(m_videoFrameQueue.mutex);
  671. bool ret = m_videoFrameQueue.cond.wait_for(lock, std::chrono::milliseconds(100), [&]() {
  672. return !m_exit && m_videoFrameQueue.size;
  673. });
  674. if (!ret)
  675. return nullptr;
  676. }
  677. int index = (m_videoFrameQueue.readIndex + m_videoFrameQueue.shown + 1) % m_maxFrameQueueSize;
  678. Decoder::MyFrame* frame = &m_videoFrameQueue.frameVec[index];
  679. return frame;
  680. }
  681. void Decoder::setNextVFrame()
  682. {
  683. std::unique_lock<std::mutex> lock(m_videoFrameQueue.mutex);
  684. if (!m_videoFrameQueue.size)
  685. return;
  686. if (!m_videoFrameQueue.shown) {
  687. m_videoFrameQueue.shown = 1;
  688. return;
  689. }
  690. av_frame_unref(&m_videoFrameQueue.frameVec[m_videoFrameQueue.readIndex].frame);
  691. m_videoFrameQueue.readIndex = (m_videoFrameQueue.readIndex + 1) % m_maxFrameQueueSize;
  692. m_videoFrameQueue.size--;
  693. }
  694. int Decoder::getRemainingVFrame()
  695. {
  696. if (!m_videoFrameQueue.size)
  697. return 0;
  698. return m_videoFrameQueue.size - m_videoFrameQueue.shown;
  699. }