diff --git a/opm/material/fluidmatrixinteractions/EclMaterialLawManager.hpp b/opm/material/fluidmatrixinteractions/EclMaterialLawManager.hpp index 05ccd9443..1ec7b9d61 100644 --- a/opm/material/fluidmatrixinteractions/EclMaterialLawManager.hpp +++ b/opm/material/fluidmatrixinteractions/EclMaterialLawManager.hpp @@ -206,28 +206,22 @@ public: MaterialLawParams& materialLawParams(int elemIdx) { - assert(0 <= elemIdx && elemIdx < (int) materialLawParams_.size()); - - int paramIdx; - if (hasElementSpecificParameters()) - paramIdx = elemIdx; + if (hasElementSpecificParameters()) { + assert(0 <= elemIdx && elemIdx < (int) materialLawParams_.size()); + return *materialLawParams_[elemIdx]; + } else - paramIdx = satnumRegionIdx_[elemIdx]; - - return *materialLawParams_[paramIdx]; + return *materialLawParams_[satnumRegionIdx_[elemIdx]]; } const MaterialLawParams& materialLawParams(int elemIdx) const { - assert(0 <= elemIdx && elemIdx < materialLawParams_.size()); - - int paramIdx; - if (hasElementSpecificParameters()) - paramIdx = elemIdx; + if (hasElementSpecificParameters()) { + assert(0 <= elemIdx && elemIdx < (int) materialLawParams_.size()); + return *materialLawParams_[elemIdx]; + } else - paramIdx = satnumRegionIdx_[elemIdx]; - - return *materialLawParams_[paramIdx]; + return *materialLawParams_[satnumRegionIdx_[elemIdx]]; } template