TargetCalculator.hpp

/*
  Copyright 2020 Equinor ASA.
 
  This file is part of the Open Porous Media project (OPM).
 
  OPM 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.
 
  OPM 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 for more details.
 
  You should have received a copy of the GNU General Public License
  along with OPM.  If not, see <http://www.gnu.org/licenses/>.
*/
 
 
#ifndef OPM_TARGETCALCULATOR_HEADER_INCLUDED
#define OPM_TARGETCALCULATOR_HEADER_INCLUDED
 
#include <opm/input/eclipse/Schedule/Group/Group.hpp>
 
#include <vector>
 
namespace Opm {
 
template<typename Scalar, typename IndexTraits> class GroupStateHelper;
 
namespace GroupStateHelpers
 {

template<typename Scalar, typename IndexTraits>
class TargetCalculator
{
public:
    using GroupStateHelperType = Opm::GroupStateHelper<Scalar, IndexTraits>;
 
    TargetCalculator(const GroupStateHelperType& groupStateHelper,
                     const std::vector<Scalar>& resv_coeff,
                     const Group& group);

    TargetCalculator(const GroupStateHelperType& groupStateHelper,
                     const std::vector<Scalar>& resv_coeff,
                     Group::ProductionCMode cmode);
 
    template <typename RateType>
    RateType calcModeRateFromRates(const std::vector<RateType>& rates) const
    {
      return calcModeRateFromRates(rates.data());
    }
 
    template <typename RateType>
    RateType calcModeRateFromRates(const RateType* rates) const;
 
private:
    Group::ProductionCMode cmode_;
    const GroupStateHelperType& groupStateHelper_;
    const std::vector<Scalar>& resv_coeff_;
};

template<typename Scalar, typename IndexTraits>
class InjectionTargetCalculator
{
public:
    using GroupStateHelperType = GroupStateHelper<Scalar, IndexTraits>;
 
    InjectionTargetCalculator(const GroupStateHelperType& groupStateHelper,
                              const Phase& injection_phase);
 
    template <typename RateVec>
    auto calcModeRateFromRates(const RateVec& rates) const
    {
        return rates[pos_];
    }
 
private:
    int pos_;
};
 
} // namespace GroupStateHelpers
 
 
} // namespace Opm
 
#endif