Browse Source

fix output of prob choice labels in scheduler

tempestpy_adaptions
Stefan Pranger 4 years ago
parent
commit
814498ada2
  1. 2
      src/storm/storage/PostScheduler.cpp
  2. 2
      src/storm/storage/Scheduler.cpp

2
src/storm/storage/PostScheduler.cpp

@ -144,7 +144,7 @@ namespace storm {
stateString << choiceProbPair.first; stateString << choiceProbPair.first;
} }
if (choiceLabelsGiven) { if (choiceLabelsGiven) {
auto choiceLabels = model->getChoiceLabeling().getLabelsOfChoice(model->getTransitionMatrix().getRowGroupIndices()[state] + choice.getDeterministicChoice());
auto choiceLabels = model->getChoiceLabeling().getLabelsOfChoice(model->getTransitionMatrix().getRowGroupIndices()[state] + choiceProbPair.first);
stateString << " {" << boost::join(choiceLabels, ", ") << "}"; stateString << " {" << boost::join(choiceLabels, ", ") << "}";
} }
stateString << ")"; stateString << ")";

2
src/storm/storage/Scheduler.cpp

@ -204,7 +204,7 @@ namespace storm {
stateString << choiceProbPair.first; stateString << choiceProbPair.first;
} }
if (choiceLabelsGiven) { if (choiceLabelsGiven) {
auto choiceLabels = model->getChoiceLabeling().getLabelsOfChoice(model->getTransitionMatrix().getRowGroupIndices()[state] + choice.getDeterministicChoice());
auto choiceLabels = model->getChoiceLabeling().getLabelsOfChoice(model->getTransitionMatrix().getRowGroupIndices()[state] + choiceProbPair.first);
stateString << " {" << boost::join(choiceLabels, ", ") << "}"; stateString << " {" << boost::join(choiceLabels, ", ") << "}";
} }
stateString << ")"; stateString << ")";

Loading…
Cancel
Save