#include "mainwindow.h"
#include "ui_mainwindow.h"
#include "rle.h"

#include <QMatrix>
#include <QDebug>
#include <QPainter>
#include <QDesktopWidget>

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;
            }

//            static bool first = true;
//            if(first) {
//                first = false;
//                int total_bytes = w*h*2;
//                for(int i = 0; i < total_bytes; i++) {
//                    printf("%02x ", _dat[i]);
//                    if(i != 0 && i % 16 == 0)
//                        printf("\n");
//                }
//                fflush(stdout);
//            }

            out = QImage(_dat, w, h, QImage::Format_RGB16);
            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;
}


MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainWindow)
{
    m_shown = false;
    m_show_bg = true;
    m_bg = QImage(":/tp-player/res/bg");
    m_pt_normal = QImage(":/tp-player/res/cursor.png");
    m_update_img = false;

    qDebug() << m_pt_normal.width() << "x" << m_pt_normal.height();

    ui->setupUi(this);

    //qRegisterMetaType<update_data*>("update_data");

    // frame-less window.
//#ifdef __APPLE__
//    setWindowFlags(Qt::FramelessWindowHint | Qt::MSWindowsFixedSizeDialogHint | Qt::Window);
//    OSXCode::fixWin(winId());
//#else
//    setWindowFlags(Qt::FramelessWindowHint | Qt::MSWindowsFixedSizeDialogHint | windowFlags());
//#endif //__APPLE__

    resize(m_bg.width(), m_bg.height());

    setWindowFlags(windowFlags()&~Qt::WindowMaximizeButtonHint);    // 禁止最大化按钮
    setFixedSize(m_bg.width(), m_bg.height());                     // 禁止拖动窗口大小

    connect(&m_thr_play, SIGNAL(signal_update_data(update_data*)), this, SLOT(on_update_data(update_data*)));
}

MainWindow::~MainWindow()
{
    m_thr_play.stop();
    m_thr_play.wait();
    delete ui;
}

void MainWindow::paintEvent(QPaintEvent *)
{
    QPainter painter(this);

    if(m_show_bg) {
        //qDebug() << "draw bg.";
        painter.setBrush(Qt::black);
        painter.drawRect(this->rect());

        int x = (rect().width() - m_bg.width()) / 2;
        int y = (rect().height() - m_bg.height()) / 2;
        painter.drawImage(x, y, m_bg);
        //painter.drawPixmap(rect(), m_bg1);
    }

    else {
        if(m_update_img)
            painter.drawImage(m_img_update_x, m_img_update_y, m_img_update, 0, 0, m_img_update_w, m_img_update_h, Qt::AutoColor);
        else
            //qDebug() << "draw pt (" << m_pt.x << "," << m_pt.y << ")";
            painter.drawImage(m_pt.x, m_pt.y, m_pt_normal);
    }


    if(!m_shown) {
        m_shown = true;
        m_thr_play.start();
    }
}

void MainWindow::on_update_data(update_data* dat) {
    if(!dat)
        return;
//    qDebug() << "slot-event: " << dat->data_type();

    if(dat->data_type() == TYPE_DATA) {
        m_show_bg = false;

        if(dat->data_len() <= sizeof(TS_RECORD_PKG)) {
            qDebug() << "invalid record package(1).";
            delete dat;
            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).";
                delete dat;
                return;
            }

            memcpy(&m_pt, dat->data_buf() + sizeof(TS_RECORD_PKG), sizeof(TS_RECORD_RDP_POINTER));
            m_update_img = false;
            update();
            //update(m_pt.x - 8, m_pt.y - 8, 32, 32);
        }
        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).";
                delete dat;
                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);

            rdpimg2QImage(m_img_update, info->width, info->height, info->bitsPerPixel, (info->format == TS_RDP_IMG_BMP) ? true : false, img_dat, img_len);
            m_img_update_x = info->destLeft;
            m_img_update_y = info->destTop;
            m_img_update_w = info->destRight - info->destLeft + 1;
            m_img_update_h = info->destBottom - info->destTop + 1;

            static int count = 0;
            qDebug() << count << "img " << ((info->format == TS_RDP_IMG_BMP) ? "+" : " ") << " (" << m_img_update_x << "," << m_img_update_y << "), [" << m_img_update.width() << "x" << m_img_update.height() << "]";
            count++;

            m_update_img = true;
            update(m_img_update_x, m_img_update_y, m_img_update_w, m_img_update_h);
        }

        delete dat;
        return;
    }


    if(dat->data_type() == TYPE_HEADER_INFO) {
        if(dat->data_len() != sizeof(TS_RECORD_HEADER)) {
            qDebug() << "invalid record header.";
            delete dat;
            return;
        }
        memcpy(&m_rec_hdr, dat->data_buf(), sizeof(TS_RECORD_HEADER));
        delete dat;

        qDebug() << "resize (" << m_rec_hdr.basic.width << "," << m_rec_hdr.basic.height << ")";
        if(m_rec_hdr.basic.width > 0 && m_rec_hdr.basic.height > 0) {
            m_win_board_w = frameGeometry().width() - geometry().width();
            m_win_board_h = frameGeometry().height() - geometry().height();

            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);

//            QDesktopWidget *desktop = QApplication::desktop(); // =qApp->desktop();也可以
//            //move((desktop->width() - this->width())/2, (desktop->height() - this->height())/2);
//            move(10, (desktop->height() - this->height())/2);
        }

        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;
    }


    delete dat;
}