#ifndef STORM_LOGIC_CUMULATIVEREWARDFORMULA_H_
#define STORM_LOGIC_CUMULATIVEREWARDFORMULA_H_

#include "storm/logic/PathFormula.h"

#include "storm/logic/TimeBound.h"
#include "storm/logic/TimeBoundType.h"

namespace storm {
    namespace logic {
        class CumulativeRewardFormula : public PathFormula {
        public:
            CumulativeRewardFormula(TimeBound const& bound, TimeBoundReference const& timeBoundReference = TimeBoundReference(TimeBoundType::Time), boost::optional<RewardAccumulation> rewardAccumulation = boost::none);
            CumulativeRewardFormula(std::vector<TimeBound> const& bounds, std::vector<TimeBoundReference> const& timeBoundReferences, boost::optional<RewardAccumulation> rewardAccumulation = boost::none);
            
            virtual ~CumulativeRewardFormula() = default;

            virtual bool isCumulativeRewardFormula() const override;
            virtual bool isRewardPathFormula() const override;
            
            bool isMultiDimensional() const;
            unsigned getDimension() const;

            virtual boost::any accept(FormulaVisitor const& visitor, boost::any const& data) const override;

            virtual void gatherReferencedRewardModels(std::set<std::string>& referencedRewardModels) const override;
            
            virtual std::ostream& writeToStream(std::ostream& out) const override;
            
            TimeBoundReference const& getTimeBoundReference() const;
            TimeBoundReference const& getTimeBoundReference(unsigned i) const;
            
            bool isBoundStrict() const;
            bool isBoundStrict(unsigned i) const;
            bool hasIntegerBound() const;
            bool hasIntegerBound(unsigned i) const;
            
            storm::expressions::Expression const& getBound() const;
            storm::expressions::Expression const& getBound(unsigned i) const;
            
            template <typename ValueType>
            ValueType getBound() const;
            
            template <typename ValueType>
            ValueType getBound(unsigned i) const;
            
            template <typename ValueType>
            ValueType getNonStrictBound() const;
            
            bool hasRewardAccumulation() const;
            RewardAccumulation const& getRewardAccumulation() const;
            
            std::shared_ptr<CumulativeRewardFormula const> restrictToDimension(unsigned i) const;
            
        private:
            static void checkNoVariablesInBound(storm::expressions::Expression const& bound);

            std::vector<TimeBoundReference> timeBoundReferences;
            std::vector<TimeBound> bounds;
            boost::optional<RewardAccumulation> rewardAccumulation;

        };
    }
}

#endif /* STORM_LOGIC_CUMULATIVEREWARDFORMULA_H_ */