capturerunner.h 如下
#ifndef CAPTURERUNNER_H
#def
ine CAPTURERUNNER_H
#
include <thread>
#
include <atomic>
#
include "../common/XThreadPool.h"
#
include "../logger/logger.h"
extern "C" {
#
include <libavcodec/avcodec.h>
#
include <libavformat/avformat.h>
#
include <libavutil/fifo.h>
}
#def
ine m
in_time_
interval 40
struct CaptureRunnerParam {
Capture* capt; // 视频捕获对象指针
AVFifo* fifo; // AVFifo帧数据队列
};
class CaptureRunner : public XTask {
public:
int
Init(void* param);
void De
init(void);
int Start(
int time);
int Stop(void);
int Run();
private:
std::atomic<bool> m_is_runn
ing;
Logger::ptr m_logger;
Capture* m_capt;
AVFifo* m_fifo;
int m_time_
interval;
};
#endif // CAPTURERUNNER_H
capturerunner.cpp如下
#
include "capturerunner.h"
#
include <chrono>
us
ing namespace std;
int CaptureRunner::
Init(void* param) {
CaptureRunnerParam* c = (CaptureRunnerParam*)param;
this->m_capt = c->capt;
this->m_fifo = c->fifo;
this->m_logger = DefaultLogger();
return 0;
}
void CaptureRunner::De
init() {
if(m_is_runn
ing) {
this->Stop();
}
return;
}
int CaptureRunner::Start(
int time) {
this->m_time_
interval = time;
Run();
return 0;
}
int CaptureRunner::Stop() {
m_is_runn
ing = false;
return 0;
}
int CaptureRunner::Run() {
if(m_is_runn
ing) {
m_logger->Error("Runn
ing now");
return -1;
}
m_is_runn
ing = true;
int ret = 0;
while(m_is_runn
ing) {
AVFrame* frame = m_capt->CaptureFrame();
if(!frame) {
m_logger->Error("readfile failed");
this_thread::sleep_for(chrono::seconds(1));
cont
inue;
}
// 将帧写入AVFifo
ret = av_fifo_write(m_fifo, &frame, 1);
if(ret < 0) {
m_logger->Error("write to fifo failed");
av_frame_free(&frame);
cont
inue;
}
if(m_time_
interval > m
in_time_
interval) {
this_thread::sleep_for(chrono::milliseconds(m_time_
interval));
} else {
this_thread::sleep_for(chrono::milliseconds(m
in_time_
interval));
}
}
return 0;
}
capt_test.cpp如下
#
include <signal.h>
#
include <atomic>
#
include "../common/XThreadPool.h"
#
include "test.h"
#
include "../capt/capture.h"
#
include "../capt/capturerunner.h"
extern "C" {
#
include <libavutil/fifo.h>
#
include <libavutil/imgutils.h>
}
us
ing namespace std;
Logger::ptr Logger2 = DefaultLogger();
class capt_write : public XTask {
public:
int Run();
void Stop();
int
Init(CaptureRunnerParam* param);
void De
init();
atomic<bool> is_runn
ing;
AVFifo* fifo;
FILE* fp;
AVPixelFormat pix_fmt;
int width, height;
};
int capt_write::
Init(CaptureRunnerParam* par) {
this->fifo = par->fifo;
this->fp = fopen("/home/kyl
in/zxk/test20241029/output.yuv", "wb");
if(!fp) {
Logger2->Error("Failed to open output file");
return -1;
}
// 从捕获对象获取参数
this->width = par->capt->m_width;
this->height = par->capt->m_height;
this->pix_fmt = par->capt->m_pixel_fmt;
return 0;
}
int capt_write::Run() {
if(is_runn
ing) {
Logger2->Error("Runn
ing now");
return -1;
}
is_runn
ing = true;
while(is_runn
ing) {
if(av_fifo_can_read(fifo) <= 0) {
Logger2->Warn
ing("Fifo empty");
this_thread::sleep_for(chrono::seconds(1));
cont
inue;
}
AVFrame* frame = nullptr;
av_fifo_read(fifo, &frame, 1);
if(!frame) {
Logger2->Warn
ing("frame is NULL");
cont
inue;
}
// 计算各分量大小
size_t y_bytes = width * height;
size_t u_bytes = 0, v_bytes = 0;
switch(pix_fmt) {
case AV_PIX_FMT_YUYV422:
y_bytes = width * height * 2; // YUYV packed格式
break;
case AV_PIX_FMT_YUV420P:
u_bytes = width * height / 4;
v_bytes = width * height / 4;
break;
case AV_PIX_FMT_BGR0:
y_bytes = width * height * 4; // BGR0 packed格式
break;
}
// 写入文件
fwrite(frame->data[0], 1, y_bytes, fp);
if(u_bytes > 0) fwrite(frame->data[1], 1, u_bytes, fp);
if(v_bytes > 0) fwrite(frame->data[2], 1, v_bytes, fp);
av_frame_free(&frame);
}
return 0;
}
void capt_write::De
init() {
if(is_runn
ing) {
Stop();
}
if(fp) {
fclose(fp);
}
}
void capt_write::Stop() {
is_runn
ing = false;
}
bool stop2 = false;
void sig
int_handler02(
int sig_num) {
stop2 = true;
}
// 初始化V4L2捕获参数
void
init_v4l2(CaptParam& c) {
c.width = 1280;
c.height = 720;
c.file_name = "/dev/video0";
c.pixel_fmt = AV_PIX_FMT_YUYV422;
c.v4l2_buffer_type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
c.v4l2_memory_type = V4L2_MEMORY_MMAP;
c.v4l2_pixel_fmt = V4L2_PIX_FMT_YUYV;
}
// 初始化文件捕获参数
void
init_file(CaptParam& c) {
c.file_name = "/home/kyl
in/zxk/test20241021/testyuv.yuv";
c.width = 1280;
c.height = 720;
c.pixel_fmt = AV_PIX_FMT_YUV420P;
}
// 初始化屏幕捕获参数
void
init_screen(CaptParam& c) {
c.width = 1920;
c.height = 1080;
c.pixel_fmt = AV_PIX_FMT_BGR0;
}
int capt_test() {
CaptParam c;
// 选择一种捕获方式初始化 (取消注释对应的行)
//
init_v4l2(c); // V4L2捕获
init_file(c); // 文件捕获
//
init_screen(c); // 屏幕捕获
// 创建AVFifo
AVFifo* fifo = av_fifo_alloc2(10 * 1024 * 1024, 1, 0); // 10MB缓冲区
// 创建捕获对象 (与上面选择的初始化方式对应)
//Capture* capt = new V4L2Capture;
Capture* capt = new FileCapture;
//Capture* capt = new ScreenCapture;
capt->
Init(&c);
// 设置任务参数
CaptureRunnerParam param;
param.capt = capt;
param.fifo = fifo;
// 创建任务
auto c1 = make_shared<CaptureRunner>();
auto c2 = make_shared<capt_write>();
c1->
Init(¶m);
c2->
Init(¶m);
// 启动线程池
XThreadPool pool;
pool.
Init(2);
pool.Start();
pool.AddTask(c1);
pool.AddTask(c2);
// 主循环
signal(SIG
INT, sig
int_handler02);
signal(SIGKILL, sig
int_handler02);
while(!stop2) {
this_thread::sleep_for(chrono::milliseconds(100));
}
// 清理
c2->De
init();
this_thread::sleep_for(chrono::milliseconds(1000));
c1->De
init();
capt->De
init();
av_fifo_freep2(&fifo);
pool.Stop();
return 0;
}
CMakeLists.txt 如下
cmake_m
inimum_required(VERSION 3.15)
project(CaptureTest)
set(CMAKE_CXX_STANDARD 11)
#添加头文件搜索路径
include_directories(
${CMAKE_SOURCE_DIR}/
include
/home/zhanghan/
include
/home/zhanghan/Log/Log-lib/
include
/usr/local/
include
)
#添加静态库路径
l
ink_directories(
/home/zhanghan/Log
)
# 查找FFmpeg相关库
f
ind_package(PkgConfig REQUIRED)
pkg_check_modules(FFMPEG REQUIRED
libavcodec
libavformat
libavutil
libswscale
libavdevice
)
# 添加FFmpeg头文件
include_directories(${FFMPEG_
INCLUDE_DIRS})
# 所有源文件
set(SOURCES
ma
in.cpp
capt_test.cpp
../logger/logger.cpp
../logger/formatter.cpp
../logger/logs
ink.cpp
../logger/asynlopper.cpp
../capt/capturerunner.cpp
../capt/capture.cpp
../common/XThreadPool.cpp
../capt/ScreenCapture.cpp
../capt/V4L2Capture.cpp
../capt/FileCapture.cpp
)
add_executable(capture_test ${SOURCES})
# 链接库
target_l
ink_libraries(capture_test
pthread
${FFMPEG_LIBRARIES}
log
)
错误信息如下
zhanghan@zhanghan-virtual-mach
ine:~/daima/2/test2/build$ make
-- Configur
ing done
-- Generat
ing done
-- Build files have been written to: /home/zhanghan/daima/2/test2/build
Consolidate compiler generated dependencies of target capture_test
[ 7%] Build
ing CXX object CMakeFiles/capture_test.dir/capt_test.cpp.o
In file
included from /home/zhanghan/daima/2/test2/capt_test.cpp:6:
/home/zhanghan/daima/2/test2/../capt/capturerunner.h:19:5: error: ‘AVFifo’ does not name a type
19 | AVFifo* fifo; // AVFifo帧数据队列
| ^~~~~~
/home/zhanghan/daima/2/test2/../capt/capturerunner.h:34:5: error: ‘AVFifo’ does not name a type
34 | AVFifo* m_fifo;
| ^~~~~~
/home/zhanghan/daima/2/test2/capt_test.cpp:25:5: error: ‘AVFifo’ does not name a type
25 | AVFifo* fifo;
| ^~~~~~
/home/zhanghan/daima/2/test2/capt_test.cpp:
In member function ‘
int capt_write::
Init(CaptureRunnerParam*)’:
/home/zhanghan/daima/2/test2/capt_test.cpp:32:11: error: ‘class capt_write’ has no member named ‘fifo’
32 | this->fifo = par->fifo;
| ^~~~
/home/zhanghan/daima/2/test2/capt_test.cpp:32:23: error: ‘struct CaptureRunnerParam’ has no member named ‘fifo’
32 | this->fifo = par->fifo;
| ^~~~
/home/zhanghan/daima/2/test2/capt_test.cpp:40:30: error: ‘
int Capture::m_width’ is
protected with
in this context
40 | this->width = par->capt->m_width;
| ^~~~~~~
In file
included from /home/zhanghan/daima/2/test2/capt_test.cpp:5:
/home/zhanghan/daima/2/test2/../capt/capture.h:87:9: note: declared
protected here
87 |
int m_width;
| ^~~~~~~
/home/zhanghan/daima/2/test2/capt_test.cpp:40:30: note: field ‘
int Capture::m_width’ can be accessed via ‘
int Capture::GetWidth() const’
40 | this->width = par->capt->m_width;
| ^~~~~~~
| GetWidth()
/home/zhanghan/daima/2/test2/capt_test.cpp:41:31: error: ‘
int Capture::m_height’ is
protected with
in this context
41 | this->height = par->capt->m_height;
| ^~~~~~~~
In file
included from /home/zhanghan/daima/2/test2/capt_test.cpp:5:
/home/zhanghan/daima/2/test2/../capt/capture.h:88:9: note: declared
protected here
88 |
int m_height;
| ^~~~~~~~
/home/zhanghan/daima/2/test2/capt_test.cpp:41:31: note: field ‘
int Capture::m_height’ can be accessed via ‘
int Capture::GetHeight() const’
41 | this->height = par->capt->m_height;
| ^~~~~~~~
| GetHeight()
/home/zhanghan/daima/2/test2/capt_test.cpp:42:32: error: ‘AVPixelFormat Capture::m_pixel_fmt’ is
protected with
in this context
42 | this->pix_fmt = par->capt->m_pixel_fmt;
| ^~~~~~~~~~~
In file
included from /home/zhanghan/daima/2/test2/capt_test.cpp:5:
/home/zhanghan/daima/2/test2/../capt/capture.h:89:19: note: declared
protected here
89 | AVPixelFormat m_pixel_fmt;
| ^~~~~~~~~~~
/home/zhanghan/daima/2/test2/capt_test.cpp:42:32: note: field ‘AVPixelFormat Capture::m_pixel_fmt’ can be accessed via ‘AVPixelFormat Capture::GetPixelFmt() const’
42 | this->pix_fmt = par->capt->m_pixel_fmt;
| ^~~~~~~~~~~
| GetPixelFmt()
/home/zhanghan/daima/2/test2/capt_test.cpp:
In member function ‘virtual
int capt_write::Run()’:
/home/zhanghan/daima/2/test2/capt_test.cpp:55:29: error: ‘fifo’ was not declared
in this
scope; did you mean ‘mkfifo’?
55 | if(av_fifo_can_read(fifo) <= 0) {
| ^~~~
| mkfifo
/home/zhanghan/daima/2/test2/capt_test.cpp:55:12: error: ‘av_fifo_can_read’ was not declared
in this
scope; did you mean ‘av_fifo_generic_read’?
55 | if(av_fifo_can_read(fifo) <= 0) {
| ^~~~~~~~~~~~~~~~
| av_fifo_generic_read
/home/zhanghan/daima/2/test2/capt_test.cpp:62:22: error: ‘fifo’ was not declared
in this
scope; did you mean ‘mkfifo’?
62 | av_fifo_read(fifo, &frame, 1);
| ^~~~
| mkfifo
/home/zhanghan/daima/2/test2/capt_test.cpp:62:9: error: ‘av_fifo_read’ was not declared
in this
scope; did you mean ‘avio_read’?
62 | av_fifo_read(fifo, &frame, 1);
| ^~~~~~~~~~~~
| avio_read
/home/zhanghan/daima/2/test2/capt_test.cpp:
In function ‘
int capt_test()’:
/home/zhanghan/daima/2/test2/capt_test.cpp:149:5: error: ‘AVFifo’ was not declared
in this
scope
149 | AVFifo* fifo = av_fifo_alloc2(10 * 1024 * 1024, 1, 0); // 10MB缓冲区
| ^~~~~~
/home/zhanghan/daima/2/test2/capt_test.cpp:149:13: error: ‘fifo’ was not declared
in this
scope; did you mean ‘mkfifo’?
149 | AVFifo* fifo = av_fifo_alloc2(10 * 1024 * 1024, 1, 0); // 10MB缓冲区
| ^~~~
| mkfifo
/home/zhanghan/daima/2/test2/capt_test.cpp:149:20: error: ‘av_fifo_alloc2’ was not declared
in this
scope; did you mean ‘av_fifo_alloc’?
149 | AVFifo* fifo = av_fifo_alloc2(10 * 1024 * 1024, 1, 0); // 10MB缓冲区
| ^~~~~~~~~~~~~~
| av_fifo_alloc
/home/zhanghan/daima/2/test2/capt_test.cpp:161:11: error: ‘struct CaptureRunnerParam’ has no member named ‘fifo’
161 | param.fifo = fifo;
| ^~~~
/home/zhanghan/daima/2/test2/capt_test.cpp:188:5: error: ‘av_fifo_freep2’ was not declared
in this
scope; did you mean ‘av_fifo_freep’?
188 | av_fifo_freep2(&fifo);
| ^~~~~~~~~~~~~~
| av_fifo_freep
如何
解决？
本文介绍了一种在Swift中使用switch case语句的方法，并通过一个UITabBar选择项的例子展示了如何通过在case后添加花括号来避免编译错误。
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
