From 18a05d269516a5e33d8460291c2f93e73d95adce Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期二, 26 十二月 2023 10:45:31 +0800
Subject: [PATCH] GetYUV format is NV12
---
csrc/worker/stream.hpp | 15 ++++++++++-----
1 files changed, 10 insertions(+), 5 deletions(-)
diff --git a/csrc/worker/stream.hpp b/csrc/worker/stream.hpp
index 6c87ff6..43b24b0 100644
--- a/csrc/worker/stream.hpp
+++ b/csrc/worker/stream.hpp
@@ -1,27 +1,32 @@
#ifndef _cffmpeg_stream_hpp_
#define _cffmpeg_stream_hpp_
-#include <list>
+#include <deque>
#include <mutex>
#include <memory>
+#include "../common.hpp"
+
namespace ffwrapper{
- class CodedData;
+ class FormatIn;
}
namespace cffmpeg_wrap{
class stream
{
private:
- std::list<std::shared_ptr<ffwrapper::CodedData> > list_avpkt_;
+ std::deque<CPacket> list_pkt_;
std::mutex mutex_avpkt_;
+ ffwrapper::FormatIn *streamRef_;
const int max_size_;
+ bool audio_;
public:
- explicit stream(const int maxSize);
+ stream(ffwrapper::FormatIn *in, const int maxSize);
~stream();
- int SetPacket(std::shared_ptr<ffwrapper::CodedData> data);
+ int SetPacket(const CPacket &pkt);
void GetPacket(unsigned char **pktData, int *size, int *key);
+ void AudioSwitch(const bool a){audio_ = a;}
};
}
--
Gitblit v1.8.0