diff --git a/src/storm/modelchecker/rpatl/helper/SparseSmgRpatlHelper.cpp b/src/storm/modelchecker/rpatl/helper/SparseSmgRpatlHelper.cpp index 3edff110e..ed535f8d4 100644 --- a/src/storm/modelchecker/rpatl/helper/SparseSmgRpatlHelper.cpp +++ b/src/storm/modelchecker/rpatl/helper/SparseSmgRpatlHelper.cpp @@ -32,8 +32,8 @@ namespace storm { // Reduce the matrix to relevant states storm::storage::SparseMatrix submatrix = transitionMatrix.getSubmatrix(true, relevantStates, relevantStates, false); - storm::storage::BitVector clippedStatesOfCoalition(statesOfCoalition.size() - psiStates.getNumberOfSetBits()); - clippedStatesOfCoalition.setClippedStatesOfCoalition(psiStates, statesOfCoalition); + storm::storage::BitVector clippedStatesOfCoalition(relevantStates.getNumberOfSetBits()); + clippedStatesOfCoalition.setClippedStatesOfCoalition(relevantStates, statesOfCoalition); clippedStatesOfCoalition.complement(); storm::modelchecker::helper::internal::GameViHelper viHelper(submatrix, clippedStatesOfCoalition); diff --git a/src/storm/storage/BitVector.cpp b/src/storm/storage/BitVector.cpp index 506613965..194e37b81 100644 --- a/src/storm/storage/BitVector.cpp +++ b/src/storm/storage/BitVector.cpp @@ -1026,12 +1026,11 @@ namespace storm { } } - void BitVector::setClippedStatesOfCoalition(BitVector psiStates, BitVector statesOfCoalition) + void BitVector::setClippedStatesOfCoalition(BitVector relevantStates, BitVector statesOfCoalition) { auto clippedStatesCounter = 0; - for(uint i = 0; i < psiStates.size(); i++) { - std::cout << i << " : " << psiStates.get(i) << " -> " << statesOfCoalition[i] << std::endl; - if(!psiStates.get(i)) { + for(uint i = 0; i < relevantStates.size(); i++) { + if(relevantStates.get(i)) { this->set(clippedStatesCounter, statesOfCoalition[i]); clippedStatesCounter++; } diff --git a/src/storm/storage/BitVector.h b/src/storm/storage/BitVector.h index 83f656a4b..3dfdd8a55 100644 --- a/src/storm/storage/BitVector.h +++ b/src/storm/storage/BitVector.h @@ -542,7 +542,7 @@ namespace storm { * @param psiStates Bitvector of states which are in the winning region. * @param statesOfCoalition Bitvector of states which belong to the coalition. */ - void setClippedStatesOfCoalition(BitVector psiStates, BitVector statesOfCoalition); + void setClippedStatesOfCoalition(BitVector relevantStates, BitVector statesOfCoalition); friend std::ostream& operator<<(std::ostream& out, BitVector const& bitVector);