106 lines
3.3 KiB
C++
106 lines
3.3 KiB
C++
#include "VideoPlayer.h"
|
|
#include "BoostLog.h"
|
|
#include <QImage>
|
|
extern "C" {
|
|
#include <libavdevice/avdevice.h>
|
|
#include <libavformat/avformat.h>
|
|
}
|
|
|
|
VideoPlayer::VideoPlayer() {
|
|
static bool init = false;
|
|
if (!init) {
|
|
avdevice_register_all();
|
|
init = true;
|
|
}
|
|
}
|
|
|
|
void VideoPlayer::setFrameCallback(FrameCallback &&callback) {
|
|
m_callback = std::move(callback);
|
|
}
|
|
|
|
void VideoPlayer::setErrorCallback(ErrorCallback &&callback) {
|
|
m_errorCallback = std::move(callback);
|
|
}
|
|
|
|
VideoPlayer::~VideoPlayer() {
|
|
close();
|
|
}
|
|
|
|
bool VideoPlayer::open(const std::string &deviceName) {
|
|
bool ret = true;
|
|
m_formatContext = avformat_alloc_context();
|
|
#ifdef WIN32
|
|
constexpr auto format = "dshow";
|
|
std::ostringstream oss;
|
|
oss << "video=" << deviceName;
|
|
auto device = oss.str();
|
|
#else
|
|
constexpr auto format = "v4l2";
|
|
auto &device = deviceName;
|
|
#endif
|
|
auto inputFormat = av_find_input_format(format);
|
|
|
|
AVDictionary *dictionary = nullptr;
|
|
// ffmpeg -f dshow -list_options true -i video="UVC Camera"
|
|
av_dict_set(&dictionary, "video_size", "800*600", 0);
|
|
|
|
int status = avformat_open_input(&m_formatContext, device.c_str(), inputFormat, &dictionary);
|
|
if (status != 0) {
|
|
char message[256] = {0};
|
|
av_make_error_string(message, sizeof(message), status);
|
|
LOG(error) << "open device[" << device << "] failed: " << status << "," << message;
|
|
ret = false;
|
|
} else {
|
|
av_dump_format(m_formatContext, 0, device.c_str(), 0);
|
|
// for (unsigned int i = 0; i < m_formatContext->nb_streams; i++) {
|
|
// AVStream *stream = m_formatContext->streams[i];
|
|
// if (stream->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
|
|
// std::cout << "当前分辨率: " << stream->codecpar->width << "x" << stream->codecpar->height << std::endl;
|
|
// }
|
|
// }
|
|
m_exit = false;
|
|
m_thread = std::thread(&VideoPlayer::run, this);
|
|
}
|
|
av_dict_free(&dictionary);
|
|
return ret;
|
|
}
|
|
|
|
void VideoPlayer::close() {
|
|
m_exit = true;
|
|
if (m_thread.joinable()) {
|
|
m_thread.join();
|
|
}
|
|
avformat_close_input(&m_formatContext);
|
|
avformat_free_context(m_formatContext);
|
|
m_formatContext = nullptr;
|
|
}
|
|
|
|
void VideoPlayer::run() {
|
|
auto packet = av_packet_alloc();
|
|
while (!m_exit) {
|
|
auto status = av_read_frame(m_formatContext, packet);
|
|
if (status == 0) {
|
|
QImage image;
|
|
image.loadFromData(packet->data, packet->size);
|
|
QTransform transform;
|
|
transform.rotate(90);
|
|
image = image.transformed(transform);
|
|
if (!image.isNull() && m_callback) m_callback(image);
|
|
} else {
|
|
char message[256] = {0};
|
|
av_make_error_string(message, sizeof(message), status);
|
|
if (m_errorCallback) m_errorCallback(status, message);
|
|
LOG(error) << "av_read_frame() failed: " << status << "," << message;
|
|
if (status == -EIO) {
|
|
break;
|
|
}
|
|
}
|
|
av_packet_unref(packet);
|
|
}
|
|
|
|
av_packet_free(&packet);
|
|
QImage image(800, 600, QImage::Format_RGB888);
|
|
image.fill(Qt::black);
|
|
if (m_callback) m_callback(image);
|
|
}
|