mainwindow.cpp

#include "mainwindow.h"
#include "ui_mainwindow.h"
#include <PDFObject.h>
#include <QLayout>
#include <QIcon>
#include <string>
#include <QMessageBox>
#include <QDropEvent>
#include <QUrl>
#include <QFileDialog>
#include <QInputDialog>
#include <changecategorydialog.h>
#include <QCompleter>
#include <manualinputdialog.h>
#include <aboutdialog.h>

#include <QDesktopServices>

MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainWindow)
{
    //QDesktopServices::openUrl(QUrl("file:///C:/Documents and Settings/All Users/Desktop", QUrl::TolerantMode));
    ui->setupUi( this);
    //ui->listWidget->setCurrentRow(0);
    //ui->listWidget_2->setCurrentRow(0);

    wordList << "title:"<<"author:"<<"keyword:"<<"tag:"<<"note:";
    initToolbar();

    initMainArea();
    initIcons();

    initTabWidget();

    initTheDownLeftCorner();
    initTheUpLeftCorner();
    initSignalsAndSlots();
    initContextMenu();
    ui->listWidget->setCurrentRow(0);
    ui->listWidget_2->setCurrentRow(0);
}

MainWindow::~MainWindow()
{
    delete myDao;
    delete tempObj;
    delete ui;
}

void MainWindow::initContextMenu()
{
    ui->tableWidget->setContextMenuPolicy(Qt::ActionsContextMenu);
    //QMessageBox::about(this,"set",ui->tableWidget->itemAt(ui->tableWidget->currentRow(),ui->tableWidget->currentColumn())->text());
//    ui->tableWidget->addAction( ui->menu_action_add_file );
    ui->tableWidget->addAction( ui->menu_action_open_file );
    ui->tableWidget->addAction( ui->menu_action_remove_file );
    ui->tableWidget->addAction( ui->menu_action_change_category );
    ui->tableWidget->addAction( ui->menu_action_addto_favorite );
    ui->tableWidget->addAction( ui->menu_action_remove_favorite );

}

void MainWindow::initToolbar()
{

    QSize myIconSize(32,32);
    ui->mainToolBar->setIconSize( myIconSize );

    //添加文档按钮,添加文献的下拉菜单
    add_doc_button = new QToolButton( this );
    add_doc_button->setDefaultAction( ui->menu_action_add_file );

    QMenu * buttonMenu = new QMenu( this );
    buttonMenu->addAction( ui->menu_action_add_file );
    buttonMenu->addAction( ui->menu_action_add_category );
    buttonMenu->addAction( ui->menu_action_manual_input );
    buttonMenu->addAction( ui->menu_action_change_category );
    add_doc_button->setMenu( buttonMenu );
    add_doc_button->setPopupMode( QToolButton::MenuButtonPopup );
    ui->mainToolBar->addWidget( add_doc_button );

    //修改分类按钮
    change_category_button = new QToolButton( this );
    change_category_button->setDefaultAction( ui->menu_action_change_category );
    ui->mainToolBar->addWidget( change_category_button );

    //删除文档按钮
    del_doc_button = new QToolButton( this );
    del_doc_button->setDefaultAction( ui->menu_action_remove_file );
    ui->mainToolBar->addWidget( del_doc_button );

    //创建分类按钮
    create_category_button = new QToolButton( this );
    create_category_button->setDefaultAction( ui->menu_action_add_category );
    ui->mainToolBar->addWidget( create_category_button );

    //移除分类按钮
    remove_category_button = new QToolButton( this );
    remove_category_button->setDefaultAction( ui->menu_action_remove_category );
    ui->mainToolBar->addWidget( remove_category_button );

    //搜索框
    QWidget* spacer = new QWidget( this );

    spacer->setSizePolicy( QSizePolicy::Expanding, QSizePolicy::Expanding );
    ui->mainToolBar->addWidget( spacer );
    searchSquare = new QLineEdit( this );

    searchSquare->setText(tr(""));
    searchSquare->setStyleSheet("QLineEdit {border:1px solid gray; border-radius:9px}");
    //searchSquare->setStyleSheet(" border-radius:10px ");

    searchSquare->setSizePolicy( QSizePolicy::Fixed,QSizePolicy::Fixed );


    //补充wordList
    //if(!myDao){
        myDao = Dao::getInstance();
    //}


    QCompleter *completer= new QCompleter(wordList);
    completer->setCompletionMode(QCompleter::PopupCompletion);
    completer->setCaseSensitivity(Qt::CaseInsensitive);
    searchSquare->setCompleter(completer);
    ui->mainToolBar->addWidget( searchSquare );
//    QWidget* spacer2 = new QWidget( this );


//    spacer2->setSizePolicy( QSizePolicy::Fixed,QSizePolicy::Expanding );
//    ui->mainToolBar->addWidget( spacer2 );
    initToolbarStyle();
    //myDao = Dao::getInstance();

}

void MainWindow::initToolbarStyle()
{
    add_doc_button->setToolButtonStyle( Qt::ToolButtonTextUnderIcon );
    del_doc_button->setToolButtonStyle( Qt::ToolButtonTextUnderIcon );
    change_category_button->setToolButtonStyle(Qt::ToolButtonTextUnderIcon);
    create_category_button->setToolButtonStyle( Qt::ToolButtonTextUnderIcon );
    remove_category_button->setToolButtonStyle( Qt::ToolButtonTextUnderIcon );
}

void MainWindow::initIcons()
{
    //工具栏图标初始化
    QIcon add_doc_icon(":/toolbaricon/img/file_add.png");
    add_doc_button->setIcon( add_doc_icon );
    QIcon del_doc_icon(":/toolbaricon/img/file_delete.png");
    del_doc_button->setIcon( del_doc_icon );
    QIcon create_category_icon(":/toolbaricon/img/folder_add.png");
    create_category_button->setIcon( create_category_icon );
    QIcon change_category_icon(":/toolbaricon/img/folder_move.png");
    change_category_button->setIcon( change_category_icon );
    QIcon remove_category_icon(":/toolbaricon/img/folder_delete.png");
    remove_category_button->setIcon( remove_category_icon );
    //菜单图标初始化
}

void MainWindow::initMainArea()
{
    ui->mainToolBar->setMovable( false );
    //将中心区域设置为gridlayout，可以让内容填满窗口
    ui->centralWidget->setLayout( ui->horizontalLayout );
    //分配splitter的拉伸比例
    ui->splitter->setStretchFactor(0,1);
    ui->splitter->setStretchFactor(1,3);


}

void MainWindow::refreshTabWidget()
{
    //int x1=ui->listWidget->currentRow();
   // int x2=ui->listWidget_2->currentRow();
    //先获得分类和，下面的过滤信息
    QString kind = ui->listWidget->currentItem()->text();
    if(kind == "All")
        kind = "";
    if(kind == "Favorite")
        kind = "true";
    if(kind == "Unsorted")
        kind = "unsorted";
    QString filter = ui->listWidget_2->currentItem()->text();
    //if(!myDao){
    myDao = Dao::getInstance();
    //}
    vector<PDFObject*> pdfVector;
    if(filter=="All"){
        pdfVector = myDao->getAllDocs(kind.toStdString());
    }else if(ui->comboBox->currentIndex()==0){
        pdfVector = myDao->getDocByAuthor(filter.toStdString(),kind.toStdString());
    }else if(ui->comboBox->currentIndex()==1){
        pdfVector = myDao->getDocByTag(filter.toStdString(),kind.toStdString());
    }else if(ui->comboBox->currentIndex()==2){
        pdfVector = myDao->getDocByKeyword(filter.toStdString(),kind.toStdString());
    }

    for(int i=0; i<pdfVector.size();i++){
        PDFObject *temp = pdfVector[i];
        int newRowIdx = ui->tableWidget->rowCount();
        ui->tableWidget->insertRow(newRowIdx);
        ui->tableWidget->setItem(newRowIdx,0,new QTableWidgetItem(QString::number(temp->getPdfID())));//数据库存储ID
        if(temp->getAuthorsList().size()>0)
            ui->tableWidget->setItem(newRowIdx,1,new QTableWidgetItem(QString::fromStdString(temp->getAuthorsList().at(0))));//作者列表
        else
            ui->tableWidget->setItem(newRowIdx,1,new QTableWidgetItem(""));//作者列表
        ui->tableWidget->setItem(newRowIdx,2,new QTableWidgetItem(QString::fromStdString(temp->getSubject())));//标题
        QString abstract = QString::fromStdString(temp->getAbstract());
        ui->tableWidget->setItem(newRowIdx,3,new QTableWidgetItem(abstract));//摘要
        ui->tableWidget->setItem(newRowIdx,4,new QTableWidgetItem(QString::fromStdString(temp->getPath())));//路径
        ui->tableWidget->setItem(newRowIdx,5,new QTableWidgetItem(QString::fromStdString(temp->getIssue())));//期刊
    }
    //ui->listWidget->setCurrentRow(x1);
    //ui->listWidget->setCurrentRow(x2);
    //设置id列隐藏
//    ui->tableWidget->setColumnHidden(0,true);
//    ui->tableWidget->horizontalHeader()->setResizeMode(QHeaderView::ResizeToContents);
    //ui->tableWidget->horizontalHeader()->setTextElideMode();
    //ui->tableWidget->setColumnWidth(1,130);
    //QToolBar * textEditorToolbar = new QToolBar(this);
    //ui->tabWidget->removeTab(1);
}

void MainWindow::initTabWidget()
{

    //先获得分类和，下面的过滤信息

    //if(!myDao){
        myDao = Dao::getInstance();
    //}

    vector<PDFObject*> pdfVector;

    pdfVector = myDao->getAllDocs("");


    for(int i=0; i<pdfVector.size();i++){
        PDFObject *temp = pdfVector[i];
        int newRowIdx = ui->tableWidget->rowCount();
        ui->tableWidget->insertRow(newRowIdx);
        ui->tableWidget->setItem(newRowIdx,0,new QTableWidgetItem(QString::number(temp->getPdfID())));//数据库存储ID
        if(temp->getAuthorsList().size()>0)
            ui->tableWidget->setItem(newRowIdx,1,new QTableWidgetItem(QString::fromStdString(temp->getAuthorsList().at(0))));//作者列表
        else
            ui->tableWidget->setItem(newRowIdx,1,new QTableWidgetItem(""));
        ui->tableWidget->setItem(newRowIdx,2,new QTableWidgetItem(QString::fromStdString(temp->getSubject())));//标题
        QString abstract = QString::fromStdString(temp->getAbstract());
        ui->tableWidget->setItem(newRowIdx,3,new QTableWidgetItem(abstract));//摘要
        ui->tableWidget->setItem(newRowIdx,4,new QTableWidgetItem(QString::fromStdString(temp->getPath())));//路径
        ui->tableWidget->setItem(newRowIdx,5,new QTableWidgetItem(QString::fromStdString(temp->getIssue())));//期刊
    }
    //设置id列隐藏
    ui->tableWidget->setColumnHidden(0,true);
    ui->tableWidget->setColumnHidden(3,true);
    ui->tableWidget->setColumnHidden(4,true);
    ui->tableWidget->horizontalHeader()->setResizeMode(QHeaderView::ResizeToContents);
    //ui->tableWidget->horizontalHeader()->setTextElideMode();
    //ui->tableWidget->setColumnWidth(1,130);
    //QToolBar * textEditorToolbar = new QToolBar(this);
    //ui->tabWidget->removeTab(1);


}

void MainWindow::initTheUpLeftCorner()
{
    ui->listWidget->clear();
    ui->listWidget->addItem(tr("All"));
    ui->listWidget->addItem(tr("Favorite"));
    ui->listWidget->addItem(tr("Unsorted"));
    //if(!myDao)
        myDao = Dao::getInstance();
    vector<string> categories = myDao->getAllClassifications();
    for(int i=0;i<categories.size();i++)
    {
        ui->listWidget->addItem(QString::fromStdString(categories[i]));
    }
    ui->listWidget->addItem(tr("-Search"));
}

void MainWindow::initTheDownLeftCorner()
{
    //顺便把左边的作者列表也初始化
    //todo
    //需要判断下拉框的状态，来选择是刷新作者列表还是tag列表，还是keywords列表
    //int originalRowIndex = ui->listWidget_2->currentRow();
    //QString originalText = ui->listWidget_2->currentItem()->text();
    vector<string>authors;
    switch(ui->comboBox->currentIndex()){
    case 0:
        authors =myDao->getAllAuthors();
        break;
    case 1:
        authors = myDao->getAllTags();
        break;
    case 2:
        authors = myDao->getAllKeywords();
        break;
    }

    ui->listWidget_2->clear();
    ui->listWidget_2->addItem("All");
    if(authors.size()>0){
        for(int i=0;i<authors.size();i++)
        {
            ui->listWidget_2->addItem(QString::fromStdString(authors[i]));
        }
    }
//    if(originalRowIndex>=ui->listWidget_2->count())
//    {
//        ui->listWidget_2->setCurrentRow(0);
//        return;
//    }
//    if(ui->listWidget_2->item(originalRowIndex)->text()!=originalText){
//        ui->listWidget_2->setCurrentRow(0);
//        return;
//    }
//    ui->listWidget_2->setCurrentRow(originalRowIndex);
        ui->listWidget_2->setCurrentRow(0);
}

void MainWindow::initSignalsAndSlots()
{
    //双击表格某单元格，根据该行id，从数据库里读取信息
    connect(ui->tableWidget,SIGNAL(cellDoubleClicked(int,int)),this,SLOT(showText(int,int)));
    connect(ui->tableWidget,SIGNAL(cellClicked(int,int)),this,SLOT(updateTheRightSideBar(int,int)));
    connect(ui->menu_action_add_file,SIGNAL(triggered()),this,SLOT(addFile()));

    connect(ui->menu_action_remove_file,SIGNAL(triggered()),this,SLOT(removeDocument()));
    connect(ui->menu_action_add_category,SIGNAL(triggered()),this,SLOT(addCategory()));
    connect(ui->menu_action_remove_category,SIGNAL(triggered()),this,SLOT(removeCategory()));
    connect(ui->menu_action_change_category,SIGNAL(triggered()),this,SLOT(changeCategory()));
    connect(ui->menu_action_manual_input,SIGNAL(triggered()),this,SLOT(manual_input()));
    connect(ui->menu_action_addto_favorite,SIGNAL(triggered()),this,SLOT(add_to_favorite()));
    connect(ui->menu_action_remove_favorite,SIGNAL(triggered()),this,SLOT(remove_outof_favorite()));
    connect(searchSquare,SIGNAL(returnPressed()),this,SLOT(search()));
    connect(ui->menu_action_about,SIGNAL(triggered()),this,SLOT(about()));

    connect(ui->menu_action_open_file,SIGNAL(triggered()),this,SLOT(openFile()));
}

void MainWindow::showText(int a, int b)
{
    QMessageBox::about(this,"Tsignal",ui->tableWidget->item(a,b)->text());
}

void MainWindow::updateTheRightSideBar(int a, int b)
{
    //QMessageBox::about(this,"Tsignal",ui->tableWidget->item(a,0)->text());
    //先根据该行的id去数据库查询构造pdfobject
    //然后再填充右边的textedit
    PDFObject * temp = myDao->getDocByID(ui->tableWidget->item(a,0)->text().toInt());
    ui->textEdit_7->setText(QString::fromStdString(temp->getIssue()));//期刊
    ui->textEdit_2->setText(QString::fromStdString(temp->getSubject()));//题目
    //作者名单列表
    vector<string> authorsList = temp->getAuthorsList();
    ui->textEdit_3->clear();
    for(int i=0;i<authorsList.size();i++)
    {
        ui->textEdit_3->append(QString::fromStdString(authorsList[i]));
    }
    //摘要
    ui->textEdit_4->setText(QString::fromStdString(temp->getAbstract()));
    //keywords
    vector<string> keywordsList = temp->getKeywordsList();
    ui->textEdit_5->clear();
    for(int i=0;i<keywordsList.size();i++)
    {
        ui->textEdit_5->append(QString::fromStdString(keywordsList[i]));
    }
    //tags
    vector<string> tagsList = temp->getTagsList();
    ui->textEdit_6->clear();
    for(int i=0;i<tagsList.size();i++)
    {
        ui->textEdit_6->append(QString::fromStdString(tagsList[i]));
    }

    //notes
    ui->textEdit->setText(QString::fromStdString(temp->getNote()));

    //delete tempObj;
    //tempObj = myDao->getDocByID(ui->tableWidget->item(a,0)->text().toInt());
    //myDao->getDocByID()
    //delete myDao;
}

void MainWindow::addFileToDatabaseAndRefresh(string path)
{


    PDFObject *obj = new PDFObject(path);
    myDao = Dao::getInstance();
    int pdfID = myDao->saveOneDoc(obj);
    if(pdfID>0)
    {
        removeAllRow();
        refreshTabWidget();
//        int newRowIdx = ui->tableWidget->rowCount();
//        ui->tableWidget->insertRow(newRowIdx);
//        string subj = obj->getSubject();
//        QString subject = QString::fromStdString(subj);
//        //先给表格添加一行
//        ui->tableWidget->setItem(newRowIdx,0,new QTableWidgetItem(QString::number(pdfID)));//数据库存储ID
//        ui->tableWidget->setItem(newRowIdx,1,new QTableWidgetItem(QString::fromStdString(obj->getAuthorsList().at(0))));//作者列表
//        ui->tableWidget->setItem(newRowIdx,2,new QTableWidgetItem(subject));//标题
//        QString abstract = QString::fromStdString(obj->getAbstract());
//        ui->tableWidget->setItem(newRowIdx,3,new QTableWidgetItem(abstract));//摘要
//        ui->tableWidget->setItem(newRowIdx,4,new QTableWidgetItem(QString::fromStdString(path)));//路径
//        ui->tableWidget->setItem(newRowIdx,5,new QTableWidgetItem(QString::fromStdString(obj->getIssue())));//期刊
    }

}

//drop时需要在表格添加内容，并且需要和数据库互动
void MainWindow::dropEvent(QDropEvent *event)
{
    const QMimeData *mimeData = event->mimeData();
    QString x  = mimeData->urls().at(0).path();

    string path = x.toStdString();
    if(path[0]=='/')
        path = path.substr(1,path.length()-1);
    addFileToDatabaseAndRefresh(path);
    initTheDownLeftCorner();
    ui->listWidget_2->setCurrentRow(0);
    event->acceptProposedAction();
}

void MainWindow::dragEnterEvent(QDragEnterEvent *event)
 {
     event->acceptProposedAction();
 }

 void MainWindow::dragMoveEvent(QDragMoveEvent *event)
 {
     event->acceptProposedAction();
 }
 void MainWindow::dragLeaveEvent(QDragLeaveEvent *event )
 {
     event->accept();
 }

void MainWindow::on_comboBox_activated(int index)
{
    //QMessageBox::about(this,"Tsignal",QString::number(index));
    ui->listWidget_2->clear();
    ui->listWidget_2->addItem("All");
    vector<string> tempContainer;
    switch(index){
    case 0:
        tempContainer = myDao->getAllAuthors();
        for(int i=0;i<tempContainer.size();i++)
        {
            ui->listWidget_2->addItem(QString::fromStdString(tempContainer[i]));
        }
        break;
    case 1:
        tempContainer = myDao->getAllTags();
        for(int i=0;i<tempContainer.size();i++)
        {
            ui->listWidget_2->addItem(QString::fromStdString(tempContainer[i]));
        }
        break;
    case 2:
        tempContainer = myDao->getAllKeywords();
        for(int i=0;i<tempContainer.size();i++)
        {
            ui->listWidget_2->addItem(QString::fromStdString(tempContainer[i]));
        }
        break;
    default:
        break;
    }
}

void MainWindow::removeAllRow()
{
    int x = ui->tableWidget->rowCount();
    for(int i=x;i>=0;i--){
        ui->tableWidget->removeRow(i);
    }
}

void MainWindow::on_listWidget_2_itemClicked(QListWidgetItem* item)
{

    if(ui->listWidget->currentItem()->text()=="-Search")
    {
        ui->listWidget_2->setCurrentRow(0);
        return;
    }
    QString kind = ui->listWidget->currentItem()->text();
    if(kind=="All"){
        kind = "";
    }else if(kind =="Favorite"){
        kind ="true";
    }else if(kind == "Unsorted"){
        kind = "unsorted";
    }

    vector<PDFObject*> pdfVector;
    removeAllRow();
    clearRightSideBar();
    myDao = Dao::getInstance();
    if(item->text()=="All")
    {
        pdfVector = myDao->getAllDocs(kind.toStdString());
    }else{
        switch(ui->comboBox->currentIndex())
        {
        case 0://根据作者查找文献，需要根据上面那个列表的状态来更改中间的表格和右边的信息
            //QMessageBox::about(this,"tf",item->text());
            pdfVector = myDao->getDocByAuthor(item->text().toStdString(),kind.toStdString());
            break;
        case 1://根据自定义tag查找文献
            //QMessageBox::about(this,"tf",item->text());
            pdfVector = myDao->getDocByTag(item->text().toStdString(),kind.toStdString());
            break;
        case 2://根据关键字查找文献
            //QMessageBox::about(this,"tf",item->text());
            pdfVector = myDao->getDocByKeyword(item->text().toStdString(),kind.toStdString());
            break;
        default:
            break;
        }
    }
    //更新表格
    for(int i=0; i<pdfVector.size();i++){
        PDFObject *temp = pdfVector[i];
        int newRowIdx = ui->tableWidget->rowCount();
        ui->tableWidget->insertRow(newRowIdx);
        ui->tableWidget->setItem(newRowIdx,0,new QTableWidgetItem(QString::number(temp->getPdfID())));//数据库存储ID
        if(temp->getAuthorsList().size()>0)
            ui->tableWidget->setItem(newRowIdx,1,new QTableWidgetItem(QString::fromStdString(temp->getAuthorsList().at(0))));//作者列表
        else
            ui->tableWidget->setItem(newRowIdx,1,new QTableWidgetItem(""));
        ui->tableWidget->setItem(newRowIdx,2,new QTableWidgetItem(QString::fromStdString(temp->getSubject())));//标题
        QString abstract = QString::fromStdString(temp->getAbstract());
        ui->tableWidget->setItem(newRowIdx,3,new QTableWidgetItem(abstract));//摘要
        ui->tableWidget->setItem(newRowIdx,4,new QTableWidgetItem(QString::fromStdString(temp->getPath())));//路径
        ui->tableWidget->setItem(newRowIdx,5,new QTableWidgetItem(QString::fromStdString(temp->getIssue())));//期刊
    }
}


void MainWindow::addFile()
{
    QString fileName;
    fileName = QFileDialog::getOpenFileName(this,"Choose File","d:/",tr("PDF Files (*.pdf)"));
    //QMessageBox::about(this,"set",fileName);
    addFileToDatabaseAndRefresh(fileName.toStdString());
    initTheDownLeftCorner();
    ui->listWidget_2->setCurrentRow(0);
}

void MainWindow::openFile()
{
    int x = ui->tableWidget->currentRow();//if not one selected , return -1
    int y = ui->tableWidget->currentColumn();
    if(x!=-1&&y!=-1){

        int pdfId = ui->tableWidget->item(x,0)->text().toInt();
        PDFObject * pdfToRead =  myDao->getDocByID(pdfId);
        QString path = QString::fromStdString(pdfToRead->getPath());
        //QMessageBox::about(this,"",path);
        QDesktopServices::openUrl(QUrl("file:///"+path, QUrl::TolerantMode));

    }else{
        QMessageBox::about(this,"Error",tr("Please Select a File!"));
    }
}

void MainWindow::clearRightSideBar()
{
    ui->textEdit->clear();
    ui->textEdit_2->clear();
    ui->textEdit_3->clear();
    ui->textEdit_4->clear();
    ui->textEdit_5->clear();
    ui->textEdit_6->clear();
    ui->textEdit_7->clear();
}


void MainWindow::addCategory()
{
    //string categoryName;
    bool ok;
    QString categoryName = QInputDialog::getText(this, tr("Input Category Name"),
                                         tr("New Category Name:"), QLineEdit::Normal,
                                         tr(""), &ok);
    if (ok && !categoryName.isEmpty()){
        //QMessageBox::about(this,"s",categoryName);
        myDao->addClassification(categoryName.toStdString());
        initTheUpLeftCorner();
        ui->listWidget->setCurrentRow(0);
        removeAllRow();
        refreshTabWidget();
    }
}

void MainWindow::removeDocument()
{
    int x = ui->tableWidget->currentRow();//if not one selected , return -1
    int y = ui->tableWidget->currentColumn();
    if(x!=-1&&y!=-1){
        //string pdfId = ui->tableWidget->itemAt(x,0)->text();
        /*
        QMessageBox::about(this,"",QString::number(x));
        QMessageBox::about(this,"",QString::number(y));
        QMessageBox::about(this,"",ui->tableWidget->item(x,0)->text());*/
        int pdfId = ui->tableWidget->item(x,0)->text().toInt();
        myDao->deleteDocByID(pdfId);//删除之后应该更新表格
        removeAllRow();
        initTabWidget();
        clearRightSideBar();
        initTheDownLeftCorner();
        //ui->listWidget_2->setCurrentRow(0);

        //initTheDownLeftCorner();
        ui->listWidget_2->setCurrentRow(0);
    }else{
        QMessageBox::about(this,"Error",tr("Please Select a File!"));
    }

}

void MainWindow::removeCategory()
{
    int x = ui->listWidget->currentRow();
    if(x>2){
    //QMessageBox::about(this,"test",QString::number(x));
        myDao->deleteClassification(ui->listWidget->item(x)->text().toStdString());
        initTheUpLeftCorner();
    }else if(x<0){
        QMessageBox::about(this,"Error",tr("Please select a category!"));
    }else{
        QMessageBox::about(this,"Error",tr("Only custom category can be removed!"));
    }
    ui->listWidget->setCurrentRow(0);
    removeAllRow();
    refreshTabWidget();
}

void MainWindow::changeCategory()
{
    int x = ui->tableWidget->currentRow();//if not one selected , return -1
    int y = ui->tableWidget->currentColumn();
    if(x!=-1&&y!=-1){
        //string pdfId = ui->tableWidget->itemAt(x,0)->text();
        /*
        QMessageBox::about(this,"",QString::number(x));
        QMessageBox::about(this,"",QString::number(y));
        QMessageBox::about(this,"",ui->tableWidget->item(x,0)->text());*/
        int pdfId = ui->tableWidget->item(x,0)->text().toInt();
        PDFObject * pObj = myDao->getDocByID(pdfId);
        ChangeCategoryDialog cdlg(this);
        QString newCategory = "";
        if(cdlg.exec()){
            //QMessageBox::about(this,"code",cdlg.ui->comboBox->currentText());
            //QMessageBox::about(this,"code",cdlg.comboBox->currentText());
            //if(cdlg.comboBox->currentText()!="Unsorted")
            if(cdlg.comboBox->currentText()!="Unsorted")
                newCategory = cdlg.comboBox->currentText();
            else newCategory = "unsorted";
            pObj->setClassification(newCategory.toStdString());
            myDao->updateClassification(pObj);
            //myDao->deleteDocByID(pdfId);//删除之后应该更新表格
            removeAllRow();
            refreshTabWidget();
            //initTabWidget();
            clearRightSideBar();
            //initTheDownLeftCorner();
        }

    }else{
        QMessageBox::about(this,"Error",tr("Please Select a File!"));
    }


}

//保存笔记信息
void MainWindow::on_pushButton_2_clicked()
{
    int x = ui->tableWidget->currentRow();//if not one selected , return -1
    int y = ui->tableWidget->currentColumn();
    if(x!=-1&&y!=-1){
        int pdfId = ui->tableWidget->item(x,0)->text().toInt();
        PDFObject * pObj = myDao->getDocByID(pdfId);
        pObj->setNote(ui->textEdit->toPlainText().toStdString());
        //QMessageBox::about(this,"s",ui->textEdit->toPlainText());
        myDao->updateNote(pObj);
        //myDao->updateOneDoc()
        delete pObj;
        pObj = NULL;
    }else{
        QMessageBox::about(this,"Error",tr("Please Select a File!"));
    }
}

//除笔记外保存所有信息
//TODO 没做完呢
void MainWindow::on_pushButton_clicked()
{
    int x = ui->tableWidget->currentRow();//if not one selected , return -1
    int y = ui->tableWidget->currentColumn();
    if(x!=-1&&y!=-1){
        int pdfId = ui->tableWidget->item(x,0)->text().toInt();
        PDFObject * pObj = myDao->getDocByID(pdfId);
        pObj->setSubject(ui->textEdit_2->toPlainText().toStdString());
        pObj->setIssue(ui->textEdit_7->toPlainText().toStdString());
        pObj->setAbstract(ui->textEdit_4->toPlainText().toStdString());
        //作者列表
        QStringList authorsList = ui->textEdit_3->toPlainText().split("\n");
        vector<string>authorsListVec;
        if(ui->textEdit_3->toPlainText().trimmed()!=""){
            for(int i=0;i<authorsList.size();i++){
                if(authorsList.at(i)!="")
                authorsListVec.push_back(authorsList.at(i).toStdString());
            }
        }
        pObj->setAuthorsList(authorsListVec);
        //tag列表
        QStringList tagList = ui->textEdit_6->toPlainText().split("\n");
        vector<string>tagListVec;
        if(ui->textEdit_6->toPlainText().trimmed()!=""){
            for(int i=0;i<tagList.size();i++){
                if(tagList.at(i)!="")
                tagListVec.push_back(tagList.at(i).toStdString());
            }
        }
        pObj->setTags(tagListVec);
        //keywords列表
        QStringList keywordsList = ui->textEdit_5->toPlainText().split("\n");
        vector<string>keywordsListVec;
        if(ui->textEdit_5->toPlainText().trimmed()!=""){
            for(int i=0;i<keywordsList.size();i++){
                if(keywordsList.at(i)!="")
                keywordsListVec.push_back(keywordsList.at(i).toStdString());
            }
        }
        pObj->setKeywords(keywordsListVec);
        myDao->updateOneDoc(pObj);
        delete pObj;
        pObj = NULL;
        removeAllRow();
        initTabWidget();
        initTheDownLeftCorner();
    }else{
        QMessageBox::about(this,"Error",tr("Please Select a File!"));
    }
}

void MainWindow::on_listWidget_itemClicked(QListWidgetItem* item)
{
    ui->listWidget_2->setCurrentRow(0);
    vector<PDFObject*> pdfVector;
    removeAllRow();
    clearRightSideBar();
    myDao = Dao::getInstance();
    //QMessageBox::about(this,"s",QString::number(ui->listWidget->currentRow()));
    switch(ui->listWidget->currentRow()){
    case 0://All
        pdfVector = myDao->getAllDocs("");
        break;
    case 1://Favorite
        pdfVector = myDao->getAllDocs("true");
        break;
    case 2://Unsorted
        pdfVector = myDao->getAllDocs("unsorted");
        break;
    default://Has a Category
        pdfVector = myDao->getAllDocs(ui->listWidget->currentItem()->text().toStdString());
        break;
    }

    //更新表格
//    for(int i=0; i<pdfVector.size();i++){
//        PDFObject *temp = pdfVector[i];
//        int newRowIdx = ui->tableWidget->rowCount();
//        ui->tableWidget->insertRow(newRowIdx);
//        ui->tableWidget->setItem(newRowIdx,0,new QTableWidgetItem(QString::number(temp->getPdfID())));//数据库存储ID
//        ui->tableWidget->setItem(newRowIdx,1,new QTableWidgetItem(QString::fromStdString(temp->getAuthorsList().at(0))));//作者列表
//        ui->tableWidget->setItem(newRowIdx,2,new QTableWidgetItem(QString::fromStdString(temp->getSubject())));//标题
//        QString abstract = QString::fromStdString(temp->getAbstract());
//        ui->tableWidget->setItem(newRowIdx,3,new QTableWidgetItem(abstract));//摘要
//        ui->tableWidget->setItem(newRowIdx,4,new QTableWidgetItem(QString::fromStdString(temp->getPath())));//路径
//        ui->tableWidget->setItem(newRowIdx,5,new QTableWidgetItem(QString::fromStdString(temp->getIssue())));//期刊
//    }
    if(ui->listWidget->currentItem()->text()!="-Search")
        refreshTabWidget();
    else
        search();

}

void MainWindow::manual_input()
{
    ManualInputDialog miDlg(this);
    if(miDlg.exec())
    {
        //QMessageBox::about(this,"ss",miDlg.abstract->toPlainText());
        PDFObject * s = new PDFObject();//miDlg.path->text().toStdString());
        //myDao = Dao::getInstance();
        //tempObj = new PDFObject();
        s->setPath(miDlg.path->text().toStdString());
        s->setSubject(miDlg.subject->toPlainText().toStdString());
        s->setIssue(miDlg.issue->toPlainText().toStdString());
        s->setAbstract(miDlg.abstract->toPlainText().toStdString());
        //textedit_3麻烦的作者
        vector<string>authorsListVec;
        QStringList authorsList = miDlg.author->toPlainText().split("\n");
        for(int i=0;i<authorsList.size();i++)
            authorsListVec.push_back(authorsList.at(i).toStdString());
        s->setAuthorsList(authorsListVec);
        //myDao = Dao::getInstance();
        int pdfID = myDao->saveOneDoc(s);
        //QMessageBox::about(this,"s",QString::number(pdfID));
        if(pdfID>0){
            removeAllRow();
            refreshTabWidget();
        }
    }
}

void MainWindow::add_to_favorite()
{
    int x = ui->tableWidget->currentRow();//if not one selected , return -1
    int y = ui->tableWidget->currentColumn();
    if(x!=-1&&y!=-1){
        int pdfId = ui->tableWidget->item(x,0)->text().toInt();
        PDFObject * pObj = myDao->getDocByID(pdfId);
        pObj->setFavor("true");
        //QMessageBox::about(this,"s",ui->textEdit->toPlainText());
        myDao->updateFavor(pObj);
        //myDao->updateOneDoc()
        delete pObj;
        pObj = NULL;
        removeAllRow();
        refreshTabWidget();
    }else{
        QMessageBox::about(this,"Error",tr("Please Select a File!"));
    }
}

void MainWindow::remove_outof_favorite()
{
    int x = ui->tableWidget->currentRow();//if not one selected , return -1
    int y = ui->tableWidget->currentColumn();
    if(x!=-1&&y!=-1){
        int pdfId = ui->tableWidget->item(x,0)->text().toInt();
        PDFObject * pObj = myDao->getDocByID(pdfId);
        pObj->setFavor("false");
        //QMessageBox::about(this,"s",ui->textEdit->toPlainText());
        myDao->updateFavor(pObj);
        //myDao->updateOneDoc()
        delete pObj;
        pObj = NULL;
        removeAllRow();
        refreshTabWidget();
    }else{
        QMessageBox::about(this,"Error",tr("Please Select a File!"));
    }
}

void MainWindow::search()
{
    if(searchSquare->text()!=""){
        //不带冒号的
        if(!searchSquare->text().contains(":")){
            vector<PDFObject*> pdfVector;
            myDao = Dao::getInstance();
            pdfVector = myDao->getDocBySubject(searchSquare->text().toStdString());
            for(int i=0;i<ui->listWidget->count();i++)
            {
                if(ui->listWidget->item(i)->text()=="-Search"){
                    ui->listWidget->setCurrentRow(i);
                    break;
                }
            }
            ui->listWidget_2->setCurrentRow(0);
            removeAllRow();
            for(int i=0; i<pdfVector.size();i++){
                PDFObject *temp = pdfVector[i];
                int newRowIdx = ui->tableWidget->rowCount();
                ui->tableWidget->insertRow(newRowIdx);
                ui->tableWidget->setItem(newRowIdx,0,new QTableWidgetItem(QString::number(temp->getPdfID())));//数据库存储ID
                if(temp->getAuthorsList().size()>0)
                    ui->tableWidget->setItem(newRowIdx,1,new QTableWidgetItem(QString::fromStdString(temp->getAuthorsList().at(0))));//作者列表
                else
                    ui->tableWidget->setItem(newRowIdx,1,new QTableWidgetItem(""));
                ui->tableWidget->setItem(newRowIdx,2,new QTableWidgetItem(QString::fromStdString(temp->getSubject())));//标题
                QString abstract = QString::fromStdString(temp->getAbstract());
                ui->tableWidget->setItem(newRowIdx,3,new QTableWidgetItem(abstract));//摘要
                ui->tableWidget->setItem(newRowIdx,4,new QTableWidgetItem(QString::fromStdString(temp->getPath())));//路径
                ui->tableWidget->setItem(newRowIdx,5,new QTableWidgetItem(QString::fromStdString(temp->getIssue())));//期刊
            }
            return;
        }

        //带冒号的
        QStringList words;

        if(searchSquare->text().contains(":"))
            words = searchSquare->text().split(":");
        //QMessageBox::about(this,"s",words.at(1));
        //"title:"<<"author:"<<"keyword:"<<"tag:"<<"note:";
        vector<PDFObject*> pdfVector;
        myDao = Dao::getInstance();
        if(words.at(0)=="keyword"){
            pdfVector = myDao->getDocByKeyword(words.at(1).toStdString(),"");
        }else if(words.at(0)=="title"){
            pdfVector = myDao->getDocBySubject(words.at(1).toStdString());
        }else if(words.at(0)=="author"){
            pdfVector = myDao->getDocByAuthor(words.at(1).toStdString(),"");
        }else if(words.at(0)=="tag"){
            pdfVector = myDao->getDocByTag(words.at(1).toStdString(),"");
        }else if(words.at(0)=="note"){
            pdfVector = myDao->getDocByNote(words.at(1).toStdString());
        }
        //ui->listWidget->setCurrentRow(0);
        for(int i=0;i<ui->listWidget->count();i++)
        {
            if(ui->listWidget->item(i)->text()=="-Search"){
                ui->listWidget->setCurrentRow(i);
                break;
            }
        }
        ui->listWidget_2->setCurrentRow(0);
        removeAllRow();
        for(int i=0; i<pdfVector.size();i++){
            PDFObject *temp = pdfVector[i];
            int newRowIdx = ui->tableWidget->rowCount();
            ui->tableWidget->insertRow(newRowIdx);
            ui->tableWidget->setItem(newRowIdx,0,new QTableWidgetItem(QString::number(temp->getPdfID())));//数据库存储ID
            if(temp->getAuthorsList().size()>0)
                ui->tableWidget->setItem(newRowIdx,1,new QTableWidgetItem(QString::fromStdString(temp->getAuthorsList().at(0))));//作者列表
            else
                ui->tableWidget->setItem(newRowIdx,1,new QTableWidgetItem(""));
            ui->tableWidget->setItem(newRowIdx,2,new QTableWidgetItem(QString::fromStdString(temp->getSubject())));//标题
            QString abstract = QString::fromStdString(temp->getAbstract());
            ui->tableWidget->setItem(newRowIdx,3,new QTableWidgetItem(abstract));//摘要
            ui->tableWidget->setItem(newRowIdx,4,new QTableWidgetItem(QString::fromStdString(temp->getPath())));//路径
            ui->tableWidget->setItem(newRowIdx,5,new QTableWidgetItem(QString::fromStdString(temp->getIssue())));//期刊
        }
    }
}
void MainWindow::about()
{
    AboutDialog ab(this);
    ab.exec();
}