538 lines
18 KiB
C++
538 lines
18 KiB
C++
#include "mainwindow.h"
|
||
#include "ui_mainwindow.h"
|
||
#include "rle.h"
|
||
|
||
#include <QMatrix>
|
||
#include <QDebug>
|
||
#include <QPainter>
|
||
#include <QDesktopWidget>
|
||
#include <QPaintEvent>
|
||
#include <QMessageBox>
|
||
#include <QDialogButtonBox>
|
||
|
||
bool rdpimg2QImage(QImage& out, int w, int h, int bitsPerPixel, bool isCompressed, uint8_t* dat, uint32_t len) {
|
||
switch(bitsPerPixel) {
|
||
case 15:
|
||
if(isCompressed) {
|
||
uint8_t* _dat = (uint8_t*)calloc(1, w*h*2);
|
||
if(!bitmap_decompress1(_dat, w, h, dat, len)) {
|
||
free(_dat);
|
||
return false;
|
||
}
|
||
out = QImage(_dat, w, h, QImage::Format_RGB555);
|
||
free(_dat);
|
||
}
|
||
else {
|
||
out = QImage(dat, w, h, QImage::Format_RGB555).transformed(QMatrix(1.0, 0.0, 0.0, -1.0, 0.0, 0.0)) ;
|
||
}
|
||
break;
|
||
case 16:
|
||
if(isCompressed) {
|
||
|
||
uint8_t* _dat = (uint8_t*)calloc(1, w*h*2);
|
||
if(!bitmap_decompress2(_dat, w, h, dat, len)) {
|
||
free(_dat);
|
||
return false;
|
||
}
|
||
|
||
// TODO: 这里需要进一步优化，直接操作QImage的buffer。
|
||
// QTime t1;
|
||
// t1.start();
|
||
|
||
out = QImage(w, h, QImage::Format_RGB16);
|
||
for(int y = 0; y < h; y++) {
|
||
for(int x = 0; x < w; x++) {
|
||
uint16 a = ((uint16*)_dat)[y * w + x];
|
||
uint8 r = ((a & 0xf800) >> 11) * 255 / 31;
|
||
uint8 g = ((a & 0x07e0) >> 5) * 255 / 63;
|
||
uint8 b = (a & 0x001f) * 255 / 31;
|
||
out.setPixelColor(x, y, QColor(r,g,b));
|
||
}
|
||
}
|
||
// qDebug("parse: %dB, %dms", len, t1.elapsed());
|
||
|
||
free(_dat);
|
||
}
|
||
else {
|
||
out = QImage(dat, w, h, QImage::Format_RGB16).transformed(QMatrix(1.0, 0.0, 0.0, -1.0, 0.0, 0.0)) ;
|
||
}
|
||
break;
|
||
case 24:
|
||
qDebug() << "--------NOT support 24";
|
||
break;
|
||
case 32:
|
||
qDebug() << "--------NOT support 32";
|
||
break;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
static inline int min(int a, int b){
|
||
return a < b ? a : b;
|
||
}
|
||
|
||
static inline int max(int a, int b){
|
||
return a > b ? a : b;
|
||
}
|
||
|
||
MainWindow::MainWindow(QWidget *parent) :
|
||
QMainWindow(parent),
|
||
ui(new Ui::MainWindow)
|
||
{
|
||
//m_shown = false;
|
||
m_show_default = true;
|
||
m_bar_shown = false;
|
||
m_bar_fade_in = false;
|
||
m_bar_fading = false;
|
||
m_bar_opacity = 1.0;
|
||
m_show_message = false;
|
||
memset(&m_pt, 0, sizeof(TS_RECORD_RDP_POINTER));
|
||
|
||
m_thr_play = nullptr;
|
||
m_play_state = PLAY_STATE_UNKNOWN;
|
||
m_thr_data = nullptr;
|
||
|
||
// m_dl = nullptr;
|
||
|
||
ui->setupUi(this);
|
||
|
||
ui->centralWidget->setMouseTracking(true);
|
||
setMouseTracking(true);
|
||
|
||
// frame-less window.
|
||
//#ifdef __APPLE__
|
||
// setWindowFlags(Qt::FramelessWindowHint | Qt::MSWindowsFixedSizeDialogHint | Qt::Window);
|
||
// OSXCode::fixWin(winId());
|
||
//#else
|
||
// setWindowFlags(Qt::FramelessWindowHint | Qt::MSWindowsFixedSizeDialogHint | windowFlags());
|
||
//#endif //__APPLE__
|
||
|
||
m_pt_normal.load(":/tp-player/res/cursor.png");
|
||
m_default_bg.load(":/tp-player/res/bg.png");
|
||
|
||
m_canvas = QPixmap(m_default_bg.width(), m_default_bg.height());
|
||
QPainter pp(&m_canvas);
|
||
pp.drawPixmap(0, 0, m_default_bg, 0, 0, m_default_bg.width(), m_default_bg.height());
|
||
|
||
|
||
setWindowFlags(windowFlags()&~Qt::WindowMaximizeButtonHint); // 禁止最大化按钮
|
||
setFixedSize(m_default_bg.width(), m_default_bg.height()); // 禁止拖动窗口大小
|
||
|
||
if(!m_bar.init(this)) {
|
||
qDebug("bar init failed.");
|
||
return;
|
||
}
|
||
|
||
|
||
// connect(&m_thr_play, SIGNAL(signal_update_data(update_data*)), this, SLOT(_do_update_data(update_data*)));
|
||
connect(&m_timer_first_run, SIGNAL(timeout()), this, SLOT(_do_first_run()));
|
||
connect(&m_timer_bar_fade, SIGNAL(timeout()), this, SLOT(_do_bar_fade()));
|
||
connect(&m_timer_bar_delay_hide, SIGNAL(timeout()), this, SLOT(_do_bar_delay_hide()));
|
||
|
||
m_timer_first_run.setSingleShot(true);
|
||
m_timer_first_run.start(500);
|
||
}
|
||
|
||
MainWindow::~MainWindow()
|
||
{
|
||
if(m_thr_play) {
|
||
m_thr_play->stop();
|
||
//m_thr_play->wait();
|
||
//qDebug() << "play thread stoped.";
|
||
|
||
disconnect(m_thr_play, SIGNAL(signal_update_data(UpdateData*)), this, SLOT(_do_update_data(UpdateData*)));
|
||
|
||
delete m_thr_play;
|
||
m_thr_play = nullptr;
|
||
}
|
||
|
||
if(m_thr_data) {
|
||
m_thr_data->stop();
|
||
disconnect(m_thr_data, SIGNAL(signal_update_data(UpdateData*)), this, SLOT(_do_update_data(UpdateData*)));
|
||
// disconnect(m_thr_data, SIGNAL(signal_download(DownloadParam*)), this, SLOT(_do_download(DownloadParam*)));
|
||
delete m_thr_data;
|
||
m_thr_data = nullptr;
|
||
}
|
||
|
||
// if(m_dl) {
|
||
// delete m_dl;
|
||
// m_dl = nullptr;
|
||
// }
|
||
|
||
delete ui;
|
||
}
|
||
|
||
void MainWindow::set_resource(const QString &res) {
|
||
m_res = res;
|
||
}
|
||
|
||
void MainWindow::_do_first_run() {
|
||
m_thr_data = new ThrData(this, m_res);
|
||
connect(m_thr_data, SIGNAL(signal_update_data(UpdateData*)), this, SLOT(_do_update_data(UpdateData*)));
|
||
// connect(m_thr_data, SIGNAL(signal_download(DownloadParam*)), this, SLOT(_do_download(DownloadParam*)));
|
||
m_thr_data->start();
|
||
|
||
_start_play_thread();
|
||
}
|
||
|
||
void MainWindow::_start_play_thread() {
|
||
if(m_thr_play) {
|
||
m_thr_play->stop();
|
||
//m_thr_play->wait();
|
||
|
||
disconnect(m_thr_play, SIGNAL(signal_update_data(UpdateData*)), this, SLOT(_do_update_data(UpdateData*)));
|
||
|
||
delete m_thr_play;
|
||
m_thr_play = nullptr;
|
||
}
|
||
|
||
m_thr_play = new ThrPlay();
|
||
connect(m_thr_play, SIGNAL(signal_update_data(UpdateData*)), this, SLOT(_do_update_data(UpdateData*)));
|
||
|
||
m_thr_play->speed(m_bar.get_speed());
|
||
m_thr_play->start();
|
||
}
|
||
|
||
void MainWindow::speed(int s) {
|
||
if(m_thr_play)
|
||
m_thr_play->speed(s);
|
||
}
|
||
|
||
void MainWindow::paintEvent(QPaintEvent *e)
|
||
{
|
||
QPainter painter(this);
|
||
|
||
if(m_show_default) {
|
||
painter.drawPixmap(e->rect(), m_default_bg, e->rect());
|
||
}
|
||
else {
|
||
painter.drawPixmap(e->rect(), m_canvas, e->rect());
|
||
|
||
QRect rcpt(m_pt_normal.rect());
|
||
rcpt.moveTo(m_pt.x - m_pt_normal.width()/2, m_pt.y-m_pt_normal.height()/2);
|
||
if(e->rect().intersects(rcpt)) {
|
||
painter.drawPixmap(m_pt.x-m_pt_normal.width()/2, m_pt.y-m_pt_normal.height()/2, m_pt_normal);
|
||
}
|
||
|
||
// {
|
||
// QRect rc_draw = e->rect();
|
||
// QRect rc(m_rc_message);
|
||
// //rc.moveTo(m_rc.left()+rc.left(), m_rc.top() + rc.top());
|
||
|
||
// int from_x = max(rc_draw.left(), rc.left()) - rc.left();
|
||
// int from_y = max(rc_draw.top(), rc.top()) - rc.top();
|
||
// int w = min(rc.right(), rc_draw.right()) - rc.left() - from_x + 1;
|
||
// int h = min(rc.bottom(), rc_draw.bottom()) - rc.top() - from_y + 1;
|
||
// int to_x = rc.left() + from_x;
|
||
// int to_y = rc.top() + from_y;
|
||
// painter.drawPixmap(to_x, to_y, m_img_message, from_x, from_y, w, h);
|
||
// }
|
||
|
||
// 绘制浮动控制窗
|
||
if(m_bar_fading) {
|
||
painter.setOpacity(m_bar_opacity);
|
||
m_bar.draw(painter, e->rect());
|
||
}
|
||
else if(m_bar_shown) {
|
||
m_bar.draw(painter, e->rect());
|
||
}
|
||
}
|
||
|
||
if(m_show_message) {
|
||
QRect rc_draw = e->rect();
|
||
QRect rc(m_rc_message);
|
||
//rc.moveTo(m_rc.left()+rc.left(), m_rc.top() + rc.top());
|
||
|
||
int from_x = max(rc_draw.left(), rc.left()) - rc.left();
|
||
int from_y = max(rc_draw.top(), rc.top()) - rc.top();
|
||
int w = min(rc.right(), rc_draw.right()) - rc.left() - from_x + 1;
|
||
int h = min(rc.bottom(), rc_draw.bottom()) - rc.top() - from_y + 1;
|
||
int to_x = rc.left() + from_x;
|
||
int to_y = rc.top() + from_y;
|
||
painter.drawPixmap(to_x, to_y, m_img_message, from_x, from_y, w, h);
|
||
}
|
||
|
||
// if(!m_shown) {
|
||
// m_shown = true;
|
||
// //m_thr_play.start();
|
||
// _start_play_thread();
|
||
// }
|
||
}
|
||
|
||
void MainWindow::pause() {
|
||
if(m_play_state != PLAY_STATE_RUNNING)
|
||
return;
|
||
m_thr_play->pause();
|
||
m_play_state = PLAY_STATE_PAUSE;
|
||
}
|
||
|
||
void MainWindow::resume() {
|
||
if(m_play_state == PLAY_STATE_PAUSE)
|
||
m_thr_play->resume();
|
||
else if(m_play_state == PLAY_STATE_STOP)
|
||
//m_thr_play->start();
|
||
_start_play_thread();
|
||
|
||
m_play_state = PLAY_STATE_RUNNING;
|
||
}
|
||
|
||
void MainWindow::_do_update_data(UpdateData* dat) {
|
||
if(!dat)
|
||
return;
|
||
|
||
UpdateDataHelper data_helper(dat);
|
||
|
||
if(dat->data_type() == TYPE_DATA) {
|
||
|
||
if(dat->data_len() <= sizeof(TS_RECORD_PKG)) {
|
||
qDebug() << "invalid record package(1).";
|
||
return;
|
||
}
|
||
|
||
TS_RECORD_PKG* pkg = (TS_RECORD_PKG*)dat->data_buf();
|
||
|
||
if(pkg->type == TS_RECORD_TYPE_RDP_POINTER) {
|
||
if(dat->data_len() != sizeof(TS_RECORD_PKG) + sizeof(TS_RECORD_RDP_POINTER)) {
|
||
qDebug() << "invalid record package(2).";
|
||
return;
|
||
}
|
||
|
||
TS_RECORD_RDP_POINTER pt;
|
||
memcpy(&pt, &m_pt, sizeof(TS_RECORD_RDP_POINTER));
|
||
|
||
// 更新虚拟鼠标信息，这样下一次绘制界面时就会在新的位置绘制出虚拟鼠标
|
||
memcpy(&m_pt, dat->data_buf() + sizeof(TS_RECORD_PKG), sizeof(TS_RECORD_RDP_POINTER));
|
||
update(m_pt.x - m_pt_normal.width()/2, m_pt.y - m_pt_normal.width()/2, m_pt_normal.width(), m_pt_normal.height());
|
||
|
||
update(pt.x - m_pt_normal.width()/2, pt.y - m_pt_normal.width()/2, m_pt_normal.width(), m_pt_normal.height());
|
||
}
|
||
else if(pkg->type == TS_RECORD_TYPE_RDP_IMAGE) {
|
||
if(dat->data_len() <= sizeof(TS_RECORD_PKG) + sizeof(TS_RECORD_RDP_IMAGE_INFO)) {
|
||
qDebug() << "invalid record package(3).";
|
||
return;
|
||
}
|
||
|
||
TS_RECORD_RDP_IMAGE_INFO* info = (TS_RECORD_RDP_IMAGE_INFO*)(dat->data_buf() + sizeof(TS_RECORD_PKG));
|
||
uint8_t* img_dat = dat->data_buf() + sizeof(TS_RECORD_PKG) + sizeof(TS_RECORD_RDP_IMAGE_INFO);
|
||
uint32_t img_len = dat->data_len() - sizeof(TS_RECORD_PKG) - sizeof(TS_RECORD_RDP_IMAGE_INFO);
|
||
|
||
QImage img_update;
|
||
rdpimg2QImage(img_update, info->width, info->height, info->bitsPerPixel, (info->format == TS_RDP_IMG_BMP) ? true : false, img_dat, img_len);
|
||
|
||
int x = info->destLeft;
|
||
int y = info->destTop;
|
||
int w = info->destRight - info->destLeft + 1;
|
||
int h = info->destBottom - info->destTop + 1;
|
||
|
||
QPainter pp(&m_canvas);
|
||
pp.drawImage(x, y, img_update, 0, 0, w, h, Qt::AutoColor);
|
||
|
||
update(x, y, w, h);
|
||
}
|
||
|
||
return;
|
||
}
|
||
|
||
else if(dat->data_type() == TYPE_PLAYED_MS) {
|
||
m_bar.update_passed_time(dat->played_ms());
|
||
return;
|
||
}
|
||
|
||
else if(dat->data_type() == TYPE_MESSAGE) {
|
||
m_show_message = true;
|
||
|
||
qDebug("1message, w=%d, h=%d", m_canvas.width(), m_canvas.height());
|
||
// if(0 == m_canvas.width()) {
|
||
// QMessageBox::warning(nullptr, QGuiApplication::applicationDisplayName(), dat->message());
|
||
// return;
|
||
// }
|
||
|
||
QPainter pp(&m_canvas);
|
||
QRect rcWin(0, 0, m_canvas.width(), m_canvas.height());
|
||
pp.drawText(rcWin, Qt::AlignLeft|Qt::TextDontPrint, dat->message(), &m_rc_message);
|
||
|
||
qDebug("2message, w=%d, h=%d", m_rc_message.width(), m_rc_message.height());
|
||
m_rc_message.setWidth(m_rc_message.width()+60);
|
||
m_rc_message.setHeight(m_rc_message.height()+60);
|
||
|
||
m_img_message = QPixmap(m_rc_message.width(), m_rc_message.height());
|
||
m_img_message.fill(Qt::transparent);
|
||
QPainter pm(&m_img_message);
|
||
pm.setPen(QColor(255,255,255,153));
|
||
pm.fillRect(m_rc_message, QColor(0,0,0,190));
|
||
|
||
QRect rcRect(m_rc_message);
|
||
rcRect.setWidth(rcRect.width()-1);
|
||
rcRect.setHeight(rcRect.height()-1);
|
||
pm.drawRect(rcRect);
|
||
|
||
QRect rcText(m_rc_message);
|
||
rcText.setLeft(30);
|
||
rcText.setTop(30);
|
||
pm.drawText(rcText, Qt::AlignLeft, dat->message());
|
||
m_rc_message.moveTo(
|
||
(m_canvas.width() - m_rc_message.width())/2,
|
||
(m_canvas.height() - m_rc_message.height())/3
|
||
);
|
||
|
||
update(m_rc_message.x(), m_rc_message.y(), m_rc_message.width(), m_rc_message.height());
|
||
|
||
return;
|
||
}
|
||
|
||
else if(dat->data_type() == TYPE_ERROR) {
|
||
QMessageBox::critical(this, QGuiApplication::applicationDisplayName(), dat->message());
|
||
QApplication::instance()->exit(0);
|
||
return;
|
||
}
|
||
|
||
// 这是播放开始时收到的第一个数据包
|
||
else if(dat->data_type() == TYPE_HEADER_INFO) {
|
||
if(dat->data_len() != sizeof(TS_RECORD_HEADER)) {
|
||
qDebug() << "invalid record header.";
|
||
return;
|
||
}
|
||
memcpy(&m_rec_hdr, dat->data_buf(), sizeof(TS_RECORD_HEADER));
|
||
|
||
qDebug() << "resize (" << m_rec_hdr.basic.width << "," << m_rec_hdr.basic.height << ")";
|
||
|
||
//if(m_canvas.width() != m_rec_hdr.basic.width && m_canvas.height() != m_rec_hdr.basic.height) {
|
||
m_canvas = QPixmap(m_rec_hdr.basic.width, m_rec_hdr.basic.height);
|
||
|
||
//m_win_board_w = frameGeometry().width() - geometry().width();
|
||
//m_win_board_h = frameGeometry().height() - geometry().height();
|
||
|
||
QDesktopWidget *desktop = QApplication::desktop(); // =qApp->desktop();也可以
|
||
qDebug("desktop w:%d,h:%d, this w:%d,h:%d", desktop->width(), desktop->height(), width(), height());
|
||
//move((desktop->width() - this->width())/2, (desktop->height() - this->height())/2);
|
||
move(10, (desktop->height() - m_rec_hdr.basic.height)/2);
|
||
|
||
//setFixedSize(m_rec_hdr.basic.width + m_win_board_w, m_rec_hdr.basic.height + m_win_board_h);
|
||
//resize(m_rec_hdr.basic.width + m_win_board_w, m_rec_hdr.basic.height + m_win_board_h);
|
||
//resize(m_rec_hdr.basic.width, m_rec_hdr.basic.height);
|
||
setFixedSize(m_rec_hdr.basic.width, m_rec_hdr.basic.height);
|
||
//}
|
||
|
||
m_canvas.fill(QColor(38, 73, 111));
|
||
|
||
m_show_default = false;
|
||
repaint();
|
||
|
||
m_bar.start(m_rec_hdr.info.time_ms, 640);
|
||
m_bar_shown = true;
|
||
m_play_state = PLAY_STATE_RUNNING;
|
||
|
||
update(m_bar.rc());
|
||
|
||
m_bar_fade_in = false;
|
||
m_bar_fading = true;
|
||
m_timer_bar_delay_hide.start(2000);
|
||
|
||
QString title;
|
||
if (m_rec_hdr.basic.conn_port == 3389)
|
||
title.sprintf("[%s] %s@%s [Teleport-RDP录像回放]", m_rec_hdr.basic.acc_username, m_rec_hdr.basic.user_username, m_rec_hdr.basic.conn_ip);
|
||
else
|
||
title.sprintf("[%s] %s@%s:%d [Teleport-RDP录像回放]", m_rec_hdr.basic.acc_username, m_rec_hdr.basic.user_username, m_rec_hdr.basic.conn_ip, m_rec_hdr.basic.conn_port);
|
||
|
||
setWindowTitle(title);
|
||
|
||
return;
|
||
}
|
||
|
||
|
||
else if(dat->data_type() == TYPE_END) {
|
||
m_bar.end();
|
||
m_play_state = PLAY_STATE_STOP;
|
||
|
||
return;
|
||
}
|
||
}
|
||
|
||
void MainWindow::_do_bar_delay_hide() {
|
||
m_bar_fading = true;
|
||
m_timer_bar_delay_hide.stop();
|
||
m_timer_bar_fade.stop();
|
||
m_timer_bar_fade.start(50);
|
||
}
|
||
|
||
void MainWindow::_do_bar_fade() {
|
||
if(m_bar_fade_in) {
|
||
if(m_bar_opacity < 1.0)
|
||
m_bar_opacity += 0.3;
|
||
if(m_bar_opacity >= 1.0) {
|
||
m_bar_opacity = 1.0;
|
||
m_bar_shown = true;
|
||
m_bar_fading = false;
|
||
m_timer_bar_fade.stop();
|
||
}
|
||
}
|
||
else {
|
||
if(m_bar_opacity > 0.0)
|
||
m_bar_opacity -= 0.2;
|
||
if(m_bar_opacity <= 0.0) {
|
||
m_bar_opacity = 0.0;
|
||
m_bar_shown = false;
|
||
m_bar_fading = false;
|
||
m_timer_bar_fade.stop();
|
||
}
|
||
}
|
||
|
||
update(m_bar.rc());
|
||
}
|
||
|
||
//void MainWindow::_do_download(DownloadParam* param) {
|
||
// qDebug("MainWindow::_do_download(). %s %s %s", param->url.toStdString().c_str(), param->sid.toStdString().c_str(), param->fname.toStdString().c_str());
|
||
|
||
// if(m_dl) {
|
||
// delete m_dl;
|
||
// m_dl = nullptr;
|
||
// }
|
||
|
||
// m_dl = new Downloader();
|
||
// m_dl->run(&m_nam, param->url, param->sid, param->fname);
|
||
//}
|
||
|
||
void MainWindow::mouseMoveEvent(QMouseEvent *e) {
|
||
if(!m_show_default) {
|
||
QRect rc = m_bar.rc();
|
||
if(e->y() > rc.top() - 20 && e->y() < rc.bottom() + 20) {
|
||
if((!m_bar_shown && !m_bar_fading) || (m_bar_fading && !m_bar_fade_in)) {
|
||
m_bar_fade_in = true;
|
||
m_bar_fading = true;
|
||
|
||
m_timer_bar_delay_hide.stop();
|
||
m_timer_bar_fade.stop();
|
||
m_timer_bar_fade.start(50);
|
||
}
|
||
|
||
if(rc.contains(e->pos()))
|
||
m_bar.onMouseMove(e->x(), e->y());
|
||
}
|
||
else {
|
||
if((m_bar_shown && !m_bar_fading) || (m_bar_fading && m_bar_fade_in)) {
|
||
m_bar_fade_in = false;
|
||
m_bar_fading = true;
|
||
m_timer_bar_fade.stop();
|
||
m_timer_bar_delay_hide.stop();
|
||
|
||
if(m_bar_opacity != 1.0)
|
||
m_timer_bar_fade.start(50);
|
||
else
|
||
m_timer_bar_delay_hide.start(1000);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
void MainWindow::mousePressEvent(QMouseEvent *e) {
|
||
// QApplication::instance()->exit(0);
|
||
// return;
|
||
if(!m_show_default) {
|
||
QRect rc = m_bar.rc();
|
||
if(rc.contains(e->pos())) {
|
||
m_bar.onMousePress(e->x(), e->y());
|
||
}
|
||
}
|
||
}
|