添加音频结构, 添加MediaParam, 增加bugList

This commit is contained in:
jie 2024-02-22 20:36:02 +08:00
parent 146a955841
commit 4d82e2b0a5
7 changed files with 74 additions and 49 deletions

0
include/audioDecoder.h Normal file
View File

View File

@ -1,14 +1,5 @@
#ifndef DECODER_H
#define DECODER_H
extern "C" {
#include "libavcodec/avcodec.h"
#include "libavformat/avformat.h"
#include "libavutil/imgutils.h"
}
#include <queue>
#include <condition_variable>
#include <mutex>
#ifndef DECODEPARAM_H
#define DECODEPARAM_H
template<typename T>
requires std::is_same_v<T, AVPacket> || std::is_same_v<T, AVFrame>
struct MediaQueue
@ -103,7 +94,23 @@ struct VideoParam
bool quit = false;
};
void InitDecoder(const char* filepath, VideoParam& param);
void RequestPacket(VideoParam& param);
void RequestFrame(VideoParam& param);
struct AudioParam
{
MediaQueue<AVPacket> packetQueue;
MediaQueue<AVFrame> frameQueue;
AVFormatContext* fmtCtx;
AVCodecContext* codecCtx;
int audioStreamIndex;
bool eof = false;
bool pause = false;
bool quit = false;
};
struct MediaParam
{
VideoParam videoParam;
AudioParam audioParam;
};
#endif

17
include/mediaDecoder.h Normal file
View File

@ -0,0 +1,17 @@
#ifndef DECODER_H
#define DECODER_H
extern "C" {
#include "libavcodec/avcodec.h"
#include "libavformat/avformat.h"
#include "libavutil/imgutils.h"
}
#include <queue>
#include <condition_variable>
#include <mutex>
#include "decodeParam.h"
void InitDecoder(const char* filepath, VideoParam& param);
void RequestPacket(MediaParam& param);
void RequestVideoFrame(MediaParam& param);
#endif

32
main.cc
View File

@ -6,7 +6,7 @@
#include <SDL2/SDL.h>
#include "util.h"
#include "videoDecoder.h"
#include "mediaDecoder.h"
#include "shaderService.h"
#include "shader.h"
using std::cout;
@ -18,14 +18,14 @@ struct OpenglVideoParam
unsigned int texs[3];
};
int InitVideo(SDL_Window*& window, const char* targetFilepath, VideoParam& videoParam, OpenglVideoParam& openglVideoParam, ShaderService*& shaderService)
int InitVideo(SDL_Window*& window, const char* targetFilepath, MediaParam& param, OpenglVideoParam& openglVideoParam, ShaderService*& shaderService)
{
InitDecoder(targetFilepath, videoParam);
InitDecoder(targetFilepath, param.videoParam);
//FIX: when app exited, the fmtCtx was freed, so need notify decode thread to stop decode and exit.
std::jthread(RequestPacket, std::ref(videoParam)).detach();
std::jthread(RequestFrame, std::ref(videoParam)).detach();
const int client_width = videoParam.width / 2;
const int client_height = videoParam.height / 2;
std::jthread(RequestPacket, std::ref(param)).detach();
std::jthread(RequestVideoFrame, std::ref(param)).detach();
const int client_width = param.videoParam.width / 2;
const int client_height = param.videoParam.height / 2;
window = SDL_CreateWindow(
"MP",
SDL_WINDOWPOS_UNDEFINED,
@ -125,10 +125,10 @@ void InitImg(SDL_Window*& window, const char* filepath, SDL_Renderer*& renderer,
texture = SDL_CreateTextureFromSurface(renderer, surface);
}
void OpenglRenderVideo(VideoParam& videoParam, const OpenglVideoParam& openglVideoParam, ShaderService* shaderService)
void OpenglRenderVideo(MediaParam& param, const OpenglVideoParam& openglVideoParam, ShaderService* shaderService)
{
AVFrame* frame = av_frame_alloc();
videoParam.frameQueue.pop(frame, true, videoParam.quit);
param.videoParam.frameQueue.pop(frame, true, param.videoParam.quit);
// TODO: TIMER
glBindTexture(GL_TEXTURE_2D, openglVideoParam.texs[0]);
glPixelStoref(GL_UNPACK_ROW_LENGTH, static_cast<float>(frame->linesize[0]));
@ -171,9 +171,8 @@ int main(int argc, char** argv)
// INIT
int client_width, client_height;
SDL_Window* window = nullptr;
VideoParam videoParam{};
MediaParam mediaParam{};
OpenglVideoParam openglVideoParam{};
ShaderService* shaderService = nullptr;
SDL_Surface* surface = nullptr;
@ -189,8 +188,8 @@ int main(int argc, char** argv)
{
case FileType::VIDEO:
{
InitVideo(window, targetFilepath, videoParam, openglVideoParam, shaderService);
const auto stream_frame_rate = videoParam.fmtCtx->streams[videoParam.videoStreamIndex]->avg_frame_rate;
InitVideo(window, targetFilepath, mediaParam, openglVideoParam, shaderService);
const auto stream_frame_rate = mediaParam.videoParam.fmtCtx->streams[mediaParam.videoParam.videoStreamIndex]->avg_frame_rate;
framerate = static_cast<double>(stream_frame_rate.den) / stream_frame_rate.num;
break;
}
@ -238,11 +237,10 @@ int main(int argc, char** argv)
switch (fileType)
{
case FileType::VIDEO:
OpenglRenderVideo(videoParam, openglVideoParam, shaderService);
OpenglRenderVideo(mediaParam, openglVideoParam, shaderService);
SDL_GL_SwapWindow(window);
std::this_thread::sleep_until(current_time + std::chrono::milliseconds(30));
current_time = std::chrono::system_clock::now();
cout << SDL_GetTicks() << '\n';
break;
case FileType::IMG:
RenderPicture(window, renderer, texture);
@ -252,8 +250,8 @@ int main(int argc, char** argv)
}
}
avcodec_close(videoParam.codecCtx);
avformat_close_input(&(videoParam.fmtCtx));
avcodec_close(mediaParam.videoParam.codecCtx);
avformat_close_input(&(mediaParam.videoParam.fmtCtx));
SDL_GL_DeleteContext(openglVideoParam.glContext);
SDL_DestroyWindow(window);
SDL_Quit();

2
src/audioDecoder.cc Normal file
View File

@ -0,0 +1,2 @@
#include"audioDecoder.h"

View File

@ -1,4 +1,4 @@
#include "videoDecoder.h"
#include "mediaDecoder.h"
#include <thread>
#include <chrono>
using namespace std::literals::chrono_literals;
@ -26,31 +26,32 @@ void InitDecoder(const char* filepath, VideoParam& param) {
param.height = codecFmt->height;
}
void RequestPacket(VideoParam& param) {
const auto& fmtCtx = param.fmtCtx;
const auto& videoStreamIndex = param.videoStreamIndex;
void RequestPacket(MediaParam& param) {
const auto& fmtCtx = param.videoParam.fmtCtx;
const auto& videoStreamIndex = param.videoParam.videoStreamIndex;
AVPacket* packet = av_packet_alloc();
while (true) {
if (param.packetQueue.isFill()) {
if (param.videoParam.packetQueue.isFill()) {
std::this_thread::sleep_for(100ms);
continue;
}
//FIX:
const int ret = av_read_frame(fmtCtx, packet);
if (param.eof) {
if (param.videoParam.eof) {
std::this_thread::sleep_for(100ms);
av_packet_unref(packet);
return;
}
if (ret == 0) {
if (packet->stream_index == videoStreamIndex) {
param.packetQueue.push(packet);
param.videoParam.packetQueue.push(packet);
av_packet_unref(packet);
}
else if (ret == AVERROR_EOF)
{
param.eof = true;
param.videoParam.eof = true;
av_packet_unref(packet);
break;
}
@ -58,28 +59,27 @@ void RequestPacket(VideoParam& param) {
av_packet_unref(packet);
}
}
else if (param.fmtCtx->pb->error == 0) {
else if (param.videoParam.fmtCtx->pb->error == 0) {
std::this_thread::sleep_for(100ms);
}
}
av_packet_unref(packet);
}
void RequestFrame(VideoParam& param) {
const auto& fmtCtx = param.fmtCtx;
const auto& codecCtx = param.codecCtx;
const auto& videoStreamIndex = param.videoStreamIndex;
void RequestVideoFrame(MediaParam& param) {
const auto& fmtCtx = param.videoParam.fmtCtx;
const auto& codecCtx = param.videoParam.codecCtx;
const auto& videoStreamIndex = param.videoParam.videoStreamIndex;
AVPacket* packet = av_packet_alloc();
AVFrame* frame = av_frame_alloc();
//frame->format = AV_PIX_FMT_YUV420P;
while (true) {
if (param.frameQueue.isFill()) {
if (param.videoParam.frameQueue.isFill()) {
std::this_thread::sleep_for(30ms);
continue;
}
if (!param.packetQueue.pop(packet, true, param.quit)) {
if (param.quit)
if (!param.videoParam.packetQueue.pop(packet, true, param.videoParam.quit)) {
if (param.videoParam.quit)
{
av_packet_unref(packet);
av_frame_unref(frame);
@ -97,7 +97,7 @@ void RequestFrame(VideoParam& param) {
{
continue;
}
param.frameQueue.push(frame);
param.videoParam.frameQueue.push(frame);
av_frame_unref(frame);
}
}

View File

@ -1,5 +1,6 @@
bug:
1. 退出时异常
2. ДЪґжР№В¶
todo:
1. 添加音频播放