Add class SegmentState

CMakeLists_files.cmake

@@ -60,6 +60,7 @@ list (APPEND MAIN_SOURCE_FILES
   opm/simulators/wells/GlobalWellInfo.cpp
   opm/simulators/wells/GroupState.cpp
   opm/simulators/wells/ParallelWellInfo.cpp
+  opm/simulators/wells/SegmentState.cpp
   opm/simulators/wells/TargetCalculator.cpp
   opm/simulators/wells/VFPHelpers.cpp
   opm/simulators/wells/VFPProdProperties.cpp
@@ -205,6 +206,7 @@ list (APPEND PUBLIC_HEADER_FILES
   opm/core/props/phaseUsageFromDeck.hpp
   opm/core/props/satfunc/RelpermDiagnostics.hpp
   opm/simulators/timestepping/SimulatorReport.hpp
+  opm/simulators/wells/SegmentState.hpp
   opm/simulators/wells/WellContainer.hpp
   opm/simulators/aquifers/AquiferInterface.hpp
   opm/simulators/aquifers/AquiferCarterTracy.hpp

opm/simulators/wells/SegmentState.cpp

@@ -0,0 +1,47 @@
+/*
+  Copyright Equinor ASA 2021
+
+  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/>.
+*/
+
+#include <opm/simulators/wells/SegmentState.hpp>
+
+#include <opm/parser/eclipse/EclipseState/Schedule/Well/WellConnections.hpp>
+#include <opm/parser/eclipse/EclipseState/Schedule/MSW/WellSegments.hpp>
+
+namespace Opm
+{
+
+SegmentState::SegmentState(int num_phases, const WellSegments& segments) :
+    rates(segments.size() * num_phases),
+    pressure(segments.size()),
+    pressure_drop_friction(segments.size()),
+    pressure_drop_hydrostatic(segments.size()),
+    pressure_drop_accel(segments.size())
+{
+
+}
+
+double SegmentState::pressure_drop(std::size_t index) const {
+    return this->pressure_drop_friction[index] + this->pressure_drop_hydrostatic[index] + this->pressure_drop_accel[index];
+}
+
+
+bool SegmentState::empty() const {
+    return this->rates.empty();
+}
+
+}

opm/simulators/wells/SegmentState.hpp

@@ -0,0 +1,49 @@
+/*
+  Copyright Equinor ASA 2021
+
+  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_SEGMENTSTATE_HEADER_INCLUDED
+#define OPM_SEGMENTSTATE_HEADER_INCLUDED
+
+#include <vector>
+
+namespace Opm
+{
+
+class WellSegments;
+class WellConnections;
+
+
+class SegmentState
+{
+public:
+    SegmentState() = default;
+    SegmentState(int num_phases, const WellSegments& segments);
+    double pressure_drop(std::size_t index) const;
+    bool empty() const;
+
+    std::vector<double> rates;
+    std::vector<double> pressure;
+    std::vector<double> pressure_drop_friction;
+    std::vector<double> pressure_drop_hydrostatic;
+    std::vector<double> pressure_drop_accel;
+};
+
+}
+
+#endif

tests/test_wellstate.cpp

@@ -26,6 +26,7 @@
 #include <opm/simulators/wells/GlobalWellInfo.hpp>
 #include <opm/simulators/wells/ParallelWellInfo.hpp>
 #include <opm/simulators/wells/WellState.hpp>
+#include <opm/simulators/wells/SegmentState.hpp>
 #include <opm/simulators/wells/WellContainer.hpp>
 #include <opm/parser/eclipse/Python/Python.hpp>
 
@@ -522,5 +523,20 @@ BOOST_AUTO_TEST_CASE(TESTWellContainer) {
     BOOST_CHECK(!wx.has_value());
 }
 
+BOOST_AUTO_TEST_CASE(TESTSegmentState) {
+    const Setup setup{ "msw.data" };
+    const auto& well = setup.sched.getWell("PROD01", 0);
+    const auto& segments = well.getSegments();
+    Opm::SegmentState ss1(3, segments);
+    Opm::SegmentState ss2;
+
+
+    ss1.pressure_drop_hydrostatic[0] = 1;
+    ss1.pressure_drop_friction[0] = 2;
+    ss1.pressure_drop_accel[0] = 3;
+
+    BOOST_CHECK_EQUAL(ss1.pressure_drop(0), 6);
+}
+
 
 BOOST_AUTO_TEST_SUITE_END()