95 lines
2.0 KiB
C++
95 lines
2.0 KiB
C++
#ifndef DECODER_H
|
|
#define DECODER_H
|
|
extern "C" {
|
|
#include "libavcodec/avcodec.h"
|
|
#include "libavformat/avformat.h"
|
|
#include "libavutil/imgutils.h"
|
|
}
|
|
#include <queue>
|
|
#include <thread>
|
|
#include <condition_variable>
|
|
#include <mutex>
|
|
|
|
struct VideoParam
|
|
{
|
|
AVFormatContext* fmtCtx;
|
|
AVCodecContext* codecCtx;
|
|
int width;
|
|
int height;
|
|
int videoStreamIndex;
|
|
};
|
|
|
|
template<typename T>
|
|
requires std::is_same_v<T, AVPacket> || std::is_same_v<T, AVFrame>
|
|
struct MediaQueue
|
|
{
|
|
std::queue<T> queue;
|
|
std::condition_variable cv;
|
|
std::mutex mut;
|
|
|
|
uint32_t size;
|
|
uint32_t count;
|
|
|
|
MediaQueue() = default;
|
|
bool push(const T* item);
|
|
bool pop(T* item, bool block = false, bool quit = false);
|
|
};
|
|
|
|
template <typename T> requires std::is_same_v<T, AVPacket> || std::is_same_v<T, AVFrame>
|
|
bool MediaQueue<T>::push(const T* item) {
|
|
if (item == nullptr) return false;
|
|
std::unique_lock lock(mut);
|
|
if constexpr (std::is_same_v<T, AVPacket>) {
|
|
auto temp = av_packet_alloc();
|
|
av_packet_ref(temp, item);
|
|
queue.push(*temp);
|
|
size += temp->size;
|
|
count++;
|
|
}else if(std::is_same_v<T, AVFrame>) {
|
|
auto temp = av_frame_alloc();
|
|
av_frame_ref(temp, item);
|
|
queue.push(*temp);
|
|
count++;
|
|
}
|
|
cv.notify_all();
|
|
return true;
|
|
}
|
|
|
|
template <typename T> requires std::is_same_v<T, AVPacket> || std::is_same_v<T, AVFrame>
|
|
bool MediaQueue<T>::pop(T* item, bool block, bool quit) {
|
|
std::unique_lock lock(mut);
|
|
while (!quit) {
|
|
if (!queue.empty()) {
|
|
T temp = queue.front();
|
|
if constexpr (std::is_same_v<T, AVPacket>) {
|
|
if (av_packet_ref(item, &temp) < 0) {
|
|
return false;
|
|
}
|
|
av_packet_unref(&temp);
|
|
}
|
|
else if (std::is_same_v<T, AVFrame>) {
|
|
if (av_frame_ref(item, &temp) < 0) {
|
|
return false;
|
|
}
|
|
av_frame_unref(&temp);
|
|
}
|
|
queue.pop();
|
|
count--;
|
|
return true;
|
|
}
|
|
else if (block) {
|
|
cv.wait(lock);
|
|
}
|
|
else {
|
|
return false;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
|
|
|
|
|
|
void InitDecoder(const char* filepath, VideoParam& param);
|
|
AVFrame* RequestFrame(VideoParam& param);
|
|
#endif |