mirror of https://github.com/tp4a/teleport
434 lines
13 KiB
C++
434 lines
13 KiB
C++
#include "mainwindow.h"
|
|
#include "ui_mainwindow.h"
|
|
|
|
#include <QMatrix>
|
|
#include <QDebug>
|
|
#include <QPainter>
|
|
#include <QDesktopWidget>
|
|
#include <QPaintEvent>
|
|
#include <QMessageBox>
|
|
#include <QDialogButtonBox>
|
|
|
|
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_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_disable_draw = false;
|
|
|
|
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_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();
|
|
|
|
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*)));
|
|
delete m_thr_data;
|
|
m_thr_data = 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*)));
|
|
m_thr_data->start(QThread::TimeCriticalPriority);
|
|
|
|
m_thr_play = new ThrPlay(this);
|
|
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::set_speed(int s) {
|
|
if(m_thr_play)
|
|
m_thr_play->speed(s);
|
|
}
|
|
|
|
void MainWindow::set_skip(bool s) {
|
|
if(m_thr_play)
|
|
m_thr_play->skip(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);
|
|
}
|
|
|
|
// 绘制浮动控制窗
|
|
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);
|
|
|
|
if(e->rect().intersects(rc)) {
|
|
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);
|
|
}
|
|
}
|
|
}
|
|
|
|
void MainWindow::pause() {
|
|
if(m_play_state != PLAY_STATE_RUNNING)
|
|
return;
|
|
m_thr_play->pause();
|
|
m_play_state = PLAY_STATE_PAUSE;
|
|
}
|
|
|
|
void MainWindow::resume(bool relocate, uint32_t ms) {
|
|
if(m_play_state == PLAY_STATE_PAUSE) {
|
|
if(relocate)
|
|
m_thr_data->restart(ms);
|
|
m_thr_play->resume(relocate, ms);
|
|
}
|
|
else if(m_play_state == PLAY_STATE_STOP) {
|
|
m_thr_data->restart(ms);
|
|
m_thr_play->resume(true, ms);
|
|
}
|
|
|
|
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_POINTER) {
|
|
TS_RECORD_RDP_POINTER pt;
|
|
memcpy(&pt, &m_pt, sizeof(TS_RECORD_RDP_POINTER));
|
|
|
|
// 更新虚拟鼠标信息，这样下一次绘制界面时就会在新的位置绘制出虚拟鼠标
|
|
memcpy(&m_pt, dat->get_pointer(), 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());
|
|
|
|
return;
|
|
}
|
|
else if(dat->data_type() == TYPE_IMAGE) {
|
|
const UpdateImages uimgs = dat->get_images();
|
|
if(uimgs.size() == 0)
|
|
return;
|
|
|
|
if(uimgs.size() > 1 && !m_disable_draw) {
|
|
// 禁止界面更新
|
|
setUpdatesEnabled(false);
|
|
}
|
|
|
|
|
|
QPainter pp(&m_canvas);
|
|
for(int i = 0; i < uimgs.size(); ++i) {
|
|
pp.drawImage(uimgs[i].x, uimgs[i].y, *(uimgs[i].img), 0, 0, uimgs[i].w, uimgs[i].h, Qt::AutoColor);
|
|
|
|
if(!m_disable_draw)
|
|
update(uimgs[i].x, uimgs[i].y, uimgs[i].w, uimgs[i].h);
|
|
}
|
|
|
|
|
|
if(uimgs.size() > 1 && !m_disable_draw) {
|
|
// 允许界面更新
|
|
setUpdatesEnabled(true);
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
else if(dat->data_type() == TYPE_PLAYED_MS) {
|
|
m_bar.update_passed_time(dat->played_ms());
|
|
return;
|
|
}
|
|
|
|
else if(dat->data_type() == TYPE_DISABLE_DRAW) {
|
|
// 禁止界面更新
|
|
m_disable_draw = true;
|
|
setUpdatesEnabled(false);
|
|
return;
|
|
}
|
|
|
|
else if(dat->data_type() == TYPE_ENABLE_DRAW) {
|
|
// 允许界面更新
|
|
m_disable_draw = false;
|
|
setUpdatesEnabled(true);
|
|
update();
|
|
return;
|
|
}
|
|
|
|
else if(dat->data_type() == TYPE_MESSAGE) {
|
|
if(dat->message().isEmpty()) {
|
|
m_show_message = false;
|
|
return;
|
|
}
|
|
|
|
m_show_message = true;
|
|
|
|
qDebug("1message, w=%d, h=%d", m_canvas.width(), m_canvas.height());
|
|
|
|
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) {
|
|
TS_RECORD_HEADER* hdr = dat->get_header();
|
|
if(hdr == nullptr)
|
|
return;
|
|
memcpy(&m_rec_hdr, hdr, sizeof(TS_RECORD_HEADER));
|
|
|
|
qDebug() << "resize (" << m_rec_hdr.basic.width << "," << m_rec_hdr.basic.height << ")";
|
|
|
|
m_canvas = QPixmap(m_rec_hdr.basic.width, m_rec_hdr.basic.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(10, (desktop->height() - m_rec_hdr.basic.height)/2);
|
|
|
|
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 = QString(LOCAL8BIT("用户 %1 访问 %2 的 %3 账号").arg(m_rec_hdr.basic.user_username, m_rec_hdr.basic.conn_ip, m_rec_hdr.basic.acc_username));
|
|
}
|
|
else {
|
|
QString _port;
|
|
_port.sprintf("%d", m_rec_hdr.basic.conn_port);
|
|
title = QString(LOCAL8BIT("用户 %1 访问 %2:%3 的 %4 账号").arg(m_rec_hdr.basic.user_username, m_rec_hdr.basic.conn_ip, _port, m_rec_hdr.basic.acc_username));
|
|
}
|
|
|
|
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::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) {
|
|
if(!m_show_default) {
|
|
QRect rc = m_bar.rc();
|
|
if(rc.contains(e->pos())) {
|
|
m_bar.onMousePress(e->x(), e->y(), e->button());
|
|
}
|
|
}
|
|
}
|
|
|
|
void MainWindow::mouseReleaseEvent(QMouseEvent *e) {
|
|
m_bar.onMouseRelease(e->x(), e->y(), e->button());
|
|
}
|
|
|