Refactoring

src/storm-dft/storage/dft/DFT.cpp

@@ -17,14 +17,13 @@ namespace storm {
     namespace storage {
 
         template<typename ValueType>
-        DFT<ValueType>::DFT(DFTElementVector const& elements, DFTElementPointer const& tle) : mElements(elements), mNrOfBEs(0), mNrOfSpares(0), mTopLevelIndex(tle->id()), mMaxSpareChildCount(0) {
+        DFT<ValueType>::DFT(DFTElementVector const& elements, DFTElementPointer const& tle) : mElements(elements), mNrOfBEs(0), mNrOfSpares(0), mNrRepresentatives(0), mTopLevelIndex(tle->id()), mMaxSpareChildCount(0) {
             // Check that ids correspond to indices in the element vector
             STORM_LOG_ASSERT(elementIndicesCorrect(), "Ids incorrect.");
-            size_t nrRepresentatives = 0;
 
             for (auto& elem : mElements) {
                 if (isRepresentative(elem->id())) {
-                    ++nrRepresentatives;
+                    ++mNrRepresentatives;
                 }
                 if(elem->isBasicElement()) {
                     ++mNrOfBEs;
@@ -81,13 +80,12 @@ namespace storm {
 
             //Reserve space for failed spares
             ++mMaxSpareChildCount;
-            size_t usageInfoBits = storm::utility::math::uint64_log2(mMaxSpareChildCount) + 1;
-            mStateVectorSize = nrElements() * 2 + mNrOfSpares * usageInfoBits + nrRepresentatives;
+            mStateVectorSize = DFTStateGenerationInfo::getStateVectorSize(nrElements(), mNrOfSpares, mNrRepresentatives, mMaxSpareChildCount);
         }
 
         template<typename ValueType>
         DFTStateGenerationInfo DFT<ValueType>::buildStateGenerationInfo(storm::storage::DFTIndependentSymmetries const& symmetries) const {
-            DFTStateGenerationInfo generationInfo(nrElements(), mMaxSpareChildCount);
+            DFTStateGenerationInfo generationInfo(nrElements(), mNrOfSpares, mNrRepresentatives, mMaxSpareChildCount);
             
             // Generate Pre and Post info for restrictions, and mutexes
             for(auto const& elem : mElements) {

src/storm-dft/storage/dft/DFT.h

@@ -58,6 +58,7 @@ namespace storm {
             DFTElementVector mElements;
             size_t mNrOfBEs;
             size_t mNrOfSpares;
+            size_t mNrRepresentatives;
             size_t mTopLevelIndex;
             size_t mStateVectorSize;
             size_t mMaxSpareChildCount;

src/storm-dft/storage/dft/DFTStateGenerationInfo.h

@@ -5,29 +5,52 @@ namespace storm {
         class DFTStateGenerationInfo {
         private:
             const size_t mUsageInfoBits;
+            const size_t stateIndexSize;
             std::map<size_t, size_t> mSpareUsageIndex; // id spare -> index first bit in state
             std::map<size_t, size_t> mSpareActivationIndex; // id spare representative -> index in state
             std::vector<size_t> mIdToStateIndex; // id -> index first bit in state
             std::map<size_t, std::vector<size_t>> mSeqRestrictionPreElements; // id -> list of restriction pre elements
             std::map<size_t, std::vector<size_t>> mSeqRestrictionPostElements; // id -> list of restriction post elements
-            std::map<size_t, std::vector<size_t>> mMutexRestrictionElements; // id -> list of elments in the same mutexes
+            std::map<size_t, std::vector<size_t>> mMutexRestrictionElements; // id -> list of elements in the same mutexes
             std::vector<std::pair<size_t, std::vector<size_t>>> mSymmetries; // pair (length of symmetry group, vector indicating the starting points of the symmetry groups)
 
         public:
 
-            DFTStateGenerationInfo(size_t nrElements, size_t maxSpareChildCount) :
-                mUsageInfoBits(storm::utility::math::uint64_log2(maxSpareChildCount) + 1), 
-                mIdToStateIndex(nrElements)
-            {
+            DFTStateGenerationInfo(size_t nrElements, size_t nrOfSpares, size_t nrRepresentatives, size_t maxSpareChildCount) :
+                    mUsageInfoBits(getUsageInfoBits(maxSpareChildCount)),
+                    stateIndexSize(getStateVectorSize(nrElements, nrOfSpares, nrRepresentatives, maxSpareChildCount)),
+                    mIdToStateIndex(nrElements) {
                 STORM_LOG_ASSERT(maxSpareChildCount < pow(2, mUsageInfoBits), "Bit length incorrect.");
             }
 
+            /*!
+             * Get number of bits required to store claiming information for spares in binary format.
+             * @param maxSpareChildCount Maximal number of children of a spare.
+             * @return Number of bits required to store claiming information.
+             */
+            static size_t getUsageInfoBits(size_t maxSpareChildCount) {
+                return storm::utility::math::uint64_log2(maxSpareChildCount) + 1;
+            }
+
+            /*!
+             * Get length of BitVector capturing DFT state.
+             * @param nrElements Number of DFT elements.
+             * @param nrOfSpares Number of Spares (needed for claiming).
+             * @param nrRepresentatives Number of representatives (needed for activation).
+             * @param maxSpareChildCount Maximal number of children of a spare.
+             * @return Length of required BitVector.
+             */
+            static size_t getStateVectorSize(size_t nrElements, size_t nrOfSpares, size_t nrRepresentatives, size_t maxSpareChildCount) {
+                return nrElements * 2 + nrOfSpares * getUsageInfoBits(maxSpareChildCount) + nrRepresentatives;
+            }
+
             size_t usageInfoBits() const {
                 return mUsageInfoBits;
             }
 
             void addStateIndex(size_t id, size_t index) {
                 STORM_LOG_ASSERT(id < mIdToStateIndex.size(), "Id invalid.");
+                STORM_LOG_ASSERT(index < stateIndexSize, "Index invalid");
                 mIdToStateIndex[id] = index;
             }
 
@@ -59,10 +82,12 @@ namespace storm {
             }
 
             void addSpareActivationIndex(size_t id, size_t index) {
+                STORM_LOG_ASSERT(index < stateIndexSize, "Index invalid");
                 mSpareActivationIndex[id] = index;
             }
 
             void addSpareUsageIndex(size_t id, size_t index) {
+                STORM_LOG_ASSERT(index < stateIndexSize, "Index invalid");
                 mSpareUsageIndex[id] = index;
             }
 
@@ -138,6 +163,8 @@ namespace storm {
             }
 
             friend std::ostream& operator<<(std::ostream& os, DFTStateGenerationInfo const& info) {
+                os << "StateGenerationInfo:" << std::endl;
+                os << "Length of state vector: " << info.stateIndexSize << std::endl;
                 os << "Id to state index:" << std::endl;
                 for (size_t id = 0; id < info.mIdToStateIndex.size(); ++id) {
                     os << id << " -> " << info.getStateIndex(id) << std::endl;