Small refactoring for ElementState

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

@@ -540,12 +540,10 @@ namespace storm {
             std::stringstream stream;
             stream << "(" << id << ") ";
             for (auto const& elem : mElements) {
-                size_t elemIndex = stateGenerationInfo.getStateIndex(elem->id());
-                int elementState = DFTState<ValueType>::getElementStateInt(status, elemIndex);
                 if (elem->isDependency()) {
-                    stream << storm::storage::toChar(static_cast<DFTDependencyState>(elementState)) << "[dep]";
+                    stream << storm::storage::toChar(DFTState<ValueType>::getDependencyState(status, stateGenerationInfo, elem->id())) << "[dep]";
                 } else {
-                    stream << storm::storage::toChar(static_cast<DFTElementState>(elementState));
+                    stream << storm::storage::toChar(DFTState<ValueType>::getElementState(status, stateGenerationInfo, elem->id()));
                     if(elem->isSpareGate()) {
                         stream << "[";
                         size_t nrUsedChild = status.getAsInt(stateGenerationInfo.getSpareUsageIndex(elem->id()), stateGenerationInfo.usageInfoBits());

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

@@ -99,19 +99,29 @@ namespace storm {
             return static_cast<DFTElementState>(getElementStateInt(id));
         }
 
+        template<typename ValueType>
+        DFTElementState DFTState<ValueType>::getElementState(storm::storage::BitVector const& state, DFTStateGenerationInfo const& stateGenerationInfo, size_t id) {
+            return static_cast<DFTElementState>(DFTState<ValueType>::getElementStateInt(state, stateGenerationInfo, id));
+        }
+        
         template<typename ValueType>
         DFTDependencyState DFTState<ValueType>::getDependencyState(size_t id) const {
             return static_cast<DFTDependencyState>(getElementStateInt(id));
         }
 
+        template<typename ValueType>
+        DFTDependencyState DFTState<ValueType>::getDependencyState(storm::storage::BitVector const& state, DFTStateGenerationInfo const& stateGenerationInfo, size_t id) {
+            return static_cast<DFTDependencyState>(DFTState<ValueType>::getElementStateInt(state, stateGenerationInfo, id));
+        }
+
         template<typename ValueType>
         int DFTState<ValueType>::getElementStateInt(size_t id) const {
             return mStatus.getAsInt(mStateGenerationInfo.getStateIndex(id), 2);
         }
         
         template<typename ValueType>
-        int DFTState<ValueType>::getElementStateInt(storm::storage::BitVector const& state, size_t indexId) {
-            return state.getAsInt(indexId, 2);
+        int DFTState<ValueType>::getElementStateInt(storm::storage::BitVector const& state, DFTStateGenerationInfo const& stateGenerationInfo, size_t id) {
+            return state.getAsInt(stateGenerationInfo.getStateIndex(id), 2);
         }
 
         template<typename ValueType>

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

@@ -153,11 +153,11 @@ namespace storm {
 
             DFTElementState getElementState(size_t id) const;
 
-            DFTDependencyState getDependencyState(size_t id) const;
+            static DFTElementState getElementState(storm::storage::BitVector const& state, DFTStateGenerationInfo const& stateGenerationInfo, size_t id);
 
-            int getElementStateInt(size_t id) const;
+            DFTDependencyState getDependencyState(size_t id) const;
 
-            static int getElementStateInt(storm::storage::BitVector const& state, size_t indexId);
+            static DFTDependencyState getDependencyState(storm::storage::BitVector const& state, DFTStateGenerationInfo const& stateGenerationInfo, size_t id);
 
             size_t getId() const;
 
@@ -372,6 +372,10 @@ namespace storm {
         private:
             void propagateActivation(size_t representativeId);
 
+            int getElementStateInt(size_t id) const;
+
+            static int getElementStateInt(storm::storage::BitVector const& state, DFTStateGenerationInfo const& stateGenerationInfo, size_t id);
+
         };
 
     }