#include "mainwidget.h"
#include <QVBoxLayout>
#include <QDebug>
#include <QScreen>
#include <QRect>
#include <QGuiApplication>
#include <QStyleOption>
#include <unistd.h>
#include <QObject>
#include <QImage>
#include <QMetaType>

#include "faceLib/common.h"
#include "opencv2/opencv.hpp"

void initDbPersons(int chan);

mainWidget::mainWidget(QWidget *parent)
    : QWidget(parent)
    , ft(0,1)
    , fe(1)
{
    qRegisterMetaType<ImgToShow>("ImgToShow");//注册ImgToShow类型
    qRegisterMetaType<std::string>("std::string");//注册ImgToShow类型
    initDbPersons(0);

    ft.setnSampleSize(640);
    printf("do Threads... \n");
    fe.run(0);
    ft.run(0);

    initParams();
    initTimer();

    QObject::connect(&ft,SIGNAL(drawImage(ImgToShow)),this,SLOT(on_ImageShow(ImgToShow)));
    QObject::connect(&ft,SIGNAL(signalAdvertise(bool)),this,SLOT(slotAdvertise(bool)));
    QObject::connect(&ft,SIGNAL(signalTips(std::string)),this,SLOT(slotTips(std::string)));
    QObject::connect(&fe,SIGNAL(signalTips(std::string)),this,SLOT(slotTips(std::string)));
}

mainWidget::~mainWidget()
{
    delete pTimer;
    delete pDateTime;

    delete pLabelVideo; //video
    delete pLabelAdv;
    delete pPixAdv; //advertise pic
    delete pStackWgt;
}

void mainWidget::initParams()
{
    QList<QScreen *> list_screen =  QGuiApplication::screens();  //多显示器
    QRect rect = list_screen.at(0)->geometry();
    qDebug() << rect.width() << rect.height();
//    this->resize(rect.width(), rect.height());

    this->resize(800, 600);

    this->setObjectName("mainWidget");
    this->setStyleSheet("QWidget#mainWidget{"
                        "border-image: url(:/res/background.png);}");
    pPixAdv = new QPixmap(":/res/adv.jpg");

    QVBoxLayout *vLayout = new QVBoxLayout(this);
    pLabelVideo = new myLabel;
    pLabelAdv = new QLabel;
    pStackWgt = new QStackedWidget(this);
    pStackWgt->addWidget(pLabelVideo);
    pStackWgt->addWidget(pLabelAdv);

    vLayout->addWidget(pStackWgt);
    vLayout->setMargin(10);
    vLayout->setStretchFactor(pStackWgt,10);
    this->setLayout(vLayout);

    this->show();//FullScreen();
}

void mainWidget::initTimer()
{
    pTimer = new QTimer;
    pTimer->setSingleShot(true);
    connect(pTimer,SIGNAL(timeout()),this,SLOT(slotClear()));
}

void mainWidget::paintEvent(QPaintEvent *)
{
    QStyleOption opt;
    opt.init(this);
    QPainter p(this);
    style()->drawPrimitive(QStyle::PE_Widget, &opt, &p, this);
}

void mainWidget::moveEvent(QMoveEvent *)
{
    this->move(QPoint(0,0));
}

void mainWidget::slotClear()
{
    pLabelVideo->setMyText(QString(""));
    pLabelVideo->setMyShow(false);
}

void mainWidget::slotAdvertise(bool value)
{
    qDebug()<< "void MainWidget::slotAdvertise()" << value;
    if (value)
    {
        std::cout << "111cpStackWgt: " << pStackWgt->width() << "x" << pStackWgt->height() << std::endl;
        QSize picSize(pStackWgt->width(),pStackWgt->height());
        QPixmap scaledPixmap = pPixAdv->scaled(picSize);
        pLabelAdv->setPixmap(scaledPixmap);
        pStackWgt->setCurrentIndex(1);
    } else {
        pStackWgt->setCurrentIndex(0);
    }
}

void initDbPersons(int chan)
{
    std::string basePath = "/data/disk2/face/pic/";
    std::vector<std::string> persons = {
            "chenshijun",
            "chenshijun2",
            "liuxin",
            "shiyajun",
            "suntianyu",
            "wangzong",
            "zhangzengfei"
    };
    cv::Mat frame;
    for (auto person:persons) {
        THFT_FaceInfo faceInfo;
        frame = cv::imread(basePath + person + ".jpg", cv::IMREAD_COLOR);
        int ret = THFT_FaceDetect(chan, frame.data, frame.cols, frame.rows, &faceInfo,
                                  1, 0);
        if (ret > 0) {
            THFI_FacePos facePos;
            BYTE *feature = new BYTE[2560];
            memcpy(&facePos, &faceInfo, sizeof(THFI_FacePos));
            ret = EF_Extract((short) chan, frame.data, frame.cols,
                             frame.rows, 3, &facePos, feature);
            if (ret == 1) {
                mFaceRec.insert(std::pair<std::string, BYTE *>(person, feature));
                printf("insert face :%s %p\n", person.c_str(), feature);
            }
        }
    }
}

void mainWidget::on_ImageShow(ImgToShow image)
{
    if((image.nWidth > 0) && (image.nHeight > 0) && (image.imgData != nullptr)){
        QImage qimg((const unsigned char*)image.imgData, image.nWidth, image.nHeight, image.nWidth * 3,
                            QImage::Format_RGB888);
        auto qpixmap = QPixmap::fromImage(qimg);
        QSize picSize(pStackWgt->width(),pStackWgt->height());
        QPixmap scaledPixmap = qpixmap.scaled(picSize);
        pLabelVideo->setMyPixmap(scaledPixmap);
        update();
        delete image.imgData;
    }
}

void mainWidget::slotTips(std::string str)
{
    pLabelVideo->setMyText(QString(str.data()));
    pLabelVideo->setMyShow(true);
    pTimer->start(2000);
}