aswi2021virtual-surreality-gitlab/db_browser/dbbrowser.cpp @ master

#include "dbbrowser.h"
#include "ui_dbbrowser.h"
#include "infodialog.h"
#include "../cv_stroj_3d/curvedialog.h"

#include <QLabel>
#include <QSqlQueryModel>
#include <QDate>
#include <QIntValidator>
#include <QList>
#include <QFile>
#include <QFileDialog>
#include <QTextStream>

#define PATIENTS_TAB    0
#define INDICATIONS_TAB 1
#define RESULTS_TAB     2

//! Konstruktor.
/*!
  V kostruktoru se inicializuje uživatelské rozhraní a připojení k databázi
*/
dbBrowser::dbBrowser(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::dbBrowser)
{
    ui->setupUi(this);
    db = new dbAccess();

    dbConnected = db->connect("localhost", "embitron", "embitron", 5432, "Cviceni");

    model = new QSqlQueryModel;

    connect(ui->exitBtn, SIGNAL(clicked()), qApp, SLOT(quit()));
    connect(ui->tabWidget, SIGNAL(currentChanged(int)), this, SLOT(tabChanged(int)));
    connect(ui->addPatientBtn, SIGNAL(clicked()), this, SLOT(addPatientClicked()));
    connect(ui->addIndicationBtn, SIGNAL(clicked()), this, SLOT(addIndicationClicked()));

    /*connect(ui->resultBtn, SIGNAL(clicked()), this, SLOT(setResult()));*/
    connect(ui->getResultsBtn, SIGNAL(clicked()), this, SLOT(getResult()));
    connect(ui->saveResultsBtn, SIGNAL(clicked()), this, SLOT(saveResults()));

    connect(ui->curveSelectBtn, SIGNAL(clicked()), this, SLOT(curveSelect()));

    statusBarText = new QLabel;
    statusBarText->setFixedSize(120,30);
    QFont font;
    font.setPixelSize(14);
    statusBarText->setFont(font);
    statusBarText->setAlignment(Qt::AlignCenter);
    if (!dbConnected) {
        statusBarText->setText(tr("DB not connected"));
        statusBarText->setStyleSheet("background-color: rgb(255,0,0)");
    } else {
        statusBarText->setText(tr("DB connected"));
        statusBarText->setStyleSheet("background-color: rgb(0,255,0)");
    }
    statusBar()->addPermanentWidget(statusBarText);

    ui->tabWidget->setCurrentIndex(0);
    ui->usersTableView->setSelectionMode(QAbstractItemView::SingleSelection);
    ui->usersTableView->setSelectionBehavior(QAbstractItemView::SelectRows);
    ui->indicationsTableView->setSelectionMode(QAbstractItemView::SingleSelection);
    ui->indicationsTableView->setSelectionBehavior(QAbstractItemView::SelectRows);

    db->getPatientsQueryModel(model);
    ui->usersTableView->setModel(model);
    ui->usersTableView->horizontalHeader()->resizeSections(QHeaderView::Stretch);

    QStringList pNames;
    QList<int> pId;
    ui->patientNameCombo->clear();
    ui->patientNameCombo->addItem(tr("---ALL---"), QVariant(0));
    if (db->getAllPatientsName(&pId, &pNames)) {
        for(int i=0; i<pId.size();i++) {
            ui->patientNameCombo->addItem(pNames.at(i), QVariant(pId.at(i)));
        }
    }
    ui->indNameCombo->clear();
    ui->indNameCombo->addItem(tr("---ALL---"), QVariant(0));
    ui->indNameCombo2->clear();
    ui->indNameCombo2->addItem(tr("---ALL---"), QVariant(0));
    pId.clear();pNames.clear();
    if (db->getAllIndicationsName(&pId, &pNames)) {
        for(int i=0; i<pId.size();i++) {
            ui->indNameCombo->addItem(pNames.at(i), QVariant(pId.at(i)));
            ui->indNameCombo2->addItem(pNames.at(i), QVariant(pId.at(i)));
        }
    }

    QIntValidator *iv = new QIntValidator(0,999, this);
    ui->timeEdit->setValidator(iv);

    cName = "";

    qsrand(QDateTime::currentMSecsSinceEpoch() / 1000);
}

//! Destruktor.
/*!
  Slouží pro zrušení vytvořených objektů
*/
dbBrowser::~dbBrowser()
{
    delete db;
    delete ui;
}

//! Změna jazyka dialogu
/*!
  Pokud se změní jazyk aplikace, funkce změní text v dialogu, aby odpovídal
  nastavenému jazyku
  \param *e Ukazatel na objekt, který obsahuje posílanou událost
*/
void dbBrowser::changeEvent(QEvent *e)
{
    QMainWindow::changeEvent(e);
    switch (e->type()) {
    case QEvent::LanguageChange:
        ui->retranslateUi(this);
        break;
    default:
        break;
    }
}

//! Info o změně zobrazované části dialogu
/*!
  Informace o změně zobrazované části dialogu. Funkce připraví nová data,
  aby zobrazované hodnoty odpovídaly požadavku.
  \param index Pořadové číslo žádané časti dialogu, která se má zobrazit
*/
void dbBrowser::tabChanged(int index)
{
    switch(index) {
    case PATIENTS_TAB:
        db->getPatientsQueryModel(model);
        ui->usersTableView->setModel(model);
        ui->usersTableView->horizontalHeader()->resizeSections(QHeaderView::Stretch);
        break;
    case INDICATIONS_TAB:
        db->getIndicationsQueryModel(model);
        ui->indicationsTableView->setModel(model);
        ui->indicationsTableView->horizontalHeader()->resizeSections(QHeaderView::Stretch);
        break;
    case RESULTS_TAB:
        getResult();
        break;
    }
}

//! Přidání pacienta do databáze
/*!
  Funkce vyčte hodnoty z editačního okna, a tyto hodnoty uloží do databáze
  jako nového pacienta
*/
void dbBrowser::addPatientClicked()
{
    infoDialog dlg;
    auto name = ui->nameEdit->text();
    auto surname = ui->surnameEdit->text();
    auto idNo = ui->idnumberEdit->text();
    auto indId = ui->indNameCombo2->currentData().toString();

    if (name.isEmpty() || surname.isEmpty() || idNo.isEmpty()) {
        // err, one of items is empty
        dlg.setInfoText(tr("Error"), tr("One item is empty. Please fill all items."));
        dlg.exec();
        return;
    }

    auto nRet = db->addNewPatient(name, surname, ui->birthdateEdit->date(), idNo, indId);

    if (nRet != ALL_OK) {
        dlg.setInfoText(tr("Error"), tr("Could not insert new patient data"));
    } else {
        dlg.setInfoText(tr("Success"), tr("OK. Patient data inserted succesfully"));
        db->getPatientsQueryModel(model);
        ui->usersTableView->setModel(model);
        ui->usersTableView->horizontalHeader()->resizeSections(QHeaderView::Stretch);

        QStringList pNames;
        QList<int> pId;
        ui->patientNameCombo->clear();
        ui->patientNameCombo->addItem(tr("---ALL---"), QVariant(0));
        if (db->getAllPatientsName(&pId, &pNames)) {
            for(int i=0; i<pId.size();i++) {
                ui->patientNameCombo->addItem(pNames.at(i), QVariant(pId.at(i)));
            }
        }
    }
    dlg.exec();
}

//! Přidání indikace do databáze
/*!
  Funkce vyčte hodnoty z editačního okna, a tyto hodnoty uloží do databáze
  jako novou indikaci
*/
void dbBrowser::addIndicationClicked()
{
    infoDialog dlg;
    auto name = ui->indNameEdit->text();
    auto time = ui->timeEdit->text().toInt();

    if (name.isEmpty() || cName.isEmpty()) {
        // err, one of items is empty
        dlg.setInfoText(tr("Error"), tr("One item is empty. Please fill all items."));
        dlg.exec();
        return;
    }

    auto nRet = db->addNewIndication(name, ui->descriptionEdit->text(), cName, time);

    if (nRet != ALL_OK) {
        dlg.setInfoText(tr("Error"), tr("Could not insert new indication data"));
    } else {
        dlg.setInfoText(tr("Success"), tr("OK. Indication data inserted succesfully"));
        db->getIndicationsQueryModel(model);
        ui->indicationsTableView->setModel(model);
        ui->indicationsTableView->horizontalHeader()->resizeSections(QHeaderView::Stretch);

        QStringList pNames;
        QList<int> pId;
        ui->indNameCombo->clear();
        ui->indNameCombo->addItem(tr("---ALL---"), QVariant(0));
        ui->indNameCombo2->clear();
        ui->indNameCombo2->addItem(tr("---ALL---"), QVariant(0));
        pId.clear();pNames.clear();
        if (db->getAllIndicationsName(&pId, &pNames)) {
            for(int i=0; i<pId.size();i++) {
                ui->indNameCombo->addItem(pNames.at(i), QVariant(pId.at(i)));
                ui->indNameCombo2->addItem(pNames.at(i), QVariant(pId.at(i)));
            }
        }
    }
    dlg.exec();
}

//! Výběr křivky pro novou indikaci
/*!
  Funkce zobrazí dialog, který slouží k výběru křiky, kterou má pacient s
  vytvářenou indikací cvičit.
*/
void dbBrowser::curveSelect()
{
    curveDialog dlg;
    int actCurveType, actCurveAngle, actCurveRot, pronSupVal;

    if (dlg.exec() == QDialog::Accepted) {
        dlg.getCurve(&actCurveType, &actCurveAngle, &actCurveRot, &pronSupVal);
        cName = QString::number(actCurveType) + "_" + QString::number(actCurveAngle) +
                "_" + QString::number(actCurveRot) + "_" + QString::number(pronSupVal);
        switch(actCurveType) {
        case CURVE_CIRCLE:
            ui->curveSelectBtn->setText(tr("Circle"));
            break;
        case CURVE_ELLIPSE:
            ui->curveSelectBtn->setText(tr("Ellipse"));
            break;
        case CURVE_KARD:
            ui->curveSelectBtn->setText(tr("Cardiotide"));
            break;
        case CURVE_LEMN:
            ui->curveSelectBtn->setText(tr("Lemniscate"));
            break;
        case CURVE_LINE:
            ui->curveSelectBtn->setText(tr("Line"));
            break;
        case CURVE_NONE:
            ui->curveSelectBtn->setText(tr("None"));
            break;
        case CURVE_EXTERN:
            ui->curveSelectBtn->setText(tr("External"));
            break;
        default:
            ui->curveSelectBtn->setText(tr(""));
        }
    }
}

//! Uložení výsledku
/*!
  Pouze testovací funkce. Funkce je použitá při testování ukládání/zobrazování výsledků cvičení.
*/
void dbBrowser::setResult()
{
    int res_count, cycles;
    QStringList arr;
    QList<int> intArr;
    QString resVals, resultList;
    infoDialog dlg;
    int vals = qrand() % 101 + 200;

    /*for(int i=0; i<vals; i++)
        intArr.append(qrand()%30+40);
    db->addResult(4, 2, 1,intArr);*/
    db->getResultsNumber(2,2, &res_count);
    while(db->getNextResult(&cycles, &resultList)) {
        resultList.chop(1);resultList.remove(0,1); //remove '{' and '}' from result string
        arr = resultList.split(",");
        resVals = "Cycles: " + QString::number(cycles);
        for(auto val: arr) {
            resVals += " ," + val;
        }
        dlg.setInfoText("2,2", resVals);
        dlg.exec();
    }
    db->getResultsNumber(2,0, &res_count);
    while(db->getNextResult(&cycles, &resultList)) {
        resultList.chop(1);resultList.remove(0,1); //remove '{' and '}' from result string
        arr = resultList.split(",");
        resVals = "Cycles: " + QString::number(cycles);
        for(auto val: arr) {
            resVals += " ," + val;
        }
        dlg.setInfoText("2,0", resVals);
        dlg.exec();
    }
    db->getResultsNumber(0,1, &res_count);
    while(db->getNextResult(&cycles, &resultList)) {
        resultList.chop(1);resultList.remove(0,1); //remove '{' and '}' from result string
        arr = resultList.split(",");
        resVals = "Cycles: " + QString::number(cycles);
        for(auto val: arr) {
            resVals += " ," + val;
        }
        dlg.setInfoText("0,1", resVals);
        dlg.exec();
    }
    db->getResultsNumber(0,0, &res_count);
    while(db->getNextResult(&cycles, &resultList)) {
        resultList.chop(1);resultList.remove(0,1); //remove '{' and '}' from result string
        arr = resultList.split(",");
        resVals = "Cycles: " + QString::number(cycles);
        for(auto val: arr) {
            resVals += " ," + val;
        }
        dlg.setInfoText("0,0", resVals);
        dlg.exec();
    }
}

//! Získání výsledků
/*!
  Funkce je použitá při testování zobrazování výsledků cvičení. Funkce vyčte z dialogu
  parametry, podle kterých je možno výsledky filtrovat.
*/
void dbBrowser::getResult()
{
    int pId = ui->patientNameCombo->currentData().toInt();
    int indId = ui->indNameCombo->currentData().toInt();

    db->getResultsQueryModel(model, pId, indId);
    ui->resultsTableView->setModel(model);
    ui->resultsTableView->horizontalHeader()->resizeSections(QHeaderView::Stretch);
}

//! Uložení výsledků
/*!
  Funkce je použitá pro uložení zobrazených výsledků cvičení. Uložený soubor
  je ve formátu CSV.
*/
void dbBrowser::saveResults()
{
    int size = ui->resultsTableView->model()->rowCount(),i;
    QString res, date;
    int pId, indId, cycles, series, resNo;
    QString filename = QFileDialog::getSaveFileName(this, tr("Save CSV"), "", tr("CSV files (.csv)"));
    if (!filename.endsWith("csv"))
        filename += ".csv";
    QFile data(filename);

    if(data.open(QFile::WriteOnly |QFile::Truncate))
    {
        QTextStream output(&data);
        output << "Patient ID; Indication ID; Series; Result number; Date; Cycles count; result\n";

        for(i=0;i<size;i++) {
            pId = ui->resultsTableView->model()->index(i,0).data().toInt();
            indId = ui->resultsTableView->model()->index(i,1).data().toInt();
            series = ui->resultsTableView->model()->index(i,2).data().toInt();
            resNo = ui->resultsTableView->model()->index(i,3).data().toInt();
            date = ui->resultsTableView->model()->index(i,4).data().toString();
            cycles = ui->resultsTableView->model()->index(i,5).data().toInt();
            res = ui->resultsTableView->model()->index(i,6).data().toString();

            //remove '{' and '}' from result string
            if (res.contains('{'))
                res.remove(0,1);
            if (res.contains('}'))
                res.chop(1);
            res.replace(',',';');
            output << pId << ";" << indId << ";" << series << ";" << resNo << ";" << date <<";" << cycles << ";" << res << "\n";
        }
        data.close();
    }
}