/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) Statoil ASA
//  Copyright (C) Ceetron Solutions AS
// 
//  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 "RigTernaryResultAccessor.h"

#include "RigResultAccessor.h"

#include <cmath> // Needed for HUGE_VAL on Linux

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
RigTernaryResultAccessor::RigTernaryResultAccessor()
{

}

//--------------------------------------------------------------------------------------------------
/// Requires at least two data objects present, asserts if more than one data accessor is nullptr
//--------------------------------------------------------------------------------------------------
void RigTernaryResultAccessor::setTernaryResultAccessors(RigResultAccessor* soil, RigResultAccessor* sgas, RigResultAccessor* swat)
{
    m_soilAccessor = soil;
    m_sgasAccessor = sgas;
    m_swatAccessor = swat;
}

//--------------------------------------------------------------------------------------------------
/// If only swat is present, soil is set to (1.0 - swat) and sgas to 0
//--------------------------------------------------------------------------------------------------
cvf::Vec2d RigTernaryResultAccessor::cellScalar(size_t gridLocalCellIndex) const
{
    double soil = 0.0;
    double sgas = 0.0;

    if (m_soilAccessor.notNull())
    { 
        soil = m_soilAccessor->cellScalar(gridLocalCellIndex);

        if (m_sgasAccessor.notNull())
        {
            sgas = m_sgasAccessor->cellScalar(gridLocalCellIndex);
        }
        else if (m_swatAccessor.notNull())
        {
            sgas = 1.0 - soil - m_swatAccessor->cellScalar(gridLocalCellIndex);
        }
        else
        {
            sgas = 1.0 - soil;
        }
    }
    else
    {
        if (m_sgasAccessor.notNull())
        {
            sgas = m_sgasAccessor->cellScalar(gridLocalCellIndex);

            if (m_swatAccessor.notNull())
            {
                soil = 1.0 - sgas - m_swatAccessor->cellScalar(gridLocalCellIndex);
            }
            else
            {
                soil = 1.0 - sgas;
            }
        }
        else if (m_swatAccessor.notNull())
        {
            soil = 1.0 - m_swatAccessor->cellScalar(gridLocalCellIndex);
        }
    }

    return cvf::Vec2d(soil, sgas);
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
cvf::Vec2d RigTernaryResultAccessor::cellFaceScalar(size_t gridLocalCellIndex, cvf::StructGridInterface::FaceType faceId) const
{
    return cellScalar(gridLocalCellIndex);
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
cvf::Vec2d RigTernaryResultAccessor::cellScalarGlobIdx(size_t globCellIndex) const
{
    double soil = 0.0;
    double sgas = 0.0;

    if (m_soilAccessor.notNull())
    {
        soil = m_soilAccessor->cellScalarGlobIdx(globCellIndex);

        if (soil == HUGE_VAL)
        {
            return cvf::Vec2d(HUGE_VAL, HUGE_VAL);
        }

        if (m_sgasAccessor.notNull())
        {
            sgas = m_sgasAccessor->cellScalarGlobIdx(globCellIndex);
        }
        else if (m_swatAccessor.notNull())
        {
            sgas = 1.0 - soil - m_swatAccessor->cellScalarGlobIdx(globCellIndex);
        }
        else
        {
            sgas = 1.0 - soil;
        }
    }
    else
    {
        if (m_sgasAccessor.notNull())
        {
            sgas = m_sgasAccessor->cellScalarGlobIdx(globCellIndex);

            if (sgas == HUGE_VAL)
            {
                return cvf::Vec2d(HUGE_VAL, HUGE_VAL);
            }

            if (m_swatAccessor.notNull())
            {
                soil = 1.0 - sgas - m_swatAccessor->cellScalarGlobIdx(globCellIndex);
            }
            else
            {
                soil = 1.0 - sgas;
            }
        }
        else if (m_swatAccessor.notNull())
        {
            double swat = m_swatAccessor->cellScalarGlobIdx(globCellIndex);
            if (swat == HUGE_VAL)
            {
                return cvf::Vec2d(HUGE_VAL, HUGE_VAL);
            }

            soil = 1.0 - m_swatAccessor->cellScalarGlobIdx(globCellIndex);
        }
    }

    return cvf::Vec2d(soil, sgas);
}