diff --git a/opm/core/simulator/WellState.hpp b/opm/core/simulator/WellState.hpp index df823324a..419353dac 100644 --- a/opm/core/simulator/WellState.hpp +++ b/opm/core/simulator/WellState.hpp @@ -127,6 +127,26 @@ namespace Opm std::vector& perfPress() { return perfpress_; } const std::vector& perfPress() const { return perfpress_; } + size_t getRestartBhpOffset() const { + return 0; + } + + size_t getRestartPerfPressOffset() const { + return bhp_.size(); + } + + size_t getRestartPerfRatesOffset() const { + return getRestartPerfPressOffset() + perfpress_.size(); + } + + size_t getRestartTemperatureOffset() const { + return getRestartPerfRatesOffset() + perfrates_.size(); + } + + size_t getRestartWellRatesOffset() const { + return getRestartTemperatureOffset() + temperature_.size(); + } + private: std::vector bhp_; std::vector temperature_;