/* Copyright Equinor ASA 2021, 2022. 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 . */ #ifndef OPM_SEGMENTSTATE_HEADER_INCLUDED #define OPM_SEGMENTSTATE_HEADER_INCLUDED #include #include namespace Opm { class WellSegments; } // namespace Opm namespace Opm { class SegmentState { public: SegmentState() = default; SegmentState(int num_phases, const WellSegments& segments); static SegmentState serializationTestObject(); double pressure_drop(std::size_t index) const; bool empty() const; void scale_pressure(double bhp); const std::vector& segment_number() const; std::size_t size() const; template void serializeOp(Serializer& serializer) { serializer(rates); serializer(dissolved_gas_rate); serializer(vaporized_oil_rate); serializer(phase_resv_rates); serializer(phase_velocity); serializer(phase_holdup); serializer(phase_viscosity); serializer(pressure); serializer(pressure_drop_friction); serializer(pressure_drop_hydrostatic); serializer(pressure_drop_accel); serializer(m_segment_number); } bool operator==(const SegmentState&) const; std::vector rates; std::vector dissolved_gas_rate; std::vector vaporized_oil_rate; /// Segment condition volume flow rates through segment (per phase) std::vector phase_resv_rates; /// Segment condition flow velocity through segment (per phase) std::vector phase_velocity; /// Segment condition holdup fractions through segment (per phase) std::vector phase_holdup; /// Segment condition phase viscosities. std::vector phase_viscosity; std::vector pressure; std::vector pressure_drop_friction; std::vector pressure_drop_hydrostatic; std::vector pressure_drop_accel; private: std::vector m_segment_number; }; } // namepace Opm #endif // OPM_SEGMENTSTATE_HEADER_INCLUDED