/////////////////////////////////////////////////////////////////////////////////
//
//  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 "RicfExportSimWellFractureCompletions.h"

#include "RicfApplicationTools.h"
#include "RicfCommandFileExecutor.h"

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

#include "RimDialogData.h"
#include "RimEclipseCase.h"
#include "RimEclipseCaseCollection.h"
#include "RimEclipseView.h"
#include "RimSimWellInViewCollection.h"
#include "RimOilField.h"
#include "RimProject.h"
#include "RimSimWellInView.h"
#include "RimSimWellInView.h"
#include "RimWellPath.h"
#include "RimWellPathCollection.h"

#include "CompletionExportCommands/RicWellPathExportCompletionDataFeatureImpl.h"

CAF_PDM_SOURCE_INIT(RicfExportSimWellFractureCompletions, "exportSimWellFractureCompletions");

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
RicfExportSimWellFractureCompletions::RicfExportSimWellFractureCompletions()
{
    RICF_InitField(&m_caseId,          "caseId",                -1,                                                     "Case ID",  "", "", "");
    RICF_InitField(&m_viewName,        "viewName",              QString(""),                                            "View Name", "", "", "");
    RICF_InitField(&m_timeStep,        "timeStep",              -1,                                                     "Time Step Index",  "", "", "");
    RICF_InitField(&m_simWellNames,    "simulationWellNames",   std::vector<QString>(),                                 "Simulation Well Names",  "", "", "");
    RICF_InitField(&m_fileSplit,       "fileSplit",             RicExportCompletionDataSettingsUi::ExportSplitType(),   "File Split",  "", "", "");
    RICF_InitField(&m_compdatExport,   "compdatExport",         RicExportCompletionDataSettingsUi::CompdatExportType(), "Compdat Export",  "", "", "");
}

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

    RimProject* project = RiaApplication::instance()->project();
    RicExportCompletionDataSettingsUi* exportSettings = project->dialogData()->exportCompletionData();
    
    exportSettings->timeStep = m_timeStep;
    exportSettings->fileSplit = m_fileSplit;
    exportSettings->compdatExport = m_compdatExport;

    {
        auto eclipseCase = TOOLS::caseFromId(m_caseId());
        if (!eclipseCase)
        {
            QString error = QString("exportSimWellCompletions: Could not find case with ID %1").arg(m_caseId());
            RiaLogging::error(error);
            return RicfCommandResponse(RicfCommandResponse::COMMAND_ERROR, error);
        }
        exportSettings->caseToApply = eclipseCase;
    }

    QString exportFolder = RicfCommandFileExecutor::instance()->getExportPath(RicfCommandFileExecutor::COMPLETIONS);
    if (exportFolder.isNull())
    {
        exportFolder = RiaApplication::instance()->createAbsolutePathFromProjectRelativePath("completions");
    }
    exportSettings->folder = exportFolder;

    std::vector<RimEclipseView*> views;
    for (Rim3dView* v : exportSettings->caseToApply->views())
    {
        RimEclipseView* eclipseView = dynamic_cast<RimEclipseView*>(v);
        if (eclipseView && eclipseView->name() == m_viewName())
        {
            views.push_back(eclipseView);
        }
    }
    if (views.empty())
    {
        QString error = QString("exportSimWellCompletions: Could not find any views named \"%1\" in the case with ID %2").arg(m_viewName).arg(m_caseId());
        RiaLogging::error(error);
        return RicfCommandResponse(RicfCommandResponse::COMMAND_ERROR, error);
    }

    RicfCommandResponse response;

    std::vector<RimSimWellInView*> simWells;
    if (m_simWellNames().empty())
    {
        for (RimEclipseView* view : views)
        {
            for (auto simWell : view->wellCollection()->wells)
            {
                if (simWell->showWell())
                {
                    simWells.push_back(simWell);
                }
            }
        }
    }
    else
    {
        for (const QString& wellPathName : m_simWellNames())
        {
            for (RimEclipseView* view : views)
            {
                RimSimWellInView* simWell = view->wellCollection()->findWell(wellPathName);
                if (simWell)
                {
                    simWells.push_back(simWell);
                }
                else
                {
                    QString warning = QString("exportSimWellCompletions: Could not find well with name %1 in view \"%2\" on case with ID %2").arg(wellPathName).arg(m_viewName).arg(m_caseId());
                    RiaLogging::warning(warning);
                    response.updateStatus(RicfCommandResponse::COMMAND_WARNING, warning);
                }
            }
        }
    }

    std::vector<RimWellPath*> wellPaths;

    RicWellPathExportCompletionDataFeatureImpl::exportCompletions(wellPaths, simWells, *exportSettings);

    return response;
}