DftNextStateGenerator.h

Go to the documentation of this file.

#pragma once
 
#include "storm/generator/NextStateGenerator.h"
#include "storm/utility/ConstantsComparator.h"
 
#include "storm-dft/storage/DFT.h"
 
namespace storm::dft {
namespace generator {

template<typename ValueType, typename StateType = uint32_t>
class DftNextStateGenerator {
    // TODO: inherit from NextStateGenerator
 
    using DFTStatePointer = std::shared_ptr<storm::dft::storage::DFTState<ValueType>>;
    using DFTElementPointer = std::shared_ptr<storm::dft::storage::elements::DFTElement<ValueType>>;
    using DFTGatePointer = std::shared_ptr<storm::dft::storage::elements::DFTGate<ValueType>>;
    using DFTRestrictionPointer = std::shared_ptr<storm::dft::storage::elements::DFTRestriction<ValueType>>;
 
   public:
    typedef std::function<StateType(DFTStatePointer const&)> StateToIdCallback;
 
    DftNextStateGenerator(storm::dft::storage::DFT<ValueType> const& dft, storm::dft::storage::DFTStateGenerationInfo const& stateGenerationInfo);
 
    bool isDeterministicModel() const;
    std::vector<StateType> getInitialStates(StateToIdCallback const& stateToIdCallback);
 
    void load(storm::storage::BitVector const& state);
    void load(DFTStatePointer const& state);

    storm::generator::StateBehavior<ValueType, StateType> expand(StateToIdCallback const& stateToIdCallback, bool takeFirstDependency);

    storm::generator::StateBehavior<ValueType, StateType> createMergeFailedState(StateToIdCallback const& stateToIdCallback);

    DFTStatePointer createInitialState() const;

    DFTStatePointer createSuccessorState(DFTStatePointer const origState, std::shared_ptr<storm::dft::storage::elements::DFTBE<ValueType> const> be) const;

    DFTStatePointer createSuccessorState(DFTStatePointer const origState,
                                         std::shared_ptr<storm::dft::storage::elements::DFTDependency<ValueType> const> dependency,
                                         bool dependencySuccessful = true) const;

    void propagateFailure(DFTStatePointer newState, std::shared_ptr<storm::dft::storage::elements::DFTBE<ValueType> const>& nextBE,
                          storm::dft::storage::DFTStateSpaceGenerationQueues<ValueType>& queues) const;

    void propagateFailsafe(DFTStatePointer newState, std::shared_ptr<storm::dft::storage::elements::DFTBE<ValueType> const>& nextBE,
                           storm::dft::storage::DFTStateSpaceGenerationQueues<ValueType>& queues) const;
 
   private:
    storm::generator::StateBehavior<ValueType, StateType> exploreState(StateToIdCallback const& stateToIdCallback, bool exploreDependencies,
                                                                       bool takeFirstDependency);

    std::pair<StateType, bool> getNewStateId(DFTStatePointer state, StateToIdCallback const& stateToIdCallback) const;
 
    // The dft used for the generation of next states.
    storm::dft::storage::DFT<ValueType> const& mDft;
 
    // General information for the state generation.
    storm::dft::storage::DFTStateGenerationInfo const& mStateGenerationInfo;
 
    // Current state
    DFTStatePointer state;
 
    // Flag indicating whether all failed states should be merged into one unique failed state.
    bool uniqueFailedState;
 
    // Flag indicating whether the model is deterministic.
    bool deterministicModel = false;
};
 
}  // namespace generator
}  // namespace storm::dft