/////////////////////////////////////////////////////////////////////////////////
//
//  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.
//
/////////////////////////////////////////////////////////////////////////////////

#pragma once

#include "RigCompletionData.h"

#include <map>
#include <vector>

class RigFractureGrid;
class RicWellPathFractureReportItem;
class RigWellPath;
class RigTransmissibilityCondenser;

class RimEclipseCase;
class RimFracture;
class RimFractureTemplate;
class RimSimWellInView;
class RimWellPath;

class QTextStream;
class QString;
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
class RicExportFractureCompletionsImpl
{
public:
    static std::vector<RigCompletionData>
        generateCompdatValuesForWellPath(RimWellPath*                                wellPath,
                                         RimEclipseCase*                             caseToApply,
                                         std::vector<RicWellPathFractureReportItem>* fractureDataForReport,
                                         QTextStream*                                outputStreamForIntermediateResultsText);

    static std::vector<RigCompletionData> generateCompdatValuesForSimWell(RimEclipseCase*         eclipseCase,
                                                                          const RimSimWellInView* well,
                                                                          QTextStream* outputStreamForIntermediateResultsText);

    static std::vector<RigCompletionData>
        generateCompdatValues(RimEclipseCase*                             caseToApply,
                              const QString&                              wellPathName,
                              const RigWellPath*                          wellPathGeometry,
                              const std::vector<const RimFracture*>&      fractures,
                              std::vector<RicWellPathFractureReportItem>* fractureDataReportItems,
                              QTextStream*                                outputStreamForIntermediateResultsText);

private:
    static std::vector<RigCompletionData>
        generateCompdatValuesConst(const RimEclipseCase*                       caseToApply,
                                   const QString&                              wellPathName,
                                   const RigWellPath*                          wellPathGeometry,
                                   const std::vector<const RimFracture*>&      fractures,
                                   std::vector<RicWellPathFractureReportItem>* fractureDataReportItems,
                                   QTextStream*                                outputStreamForIntermediateResultsText);

    static bool checkForStimPlanConductivity(const RimFractureTemplate* fracTemplate, const RimFracture* fracture);

    static void calculateInternalFractureTransmissibilities(const RigFractureGrid* fractureGrid, double cDarcyInCorrectUnit, RigTransmissibilityCondenser &transCondenser);
    static void calculateFractureToWellTransmissibilities(const RimFractureTemplate* fracTemplate, const RigFractureGrid* fractureGrid, const RimFracture* fracture, double cDarcyInCorrectUnit, const RigWellPath* wellPathGeometry, RigTransmissibilityCondenser &transCondenser);

    static std::map<size_t, double>       calculateMatrixToWellTransmissibilities(RigTransmissibilityCondenser &transCondenser);
    static std::vector<RigCompletionData> generateCompdatValuesForFracture(const std::map<size_t, double>& matrixToWellTransmissibilites, const QString& wellPathName, const RimEclipseCase* caseToApply, const RimFracture* fracture, const RimFractureTemplate* fracTemplate);

    static void computeNonDarcyFlowParameters(const RimFracture* fracture, std::vector<RigCompletionData> allCompletionsForOneFracture);

    static double sumUpCellAreas(const std::map<size_t, double>& cellAreas);
    static double sumUpTransmissibilities(const std::vector<RigCompletionData>& allCompletionsForOneFracture);

    static void calculateAndSetLengthsAndConductivity(const RimFractureTemplate* fracTemplate, double area, RicWellPathFractureReportItem &reportItem);
    static void calculateAndSetAreaWeightedTransmissibility(const RimEclipseCase* caseToApply, std::map<size_t, double> cellAreas, double area, RicWellPathFractureReportItem &reportItem);

    static void outputIntermediateResultsText(QTextStream* outputStreamForIntermediateResultsText, const RimFracture* fracture, RigTransmissibilityCondenser &transCondenser, const RigMainGrid* mainGrid, const RigFractureGrid* fractureGrid);
};