#include "ModelCheckerHelper.h"

#include "storm/utility/macros.h"

namespace storm {
    namespace modelchecker {
        namespace helper {
            
            template <typename ValueType, storm::models::ModelRepresentation ModelRepresentation>
            void ModelCheckerHelper<ValueType, ModelRepresentation>::setRelevantStates(StateSet const& relevantStates) {
                _relevantStates = relevantStates;
            }
            
            template <typename ValueType, storm::models::ModelRepresentation ModelRepresentation>
            void ModelCheckerHelper<ValueType, ModelRepresentation>::clearRelevantStates() {
                _relevantStates = boost::none;
            }
            
            template <typename ValueType, storm::models::ModelRepresentation ModelRepresentation>
            bool ModelCheckerHelper<ValueType, ModelRepresentation>::hasRelevantStates() const {
                return _relevantStates.is_initialized();
            }
            
            template <typename ValueType, storm::models::ModelRepresentation ModelRepresentation>
            boost::optional<typename ModelCheckerHelper<ValueType, ModelRepresentation>::StateSet> const& ModelCheckerHelper<ValueType, ModelRepresentation>::getOptionalRelevantStates() const {
                return _relevantStates;
            }
            
            template <typename ValueType, storm::models::ModelRepresentation ModelRepresentation>
            typename ModelCheckerHelper<ValueType, ModelRepresentation>::StateSet const& ModelCheckerHelper<ValueType, ModelRepresentation>::getRelevantStates() const {
                STORM_LOG_ASSERT(hasRelevantStates(), "Retrieving relevant states although none have been set.");
                return _relevantStates.get();
            }
            
            template class ModelCheckerHelper<double, storm::models::ModelRepresentation::Sparse>;
            template class ModelCheckerHelper<storm::RationalNumber, storm::models::ModelRepresentation::Sparse>;
            template class ModelCheckerHelper<storm::RationalFunction, storm::models::ModelRepresentation::Sparse>;
            
            template class ModelCheckerHelper<double, storm::models::ModelRepresentation::DdSylvan>;
            template class ModelCheckerHelper<storm::RationalNumber, storm::models::ModelRepresentation::DdSylvan>;
            template class ModelCheckerHelper<storm::RationalFunction, storm::models::ModelRepresentation::DdSylvan>;
            
            template class ModelCheckerHelper<double, storm::models::ModelRepresentation::DdCudd>;
            
        }
    }
}