diff --git a/src/storm/modelchecker/results/ExplicitQuantitativeCheckResult.cpp b/src/storm/modelchecker/results/ExplicitQuantitativeCheckResult.cpp index af654b505..b3c83efb6 100644 --- a/src/storm/modelchecker/results/ExplicitQuantitativeCheckResult.cpp +++ b/src/storm/modelchecker/results/ExplicitQuantitativeCheckResult.cpp @@ -234,7 +234,13 @@ namespace storm { template std::shared_ptr::IndexType>> const& ExplicitQuantitativeCheckResult::getShield() const { - STORM_LOG_THROW(this->hasScheduler(), storm::exceptions::InvalidOperationException, "Unable to retrieve non-existing scheduler."); + STORM_LOG_THROW(this->hasShield(), storm::exceptions::InvalidOperationException, "Unable to retrieve non-existing shield."); + return shield.get(); + } + + template + std::shared_ptr::IndexType>>& ExplicitQuantitativeCheckResult::getShield() { + STORM_LOG_THROW(this->hasShield(), storm::exceptions::InvalidOperationException, "Unable to retrieve non-existing shield."); return shield.get(); } diff --git a/src/storm/modelchecker/results/ExplicitQuantitativeCheckResult.h b/src/storm/modelchecker/results/ExplicitQuantitativeCheckResult.h index 5375a7902..d434e10b1 100644 --- a/src/storm/modelchecker/results/ExplicitQuantitativeCheckResult.h +++ b/src/storm/modelchecker/results/ExplicitQuantitativeCheckResult.h @@ -85,6 +85,7 @@ namespace storm { void setShield(std::unique_ptr> shield); std::shared_ptr> const& getShield() const; + std::shared_ptr>& getShield(); storm::json toJson(boost::optional const& stateValuations = boost::none) const;