129 lines
2.6 KiB
C++
129 lines
2.6 KiB
C++
#ifndef DECODEPARAM_H
|
|
#define DECODEPARAM_H
|
|
|
|
extern "C" {
|
|
#include "libavcodec/avcodec.h"
|
|
#include "libavformat/avformat.h"
|
|
#include "libavutil/imgutils.h"
|
|
}
|
|
#include <queue>
|
|
#include <condition_variable>
|
|
#include <mutex>
|
|
#include <chrono>
|
|
using namespace std::literals::chrono_literals;
|
|
|
|
template<typename T>
|
|
requires std::is_same_v<T, AVPacket> || std::is_same_v<T, AVFrame>
|
|
struct MediaQueue
|
|
{
|
|
static constexpr int MAX_SIZE = 30;
|
|
bool full = false;
|
|
std::queue<T*> queue;
|
|
std::condition_variable cv;
|
|
std::mutex mut;
|
|
|
|
uint32_t size;
|
|
uint32_t count;
|
|
|
|
MediaQueue() = default;
|
|
bool isFill() const { return full; }
|
|
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;
|
|
if (count >= MAX_SIZE)
|
|
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);
|
|
size += temp->size;
|
|
queue.push(temp);
|
|
}
|
|
else if (std::is_same_v<T, AVFrame>) {
|
|
auto temp = av_frame_alloc();
|
|
av_frame_ref(temp, item);
|
|
queue.push(temp);
|
|
}
|
|
count++;
|
|
if (count >= MAX_SIZE) {
|
|
full = true;
|
|
}
|
|
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()) {
|
|
if constexpr (std::is_same_v<T, AVPacket>) {
|
|
AVPacket* temp = queue.front();
|
|
if (av_packet_ref(item, temp) < 0) {
|
|
return false;
|
|
}
|
|
av_packet_unref(temp);
|
|
}
|
|
else if (std::is_same_v<T, AVFrame>) {
|
|
AVFrame* temp = queue.front();
|
|
if (av_frame_ref(item, temp) < 0) {
|
|
return false;
|
|
}
|
|
av_frame_free(&temp);
|
|
delete temp;
|
|
}
|
|
queue.pop();
|
|
count--;
|
|
full = false;
|
|
return true;
|
|
}
|
|
else if (block) {
|
|
cv.wait(lock);
|
|
}
|
|
else {
|
|
return false;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
struct VideoParam
|
|
{
|
|
MediaQueue<AVPacket> packetQueue;
|
|
MediaQueue<AVFrame> frameQueue;
|
|
AVCodecContext* codecCtx;
|
|
int width;
|
|
int height;
|
|
int videoStreamIndex;
|
|
|
|
bool eof = false;
|
|
|
|
bool pause = false;
|
|
bool quit = false;
|
|
};
|
|
|
|
struct AudioParam
|
|
{
|
|
MediaQueue<AVPacket> packetQueue;
|
|
AVCodecContext* codecCtx;
|
|
int audioStreamIndex;
|
|
static constexpr int MAX_BUFFER_SIZE = 192000;
|
|
uint8_t* buffer = new uint8_t[MAX_BUFFER_SIZE];
|
|
uint32_t bufferSize = 0;
|
|
uint32_t bufferIndex = 0;
|
|
bool eof = false;
|
|
bool pause = false;
|
|
bool quit = false;
|
|
};
|
|
|
|
struct MediaParam
|
|
{
|
|
VideoParam videoParam;
|
|
AudioParam audioParam;
|
|
AVFormatContext* fmtCtx;
|
|
};
|
|
#endif |