/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2026 Equinor 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.
//
/////////////////////////////////////////////////////////////////////////////////

#pragma once

#include "RimGenericJob.h"

#include "cafPdmPtrArrayField.h"
#include "cafPdmPtrField.h"
#include "cafPdmUiTreeAttributes.h"

#include <QDateTime>

#include <string>
#include <vector>

class RimReservoirGridEnsemble;
class RimEclipseCase;
class RimOpmFlowJob;
class RimOpmFlowJobSettings;
class RimJobWellSettings;
class RimEnsembleFileSet;

//==================================================================================================
///
//==================================================================================================
class RimEnsembleJob : public RimGenericJob
{
    CAF_PDM_HEADER_INIT;

public:
    RimEnsembleJob();
    ~RimEnsembleJob() override;

    void setInputEnsemble( RimReservoirGridEnsemble* ensemble );

    bool              execute() override;
    void              stop() override;
    double            percentageDone() const override;
    const QStringList jobLog() const override;
    bool              matchesKeyValue( const QString& key, const QString& value ) const override;

    static QString jobInputFileKey();

    void setFinished( bool runOk ) override;

    void initAfterCopy();

protected:
    struct RealizationInfo
    {
        RimEclipseCase* inputCase;
        std::string     realizationInputDeckName;
        std::string     realizationOutputDir;
        std::string     outputDeckName;
    };

    QList<caf::PdmOptionItemInfo> calculateValueOptions( const caf::PdmFieldHandle* fieldNeedingOptions ) override;

    void                                   defineUiOrdering( QString uiConfigName, caf::PdmUiOrdering& uiOrdering ) override;
    std::vector<caf::PdmUiTreeViewItemTag> treeItemTags() override;
    void fieldChangedByUi( const caf::PdmFieldHandle* changedField, const QVariant& oldValue, const QVariant& newValue ) override;
    void initAfterRead() override;

    std::vector<RealizationInfo> setUpSelectedRealizations() const;
    std::string                  outputIteration() const;

    std::vector<QString> dateStrings() const;

    void subJobCompleted( const caf::SignalEmitter* emitter, bool runOk );

private:
    std::vector<std::string> selectedRealizationFileNames() const;

private:
    caf::PdmPtrField<RimReservoirGridEnsemble*> m_inputEnsemble;
    caf::PdmPtrArrayField<RimEclipseCase*>      m_selectedRealizations;
    caf::PdmField<int>                          m_outputIterationNumber;
    caf::PdmChildArrayField<RimOpmFlowJob*>     m_subJobs;
    caf::PdmChildField<RimOpmFlowJobSettings*>  m_jobSettings;
    caf::PdmChildField<RimJobWellSettings*>     m_jobWellSettings;
    caf::PdmPtrField<RimEnsembleFileSet*>       m_outputEnsembleFileSet;
    caf::PdmField<QString>                      m_outputEnsembleName;

    caf::PdmField<std::vector<QDateTime>> m_datesInInputDeck;
    caf::PdmField<std::vector<QString>>   m_wellGroupsInInputDeck;

    caf::PdmField<bool> m_createGridEnsemble;
    caf::PdmField<bool> m_createSummaryEnsemble;

    std::vector<std::string> m_expectedOutputFiles;

    std::vector<std::string> m_jobLog;
    int                      m_subJobsCompleted;
    int                      m_subJobsStarted;
};