#include "qqtmptablewidget.h"
#include "ui_qqtmptablewidget.h"
#include "qqtdefine.h"
#include "qqtgui.h"
#include "qqtobjectfactory.h"

QQTMPTableWidget::QQTMPTableWidget(QWidget *parent) :
    QWidget(parent),
    ui(new Ui::QQTMPTableWidget)
{
    ui->setupUi(this);

    m_db = newDatabaseConn();

    m_pageNum = 0;
    m_numPerPage = 14;
    selectionMode = QAbstractItemView::MultiSelection;
    altColor = true;
    resizeMode = QHeaderView::ResizeToContents;

    ui->btnJump->setFixedWidth(30);
    ui->btnLeft->setFixedWidth(30);
    ui->btnRight->setFixedWidth(30);
    ui->btnLeftHead->setFixedWidth(30);
    ui->btnRightHead->setFixedWidth(30);
    ui->leNum->setFixedWidth(40);
    ui->btnJump->setHidden(true);
    ui->leNum->setHidden(true);
    ui->btnRightHead->setHidden(true);
    ui->btnRight->setHidden(true);
    ui->btnLeftHead->setHidden(true);
    ui->btnLeft->setHidden(true);

}

QQTMPTableWidget::~QQTMPTableWidget()
{
    delete ui;
}

void QQTMPTableWidget::setDB(QString db)
{
    m_name = db;
    setDatabaseName(m_db, m_name);
}

void QQTMPTableWidget::setTable(QString table)
{
    m_table = table;
}

void QQTMPTableWidget::setRelation(int column, const QSqlRelation &relation)
{
    m_columnRelation.insert(column, relation);
}

void QQTMPTableWidget::query(QString filter)
{
    ptime(); //3ms

    //high?
    QQTSleep(478);

    QSqlQuery query(m_db);
    query.exec(QString("select count(*) from %1").arg(m_table));

    int num = 0;
    if(query.next())
    {
        num = query.value(0).toInt();
        pline() << "record num:" << num;
    }
    query.finish();

    int pageNum = 0;
    if(num%m_numPerPage>0)
        pageNum = num / m_numPerPage+1;
    else
        pageNum = num / m_numPerPage;

    m_pageNum = pageNum;

    int pix = 0;

#if 0
    /*
     * 首次做检查，每隔10张*14条
     */
    if(m_pageNum > ui->stWidgetPage->count())
        pix = m_pageNum - ui->stWidgetPage->count() + 10；
#else
    /*
     * 每次做检查，每次的延迟比较均匀 每隔1张*14条
     */
    pix = m_pageNum - ui->stWidgetPage->count();
#endif

    // pix >= 1 start work
    for(int i = 0; i < pix; i++)
    {
        //ptime();//89ms
        QQTTableWidget* page = new QQTTableWidget(this);
        //ptime();//2ms
        page->setDB(m_name);
        //ptime();//8ms
        page->setTable(m_table);
        //ptime();//14ms
        //query
        //ptime();//3ms
        QAbstractItemModel* m_model = page->model();
        for(int i = 0; i < m_model->columnCount(); i++)
            m_model->setHeaderData(
                        i, Qt::Horizontal,
                        m_headerData.value(i, m_model->headerData(i, Qt::Horizontal).toString()));
        //ptime();//1ms
        page->setSelectionMode(selectionMode);
        page->setAlternatingRowColors(altColor);
#ifdef __QT5__
        page->horizontalHeader()->setSectionResizeMode(resizeMode);
#else
        page->horizontalHeader()->setResizeMode(resizeMode);
#endif
        //ptime();//0ms
        for(int i = 0; i < m_model->columnCount(); i++)
#ifdef __QT5__
            page->horizontalHeader()->setSectionResizeMode(i, m_resizeMode.value(i, resizeMode));
#else
            page->horizontalHeader()->setResizeMode(i, m_resizeMode.value(i, resizeMode));
#endif
        //ptime();//QHash(338ms) QMap(372ms) 400ms(QHash等几乎不耗时)
        for(int i = 0; i < m_model->columnCount(); i++)
            page->setColumnWidth(i, m_columnWidth.value(i));
        //ptime();
        for(int i = 0; i < m_model->columnCount(); i++)
            page->setColumnHidden(i, m_columnHidden.value(i));
        //ptime();//219ms
        ui->stWidgetPage->addWidget(page);
        //ptime();
    }

    for(int i = 0; i < m_pageNum; i++)
    {
        QQTTableWidget* page = (QQTTableWidget*)(ui->stWidgetPage->widget(i));
        page->query(QString("%1 limit %2 offset %3")
                    .arg(filter)
                    .arg(m_numPerPage)
                    .arg(i*m_numPerPage));
    }
}

int QQTMPTableWidget::pageNum()
{
    return m_pageNum;
}

int QQTMPTableWidget::currentPage()
{
    return ui->stWidgetPage->currentIndex()+1;
}

void QQTMPTableWidget::setCurrentPage(int index)
{
    if(index < 1 || index > m_pageNum)
        return;
    ui->stWidgetPage->setCurrentIndex(index-1);
    ui->lbPos->setText(QString("%1/%2").arg(index).arg(m_pageNum));
}

void QQTMPTableWidget::setRecordNumPerPage(int num)
{
    m_numPerPage = num;
}

void QQTMPTableWidget::setColumnHidden(int key, bool value)
{
    m_columnHidden.insert(key, value);
}

void QQTMPTableWidget::setSelectionMode(QAbstractItemView::SelectionMode mode)
{
    selectionMode = mode;
}

void QQTMPTableWidget::setResizeMode(QHeaderView::ResizeMode mode)
{
    resizeMode = mode;
}

void QQTMPTableWidget::setResizeMode(int column, QHeaderView::ResizeMode mode)
{
    m_resizeMode.insert(column, mode);
}

void QQTMPTableWidget::setAlternatingRowColors(bool alt)
{
    altColor = alt;
}

void QQTMPTableWidget::setHeaderData(int i, Qt::Orientation, QString data)
{
    m_headerData.insert(i, data);
}

void QQTMPTableWidget::setColumnWidth(int column, int width)
{
    m_columnWidth.insert(column, width);
}

void QQTMPTableWidget::on_btnLeft_clicked()
{
    int index = ui->stWidgetPage->currentIndex();
    if(index > 0)
        index--;
    ui->stWidgetPage->setCurrentIndex(index);
    ui->lbPos->setText(QString("%1/%2").arg(index+1).arg(m_pageNum));
}

void QQTMPTableWidget::on_btnRight_clicked()
{
    int index = ui->stWidgetPage->currentIndex();
    if(index < m_pageNum-1)
        index++;
    ui->stWidgetPage->setCurrentIndex(index);
    ui->lbPos->setText(QString("%1/%2").arg(index+1).arg(m_pageNum));
}

void QQTMPTableWidget::on_btnJump_clicked()
{
    int num = ui->leNum->text().toInt();
    if(num <= m_pageNum && num > 0)
    {
        ui->stWidgetPage->setCurrentIndex(num-1);
        ui->lbPos->setText(QString("%1/%2").arg(num).arg(m_pageNum));
    }
}

void QQTMPTableWidget::on_btnLeftHead_clicked()
{
    ui->stWidgetPage->setCurrentIndex(0);
    ui->lbPos->setText(QString("%1/%2").arg(1).arg(m_pageNum));
}

void QQTMPTableWidget::on_btnRightHead_clicked()
{
    int index = m_pageNum-1;
    ui->stWidgetPage->setCurrentIndex(index);
    ui->lbPos->setText(QString("%1/%2").arg(index+1).arg(m_pageNum));
}

void QQTMPTableWidget::selectedRows(int column, QVector<QStringList> &strl)
{
    for(int i = 0; i < m_pageNum; i++)
    {
        QQTTableWidget* page = (QQTTableWidget*)ui->stWidgetPage->widget(i);
        QMap<int, QStringList> ids;
        page->selectedRows(column, ids);
        QMapIterator<int, QStringList> itor(ids);
        while (itor.hasNext()) {
            itor.next();
            strl.append(itor.value());
        }
    }
    return;
}

QQTTableWidget* QQTMPTableWidget::selectedRows(int column)
{
    QString sectionName;
    QSqlQuery query(m_db);
    query.exec(QString("select * from %1 limit 0").arg(m_table));
    sectionName = query.record().fieldName(column);
    query.finish();
    pline() << sectionName;

    QVector<QStringList> lid;
    selectedRows(column, lid);

    static QQTTableWidget* page = new QQTTableWidget(this);
    page->setDB(m_name);
    page->setTable(m_table);

    QString excp;
    for(int i = 0; i < lid.count() - 1; i++)
        excp += QString("%1 = '%2' or ").arg(sectionName).arg(lid[i].at(column));
    excp += QString("%1 = '%2'").arg(sectionName).arg(lid.last().at(column));
    page->query(excp);

    QAbstractItemModel* m_model = page->model();
    for(int i = 0; i < m_model->columnCount(); i++)
        m_model->setHeaderData(
                    i, Qt::Horizontal,
                    m_headerData.value(i, m_model->headerData(i, Qt::Horizontal).toString()));
    page->setSelectionMode(selectionMode);
    page->setAlternatingRowColors(altColor);
#ifdef __QT5__
    page->horizontalHeader()->setSectionResizeMode(resizeMode);
#else
    page->horizontalHeader()->setResizeMode(resizeMode);
#endif
    for(int i = 0; i < m_model->columnCount(); i++)
    {
        page->setColumnHidden(i, m_columnHidden.value(i));
#ifdef __QT5__
        page->horizontalHeader()->setSectionResizeMode(i, m_resizeMode.value(i, resizeMode));
#else
        page->horizontalHeader()->setResizeMode(i, m_resizeMode.value(i, resizeMode));
#endif
        page->setColumnWidth(i, m_columnWidth.value(i));
    }

    return page;
}


void QQTMPTableWidget::removeSelectedRows(int column)
{
    for(int i = 0; i < m_pageNum; i++)
    {
        QQTTableWidget* page = (QQTTableWidget*)ui->stWidgetPage->widget(i);
        QMap<int, QStringList> ids;
        page->selectedRows(column, ids);
        page->removeRows(column, ids.values());
    }
}