Browse Source

Started refactoring ExplicitModelAdapter to finally make it nice.

Former-commit-id: 6df7e5d9fa
tempestpy_adaptions
dehnert 11 years ago
parent
commit
e2b0c4f1aa
  1. 1014
      src/adapters/ExplicitModelAdapter.cpp
  2. 128
      src/adapters/ExplicitModelAdapter.h
  3. 8
      src/ir/Program.cpp
  4. 14
      src/ir/Program.h
  5. 9
      src/models/AbstractModel.h
  6. 5
      src/models/AtomicPropositionsLabeling.h
  7. 2
      src/storage/SparseMatrix.h
  8. 7
      src/storm.cpp

1014
src/adapters/ExplicitModelAdapter.cpp
File diff suppressed because it is too large
View File

128
src/adapters/ExplicitModelAdapter.h

@ -59,22 +59,31 @@ namespace storm {
class ExplicitModelAdapter {
public:
/*!
* Initializes the adapter with the given program.
*
* @param program The program from which to build the explicit model.
*/
ExplicitModelAdapter(storm::ir::Program program);
/*!
* Destroys the adapter.
*/
~ExplicitModelAdapter();
// A structure holding information about the reachable state space.
struct StateInformation {
std::vector<StateType*> reachableStates;
std::unordered_map<StateType, uint_fast64_t, StateHash, StateCompare> stateToIndexMap;
};
// A structure holding the individual components of a model.
struct ModelComponents {
ModelComponents() : transitionMatrix(), stateLabeling(), nondeterministicChoiceIndices(), stateRewards(), transitionRewardMatrix(), choiceLabeling() {
// Intentionally left empty.
}
storm::storage::SparseMatrix<double> transitionMatrix;
storm::models::AtomicPropositionsLabeling stateLabeling;
std::vector<uint_fast64_t> nondeterministicChoiceIndices;
std::vector<double> stateRewards;
storm::storage::SparseMatrix<double> transitionRewardMatrix;
std::vector<std::list<uint_fast64_t>> choiceLabeling;
};
/*!
* Convert the program given at construction time to an abstract model. The type of the model is the one
* specified in the program. The given reward model name selects the rewards that the model will contain.
*
* @param program The program to translate.
* @param constantDefinitionString A string that contains a comma-separated definition of all undefined
* constants in the model.
* @param rewardModelName The name of reward model to be added to the model. This must be either a reward
@ -82,27 +91,9 @@ namespace storm {
* rewards.
* @return The explicit model that was given by the probabilistic program.
*/
std::shared_ptr<storm::models::AbstractModel<double>> getModel(std::string const& constantDefinitionString = "", std::string const& rewardModelName = "");
static std::shared_ptr<storm::models::AbstractModel<double>> translateProgram(storm::ir::Program program, std::string const& constantDefinitionString = "", std::string const& rewardModelName = "");
private:
/*!
* Private copy constructor to disable copy-construction (and move-construction) of this class.
*
* @param other The object to copy-construct from.
*/
ExplicitModelAdapter(ExplicitModelAdapter const& other);
/*!
* Private copy-assignment to disable copy-assignment (and move-assignment) of this class.
*
* @param other The object to copy-assign from.
*/
ExplicitModelAdapter operator=(ExplicitModelAdapter const& other);
/*!
* The precision that is to be used for sanity checks internally.
*/
double precision;
/*!
* Sets some boolean variable in the given state object.
*
@ -133,29 +124,50 @@ namespace storm {
* Applies an update to the given state and returns the resulting new state object. This methods does not
* modify the given state but returns a new one.
*
* @param program The program in which the variables of the update are declared.
* @params state The state to which to apply the update.
* @params update The update to apply.
* @return The resulting state.
*/
StateType* applyUpdate(StateType const* state, storm::ir::Update const& update) const;
static StateType* applyUpdate(storm::ir::Program const& program, StateType const* state, storm::ir::Update const& update);
/*!
* Applies an update to the given state and returns the resulting new state object. The update is evaluated
* over the variable values of the given base state. This methods does not modify the given state but
* returns a new one.
*
* @params state The state to which to apply the update.
* @params baseState The state used for evaluating the update.
* @params update The update to apply.
* @param program The program in which the variables of the update are declared.
* @param state The state to which to apply the update.
* @param baseState The state used for evaluating the update.
* @param update The update to apply.
* @return The resulting state.
*/
StateType* applyUpdate(StateType const* state, StateType const* baseState, storm::ir::Update const& update) const;
static StateType* applyUpdate(storm::ir::Program const& program, StateType const* state, StateType const* baseState, storm::ir::Update const& update);
/*!
* Prepares internal data structures associated with the variables in the program that are used during the
* translation.
* Defines the undefined constants of the given program using the given string.
*
* @param program The program in which to define the constants.
* @param constantDefinitionString A comma-separated list of constant definitions.
*/
void initializeVariables();
static void defineUndefinedConstants(storm::ir::Program& program, std::string const& constantDefinitionString);
/*!
* Undefines all previously defined constants in the given program.
*
* @param program The program in which to undefine the constants.
*/
static void undefineUndefinedConstants(storm::ir::Program& program);
/*!
* Explores the state space of the given program and returns the components of the model as a result.
*
* @param program The program whose state space to explore.
* @param rewardModelName The name of the reward model that is to be considered. If empty, no reward model
* is considered.
* @return A structure containing the components of the resulting model.
*/
static ModelComponents buildModelComponents(storm::ir::Program const& program, std::string const& rewardModelName);
/*!
* Retrieves the state rewards for every reachable state based on the given state rewards.
@ -163,7 +175,7 @@ namespace storm {
* @param rewards The rewards to use.
* @return The reward values for every (reachable) state.
*/
std::vector<double> getStateRewards(std::vector<storm::ir::StateReward> const& rewards);
// static std::vector<double> getStateRewards(std::vector<storm::ir::StateReward> const& rewards);
/*!
* Computes the labels for every reachable state based on a list of available labels.
@ -171,7 +183,7 @@ namespace storm {
* @param labels A mapping from label names to boolean expressions to use for the labeling.
* @return The resulting labeling.
*/
storm::models::AtomicPropositionsLabeling getStateLabeling(std::map<std::string, std::shared_ptr<storm::ir::expressions::BaseExpression>> labels);
// static storm::models::AtomicPropositionsLabeling getStateLabeling(std::map<std::string, std::shared_ptr<storm::ir::expressions::BaseExpression>> labels);
/*!
* Retrieves all commands that are labeled with the given label and enabled in the given state, grouped by
@ -188,14 +200,14 @@ namespace storm {
* @param action The action label to select.
* @return A list of lists of active commands or nothing.
*/
boost::optional<std::vector<std::list<storm::ir::Command>>> getActiveCommandsByAction(StateType const* state, std::string const& action);
// static boost::optional<std::vector<std::list<storm::ir::Command>>> getActiveCommandsByAction(StateType const* state, std::string const& action);
/*!
* Generates the initial state.
*
* @return The initial state.
*/
StateType* getInitialState();
// static StateType* getInitialState();
/*!
* Retrieves the state id of the given state. If the state has not been encountered yet, it will be added to
@ -206,7 +218,7 @@ namespace storm {
* @param state A pointer to a state for which to retrieve the index. This must not be used after the call.
* @return The state id of the given state.
*/
uint_fast64_t getOrAddStateIndex(StateType* state);
// static uint_fast64_t getOrAddStateIndex(StateType* state);
/*!
* Expands all unlabeled (i.e. independent) transitions of the given state and adds them to the transition list.
@ -214,7 +226,7 @@ namespace storm {
* @param stateIndex The state to expand.
* @params transitionList The current list of transitions for this state.
*/
void addUnlabeledTransitions(uint_fast64_t stateIndex, std::list<std::pair<std::pair<std::string, std::list<uint_fast64_t>>, std::map<uint_fast64_t, double>>>& transitionList);
// static void addUnlabeledTransitions(uint_fast64_t stateIndex, std::list<std::pair<std::pair<std::string, std::list<uint_fast64_t>>, std::map<uint_fast64_t, double>>>& transitionList);
/*!
* Expands all labeled (i.e. synchronizing) transitions of the given state and adds them to the transition list.
@ -222,46 +234,26 @@ namespace storm {
* @param stateIndex The index of the state to expand.
* @param transitionList The current list of transitions for this state.
*/
void addLabeledTransitions(uint_fast64_t stateIndex, std::list<std::pair<std::pair<std::string, std::list<uint_fast64_t>>, std::map<uint_fast64_t, double>>>& transitionList);
// static void addLabeledTransitions(uint_fast64_t stateIndex, std::list<std::pair<std::pair<std::string, std::list<uint_fast64_t>>, std::map<uint_fast64_t, double>>>& transitionList);
/*!
* Builds the transition matrix of a deterministic model from the current list of transitions.
*
* @return The transition matrix.
*/
storm::storage::SparseMatrix<double> buildDeterministicMatrix();
// static storm::storage::SparseMatrix<double> buildDeterministicMatrix();
/*!
* Builds the transition matrix of a nondeterministic model from the current list of transitions.
*
* @return result The transition matrix.
*/
storm::storage::SparseMatrix<double> buildNondeterministicMatrix();
// static storm::storage::SparseMatrix<double> buildNondeterministicMatrix();
/*!
* Generate the (internal) list of all transitions of the model.
*/
void buildTransitionMap();
/*!
* Clear all members that are initialized during the computation.
*/
void clearInternalState();
/*!
* Defines the undefined constants of the program using the given string.
*
* @param constantDefinitionString A comma-separated list of constant definitions.
*/
void defineUndefinedConstants(std::string const& constantDefinitionString);
/*!
* Sets all values of program constants to undefined again.
*/
void undefineUndefinedConstants();
// Program that is to be converted.
storm::ir::Program program;
// void buildTransitionMap();
// List of all boolean variables.
std::vector<storm::ir::BooleanVariable> booleanVariables;

8
src/ir/Program.cpp

@ -215,5 +215,13 @@ namespace storm {
return this->doubleUndefinedConstantExpressions;
}
uint_fast64_t Program::getGlobalIndexOfBooleanVariable(std::string const& variableName) const {
return this->globalBooleanVariableToIndexMap.at(variableName);
}
uint_fast64_t Program::getGlobalIndexOfIntegerVariable(std::string const& variableName) const {
return this->globalIntegerVariableToIndexMap.at(variableName);
}
} // namespace ir
} // namepsace storm

14
src/ir/Program.h

@ -233,6 +233,20 @@ namespace storm {
*/
std::map<std::string, std::shared_ptr<storm::ir::expressions::DoubleConstantExpression>> const& getDoubleUndefinedConstantExpressionsMap() const;
/*!
* Retrieves the global index of the given boolean variable.
*
* @param variableName The name of the boolean variable whose index to retrieve.
*/
uint_fast64_t getGlobalIndexOfBooleanVariable(std::string const& variableName) const;
/*!
* Retrieves the global index of the integer boolean variable.
*
* @param variableName The name of the integer variable whose index to retrieve.
*/
uint_fast64_t getGlobalIndexOfIntegerVariable(std::string const& variableName) const;
private:
// The type of the model.
ModelType modelType;

9
src/models/AbstractModel.h

@ -300,6 +300,15 @@ class AbstractModel: public std::enable_shared_from_this<AbstractModel<T>> {
return this->getTransitionMatrix().getNonZeroEntryCount();
}
/*!
* Retrieves the initial states of the model.
*
* @return The initial states of the model represented by a bit vector.
*/
storm::storage::BitVector const& getInitialStates() const {
return this->getLabeledStates("init");
}
/*!
* Returns a bit vector in which exactly those bits are set to true that
* correspond to a state labeled with the given atomic proposition.

5
src/models/AtomicPropositionsLabeling.h

@ -32,15 +32,14 @@ namespace models {
*/
class AtomicPropositionsLabeling {
public:
public:
/*!
* Constructs an empty atomic propositions labeling for the given number of states and amount of atomic propositions.
*
* @param stateCount The number of states of the model.
* @param apCountMax The number of atomic propositions.
*/
AtomicPropositionsLabeling(const uint_fast64_t stateCount, uint_fast64_t const apCountMax)
AtomicPropositionsLabeling(const uint_fast64_t stateCount = 0, uint_fast64_t const apCountMax = 0)
: stateCount(stateCount), apCountMax(apCountMax), apsCurrent(0), singleLabelings() {
singleLabelings.reserve(apCountMax);
}

2
src/storage/SparseMatrix.h

@ -351,7 +351,7 @@ public:
*
* @param other The Matrix from which to copy the content
*/
storm::storage::SparseMatrix<T>& operator=(const SparseMatrix<T> & other) {
storm::storage::SparseMatrix<T>& operator=(SparseMatrix<T> const& other) {
this->rowCount = other.rowCount;
this->colCount = other.colCount;
this->nonZeroEntryCount = other.nonZeroEntryCount;

7
src/storm.cpp

@ -330,10 +330,9 @@ int main(const int argc, const char* argv[]) {
delete modelchecker;
}
} else if (s->isSet("symbolic")) {
std::string const arg = s->getOptionByLongName("symbolic").getArgument(0).getValueAsString();
storm::adapters::ExplicitModelAdapter adapter(storm::parser::PrismParserFromFile(arg));
std::string const constants = s->getOptionByLongName("constants").getArgument(0).getValueAsString();
std::shared_ptr<storm::models::AbstractModel<double>> model = adapter.getModel(constants);
std::string const& programFile = s->getOptionByLongName("symbolic").getArgument(0).getValueAsString();
std::string const& constants = s->getOptionByLongName("constants").getArgument(0).getValueAsString();
std::shared_ptr<storm::models::AbstractModel<double>> model = storm::adapters::ExplicitModelAdapter::translateProgram(storm::parser::PrismParserFromFile(programFile), constants);
model->printModelInformationToStream(std::cout);
// Enable the following lines to test the MinimalLabelSetGenerator.
Loading…
Cancel
Save