mirror of
https://github.com/kunkundi/crossdesk.git
synced 2025-10-26 20:25:34 +08:00
144 lines
2.9 KiB
C++
144 lines
2.9 KiB
C++
#include "record_desktop_wgc.h"
|
|
|
|
#include "error_define.h"
|
|
#include "log_helper.h"
|
|
#include "system_error.h"
|
|
#include "utils_string.h"
|
|
|
|
BOOL WINAPI EnumMonitorProc(HMONITOR hmonitor, HDC hdc, LPRECT lprc,
|
|
LPARAM data) {
|
|
MONITORINFOEX info_ex;
|
|
info_ex.cbSize = sizeof(MONITORINFOEX);
|
|
|
|
GetMonitorInfo(hmonitor, &info_ex);
|
|
|
|
if (info_ex.dwFlags == DISPLAY_DEVICE_MIRRORING_DRIVER) return true;
|
|
|
|
if (info_ex.dwFlags & MONITORINFOF_PRIMARY) {
|
|
*(HMONITOR *)data = hmonitor;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
HMONITOR GetPrimaryMonitor() {
|
|
HMONITOR hmonitor = nullptr;
|
|
|
|
::EnumDisplayMonitors(NULL, NULL, EnumMonitorProc, (LPARAM)&hmonitor);
|
|
|
|
return hmonitor;
|
|
}
|
|
|
|
namespace am {
|
|
|
|
record_desktop_wgc::record_desktop_wgc() {}
|
|
|
|
record_desktop_wgc::~record_desktop_wgc() {
|
|
stop();
|
|
clean_up();
|
|
}
|
|
|
|
int record_desktop_wgc::init(const RECORD_DESKTOP_RECT &rect, const int fps) {
|
|
int error = AE_NO;
|
|
if (_inited == true) return error;
|
|
|
|
_fps = fps;
|
|
_rect = rect;
|
|
_start_time = av_gettime_relative();
|
|
_time_base = {1, AV_TIME_BASE};
|
|
_pixel_fmt = AV_PIX_FMT_BGRA;
|
|
|
|
do {
|
|
if (!module_.is_supported()) {
|
|
error = AE_UNSUPPORT;
|
|
break;
|
|
}
|
|
|
|
session_ = module_.create_session();
|
|
if (!session_) {
|
|
error = AE_WGC_CREATE_CAPTURER_FAILED;
|
|
break;
|
|
}
|
|
|
|
session_->register_observer(this);
|
|
|
|
error = session_->initialize(GetPrimaryMonitor());
|
|
|
|
_inited = true;
|
|
} while (0);
|
|
|
|
if (error != AE_NO) {
|
|
al_debug("%s,last error:%s", err2str(error),
|
|
system_error::error2str(GetLastError()).c_str());
|
|
}
|
|
|
|
return error;
|
|
}
|
|
|
|
int record_desktop_wgc::start() {
|
|
if (_running == true) {
|
|
al_warn("record desktop duplication is already running");
|
|
return AE_NO;
|
|
}
|
|
|
|
if (_inited == false) {
|
|
return AE_NEED_INIT;
|
|
}
|
|
|
|
_running = true;
|
|
session_->start();
|
|
|
|
return AE_NO;
|
|
}
|
|
|
|
int record_desktop_wgc::pause() {
|
|
_paused = true;
|
|
if (session_) session_->pause();
|
|
return AE_NO;
|
|
}
|
|
|
|
int record_desktop_wgc::resume() {
|
|
_paused = false;
|
|
if (session_) session_->resume();
|
|
return AE_NO;
|
|
}
|
|
|
|
int record_desktop_wgc::stop() {
|
|
_running = false;
|
|
|
|
if (session_) session_->stop();
|
|
|
|
return AE_NO;
|
|
}
|
|
|
|
void record_desktop_wgc::on_frame(const wgc_session::wgc_session_frame &frame) {
|
|
al_debug("wgc on frame");
|
|
AVFrame *av_frame = av_frame_alloc();
|
|
|
|
av_frame->pts = av_gettime_relative();
|
|
av_frame->pkt_dts = av_frame->pts;
|
|
// av_frame->pkt_pts = av_frame->pts;
|
|
|
|
av_frame->width = frame.width;
|
|
av_frame->height = frame.height;
|
|
av_frame->format = AV_PIX_FMT_BGRA;
|
|
av_frame->pict_type = AV_PICTURE_TYPE_NONE;
|
|
av_frame->pkt_size = frame.width * frame.height * 4;
|
|
|
|
av_image_fill_arrays(av_frame->data, av_frame->linesize, frame.data,
|
|
AV_PIX_FMT_BGRA, frame.width, frame.height, 1);
|
|
|
|
if (_on_data) _on_data(av_frame);
|
|
|
|
av_frame_free(&av_frame);
|
|
}
|
|
|
|
void record_desktop_wgc::clean_up() {
|
|
_inited = false;
|
|
|
|
if (session_) session_->release();
|
|
|
|
session_ = nullptr;
|
|
}
|
|
|
|
} // namespace am
|