89 lines
2.7 KiB
C++
89 lines
2.7 KiB
C++
/*
|
|
Copyright 2019 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 GUIDE_RATE_CONFIG_HPP
|
|
#define GUIDE_RATE_CONFIG_HPP
|
|
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <memory>
|
|
|
|
#include <opm/parser/eclipse/EclipseState/Schedule/Group/GuideRateModel.hpp>
|
|
#include <opm/parser/eclipse/EclipseState/Schedule/Group/Group.hpp>
|
|
#include <opm/parser/eclipse/EclipseState/Schedule/Well/Well.hpp>
|
|
|
|
namespace Opm {
|
|
|
|
|
|
class GuideRateConfig {
|
|
public:
|
|
|
|
struct WellTarget {
|
|
double guide_rate;
|
|
Well::GuideRateTarget target;
|
|
double scaling_factor;
|
|
|
|
bool operator==(const WellTarget& data) const {
|
|
return guide_rate == data.guide_rate &&
|
|
target == data.target &&
|
|
scaling_factor == data.scaling_factor;
|
|
}
|
|
};
|
|
|
|
struct GroupTarget {
|
|
double guide_rate;
|
|
Group::GuideRateTarget target;
|
|
|
|
bool operator==(const GroupTarget& data) const {
|
|
return guide_rate == data.guide_rate &&
|
|
target == data.target;
|
|
}
|
|
};
|
|
|
|
GuideRateConfig() = default;
|
|
GuideRateConfig(std::shared_ptr<GuideRateModel> model,
|
|
const std::unordered_map<std::string,WellTarget>& well,
|
|
const std::unordered_map<std::string,GroupTarget>& group);
|
|
|
|
const GuideRateModel& model() const;
|
|
bool has_model() const;
|
|
bool update_model(const GuideRateModel& model);
|
|
void update_well(const Well& well);
|
|
void update_group(const Group& group);
|
|
const WellTarget& well(const std::string& well) const;
|
|
const GroupTarget& group(const std::string& group) const;
|
|
bool has_well(const std::string& well) const;
|
|
bool has_group(const std::string& group) const;
|
|
|
|
std::shared_ptr<GuideRateModel> getModel() const;
|
|
const std::unordered_map<std::string, WellTarget>& getWells() const;
|
|
const std::unordered_map<std::string, GroupTarget>& getGroups() const;
|
|
|
|
bool operator==(const GuideRateConfig& data) const;
|
|
|
|
private:
|
|
std::shared_ptr<GuideRateModel> m_model;
|
|
std::unordered_map<std::string, WellTarget> wells;
|
|
std::unordered_map<std::string, GroupTarget> groups;
|
|
};
|
|
|
|
}
|
|
|
|
#endif
|