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

#include "RimValveTemplate.h"

#include "RimWellPathAicdParameters.h"
#include "RimWellPathValve.h"

#include "cafPdmUiTreeOrdering.h"

CAF_PDM_SOURCE_INIT(RimValveTemplate, "ValveTemplate");


//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimValveTemplate::RimValveTemplate()
{
    CAF_PDM_InitObject("Valve Template", ":/ICDValve16x16.png", "", "");

    CAF_PDM_InitField(&m_valveTemplateUnit, "UnitSystem", caf::AppEnum<RiaEclipseUnitTools::UnitSystem>(RiaEclipseUnitTools::UNITS_UNKNOWN), "Units System", "", "", "");
    m_valveTemplateUnit.uiCapability()->setUiReadOnly(true);

    CAF_PDM_InitFieldNoDefault(&m_type, "CompletionType", "Type", "", "", "");
    m_type = RiaDefines::ICD;
    CAF_PDM_InitField(&m_userLabel, "UserLabel", QString("Template"), "Name", "", "", "");
    
    this->setName(fullLabel());

    CAF_PDM_InitField(&m_orificeDiameter, "OrificeDiameter", 8.0, "Orifice Diameter [mm]", "", "", "");
    CAF_PDM_InitField(&m_flowCoefficient, "FlowCoefficient", 0.7, "Flow Coefficient", "", "", "");

    CAF_PDM_InitFieldNoDefault(&m_aicdParameters, "AICDParameters", "AICD Parameters", "", "", "");
    m_aicdParameters = new RimWellPathAicdParameters;
    m_aicdParameters.uiCapability()->setUiTreeHidden(true);
    m_aicdParameters.uiCapability()->setUiTreeChildrenHidden(true);

}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimValveTemplate::~RimValveTemplate()
{
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimValveTemplate::loadDataAndUpdate()
{

}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimValveTemplate::setUnitSystem(RiaEclipseUnitTools::UnitSystemType unitSystem)
{
    m_valveTemplateUnit = unitSystem;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimValveTemplate::setDefaultValuesFromUnits()
{
    if (m_valveTemplateUnit == RiaEclipseUnitTools::UNITS_METRIC)
    {
        m_orificeDiameter = 8;
    }
    else if (m_valveTemplateUnit == RiaEclipseUnitTools::UNITS_FIELD)
    {
        m_orificeDiameter = 0.315;
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RiaDefines::WellPathComponentType RimValveTemplate::type() const
{
    return m_type();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimValveTemplate::setType(RiaDefines::WellPathComponentType type)
{
    CAF_ASSERT(type == RiaDefines::ICD || type == RiaDefines::AICD || RiaDefines::ICV);

    m_type = type;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RiaEclipseUnitTools::UnitSystemType RimValveTemplate::templateUnits() const
{
    return m_valveTemplateUnit;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
double RimValveTemplate::orificeDiameter() const
{
    return m_orificeDiameter;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
double RimValveTemplate::flowCoefficient() const
{
    return m_flowCoefficient;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
const RimWellPathAicdParameters* RimValveTemplate::aicdParameters() const
{
    return m_aicdParameters;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QString RimValveTemplate::typeLabel() const
{
    return m_type().uiText();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QString RimValveTemplate::fullLabel() const
{
    QString label = QString("%1: %2").arg(typeLabel()).arg(m_userLabel());
    return label;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimValveTemplate::setUserLabel(const QString& userLabel)
{
    m_userLabel = userLabel;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
QList<caf::PdmOptionItemInfo> RimValveTemplate::calculateValueOptions(const caf::PdmFieldHandle* fieldNeedingOptions,
                                                                      bool*                      useOptionsOnly)
{
    QList<caf::PdmOptionItemInfo> options;

    if (fieldNeedingOptions == &m_type)
    {
        std::set<RiaDefines::WellPathComponentType> supportedTypes = {RiaDefines::ICD, RiaDefines::AICD, RiaDefines::ICV};
        for (RiaDefines::WellPathComponentType type : supportedTypes)
        {
            options.push_back(caf::PdmOptionItemInfo(CompletionTypeEnum::uiText(type), type));
        }
    }
    return options;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimValveTemplate::defineUiOrdering(QString uiConfigName, caf::PdmUiOrdering& uiOrdering)
{
    uiOrdering.add(&m_type);
    uiOrdering.add(&m_userLabel);
    uiOrdering.add(&m_valveTemplateUnit);
    if (m_type() == RiaDefines::ICV || m_type() == RiaDefines::ICD)
    {
        if (m_valveTemplateUnit == RiaEclipseUnitTools::UNITS_METRIC)
        {
            m_orificeDiameter.uiCapability()->setUiName("Orifice Diameter [mm]");
        }
        else if (m_valveTemplateUnit == RiaEclipseUnitTools::UNITS_FIELD)
        {
            m_orificeDiameter.uiCapability()->setUiName("Orifice Diameter [in]");
        }
    }

    if (m_type() == RiaDefines::ICV || m_type() == RiaDefines::ICD)
    {
        caf::PdmUiGroup* group = uiOrdering.addNewGroup("MSW Valve Parameters");
        group->add(&m_orificeDiameter);
        group->add(&m_flowCoefficient);
    }
    else
    {
        caf::PdmUiGroup* group = uiOrdering.addNewGroup("MSW AICD Parameters");
        m_aicdParameters->uiOrdering(uiConfigName, *group);
    }
  
    bool readOnly = uiConfigName == QString("InsideValve");
    m_type.uiCapability()->setUiReadOnly(readOnly);
    m_userLabel.uiCapability()->setUiReadOnly(readOnly);
    m_orificeDiameter.uiCapability()->setUiReadOnly(readOnly);
    m_flowCoefficient.uiCapability()->setUiReadOnly(readOnly);

    uiOrdering.skipRemainingFields(true);
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimValveTemplate::fieldChangedByUi(const caf::PdmFieldHandle* changedField,
                                        const QVariant&            oldValue,
                                        const QVariant&            newValue)
{
    if (changedField == &m_type || changedField == &m_userLabel)
    {
        this->setName(fullLabel());
    }
    if (changedField == &m_type)
    {
        std::vector<caf::PdmFieldHandle*> referringFields;
        this->referringPtrFields(referringFields);
        for (caf::PdmFieldHandle* field : referringFields)
        {
            RimWellPathValve* valve = dynamic_cast<RimWellPathValve*>(field->ownerObject());
            valve->templateUpdated();
        }
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RimValveTemplate::defineUiTreeOrdering(caf::PdmUiTreeOrdering& uiTreeOrdering, QString uiConfigName /*= ""*/)
{
    this->setName(fullLabel());
    if (m_type() == RiaDefines::ICV)
    {
        this->setUiIconFromResourceString(":/ICVValve16x16.png");
    }
    else if (m_type() == RiaDefines::ICD)
    {
        this->setUiIconFromResourceString(":/ICDValve16x16.png");
    }
    else if (m_type() == RiaDefines::AICD)
    {
        this->setUiIconFromResourceString(":/AICDValve16x16.png");
    }

}