/////////////////////////////////////////////////////////////////////////////////
//
//  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.
//
/////////////////////////////////////////////////////////////////////////////////
#pragma once

#include "RicMswExportInfo.h"
#include "RigCompletionData.h"

class RicExportCompletionDataSettingsUi;
class RifEclipseDataTableFormatter;
class RimEclipseCase;
class RimFishbonesMultipleSubs;
class RimPerforationInterval;
class RimWellPath;
class RimWellPathValve;
class RimWellPathFracture;
class SubSegmentIntersectionInfo;

class QFile;

class RicWellPathExportMswCompletionsImpl
{
public:
    static void exportWellSegmentsForAllCompletions(const RicExportCompletionDataSettingsUi& exportSettings,
                                                    const std::vector<RimWellPath*>&         wellPaths);

    static void exportWellSegmentsForFractures(RimEclipseCase*                          eclipseCase,
                                               std::shared_ptr<QFile>                   exportFile,
                                               const RimWellPath*                       wellPath,
                                               const std::vector<RimWellPathFracture*>& fractures);

    static void exportWellSegmentsForFishbones(RimEclipseCase*                               eclipseCase,
                                               std::shared_ptr<QFile>                        exportFile,
                                               const RimWellPath*                            wellPath,
                                               const std::vector<RimFishbonesMultipleSubs*>& fishbonesSubs);

    static void exportWellSegmentsForPerforations(RimEclipseCase*                                   eclipseCase,
                                                  std::shared_ptr<QFile>                            exportFile,
                                                  const RimWellPath*                                wellPath,
                                                  int                                               timeStep,
                                                  const std::vector<const RimPerforationInterval*>& perforationIntervals);

    static RicMswExportInfo generateFishbonesMswExportInfo(const RimEclipseCase* caseToApply,
                                                           const RimWellPath*    wellPath,
                                                           bool                  enableSegmentSplitting);

private:
    static RicMswExportInfo generateFishbonesMswExportInfo(const RimEclipseCase*                         caseToApply,
                                                           const RimWellPath*                            wellPath,
                                                           const std::vector<RimFishbonesMultipleSubs*>& fishbonesSubs,
                                                           bool                                          enableSegmentSplitting);

    static RicMswExportInfo generateFracturesMswExportInfo(RimEclipseCase* caseToApply, const RimWellPath* wellPath);

    static RicMswExportInfo generateFracturesMswExportInfo(RimEclipseCase*                          caseToApply,
                                                           const RimWellPath*                       wellPath,
                                                           const std::vector<RimWellPathFracture*>& fractures);

    static RicMswExportInfo
        generatePerforationsMswExportInfo(RimEclipseCase*                                   eclipseCase,
                                          const RimWellPath*                                wellPath,
                                          int                                               timeStep,
                                          const std::vector<const RimPerforationInterval*>& perforationIntervals);

    static void generateWelsegsTable(RifEclipseDataTableFormatter& formatter, const RicMswExportInfo& exportInfo);

    static void generateWelsegsSegments(RifEclipseDataTableFormatter&                      formatter,
                                        const RicMswExportInfo&                            exportInfo,
                                        const std::set<RigCompletionData::CompletionType>& exportCompletionTypes);
    static void generateWelsegsCompletionCommentHeader(RifEclipseDataTableFormatter&     formatter,
                                                       RigCompletionData::CompletionType completionType);
    static void generateCompsegTables(RifEclipseDataTableFormatter& formatter, const RicMswExportInfo& exportInfo);
    static void generateCompsegTable(RifEclipseDataTableFormatter&                      formatter,
                                     const RicMswExportInfo&                            exportInfo,
                                     bool                                               exportSubGridIntersections,
                                     const std::set<RigCompletionData::CompletionType>& exportCompletionTypes);
    static void generateCompsegHeader(RifEclipseDataTableFormatter&     formatter,
                                      const RicMswExportInfo&           exportInfo,
                                      RigCompletionData::CompletionType completionType,
                                      bool                              exportSubGridIntersections);
    static void generateWsegvalvTable(RifEclipseDataTableFormatter& formatter, const RicMswExportInfo& exportInfo);
    static void generateWsegAicdTable(RifEclipseDataTableFormatter& formatter, const RicMswExportInfo& exportInfo);

private:
    typedef std::vector<std::shared_ptr<RicMswSegment>> MainBoreSegments;
    typedef std::map<std::shared_ptr<RicMswCompletion>, std::set<std::pair<const RimWellPathValve*, size_t>>>
        ValveContributionMap;

    static MainBoreSegments createMainBoreSegmentsForPerforations(const std::vector<SubSegmentIntersectionInfo>&    subSegIntersections,
                                                   const std::vector<const RimPerforationInterval*>& perforationIntervals,
                                                   const RimWellPath*                                wellPath,
                                                   int                                               timeStep,
                                                   RimEclipseCase*                                   eclipseCase,
                                                   bool*                                             foundSubGridIntersections);

    static void createValveCompletions(std::vector<std::shared_ptr<RicMswSegment>>& mainBoreSegments,
                                            const std::vector<const RimPerforationInterval*>& perforationIntervals,
                                            RiaEclipseUnitTools::UnitSystem unitSystem);

    static void assignValveContributionsToSuperICDsOrAICDs(const std::vector<std::shared_ptr<RicMswSegment>>& mainBoreSegments,
                                                      const std::vector<const RimPerforationInterval*>&  perforationIntervals,
                                                      RiaEclipseUnitTools::UnitSystem                    unitSystem);

    static void moveIntersectionsToICVs(const std::vector<std::shared_ptr<RicMswSegment>>& mainBoreSegments,
                                        const std::vector<const RimPerforationInterval*>&  perforationIntervals,
                                        RiaEclipseUnitTools::UnitSystem                    unitSystem);

    static void moveIntersectionsToSuperICDsOrAICDs(MainBoreSegments mainBoreSegments);

    static void assignFishbonesLateralIntersections(const RimEclipseCase*           caseToApply,
                                                    const RimFishbonesMultipleSubs* fishbonesSubs,
                                                    std::shared_ptr<RicMswSegment>  location,
                                                    bool*                           foundSubGridIntersections,
                                                    double                          maxSegmentLength);

    static void assignFractureIntersections(const RimEclipseCase*                 caseToApply,
                                            const RimWellPathFracture*            fracture,
                                            const std::vector<RigCompletionData>& completionData,
                                            std::shared_ptr<RicMswSegment>        location,
                                            bool*                                 foundSubGridIntersections);

    static std::vector<RigCompletionData> generatePerforationIntersections(const RimWellPath*            wellPath,
                                                                           const RimPerforationInterval* perforationInterval,
                                                                           int                           timeStep,
                                                                           RimEclipseCase*               eclipseCase);

    static void assignPerforationIntersections(const std::vector<RigCompletionData>& completionData,
                                               std::shared_ptr<RicMswCompletion> perforationCompletion,
                                               const SubSegmentIntersectionInfo& cellIntInfo,
                                               double overlapStart,
                                               double overlapEnd,
                                               bool* foundSubGridIntersections);

    static void assignBranchAndSegmentNumbers(const RimEclipseCase*          caseToApply,
                                              std::shared_ptr<RicMswSegment> location,
                                              int*                           branchNum,
                                              int*                           segmentNum);
    static void assignBranchAndSegmentNumbers(const RimEclipseCase* caseToApply, RicMswExportInfo* exportInfo);    

};