/*
    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
*/

#define QT_NO_CAST_TO_ASCII

#include <dcmtk/dcmdata/dcdeftag.h>
#include <dcmtk/dcmdata/dcstack.h>
#include <dcmtk/dcmdata/dcitem.h>
#include <dcmtk/dcmdata/dcelem.h>

#include <QtDcmManager.h>
#include <QtDcmMoveDicomdir.h>
#include <QtDcmConvert.h>

class QtDcmMoveDicomdirPrivate
{

public:
    QString outputDir;
    QString importDir;
    DcmItem * dcmObject;
    DcmStack dicomdirItems;
    QStringList filenames;
    QStringList series;
    QtDcmMoveDicomdir::eMoveMode mode;
    int index;
    QString uid;
};

QtDcmMoveDicomdir::QtDcmMoveDicomdir ( QObject * parent ) 
    : QThread(parent),
      d ( new QtDcmMoveDicomdirPrivate )
{
    d->mode = QtDcmMoveDicomdir::IMPORT;
}

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

void QtDcmMoveDicomdir::setMode ( QtDcmMoveDicomdir::eMoveMode mode )
{
    d->mode = mode;
}

void QtDcmMoveDicomdir::setDcmItem ( DcmItem * item )
{
    d->dcmObject = item;
}

void QtDcmMoveDicomdir::setSeries ( const QStringList & series )
{
    d->series = series;
}

void QtDcmMoveDicomdir::setIndex ( int index )
{
    d->index = index;
}

void QtDcmMoveDicomdir::setImageId ( const QString & uid)
{
    d->uid = uid;
}


void QtDcmMoveDicomdir::setOutputDir ( const QString & dir )
{
    d->outputDir = dir;
}

void QtDcmMoveDicomdir::setImportDir ( const QString & dir )
{
    d->importDir = dir;
}

void QtDcmMoveDicomdir::run()
{
    int step = ( int ) ( 100.0 / d->series.size() );
    int progress = 0;

    for ( int s = 0; s < d->series.size(); s++ ) {
        QDir serieDir ( d->outputDir + QDir::separator() + d->series.at ( s ) );

        if ( !serieDir.exists() )
            QDir ( d->outputDir ).mkdir ( d->series.at ( s ) );

        d->filenames.clear();

        bool proceed = false;
        bool proceedIndex = false;

        static const OFString Patient ( "PATIENT" );
        static const OFString Study ( "STUDY" );
        static const OFString Series ( "SERIES" );
        static const OFString Image ( "IMAGE" );

        // Loading all the dicomdir items in a stack
        DcmStack itemsTmp;

        if ( !d->dcmObject->findAndGetElements ( DCM_Item, itemsTmp ).good() )
            return;

        while ( itemsTmp.card() > 0 ) {
            d->dicomdirItems.push ( itemsTmp.top() );
            itemsTmp.pop();
        }

        OFString strName;
        OFString strDate;
        OFString strDesc;

        //Unstacking and loading the different lists

        while ( d->dicomdirItems.card() > 0 ) {
            DcmItem* lobj = ( DcmItem* ) d->dicomdirItems.top();
            DcmStack dirent;

            OFCondition condition = lobj->findAndGetElements ( DCM_DirectoryRecordType, dirent );

            if ( !condition.good() ) {
                d->dicomdirItems.pop();
                continue;
            }

            while ( dirent.card() ) {
                DcmElement* elt = ( DcmElement* ) dirent.top();
                OFString cur;
                elt->getOFStringArray ( cur );

                if ( cur ==Patient ) {
                    DcmElement* lelt;

                    if ( lobj->findAndGetElement ( DCM_PatientName, lelt ).good() )
                        lelt->getOFStringArray ( strName );
                }

                if ( cur == Study ) {
                    DcmElement* lelt;

                    if ( lobj->findAndGetElement ( DCM_StudyDate, lelt ).good() )
                        lelt->getOFStringArray ( strDate );
                }

                if ( cur == Series ) {
                    DcmElement* lelt;

                    if ( lobj->findAndGetElement ( DCM_SeriesInstanceUID, lelt ).good() ) {
                        OFString strID;
                        lelt->getOFStringArray ( strID );
                        proceed = ( QString ( strID.c_str() ) == d->series.at ( s ) );
                    }

                    if ( proceed ) {
                        if ( lobj->findAndGetElement ( DCM_SeriesDescription, lelt ).good() )
                            lelt->getOFStringArray ( strDesc );
                    }
                }

                if ( ( cur == Image ) && proceed ) {
                    DcmElement* lelt;

                    if ( lobj->findAndGetElement ( DCM_ReferencedSOPInstanceUIDInFile, lelt ).good() ) {
                        OFString strNumber;
                        lelt->getOFStringArray ( strNumber );

                        if ( d->mode == QtDcmMoveDicomdir::PREVIEW ) {
                            proceedIndex = ( QString ( strNumber.c_str() ) == d->uid );
                        }
                    }

//                     if ( lobj->findAndGetElement ( DCM_InstanceNumber, lelt ).good() )
//                     {
//                         OFString strNumber;
//                         lelt->getOFStringArray ( strNumber );
// 
//                         if ( d->mode == QtDcmMoveDicomdir::PREVIEW )
//                             proceedIndex = ( QString ( strNumber.c_str() ).toInt() == d->index );
//                     }

                    if ( lobj->findAndGetElement ( DCM_ReferencedFileID, lelt ).good() ) {
                        OFString strFilename;
                        lelt->getOFStringArray ( strFilename );

                        if ( d->mode == QtDcmMoveDicomdir::IMPORT ) {
                            d->filenames.append ( this->fixFilename ( QString ( strFilename.c_str() ) ) );
                        }
                        else {
                            if ( proceedIndex ) {
                                d->filenames.append ( this->fixFilename ( QString ( strFilename.c_str() ) ) );
                            }
                        }

                    }

                    if ( lobj->findAndGetElement ( DCM_SeriesDescription, lelt ).good() )
                        lelt->getOFStringArray ( strDesc );
                }

                dirent.pop();
            }

            d->dicomdirItems.pop();
        }

        d->dicomdirItems.clear();

        if ( d->mode == QtDcmMoveDicomdir::IMPORT ) {
            for ( int i = 0; i < d->filenames.size(); i++ ) {
                QFile image ( d->filenames.at ( i ) );

                if ( image.exists() ) {
                    QString zeroStr;
                    zeroStr.fill ( QChar ( '0' ), 5 - QString::number ( i ).size() );
                    QString newFile(serieDir.absolutePath() + QDir::separator() + "ima" + zeroStr + QString::number ( i ));
                    image.copy(newFile);
                    QFile(newFile).setPermissions(QFileDevice::WriteOwner);
                    
                    emit updateProgress ( progress + ( int ) ( ( ( float ) ( step * ( i + 1 ) / d->filenames.size() ) ) ) );
                }
            }

            progress += step;
            emit updateProgress(progress);
            emit serieMoved ( serieDir.absolutePath(), d->series.at ( s ) , s );
        }
        else {
            if ( !d->filenames.isEmpty() ) {
                QFile filename ( d->filenames.first() );
                emit previewSlice ( filename.fileName() );
            }
        }
    }
}

QString QtDcmMoveDicomdir::fixFilename ( const QString & name ) const
{
    QString tmpName(name);
    const QString basename = QFileInfo ( QtDcmManager::instance()->dicomdir() ).path();
    tmpName.replace ( QChar ( '/' ), QDir::separator() ).replace ( QChar ( '\\' ), QDir::separator() );
    QString filename = basename + QDir::separator() + tmpName.toUpper();

    if ( ! QFile ( filename ).exists() ) {
        filename = basename + QDir::separator() + tmpName.toLower();
    }

    return filename;
}