OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2025-02-25 18:55:30 -06:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3179 from joakim-hove/groupstate-dump

Browse Source 
Add json dump to GroupState
This commit is contained in:
Joakim Hove
2021-04-27 07:52:11 +02:00
committed by GitHub
parent 6cbe5ae8f4 5a4190bbea
commit 42a6505cf1
3 changed files with 88 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
opm/simulators/wells/GroupState.cpp
View File

@@ -17,7 +17,9 @@
along with OPM. If not, see <http://www.gnu.org/licenses/>.
*/
#include <iterator>
#include <opm/json/JsonObject.hpp>
#include <opm/simulators/wells/GroupState.hpp>
@@ -234,4 +236,72 @@ Group::InjectionCMode GroupState::injection_control(const std::string& gname, Ph
return group_iter->second;
}
//-------------------------------------------------------------------------
namespace {
template <typename T>
void dump_vector(Json::JsonObject& root, const std::string& key, const std::vector<T>& data) {
auto data_obj = root.add_array(key);
for (const auto& rate : data)
data_obj.add(rate);
}
template <typename T>
void dump_groupmap(Json::JsonObject& root, const std::string& key, const std::map<std::string, std::vector<T>>& group_data) {
auto map_obj = root.add_object(key);
for (const auto& [gname, data] : group_data)
dump_vector(map_obj, gname, data);
}
template <typename T>
void dump_groupmap(Json::JsonObject& root, const std::string& key, const std::map<std::string, T>& group_data) {
auto map_obj = root.add_object(key);
for (const auto& [gname, value] : group_data)
map_obj.add_item(gname, value);
}
}
std::string GroupState::dump() const
{
Json::JsonObject root;
dump_groupmap(root, "production_rates", this->m_production_rates);
dump_groupmap(root, "prod_red_rates", this->prod_red_rates);
dump_groupmap(root, "inj_red_rates", this->inj_red_rates);
dump_groupmap(root, "inj_resv_rates", this->inj_resv_rates);
dump_groupmap(root, "inj_potentials", this->inj_potentials);
dump_groupmap(root, "inj_rein_rates", this->inj_rein_rates);
dump_groupmap(root, "vrep_rate", this->inj_vrep_rate);
dump_groupmap(root, "grat_sales_target", this->m_grat_sales_target);
{
std::map<std::string, int> int_controls;
for (const auto& [gname, control] : this->production_controls)
int_controls.insert({gname, static_cast<int>(control)});
dump_groupmap(root, "production_controls", int_controls);
}
{
std::map<std::string, std::vector<std::pair<Opm::Phase, Group::InjectionCMode>>> inj_cntrl;
for (const auto& [phase_name, cmode] : this->injection_controls) {
const auto& [phase, gname] = phase_name;
inj_cntrl[gname].emplace_back(phase, cmode);
}
auto map_obj = root.add_object("injection_controls");
for (const auto& [gname, phase_cmode] : inj_cntrl) {
auto group_array = map_obj.add_array(gname);
for (const auto& [phase, cmode] : phase_cmode) {
auto control_pair = group_array.add_array();
control_pair.add(static_cast<int>(phase));
control_pair.add(static_cast<int>(cmode));
}
}
}
return root.to_string();
}
}

3
opm/simulators/wells/GroupState.hpp
View File

@@ -147,6 +147,9 @@ public:
throw std::logic_error("Internal size mismatch when distributing groupData");
}
std::string dump() const;
private:
std::size_t num_phases;
std::map<std::string, std::vector<double>> m_production_rates;