You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
4.3 KiB
73 lines
4.3 KiB
#include "storm/shields/OptimalShield.h"
|
|
|
|
#include <algorithm>
|
|
|
|
namespace tempest {
|
|
namespace shields {
|
|
|
|
template<typename ValueType, typename IndexType>
|
|
OptimalShield<ValueType, IndexType>::OptimalShield(std::vector<IndexType> const& rowGroupIndices, std::vector<ValueType> const& choiceValues, std::shared_ptr<storm::logic::ShieldExpression const> const& shieldingExpression, storm::OptimizationDirection optimizationDirection, storm::storage::BitVector relevantStates, boost::optional<storm::storage::BitVector> coalitionStates) : AbstractShield<ValueType, IndexType>(rowGroupIndices, shieldingExpression, optimizationDirection, relevantStates, coalitionStates), choiceValues(choiceValues) {
|
|
// Intentionally left empty.
|
|
}
|
|
|
|
template<typename ValueType, typename IndexType>
|
|
storm::storage::PostScheduler<ValueType> OptimalShield<ValueType, IndexType>::construct() {
|
|
if (this->getOptimizationDirection() == storm::OptimizationDirection::Minimize) {
|
|
if(this->shieldingExpression->isRelative()) {
|
|
return constructWithCompareType<storm::utility::ElementLessEqual<ValueType>, true>();
|
|
} else {
|
|
return constructWithCompareType<storm::utility::ElementLessEqual<ValueType>, false>();
|
|
}
|
|
} else {
|
|
if(this->shieldingExpression->isRelative()) {
|
|
return constructWithCompareType<storm::utility::ElementGreaterEqual<ValueType>, true>();
|
|
} else {
|
|
return constructWithCompareType<storm::utility::ElementGreaterEqual<ValueType>, false>();
|
|
}
|
|
}
|
|
}
|
|
|
|
template<typename ValueType, typename IndexType>
|
|
template<typename Compare, bool relative>
|
|
storm::storage::PostScheduler<ValueType> OptimalShield<ValueType, IndexType>::constructWithCompareType() {
|
|
tempest::shields::utility::ChoiceFilter<ValueType, Compare, relative> choiceFilter;
|
|
storm::storage::PostScheduler<ValueType> shield(this->rowGroupIndices.size() - 1, this->computeRowGroupSizes());
|
|
auto choice_it = this->choiceValues.begin();
|
|
if(this->coalitionStates.is_initialized()) {
|
|
this->relevantStates &= this->coalitionStates.get();
|
|
}
|
|
for(uint state = 0; state < this->rowGroupIndices.size() - 1; state++) {
|
|
uint rowGroupSize = this->rowGroupIndices[state + 1] - this->rowGroupIndices[state];
|
|
if(this->relevantStates.get(state)) {
|
|
auto maxProbabilityIndex = std::max_element(choice_it, choice_it + rowGroupSize) - choice_it;
|
|
ValueType maxProbability = *(choice_it + maxProbabilityIndex);
|
|
if(!relative && !choiceFilter(maxProbability, maxProbability, this->shieldingExpression->getValue())) {
|
|
STORM_LOG_WARN("No shielding action possible with absolute comparison for state with index " << state);
|
|
shield.setChoice(storm::storage::PostSchedulerChoice<ValueType>(), state, 0);
|
|
choice_it += rowGroupSize;
|
|
continue;
|
|
}
|
|
storm::storage::PostSchedulerChoice<ValueType> choiceMapping;
|
|
for(uint choice = 0; choice < rowGroupSize; choice++, choice_it++) {
|
|
if(choiceFilter(*choice_it, maxProbability, this->shieldingExpression->getValue())) {
|
|
choiceMapping.addChoice(choice, choice);
|
|
} else {
|
|
choiceMapping.addChoice(choice, maxProbabilityIndex);
|
|
}
|
|
}
|
|
shield.setChoice(choiceMapping, state, 0);
|
|
} else {
|
|
shield.setChoice(storm::storage::PostSchedulerChoice<ValueType>(), state, 0);
|
|
choice_it += rowGroupSize;
|
|
}
|
|
}
|
|
return shield;
|
|
}
|
|
|
|
// Explicitly instantiate appropriate classes
|
|
template class OptimalShield<double, typename storm::storage::SparseMatrix<double>::index_type>;
|
|
#ifdef STORM_HAVE_CARL
|
|
template class OptimalShield<storm::RationalNumber, typename storm::storage::SparseMatrix<storm::RationalNumber>::index_type>;
|
|
#endif
|
|
}
|
|
}
|