/*=========================================================================

 medInria

 Copyright (c) INRIA 2013 - 2020. All rights reserved.
 See LICENSE.txt for details.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

=========================================================================*/

#include "vtkDataMesh4DReader.h"
#include "vtkDataManagerReader.h"

#include <medAbstractData.h>
#include <medAbstractDataFactory.h>

#include <vtkDataManager.h>
#include <vtkFieldData.h>
#include <vtkMetaDataSetSequence.h>
#include <vtkSmartPointer.h>

const char vtkDataMesh4DReader::ID[] = "vtkDataMesh4DReader";

vtkDataMesh4DReader::vtkDataMesh4DReader(): vtkDataMeshReaderBase()
{
    this->reader = vtkDataManagerReader::New();
}

vtkDataMesh4DReader::~vtkDataMesh4DReader()
{
    this->reader->Delete();
}

QStringList vtkDataMesh4DReader::handled() const
{
    return QStringList() << "vtkDataMesh4D";
}

QStringList vtkDataMesh4DReader::s_handled()
{
    return QStringList() << "vtkDataMesh4D";
}

bool vtkDataMesh4DReader::canRead (const QString& path)
{
    return this->reader->CanReadFile (path.toUtf8().constData());
}

bool vtkDataMesh4DReader::readInformation (const QString& path)
{
  
    medAbstractData *medData = dynamic_cast<medAbstractData*>(this->data());
    this->reader->SetFileName (path.toUtf8().constData());
  
    if (!medData)
    {
        medData = medAbstractDataFactory::instance()->create("vtkDataMesh4D");
        if (medData)
        {
            this->setData ( medData );
        }
    }

    medData->addMetaData ("FilePath", QStringList() << path);
    medData->identifier() = "vtkDataMesh4D";
    
    return true;
}

bool vtkDataMesh4DReader::read (const QString& path)
{
    this->setProgress (0);

    this->readInformation ( path );

    this->setProgress (50);

    qDebug().noquote() << "Can read with: " << this->identifier();

    if (medAbstractData *medData = dynamic_cast<medAbstractData*>(this->data()) )
    {
        if (!(medData->identifier()=="vtkDataMesh4D"))
        {
            return false;
        }

        this->reader->SetFileName (path.toUtf8().constData());
        this->reader->Update();

        vtkMetaDataSetSequence* sequence = vtkMetaDataSetSequence::SafeDownCast (this->reader->GetOutput()->GetMetaDataSet ((unsigned int)0));
        if (sequence && sequence->GetNumberOfMetaDataSets() > 0)
        {
            if (!extractMetaData(sequence->GetMetaDataSet(0)))
            {
                qDebug() << metaObject()->className() << ": no metadata found in " << path;
            }

            // Remove field from top container of the sequence
            vtkSmartPointer<vtkFieldData> newFieldData = vtkSmartPointer<vtkFieldData>::New();
            sequence->GetDataSet()->SetFieldData(newFieldData);

            medData->setData (sequence);
        }
        else
        {
            return false;
        }
    }

    this->setProgress (100);
    return true;
}

bool vtkDataMesh4DReader::extractMetaData(vtkMetaDataSet* dataSet)
{
    bool result = false;

    if (extractMetaDataFromFieldData(dataSet))
    {
        result = true;
    }

    return result;
}

bool vtkDataMesh4DReader::registered()
{
    return medAbstractDataFactory::instance()->registerDataReaderType(
          "vtkDataMesh4DReader",
          vtkDataMesh4DReader::s_handled(),
          createVtkDataMesh4DReader);
}

QString vtkDataMesh4DReader::identifier() const
{
    return ID;
}

QString vtkDataMesh4DReader::description() const
{
    return "Reader for vtk 4D meshes";
}

// /////////////////////////////////////////////////////////////////
// Type instantiation
// /////////////////////////////////////////////////////////////////

dtkAbstractDataReader *createVtkDataMesh4DReader()
{
    return new vtkDataMesh4DReader;
}