diff --git a/opm/simulators/wells/GroupState.cpp b/opm/simulators/wells/GroupState.cpp index 87fc18951..531f8f867 100644 --- a/opm/simulators/wells/GroupState.cpp +++ b/opm/simulators/wells/GroupState.cpp @@ -17,7 +17,9 @@ along with OPM. If not, see . */ +#include +#include #include @@ -234,4 +236,72 @@ Group::InjectionCMode GroupState::injection_control(const std::string& gname, Ph return group_iter->second; } + +//------------------------------------------------------------------------- + +namespace { + +template +void dump_vector(Json::JsonObject& root, const std::string& key, const std::vector& data) { + auto data_obj = root.add_array(key); + for (const auto& rate : data) + data_obj.add(rate); +} + + +template +void dump_groupmap(Json::JsonObject& root, const std::string& key, const std::map>& group_data) { + auto map_obj = root.add_object(key); + for (const auto& [gname, data] : group_data) + dump_vector(map_obj, gname, data); +} + + +template +void dump_groupmap(Json::JsonObject& root, const std::string& key, const std::map& 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 int_controls; + for (const auto& [gname, control] : this->production_controls) + int_controls.insert({gname, static_cast(control)}); + dump_groupmap(root, "production_controls", int_controls); + } + { + std::map>> 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(phase)); + control_pair.add(static_cast(cmode)); + } + } + } + return root.to_string(); +} + + } diff --git a/opm/simulators/wells/GroupState.hpp b/opm/simulators/wells/GroupState.hpp index 6b2115fbc..4647ba007 100644 --- a/opm/simulators/wells/GroupState.hpp +++ b/opm/simulators/wells/GroupState.hpp @@ -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> m_production_rates; diff --git a/tests/test_GroupState.cpp b/tests/test_GroupState.cpp index 3a5d6df52..f549d6b8a 100644 --- a/tests/test_GroupState.cpp +++ b/tests/test_GroupState.cpp @@ -20,6 +20,8 @@ #include #include +#include + #define BOOST_TEST_MODULE GroupStateTest #include @@ -67,3 +69,16 @@ BOOST_AUTO_TEST_CASE(GroupStateCreate) { BOOST_CHECK(gs2 == gs); } + +BOOST_AUTO_TEST_CASE(GroupStateDump) { + std::size_t num_phases{3}; + GroupState gs(num_phases); + std::vector rates{0,1,2}; + gs.update_production_rates("AGROUP", rates); + gs.update_injection_rein_rates("CGROUP", rates); + gs.injection_control("AGROUP", Phase::WATER, Group::InjectionCMode::RATE); + gs.injection_control("AGROUP", Phase::GAS, Group::InjectionCMode::RATE); + + auto json_string = gs.dump(); + Json::JsonObject json_gs(json_string); +}