diff --git a/opm/core/simulator/BlackoilState.hpp b/opm/core/simulator/BlackoilState.hpp index 51c6c21cb..905b09434 100644 --- a/opm/core/simulator/BlackoilState.hpp +++ b/opm/core/simulator/BlackoilState.hpp @@ -30,6 +30,12 @@ namespace Opm { +enum HydroCarbonState { + GasOnly = 0, + GasAndOil = 1, + OilOnly = 2 +}; + /// Simulator state for a blackoil simulator. class BlackoilState : public SimulationDataContainer { diff --git a/opm/core/utility/initHydroCarbonState.hpp b/opm/core/utility/initHydroCarbonState.hpp new file mode 100644 index 000000000..c0be0f685 --- /dev/null +++ b/opm/core/utility/initHydroCarbonState.hpp @@ -0,0 +1,42 @@ +#ifndef INITHYDROCARBONSTATE_HPP +#define INITHYDROCARBONSTATE_HPP + +#include "opm/core/simulator/BlackoilState.hpp" + +namespace Opm +{ + +void initHydroCarbonState(BlackoilState& state, const PhaseUsage& pu, const int num_cells) { + enum { Oil = BlackoilPhases::Liquid, Gas = BlackoilPhases::Vapour, Water = BlackoilPhases::Aqua }; + // hydrocarbonstate is only used when gas and oil is present + assert(pu.phase_used[Oil]); + if (!pu.phase_used[Gas]) { + return; // do nothing + } + std::vector& hydroCarbonState = state.hydroCarbonState(); + const int np = pu.num_phases; + hydroCarbonState.resize(num_cells); + std::fill(hydroCarbonState.begin(), hydroCarbonState.end(), HydroCarbonState::GasAndOil); + + // set hydrocarbon state + const double epsilon = std::sqrt(std::numeric_limits::epsilon()); + const std::vector& saturation = state.saturation(); + for (int c = 0; c < num_cells; ++c) { + if (pu.phase_used[Water]) { + if ( saturation[c*np + pu.phase_pos[ Water ]] > (1.0 - epsilon)) { + continue; // cases (almost) filled with water is treated as GasAndOil case; + } + } + if ( saturation[c*np + pu.phase_pos[ Gas ]] == 0.0) { + hydroCarbonState[c] = HydroCarbonState::OilOnly; + continue; + } + if ( saturation[c*np + pu.phase_pos[ Oil ]] == 0.0) { + hydroCarbonState[c] = HydroCarbonState::GasOnly; + } + } +} + + +} // namespace Opm +#endif // INITHYDROCARBONSTATE_HPP