2 changed files with 192 additions and 0 deletions
			
			
		- 
					165src/storm-pomdp/analysis/QualitativeAnalysis.cpp
- 
					27src/storm-pomdp/analysis/QualitativeAnalysis.h
| @ -0,0 +1,165 @@ | |||
| #include "storm-pomdp/analysis/QualitativeAnalysis.h"
 | |||
| 
 | |||
| #include "storm/utility/macros.h"
 | |||
| #include "storm/utility/graph.h"
 | |||
| #include "storm/models/sparse/Pomdp.h"
 | |||
| #include "storm/modelchecker/propositional/SparsePropositionalModelChecker.h"
 | |||
| #include "storm/modelchecker/results/ExplicitQualitativeCheckResult.h"
 | |||
| 
 | |||
| #include "storm/exceptions/InvalidPropertyException.h"
 | |||
| #include "storm/exceptions/NotImplementedException.h"
 | |||
| 
 | |||
| namespace storm { | |||
|     namespace analysis { | |||
| 
 | |||
|         template<typename ValueType> | |||
|         QualitativeAnalysis<ValueType>::QualitativeAnalysis(storm::models::sparse::Pomdp<ValueType> const& pomdp) : pomdp(pomdp) { | |||
|             // Intentionally left empty
 | |||
|         } | |||
|      | |||
|         template<typename ValueType> | |||
|         storm::storage::BitVector QualitativeAnalysis<ValueType>::analyseProb0(storm::logic::ProbabilityOperatorFormula const& formula) const { | |||
|             return analyseProb0or1(formula, true); | |||
|         } | |||
|          | |||
|         template<typename ValueType> | |||
|         storm::storage::BitVector QualitativeAnalysis<ValueType>::analyseProb1(storm::logic::ProbabilityOperatorFormula const& formula) const { | |||
|             return analyseProb0or1(formula, false); | |||
|         } | |||
|          | |||
|         template<typename ValueType> | |||
|         storm::storage::BitVector QualitativeAnalysis<ValueType>::analyseProb0or1(storm::logic::ProbabilityOperatorFormula const& formula, bool prob0) const { | |||
|             // check whether the property is minimizing or maximizing
 | |||
|             STORM_LOG_THROW(formula.hasOptimalityType() || formula.hasBound(), storm::exceptions::InvalidPropertyException, "The formula " << formula << " does not specify whether to minimize or maximize."); | |||
|             bool minimizes = (formula.hasOptimalityType() && storm::solver::minimize(formula.getOptimalityType())) || (formula.hasBound() && storm::logic::isLowerBound(formula.getBound().comparisonType)); | |||
|              | |||
|             std::shared_ptr<storm::logic::Formula const> subformula = formula.getSubformula().asSharedPointer(); | |||
|             // If necessary, convert the subformula to a more general case
 | |||
|             if (subformula->isEventuallyFormula()) { | |||
|                 subformula = std::make_shared<storm::logic::UntilFormula>(storm::logic::Formula::getTrueFormula(), subformula->asEventuallyFormula().getSubformula().asSharedPointer()); | |||
|             } | |||
|              | |||
|             if (subformula->isUntilFormula()) { | |||
|                 if (minimizes && prob0) { | |||
|                     return analyseProb0Min(subformula->asUntilFormula()); | |||
|                 } else if (minimizes && !prob0){ | |||
|                     return analyseProb1Min(subformula->asUntilFormula()); | |||
|                 } else if (!minimizes && prob0){ | |||
|                     return analyseProb0Max(subformula->asUntilFormula()); | |||
|                 } else if (!minimizes && !prob0){ | |||
|                     return analyseProb1Max(subformula->asUntilFormula()); | |||
|                 } | |||
|             } | |||
|             STORM_LOG_THROW(false, storm::exceptions::InvalidPropertyException, "Prob0or1 analysis is not supported for the property " << formula); | |||
|         } | |||
|      | |||
|      | |||
|         template<typename ValueType> | |||
|         storm::storage::BitVector QualitativeAnalysis<ValueType>::analyseProb0Max(storm::logic::UntilFormula const& formula) const { | |||
|             return storm::utility::graph::performProb0A(pomdp.getBackwardTransitions(), checkPropositionalFormula(formula.getLeftSubformula()), checkPropositionalFormula(formula.getRightSubformula())); | |||
|         } | |||
|          | |||
|         template<typename ValueType> | |||
|         storm::storage::BitVector QualitativeAnalysis<ValueType>::analyseProb0Min(storm::logic::UntilFormula const& formula) const { | |||
|             STORM_LOG_THROW(false, storm::exceptions::NotImplementedException, "Prob0 analysis is currently not implemented for minimizing properties."); | |||
|         } | |||
|          | |||
|         template<typename ValueType> | |||
|         storm::storage::BitVector QualitativeAnalysis<ValueType>::analyseProb1Max(storm::logic::UntilFormula const& formula) const { | |||
|             // We consider the states that satisfy the formula with prob.1 under arbitrary schedulers as goal states.
 | |||
|             storm::storage::BitVector goalStates = storm::utility::graph::performProb1A(pomdp.getTransitionMatrix(), pomdp.getTransitionMatrix().getRowGroupIndices(), pomdp.getBackwardTransitions(), checkPropositionalFormula(formula.getLeftSubformula()), checkPropositionalFormula(formula.getRightSubformula())); | |||
|              | |||
|             // Now find a set of observations such that there is a memoryless scheduler inducing prob. 1 for each state whose observation is in the set.
 | |||
|             storm::storage::BitVector candidateStates = goalStates | checkPropositionalFormula(formula.getLeftSubformula()); | |||
|             storm::storage::BitVector candidateActions = pomdp.getTransitionMatrix().getRowFilter(candidateStates); | |||
|             storm::storage::BitVector candidateObservations(pomdp.getNrObservations(), true); | |||
|              | |||
|              | |||
|             bool converged = false; | |||
|             while (!converged) { | |||
|                 converged = true; | |||
|                  | |||
|                 // Get the candidate states that can reach the goal with prob1 via candidate actions
 | |||
|                 storm::storage::BitVector newCandidates; | |||
|                 if (candidateActions.full()) { | |||
|                     newCandidates = storm::utility::graph::performProb1E(pomdp.getTransitionMatrix(), pomdp.getTransitionMatrix().getRowGroupIndices(), pomdp.getBackwardTransitions(), candidateStates, goalStates); | |||
|                 } else { | |||
|                     storm::storage::SparseMatrix<ValueType> filteredTransitions(pomdp.getTransitionMatrix().filterEntries(candidateActions)); | |||
|                     newCandidates = storm::utility::graph::performProb1E(filteredTransitions, filteredTransitions.getRowGroupIndices(), filteredTransitions.transpose(true), candidateStates, goalStates); | |||
|                 } | |||
|                 if (candidateStates != newCandidates) { | |||
|                     converged = false; | |||
|                     candidateStates = std::move(newCandidates); | |||
|                 } | |||
|                  | |||
|                 // Unselect all observations that have a non-candidate state
 | |||
|                 for (uint64_t state = candidateStates.getNextUnsetIndex(0); state < candidateStates.size(); state = candidateStates.getNextUnsetIndex(state + 1)) { | |||
|                     candidateObservations.set(pomdp.getObservation(state), false); | |||
|                 } | |||
|                  | |||
|                 // update the candidate actions to the set of actions that stay inside the candidate state set
 | |||
|                 std::vector<storm::storage::BitVector> candidateActionsPerObservation(pomdp.getNrObservations()); | |||
|                 for (auto const& state : candidateStates) { | |||
|                     auto& candidateActionsAtState = candidateActionsPerObservation[pomdp.getObservation(state)]; | |||
|                     if (candidateActionsAtState.size() == 0) { | |||
|                         candidateActionsAtState.resize(pomdp.getNumberOfChoices(state), true); | |||
|                     } | |||
|                     STORM_LOG_ASSERT(candidateActionsAtState.size() == pomdp.getNumberOfChoices(state), "State " + std::to_string(state) + " has " + std::to_string(pomdp.getNumberOfChoices(state)) + " actions, different from other with same observation (" + std::to_string(candidateActionsAtState.size()) + ")." ); | |||
|                     for (auto const& action : candidateActionsAtState) { | |||
|                         for (auto const& entry : pomdp.getTransitionMatrix().getRow(state, action)) { | |||
|                             if (!candidateStates.get(entry.getColumn())) { | |||
|                                 candidateActionsAtState.set(action, false); | |||
|                                 break; | |||
|                             } | |||
|                         } | |||
|                     } | |||
|                 } | |||
|                  | |||
|                 // Unselect all observations without such an action
 | |||
|                 for (auto const& o : candidateObservations) { | |||
|                     if (candidateActionsPerObservation[o].empty()) { | |||
|                         candidateObservations.set(o, false); | |||
|                     } | |||
|                 } | |||
|                  | |||
|                 // only keep the candidate states with a candidateObservation
 | |||
|                 for (auto const& state : candidateStates) { | |||
|                     if (!candidateObservations.get(pomdp.getObservation(state)) && !goalStates.get(state)) { | |||
|                         candidateStates.set(state, false); | |||
|                         converged = false; | |||
|                     } | |||
|                 } | |||
|                  | |||
|                 // Only keep the candidate actions originating from a candidateState. Also transform the representation of candidate actions
 | |||
|                 candidateActions.clear(); | |||
|                 for (auto const& state : candidateStates) { | |||
|                     uint64_t offset = pomdp.getTransitionMatrix().getRowGroupIndices()[state]; | |||
|                     for (auto const& action : candidateActionsPerObservation[pomdp.getObservation(state)]) { | |||
|                         candidateActions.set(offset + action); | |||
|                     } | |||
|                 } | |||
|             } | |||
|              | |||
|             assert(goalStates.isSubsetOf(candidateStates)); | |||
|              | |||
|             return candidateStates; | |||
|              | |||
|         } | |||
|          | |||
|         template<typename ValueType> | |||
|         storm::storage::BitVector QualitativeAnalysis<ValueType>::analyseProb1Min(storm::logic::UntilFormula const& formula) const { | |||
|             return storm::utility::graph::performProb1A(pomdp.getTransitionMatrix(), pomdp.getTransitionMatrix().getRowGroupIndices(), pomdp.getBackwardTransitions(), checkPropositionalFormula(formula.getLeftSubformula()), checkPropositionalFormula(formula.getRightSubformula())); | |||
|         } | |||
|      | |||
|         template<typename ValueType> | |||
|         storm::storage::BitVector QualitativeAnalysis<ValueType>::checkPropositionalFormula(storm::logic::Formula const& propositionalFormula) const { | |||
|             storm::modelchecker::SparsePropositionalModelChecker<storm::models::sparse::Mdp<ValueType>> mc(pomdp); | |||
|             STORM_LOG_THROW(mc.canHandle(propositionalFormula), storm::exceptions::InvalidPropertyException, "Propositional model checker can not handle formula " << propositionalFormula); | |||
|             return mc.check(propositionalFormula)->asExplicitQualitativeCheckResult().getTruthValuesVector(); | |||
|         } | |||
| 
 | |||
| 
 | |||
|         template class QualitativeAnalysis<storm::RationalNumber>; | |||
|          | |||
|     } | |||
| } | |||
| @ -0,0 +1,27 @@ | |||
| #include "storm/models/sparse/Pomdp.h" | |||
| #include "storm/storage/BitVector.h" | |||
| #include "storm/logic/Formulas.h" | |||
| namespace storm { | |||
|     namespace analysis { | |||
|         template<typename ValueType> | |||
|         class QualitativeAnalysis { | |||
|         public: | |||
| 
 | |||
|             QualitativeAnalysis(storm::models::sparse::Pomdp<ValueType> const& pomdp); | |||
|             storm::storage::BitVector analyseProb0(storm::logic::ProbabilityOperatorFormula const& formula) const; | |||
|             storm::storage::BitVector analyseProb1(storm::logic::ProbabilityOperatorFormula const& formula) const; | |||
|              | |||
|         private: | |||
|             storm::storage::BitVector analyseProb0or1(storm::logic::ProbabilityOperatorFormula const& formula, bool prob0) const; | |||
|             storm::storage::BitVector analyseProb0Max(storm::logic::UntilFormula const& formula) const; | |||
|             storm::storage::BitVector analyseProb0Min(storm::logic::UntilFormula const& formula) const; | |||
|             storm::storage::BitVector analyseProb1Max(storm::logic::UntilFormula const& formula) const; | |||
|             storm::storage::BitVector analyseProb1Min(storm::logic::UntilFormula const& formula) const; | |||
|              | |||
|             storm::storage::BitVector checkPropositionalFormula(storm::logic::Formula const& propositionalFormula) const; | |||
|              | |||
|             storm::models::sparse::Pomdp<ValueType> const& pomdp; | |||
|         }; | |||
|     } | |||
| } | |||
| 
 | |||
						Write
						Preview
					
					
					Loading…
					
					Cancel
						Save
					
		Reference in new issue