/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2021 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 "RiaDefines.h"

#include "RigHistogramData.h"

#include "cvfObject.h"

#include <functional>
#include <vector>

class RimEnsembleFractureStatistics;
class RigFractureGrid;
class RigStimPlanFractureDefinition;

//==================================================================================================
///
///
//==================================================================================================
class RigEnsembleFractureStatisticsCalculator
{
public:
    enum class PropertyType
    {
        HEIGHT,
        AREA,
        WIDTH,
        XF,
        KFWF,
        PERMEABILITY,
        FORMATION_DIP
    };

    static RigHistogramData
        createStatisticsData( const RimEnsembleFractureStatistics* esf, PropertyType propertyType, int numBins );

    static RigHistogramData
        createStatisticsData( const std::vector<cvf::ref<RigStimPlanFractureDefinition>>& fractureDefinitions,
                              PropertyType                                                propertyType,
                              int                                                         numBins );

    static std::vector<cvf::ref<RigStimPlanFractureDefinition>>
        removeZeroWidthDefinitions( const std::vector<cvf::ref<RigStimPlanFractureDefinition>>& fractureDefinitions );

    static std::vector<double>
        calculateProperty( const std::vector<cvf::ref<RigStimPlanFractureDefinition>>& fractureDefinitions,
                           PropertyType                                                propertyType );

private:
    static std::vector<double>
        calculateGridStatistics( const std::vector<cvf::ref<RigStimPlanFractureDefinition>>& fractureDefinitions,
                                 double( func )( cvf::cref<RigFractureGrid> ) );

    static std::vector<double>
        calculateAreaWeightedStatistics( const std::vector<cvf::ref<RigStimPlanFractureDefinition>>& fractureDefinitions,
                                         double( func )( cvf::cref<RigFractureGrid>,
                                                         cvf::cref<RigFractureGrid>,
                                                         RiaDefines::EclipseUnitSystem,
                                                         const QString& ) );

    static std::vector<double>
        calculateFormationDip( const std::vector<cvf::ref<RigStimPlanFractureDefinition>>& fractureDefinitions );

    static double calculateHeight( cvf::cref<RigFractureGrid> fractureGrid );
    static double calculateArea( cvf::cref<RigFractureGrid> fractureGrid );
    static double calculateXf( cvf::cref<RigFractureGrid> fractureGrid );
    static double calculateKfWf( cvf::cref<RigFractureGrid> fractureGrid );

    static double calculateAreaWeightedWidth( cvf::cref<RigFractureGrid>    conductivityGrid,
                                              cvf::cref<RigFractureGrid>    widthGrid,
                                              RiaDefines::EclipseUnitSystem widthUnitSystem,
                                              const QString&                widthUnit );
    static double calculateAreaWeightedPermeability( cvf::cref<RigFractureGrid>    conductivityGrid,
                                                     cvf::cref<RigFractureGrid>    widthGrid,
                                                     RiaDefines::EclipseUnitSystem widthUnitSystem,
                                                     const QString&                widthUnit );

    static double convertUnit( double value, RiaDefines::EclipseUnitSystem unitSystem, const QString& unitName );
};