/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2017 Statoil ASA
//
//  ResInsight is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
//
//  ResInsight 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 General Public License at <http://www.gnu.org/licenses/gpl.html>
//  for more details.
//
/////////////////////////////////////////////////////////////////////////////////

#include "RicfExportProperty.h"

#include "RiaApplication.h"
#include "RiaLogging.h"

#include "../Commands/ExportCommands/RicEclipseCellResultToFileImpl.h"
#include "RicfApplicationTools.h"
#include "RicfCommandFileExecutor.h"

#include "RifEclipseInputFileTools.h"

#include "RigCaseCellResultsData.h"
#include "RigEclipseCaseData.h"
#include "RimEclipseCase.h"
#include "RimEclipseCaseCollection.h"
#include "RimEclipseCellColors.h"
#include "RimEclipseView.h"
#include "RimProject.h"

#include "cafUtils.h"

#include <QDir>

CAF_PDM_SOURCE_INIT(RicfExportProperty, "exportProperty");

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RicfExportProperty::RicfExportProperty()
{
    // clang-format off
    RICF_InitField(&m_caseId,           "caseId",           -1, "Case ID", "", "", "");
    RICF_InitField(&m_timeStepIndex,    "timeStep",         -1, "Time Step Index", "", "", "");
    RICF_InitField(&m_propertyName,     "property",         QString(), "Property Name", "", "", "");
    RICF_InitField(&m_eclipseKeyword,   "eclipseKeyword",   QString(), "Eclipse Keyword", "", "", "");
    RICF_InitField(&m_undefinedValue,   "undefinedValue",   0.0, "Undefined Value", "", "", "");
    RICF_InitField(&m_exportFileName,   "exportFile",       QString(), "Export FileName", "", "", "");
    // clang-format on
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RicfCommandResponse RicfExportProperty::execute()
{
    using TOOLS = RicfApplicationTools;

    RimEclipseCase* eclipseCase = TOOLS::caseFromId(m_caseId());
    {
        if (!eclipseCase)
        {
            QString error = QString("exportProperty: Could not find case with ID %1").arg(m_caseId());
            RiaLogging::error(error);
            return RicfCommandResponse(RicfCommandResponse::COMMAND_ERROR, error);
        }

        if (!eclipseCase->eclipseCaseData())
        {
            if (!eclipseCase->openReserviorCase())
            {
                QString error = QString("exportProperty: Could not find eclipseCaseData with ID %1").arg(m_caseId());
                RiaLogging::error(error);
                return RicfCommandResponse(RicfCommandResponse::COMMAND_ERROR, error);
            }
        }
    }

    RigEclipseCaseData* eclipseCaseData = eclipseCase->eclipseCaseData();

    RigCaseCellResultsData* cellResultsData = eclipseCaseData->results(RiaDefines::MATRIX_MODEL);

    if (!cellResultsData->ensureKnownResultLoaded(RigEclipseResultAddress(m_propertyName)))
    {
        QString error = QString("exportProperty: Could not find result property : %1").arg(m_propertyName());
        RiaLogging::error(error);
        return RicfCommandResponse(RicfCommandResponse::COMMAND_ERROR, error);
    }

    QString filePath = m_exportFileName;
    if (filePath.isNull())
    {
        QDir    propertiesDir(RicfCommandFileExecutor::instance()->getExportPath(RicfCommandFileExecutor::PROPERTIES));
        QString fileName = QString("%1-%2").arg(eclipseCase->caseUserDescription()).arg(m_propertyName);
        fileName         = caf::Utils::makeValidFileBasename(fileName);
        filePath         = propertiesDir.filePath(fileName);
    }

    QString eclipseKeyword = m_eclipseKeyword;
    if (eclipseKeyword.isNull())
    {
        eclipseKeyword = m_propertyName;
    }

    QString errMsg;
    if (!RicEclipseCellResultToFileImpl::writePropertyToTextFile(
        filePath, eclipseCase->eclipseCaseData(), m_timeStepIndex, m_propertyName, eclipseKeyword, m_undefinedValue,
        &errMsg))
    {
        return RicfCommandResponse(RicfCommandResponse::COMMAND_ERROR, errMsg);
    }

    return RicfCommandResponse();
}