ResInsight/ApplicationCode/Commands/CompletionExportCommands/RicMswCompletions.h

/////////////////////////////////////////////////////////////////////////////////
//
//  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 "RicMswSubSegment.h"

#include "RigCompletionData.h"

#include "RimWellPathAicdParameters.h"

#include "cvfMath.h"

#include <QString>
#include <memory>

class RimWellPathValve;

//==================================================================================================
///
//==================================================================================================
class RicMswCompletion
{
public:
    RicMswCompletion( const QString& label, size_t index = cvf::UNDEFINED_SIZE_T, int branchNumber = cvf::UNDEFINED_INT );

    virtual RigCompletionData::CompletionType completionType() const = 0;

    const QString& label() const;
    size_t         index() const;
    int            branchNumber() const;
    void           setBranchNumber( int branchNumber );

    void addSubSegment( std::shared_ptr<RicMswSubSegment> subSegment );

    std::vector<std::shared_ptr<RicMswSubSegment>>&       subSegments();
    const std::vector<std::shared_ptr<RicMswSubSegment>>& subSegments() const;

    void setLabel( const QString& label );

private:
    QString m_label;
    size_t  m_index;
    int     m_branchNumber;

    std::vector<std::shared_ptr<RicMswSubSegment>> m_subSegments;
};

class RicMswFishbones : public RicMswCompletion
{
public:
    RicMswFishbones( const QString& label, size_t index = cvf::UNDEFINED_SIZE_T, int branchNumber = cvf::UNDEFINED_INT )
        : RicMswCompletion( label, index, branchNumber )
    {
    }

    RigCompletionData::CompletionType completionType() const override { return RigCompletionData::FISHBONES; }
};

//==================================================================================================
///
//==================================================================================================
class RicMswFracture : public RicMswCompletion
{
public:
    RicMswFracture( const QString& label, size_t index = cvf::UNDEFINED_SIZE_T, int branchNumber = cvf::UNDEFINED_INT );
    RigCompletionData::CompletionType completionType() const override;
};

//==================================================================================================
///
//==================================================================================================
class RicMswPerforation : public RicMswCompletion
{
public:
    RicMswPerforation( const QString& label, size_t index = cvf::UNDEFINED_SIZE_T, int branchNumber = cvf::UNDEFINED_INT );
    RigCompletionData::CompletionType completionType() const override;
};

//==================================================================================================
///
//==================================================================================================
class RicMswValve : public RicMswCompletion
{
public:
    RicMswValve( const QString& label, const RimWellPathValve* wellPathValve );

    virtual ~RicMswValve() {}

    const RimWellPathValve* wellPathValve() const;

private:
    const RimWellPathValve* m_wellPathValve;
};

//==================================================================================================
///
//==================================================================================================
class RicMswWsegValve : public RicMswValve
{
public:
    RicMswWsegValve( const QString& label, const RimWellPathValve* wellPathValve );

    double flowCoefficient() const;
    double area() const;
    void   setFlowCoefficient( double icdFlowCoefficient );
    void   setArea( double icdArea );

private:
    double m_flowCoefficient;
    double m_area;
};

//==================================================================================================
///
//==================================================================================================
class RicMswFishbonesICD : public RicMswWsegValve
{
public:
    RicMswFishbonesICD( const QString& label, const RimWellPathValve* wellPathValve );
    RigCompletionData::CompletionType completionType() const override;
};

//==================================================================================================
///
//==================================================================================================
class RicMswPerforationICD : public RicMswWsegValve
{
public:
    RicMswPerforationICD( const QString& label, const RimWellPathValve* wellPathValve );
    RigCompletionData::CompletionType completionType() const override;
};

//==================================================================================================
///
//==================================================================================================
class RicMswPerforationICV : public RicMswWsegValve
{
public:
    RicMswPerforationICV( const QString& label, const RimWellPathValve* wellPathValve );
    RigCompletionData::CompletionType completionType() const override;
};

//==================================================================================================
///
//==================================================================================================
class RicMswPerforationAICD : public RicMswValve
{
public:
    RicMswPerforationAICD( const QString& label, const RimWellPathValve* wellPathValve );
    RigCompletionData::CompletionType completionType() const override;

    bool   isValid() const;
    void   setIsValid( bool valid );
    bool   isOpen() const;
    void   setIsOpen( bool deviceOpen );
    double length() const;
    void   setLength( double length );
    double flowScalingFactor() const;
    void   setflowScalingFactor( double scalingFactor );

    const std::array<double, AICD_NUM_PARAMS>& values() const;
    std::array<double, AICD_NUM_PARAMS>&       values();

private:
    bool                                m_valid;
    bool                                m_deviceOpen;
    std::array<double, AICD_NUM_PARAMS> m_parameters;
    double                              m_length;
    double                              m_flowScalingFactor;
};