/*
    QtDcm is a C++ Qt based library for communication and conversion of Dicom images.
    Copyright (C) 2011  Alexandre Abadie <Alexandre.Abadie@univ-rennes1.fr>

    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
    version 2.1 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public
    License along with this library; if not, write to the Free Software
    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
*/

#include "QtDcm.h"

#include <QFileDialog>

#include <QtDcmPatient.h>
#include <QtDcmStudy.h>
#include <QtDcmSerie.h>
#include <QtDcmServer.h>
#include <QtDcmImage.h>
#include <QtDcmPreferences.h>
#include <QtDcmPreferencesDialog.h>
#include <QtDcmManager.h>

class QtDcmPrivate
{

public:
    int mode;
    QList<QString> imagesList; /** Contains the images filenames of the current serie (i.e selected in the treewidget)*/
    QString currentSerieId; /** Id of the current selected serie */
    QDate beginDate, endDate; /** Begin and end for Q/R retrieve parameters */
    QMap<QString, QList<QString> > selectedSeries;
};

QtDcm::QtDcm ( QWidget *parent ) 
    : QWidget ( parent ), d ( new QtDcmPrivate )
{
    QTextCodec::setCodecForLocale( QTextCodec::codecForName ( "iso" ) );
    setupUi ( this );
    d->mode = QtDcm::CD_MODE;

    //Initialize QTreeWidgetPatients
    treeWidgetPatients->setColumnWidth ( 0, 400 );
    treeWidgetPatients->setColumnWidth ( 1, 100 );
    treeWidgetPatients->setColumnWidth ( 2, 100 );
    const QStringList labelsPatients = QStringList() 
            << "Patients name" 
            << "ID" 
            << "Birthdate" 
            << "Sex";
    
    treeWidgetPatients->setHeaderLabels ( labelsPatients );
    treeWidgetPatients->setContextMenuPolicy ( Qt::CustomContextMenu );

    //Initialize QTreeWidgetSeries
    treeWidgetStudies->setColumnWidth ( 0, 200 );
    treeWidgetStudies->setColumnWidth ( 1, 100 );
    const QStringList labelsStudies = QStringList() 
            << "Studies description" 
            << "UID" 
            << "Date";
    
    treeWidgetStudies->setHeaderLabels ( labelsStudies );
    treeWidgetStudies->setContextMenuPolicy ( Qt::CustomContextMenu );

    //Initialize QTreeWidgetSeries
    treeWidgetSeries->setColumnWidth ( 0, 230 );
    treeWidgetSeries->setColumnWidth ( 1, 100 );
    treeWidgetSeries->setColumnWidth ( 2, 100 );
    const QStringList labelsSeries = QStringList()
            << "Series description" 
            << "Modality" 
            << "UID"
            << "Date";
    
    treeWidgetSeries->setHeaderLabels ( labelsSeries );
    treeWidgetSeries->setContextMenuPolicy ( Qt::CustomContextMenu );

    //Initialize widgets
    QDate currentDate = QDate::currentDate();
    startDateEdit->setDate (  currentDate.addYears(-100) );
    endDateEdit->setDate ( currentDate );
    
    QtDcmManager::instance()->setPatientsTreeWidget ( treeWidgetPatients );
    QtDcmManager::instance()->setStudiesTreeWidget ( treeWidgetStudies );
    QtDcmManager::instance()->setSeriesTreeWidget ( treeWidgetSeries );
    QtDcmManager::instance()->setStartDate ( QDate() );
    QtDcmManager::instance()->setEndDate ( QDate() );

    initConnections();
}

QtDcm::~QtDcm()
{
  delete d;
  d = NULL;
}

void QtDcm::initConnections()
{
    // Initialize connections
    connect ( treeWidgetPatients,    &QTreeWidget::itemClicked, 
              this,                  &QtDcm::onPatientItemClicked);
    connect ( treeWidgetStudies,     &QTreeWidget::itemClicked, 
              this,                  &QtDcm::onStudyItemClicked);
    connect ( treeWidgetSeries,      &QTreeWidget::itemClicked, 
              this,                  &QtDcm::onSerieItemClicked);
    connect ( nameEdit,              &QLineEdit::returnPressed, 
              this,                  &QtDcm::onPatientNameTextChanged);
    connect ( serieDescriptionEdit,  &QLineEdit::returnPressed, 
              this,                  &QtDcm::onSerieDescriptionTextChanged);
    connect ( studyDescriptionEdit,  &QLineEdit::returnPressed, 
              this,                  &QtDcm::onStudyDescriptionTextChanged);
    connect ( searchButton,          &QPushButton::clicked, 
              this,                  &QtDcm::onPacsSearchButtonClicked);
    connect ( cdromButton,           &QPushButton::clicked, 
              this,                  &QtDcm::onDicomMediaButtonClicked);
    connect ( patientSexComboBox,    static_cast<void (QComboBox::*)(int)>(&QComboBox::currentIndexChanged), 
              this,                  &QtDcm::onCurrentGenderChanged);
    connect ( serieModalityComboBox, static_cast<void (QComboBox::*)(int)>(&QComboBox::currentIndexChanged), 
              this,                  &QtDcm::onCurrentModalityChanged);
    connect ( pacsComboBox,          static_cast<void (QComboBox::*)(int)>(&QComboBox::currentIndexChanged), 
              this,                  &QtDcm::onCurrentPacsChanged);
    connect ( startDateEdit,         &QDateEdit::dateChanged, 
              this,                  &QtDcm::onStartDateChanged);
    connect ( endDateEdit,           &QDateEdit::dateChanged, 
              this,                  &QtDcm::onEndDateChanged);
}

void QtDcm::updatePacsComboBox()
{
    pacsComboBox->blockSignals ( true );
    pacsComboBox->clear();

    for ( int i = 0; i < QtDcmPreferences::instance()->servers().size(); i++ ) {
        pacsComboBox->addItem ( QtDcmPreferences::instance()->servers().at ( i ).name() );
    }
    
    pacsComboBox->blockSignals ( false );
}

void QtDcm::onPacsSearchButtonClicked()
{
    d->mode = QtDcm::PACS_MODE;
    treeWidgetPatients->clear();
    treeWidgetStudies->clear();
    treeWidgetSeries->clear();
    QtDcmManager::instance()->setCurrentPacs ( pacsComboBox->currentIndex() );
    QtDcmManager::instance()->findPatientsScu();
}

void QtDcm::clearDisplay()
{
    treeWidgetPatients->clear();
    treeWidgetStudies->clear();
    treeWidgetSeries->clear();
}

void QtDcm::onPatientItemClicked ( QTreeWidgetItem * item, int column )
{
    if (!item) {
        return;
    }

    QtDcmManager::instance()->clearSerieInfo();
    QtDcmManager::instance()->clearPreview();
    treeWidgetStudies->clear();
    treeWidgetSeries->clear();
    QtDcmManager::instance()->clearDataToImport();
    QtDcmManager::instance()->clearPatientDataToFetch();
    QtDcmManager::instance()->clearSeriesDataToFetch();
    if ( d->mode == QtDcm::PACS_MODE )
    {
        for (QTreeWidgetItem *ptItem : treeWidgetPatients->selectedItems())
        {
            QtDcmManager::instance()->addPatientDataToFetch(ptItem->text(1), ptItem->text(0), 
                                                            ptItem->text(2), ptItem->text(3));
            QtDcmManager::instance()->findStudiesScu (ptItem->text(1), ptItem->text ( 0 ) );
            QtDcmManager::instance()->addDataToImport(ptItem->text(1), "PATIENT");
        }
    }
    else
    {
        QtDcmManager::instance()->findStudiesDicomdir ( item->text ( 0 ) );
    }
    
    QtDcmManager::instance()->clearPreview();

}

void QtDcm::onStudyItemClicked(QTreeWidgetItem * item, int column )
{
    if (!item) {
        return;
    }

    QtDcmManager::instance()->clearSerieInfo();
    QtDcmManager::instance()->clearPreview();
    QtDcmManager::instance()->clearDataToImport();
    QtDcmManager::instance()->clearSeriesDataToFetch();
    treeWidgetSeries->clear();

    if ( d->mode == QtDcm::PACS_MODE ) 
    {
        for (QTreeWidgetItem *stItem: treeWidgetStudies->selectedItems())
        {
            QtDcmManager::instance()->findSeriesScu ( stItem->data ( 1, 0 ).toString() );
            QtDcmManager::instance()->addDataToImport(stItem->data ( 1, 0 ).toString(), "STUDY");
        }
    }
    else
    {
        QtDcmManager::instance()->findSeriesDicomdir ( treeWidgetPatients->currentItem()->text ( 0 ), item->data ( 1, 0 ).toString() );
    }
    QtDcmManager::instance()->clearPreview();
}

void QtDcm::onSerieItemClicked ( QTreeWidgetItem * item, int column )
{
    if (!item)
    {
        return;
    }

    QtDcmManager::instance()->clearListOfImages();
    QtDcmManager::instance()->clearDataToImport();

    if ( d->mode == QtDcm::CD_MODE )
        QtDcmManager::instance()->findImagesDicomdir ( item->text ( 2 ) );
    else
    {
        int elementCount = 0;
        for (QTreeWidgetItem *current : treeWidgetSeries->selectedItems())
        {
            QtDcmManager::instance()->findImagesScu ( current->text ( 2 ) );
            elementCount = QtDcmManager::instance()->listOfImages().size();
            QtDcmManager::instance()->addDataToImport ( current->text ( 2 ), "SERIES" );
        }
        QString institution = item->data ( 5, 0 ).toString();
        QString opName = item->data ( 6, 0 ).toString();
        QtDcmManager::instance()->updateSerieInfo ( QString::number ( elementCount ), institution, opName );
        QtDcmManager::instance()->getPreviewFromSelectedSerie ( item->text ( 3 ), elementCount / 2 );
    }

}

void QtDcm::onDicomMediaButtonClicked()
{
    this->openDicomdir();
}

void QtDcm::loadPatientsFromDicomdir()
{
    this->clearDisplay();
    QtDcmManager::instance()->loadDicomdir();
}

void QtDcm::queryPACS()
{
    this->onPacsSearchButtonClicked();
}

void QtDcm::onCurrentModalityChanged ( int index )
{
    switch ( index )
    {
    case ALL_MODALITIES://*
        QtDcmManager::instance()->setModality ( "" );
        break;

    case MR_MODALITY:
        QtDcmManager::instance()->setModality ( "MR" );
        break;

    case CT_MODALITY:
        QtDcmManager::instance()->setModality ( "CT" );
        break;

    case PET_MODALITY:
        QtDcmManager::instance()->setModality ( "PET" );
        break;
    default:
        qWarning() << "Modality not supported: " << serieModalityComboBox->currentText();
        break;
    }

    treeWidgetSeries->clear();
    if ( d->mode == QtDcm::PACS_MODE )
    {
        findSeriesFromStudyRows();
    }
    else
    {
        if ( treeWidgetPatients->currentItem() && treeWidgetStudies->currentItem() ) {
            QtDcmManager::instance()->findSeriesDicomdir ( treeWidgetPatients->currentItem()->text ( 0 ), treeWidgetStudies->currentItem()->data ( 1, 0 ).toString() );
        }
    }
}

void QtDcm::onCurrentGenderChanged ( int index )
{
    switch ( index )
    {
    case ALL_GENDER:
        QtDcmManager::instance()->setPatientGender( "*" );
        this->queryPACS();
        break;

    case M_GENDER:
        QtDcmManager::instance()->setPatientGender( "M" );
        this->queryPACS();
        break;

    case F_GENDER:
        QtDcmManager::instance()->setPatientGender( "F" );
        this->queryPACS();
        break;
    default:
        qWarning() << "Gender not supported: " << patientSexComboBox->currentText();
        break;
    }
}

void QtDcm::onCurrentPacsChanged ( int index )
{
    QtDcmManager::instance()->setCurrentPacs ( index );
    this->onPacsSearchButtonClicked();
}

void QtDcm::onStartDateChanged(const QDate &startdate)
{   
    if ( startdate > endDateEdit->date() ) {
        startDateEdit->blockSignals(true);
        startDateEdit->setDate ( endDateEdit->date() );
        startDateEdit->blockSignals(false);
    }
    QtDcmManager::instance()->setStartDate ( startDateEdit->date() );
    changeDate();
}

void QtDcm::onEndDateChanged (const QDate &enddate)
{   
    if ( enddate < startDateEdit->date() ) {
        endDateEdit->blockSignals(true);
        endDateEdit->setDate(startDateEdit->date() );
        endDateEdit->blockSignals(false);
    }
    QtDcmManager::instance()->setEndDate ( endDateEdit->date() );
    changeDate();
}

void QtDcm::changeDate()
{
    treeWidgetStudies->clear();
    treeWidgetSeries->clear();

    if ( treeWidgetPatients->currentItem() ) {
        if ( d->mode == QtDcm::PACS_MODE ) {
            QtDcmManager::instance()->findStudiesScu ( treeWidgetPatients->currentItem()->text(1), treeWidgetPatients->currentItem()->text ( 0 ) );
            findSeriesFromStudyRows();
        }
        else {
            qDebug() << "Date filtering not available in CD-Rom mode";
        }
    }
}

void QtDcm::editPreferences()
{
    //Launch a dialog window for editing PACS settings
    QtDcmPreferencesDialog dialog( this );
    dialog.readPreferences();
    if ( dialog.exec() ) {
        dialog.updatePreferences();
    }
}

void QtDcm::openDicomdir()
{
    this->clearDisplay();
    d->mode = QtDcm::CD_MODE;
    // Open a QFileDialog for choosing a Dicomdir
    QFileDialog dialog(this);
    dialog.setWindowTitle ( tr ( "Open dicomdir" ) );
    dialog.setNameFilters(QStringList() << "Dicomdir files (dicomdir* DICOMDIR*)");
    
    // Trying to open directly on one of the available drives
    if (!QDir::drives().isEmpty()) {
        dialog.setDirectory ( QDir::drives().first().absoluteDir() );
    }
    
    QString fileName;
    if ( dialog.exec() ) {
        fileName = dialog.selectedFiles() [0];
    }

    if ( !fileName.isEmpty() ) {  // A file has been chosen{
        if (QString::compare(fileName, "dicomdir", Qt::CaseInsensitive)) {
            QtDcmManager::instance()->setDicomdir ( fileName );
            this->loadPatientsFromDicomdir();
        }
    }
}

void QtDcm::onPatientNameTextChanged ()
{
    QString pName = nameEdit->text();

    if (pName.isEmpty() ) {
        QtDcmManager::instance()->setPatientName ( "" );
    }
    else {
        QtDcmManager::instance()->setPatientName ( "*" + pName + "*" );
    }

    if ( d->mode == QtDcm::PACS_MODE ) {
        this->onPacsSearchButtonClicked();
    }
}

void QtDcm::onStudyDescriptionTextChanged ()
{
    QString description = studyDescriptionEdit->text();
    if ( description.isEmpty() ) {
        QtDcmManager::instance()->setStudyDescription ( "" );
    }
    else {
        QtDcmManager::instance()->setStudyDescription ( "*" + description + "*" );
    }

    if ( d->mode == QtDcm::PACS_MODE ) {
        treeWidgetStudies->clear();
        treeWidgetSeries->clear();
    }
    
    if ( treeWidgetPatients->currentItem() && d->mode == QtDcm::PACS_MODE ) {
        QtDcmManager::instance()->findStudiesScu (treeWidgetPatients->currentItem()->text(1), treeWidgetPatients->currentItem()->text ( 0 ) );
        findSeriesFromStudyRows();
    }
}

void QtDcm::onSerieDescriptionTextChanged ()
{
    QString description = serieDescriptionEdit->text();
    if ( description.isEmpty() ) {
        QtDcmManager::instance()->setSerieDescription ( "" );
    }
    else {
        QtDcmManager::instance()->setSerieDescription ( "*" + description + "*" );
    }

    if ( d->mode == QtDcm::PACS_MODE ) {
        treeWidgetSeries->clear();
    }

    if ( d->mode == QtDcm::PACS_MODE )
    {
        findSeriesFromStudyRows();
    }
}

void QtDcm::findSeriesFromStudyRows()
{
    for (int row=0; row < treeWidgetStudies->topLevelItemCount(); row++)
    {
        QTreeWidgetItem *stItem = treeWidgetStudies->topLevelItem(row);
        QtDcmManager::instance()->findSeriesScu ( stItem->data ( 1, 0 ).toString() );
    }
}