#ifndef STORM_LOGIC_VARIABLESUBSTITUTIONVISITOR_H_
#define STORM_LOGIC_VARIABLESUBSTITUTIONVISITOR_H_

#include <map>

#include "src/logic/CloneVisitor.h"

#include "src/storage/expressions/Expression.h"

namespace storm {
    namespace logic {
        
        class VariableSubstitutionVisitor : public CloneVisitor {
        public:
            VariableSubstitutionVisitor(std::map<storm::expressions::Variable, storm::expressions::Expression> const& substitution);
            
            std::shared_ptr<Formula> substitute(Formula const& f) const;
            
            virtual boost::any visit(AtomicExpressionFormula const& f, boost::any const& data) const override;
            
        private:
            std::map<storm::expressions::Variable, storm::expressions::Expression> const& substitution;
        };
        
    }
}


#endif /* STORM_LOGIC_VARIABLESUBSTITUTIONVISITOR_H_ */