mainwindow.cpp

#include "mainwindow.h"
#include "ui_mainwindow.h"
#include "Book.h"
#include "CD.h"
#include "Movie.h"
#include "addDocument.h"
#include "modifyDocument.h"
#include <iostream>
#include <fstream>
#include <QString>
#include <QMessageBox>
#include <algorithm>

MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainWindow)
{
    ui->setupUi(this);
    initMainWindow();
    initData();
}

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

void MainWindow::initMainWindow()
{
    ui->tw_tab->verticalHeader()->hide();
    ui->tw_tab->setColumnWidth(0,250);
    ui->tw_tab->setColumnWidth(1,175);
    ui->tw_tab->setColumnWidth(2,170);
}

void MainWindow::initData()
{
    std::ifstream monFichier;
    std::string extension = "./sauve.txt";
    std::vector<std::string> myLine;
    std::string title, autor, editor, style, resume, actor;
    int piste, editorYear;

    monFichier.open(extension.c_str());
    lib.clear();

    if(monFichier)
    {
        while(monFichier)
        {
            std::string ligne;
            getline(monFichier, ligne);
            myLine = split(ligne, "//");
            if (myLine[0] == "<Book>")
            {
                title = myLine[1];
                autor = myLine[2];
                resume = myLine[3];
                editor = myLine[4];
                editorYear = atoi(myLine[5].c_str());
                Book* book = new Book(title, autor, resume, editorYear, editor);
                lib.push_back(book);
                myLine.clear();
            }
            else if (myLine[0] == "<CD>")
            {
                title = myLine[1];
                autor = myLine[2];
                style = myLine[3];
                piste = atoi(myLine[4].c_str());
                CD* cd = new CD(title, autor, style, piste);
                lib.push_back(cd);
                myLine.clear();
            }
            else if (myLine[0] == "<Movie>")
            {
                title = myLine[1];
                autor = myLine[2];
                resume = myLine[3];
                style = myLine[4];
                actor = myLine[5];
                Movie* movie = new Movie(title, autor, resume, style, actor);
                lib.push_back(movie);
                myLine.clear();
            }
        }
    }
    else
    {
        std::cout<<"ERREUR: Impossible d'ouvrir le fichier "<<extension<<std::endl;
    }
    monFichier.close();

    showList();
}

void MainWindow::showList()
{
    long i;

    ui->tw_tab->clearContents();
    ui->tw_tab->setRowCount(0);

    for(i=0;i<lib.size();i++)
    {
        ui->tw_tab->setRowCount(ui->tw_tab->rowCount()+1);
        ui->tw_tab->setItem(ui->tw_tab->rowCount()-1, 0 , new QTableWidgetItem(QString::fromStdString(lib.at(i)->getTitle())));
        ui->tw_tab->setItem(ui->tw_tab->rowCount()-1, 1 , new QTableWidgetItem(QString::fromStdString(lib.at(i)->getAutor())));
        Book* b = dynamic_cast<Book*>(lib.at(i));
        CD* c = dynamic_cast<CD*>(lib.at(i));
        QString var;

        if(b)
        {
            var = "Book";
        }
        else if(c)
        {
            var = "CD";
        }
        else
        {
            var = "Movie";
        }
        ui->tw_tab->setItem(ui->tw_tab->rowCount()-1, 2 , new QTableWidgetItem(var));
    }
}

std::vector<std::string> MainWindow::split(std::string chaineInit, std::string chaineSep)
{
    std::vector<std::string> chaineSplit;
    size_t pos;

    do{
        pos = chaineInit.find(chaineSep);
        if (pos!=std::string::npos){
            chaineSplit.push_back(chaineInit.substr(0,pos));
            chaineInit = chaineInit.substr(pos+chaineSep.length());
        }else{
            chaineSplit.push_back(chaineInit);
        }
    }while(pos!=std::string::npos);

    return (chaineSplit);
}

void MainWindow::on_actionQuitter_triggered()
{
    this->close();
}

void MainWindow::on_btn_addDoc_clicked()
{
    addDocument doc(&lib,this);
    doc.exec();
    if(doc.close())
    {
        showList();
    }
}

void MainWindow::on_actionAjouter_triggered()
{
    addDocument doc(&lib, this);
    doc.exec();

    if(doc.close())
    {
        showList();
    }
}

void MainWindow::on_btn_dellDoc_clicked()
{
    while(ui->tw_tab->selectedItems().count() > 0)
    {
        QString title;

        int col = 0;
        int lig = ui->tw_tab->currentIndex().row();
        title = ui->tw_tab->item(lig,col)->data(0).toString();
        lib.remove(ui->tw_tab->selectedItems().at(0)->row());
        dell(title.toStdString());
        ui->tw_tab->removeRow(ui->tw_tab->selectedItems().at(0)->row());
    }

    ui->btn_dellDoc->setEnabled(false);
}

void MainWindow::on_tw_tab_cellClicked(int row, int column)
{
    ui->btn_dellDoc->setEnabled(true);
}

void MainWindow::dell(std::string title)
{
    std::ifstream monFichier;
    std::ofstream newFichier;
    std::string extension = "./sauve.txt";
    std::string newName = "./sauve2.txt";
    std::vector<std::string> myLine;
    std::string ligne;

    monFichier.open(extension.c_str());

    if(monFichier)
    {
        newFichier.open(newName.c_str());
        if(newFichier)
        {
            while(getline(monFichier,ligne))
            {
                myLine = split(ligne, "//");
                if (myLine[1] == title)
                {
                    //it's the doc to delete
                }
                else
                {
                    //it's a doc to keep
                    newFichier<<ligne<<std::endl;
                }
            }
        }
        else
        {
             std::cout<<"ERREUR: Impossible d'ouvrir le fichier "<<newName<<std::endl;
        }
        newFichier.close();
        monFichier.close();
        //Delete the orl file
        remove(extension.c_str());
        //Rename the new file
        rename(newName.c_str(), extension.c_str());
    }
    else
    {
        std::cout<<"ERREUR: Impossible d'ouvrir le fichier "<<extension<<std::endl;
    }
}

void MainWindow::on_actionSupprimer_triggered()
{
    on_btn_dellDoc_clicked();
}

void MainWindow::on_actionExporter_en_HTML_triggered()
{
    exportHTML();
    QMessageBox::information(this, tr("Export HTML"), tr("Export ending. File index.html creating."), QMessageBox::Ok);
}

void MainWindow::exportHTML()
{
    std::ofstream f("index.html");
    if(f)
    {
        f<<"<!DOCTYPE html>"<<std::endl;
        f<<"<html>"<<std::endl;
        f<<"<head>"<<"<h1>Liste des Documents</h1>"<<"</head>"<<std::endl;
        f<<"<body>"<<std::endl;
        f<<"<table border=\"1\";>"<<std::endl;
        f<<"<tr><th><big>Title</big></th><th><big>Autor</big></th></th><th><big>Type</big></th></tr>"<<std::endl;
        for(unsigned int i = 0;i<lib.size();i++)
        {
            Book* b = dynamic_cast<Book*>(lib[i]);
            CD* c = dynamic_cast<CD*>(lib[i]);

            f<<"<tr><th>"<<lib[i]->getTitle()<<"</th><th>"<<lib[i]->getAutor()<<"</th>";

            if(b)
            {
                f<<"<th>Book</th></tr>" << std::endl;
            }
            else if (c)
            {
                f<<"<th>CD</th></tr>" << std::endl;
            }
            else
            {
                f<<"<th>Movie</th></tr>" << std::endl;
            }
        }

        f<<"</table>"<<std::endl;
        f<<"</body>"<<std::endl;
        f<<"</html>"<<std::endl;
        f.close();
    }
}

void MainWindow::on_actionTrie_alphabetique_triggered()
{
    long i,j;

    for(i=lib.size()-1;i>=0;i--)
    {
        for(j=0;j<i;j++)
        {
            if(lib.at(j)->getTitle() > lib.at(j+1)->getTitle())
            {
                Document* doc = lib.at(j);
                lib[j] = lib.at(j+1);
                lib[j+1] = doc;
            }
        }
    }
    showList();
}

void MainWindow::on_actionTrie_par_type_triggered()
{
    long i,j,k;
    QVector<Document*> libB;
    QVector<Document*> libC;
    QVector<Document*> libM;

    //Add books
    for(k=0;k<lib.size();k++)
    {
         Book* b1 = dynamic_cast<Book*>(lib.at(k));
         if(b1)
         {
             libB.push_back(b1);
         }
    }
    //Short books
    for(i=libB.size()-1;i>=0;i--)
    {
        for(j=0;j<i;j++)
        {
            if(libB.at(j)->getTitle() > libB.at(j+1)->getTitle())
            {
                Document* doc = libB.at(j);
                libB[j] = libB.at(j+1);
                libB[j+1] = doc;
            }
        }
    }

    //Add CD
    for(k=0;k<lib.size();k++)
    {
         CD* c1 = dynamic_cast<CD*>(lib.at(k));
         if(c1)
         {
             libC.push_back(c1);
         }
    }
    //short cd
    for(i=libC.size()-1;i>=0;i--)
    {
        for(j=0;j<i;j++)
        {
            if(libC.at(j)->getTitle() > libC.at(j+1)->getTitle())
            {
                Document* doc = libC.at(j);
                libC[j] = libC.at(j+1);
                libC[j+1] = doc;
            }
        }
    }

    //add movie
    for(k=0;k<lib.size();k++)
    {
         Movie* m1 = dynamic_cast<Movie*>(lib.at(k));
         if(m1)
         {
             libM.push_back(m1);
         }
    }
    //short movie
    for(i=libM.size()-1;i>=0;i--)
    {
        for(j=0;j<i;j++)
        {
            if(libM.at(j)->getTitle() > libM.at(j+1)->getTitle())
            {
                Document* doc = lib.at(j);
                libM[j] = libM.at(j+1);
                libM[j+1] = doc;
            }
        }
    }

    lib.clear();
    lib = libB + libC + libM;
    showList();
}

void MainWindow::on_actionTrie_par_auteur_triggered()
{
    long i,j;

    for(i=lib.size()-1;i>=0;i--)
    {
        for(j=0;j<i;j++)
        {
            if(lib.at(j)->getAutor() > lib.at(j+1)->getAutor())
            {
                Document* doc = lib.at(j);
                lib[j] = lib.at(j+1);
                lib[j+1] = doc;
            }
        }
    }
    showList();
}

void MainWindow::on_btn_modify_clicked()
{
    Document* mydoc;
    int col = 0;
    QString title;
    int lig = ui->tw_tab->currentIndex().row();
    title = ui->tw_tab->item(lig,col)->data(0).toString();

    mydoc = search(title);

    modifyDocument doc(&lib,mydoc,this);
    doc.exec();
    if(doc.close())
    {
        showList();
        copyLibrary();
    }
}

Document* MainWindow::search(QString title)
{
    unsigned int i=0;
    bool ok = false;
    Document* doc = new Document("Inconnu", "Inconnu");
    while((i<lib.size()) && (ok == false))
    {
        if(lib[i]->getTitle() == title.toStdString())
        {
            ok = true;
        }
        i++;
    }

    if(ok == false)
    {
        return doc;
    }
    else
    {
        return lib[i-1];
    }
}

void MainWindow::copyLibrary()
{
    std::ofstream f("sauve.txt");
    if(f)
    {
        for(unsigned int i = 0;i<lib.size();i++)
        {
            Book* b = dynamic_cast<Book*>(lib[i]);
            CD* c = dynamic_cast<CD*>(lib[i]);
            Movie* m = dynamic_cast<Movie*>(lib[i]);

            if(b)
            {
                f<<"<Book>";
                f<<"//";
                f<<b->getTitle();
                f<<"//";
                f<<b->getAutor();
                f<<"//";
                f<<b->getResume();
                f<<"//";
                f<<b->getEditor();
                f<<"//";
                f<<b->getEditorYear();
                f<<"//"<<std::endl;
            }
            else if (c)
            {
                f<<"<CD>";
                f<<"//";
                f<<c->getTitle();
                f<<"//";
                f<<c->getAutor();
                f<<"//";
                f<<c->getStyle();
                f<<"//";
                f<<c->getPisteNumber();
                f<<"//"<<std::endl;
            }
            else
            {
                f<<"<Movie>";
                f<<"//";
                f<<m->getTitle();
                f<<"//";
                f<<m->getAutor();
                f<<"//";
                f<<m->getResume();
                f<<"//";
                f<<m->getStyle();
                f<<"//";
                f<<m->getActor();
                f<<"//"<<std::endl;
            }
        }
        f.close();
    }
}