#pragma once

#include <functional>

#include "storm/solver/SolverGuarantee.h"
#include "storm/storage/BitVector.h"

namespace storm {
    namespace solver {
        template<typename ValueType>
        class TerminationCondition {
        public:
            virtual ~TerminationCondition() = default;

            /*!
             * Retrieves whether the guarantee provided by the solver for the current result is sufficient to terminate.
             */
            virtual bool terminateNow(std::vector<ValueType> const& currentValues, SolverGuarantee const& guarantee = SolverGuarantee::None) const;
            virtual bool terminateNow(std::function<ValueType(uint64_t const&)> const& valueGetter, SolverGuarantee const& guarantee = SolverGuarantee::None) const = 0;
            
            /*!
             * Retrieves whether the termination criterion requires the given guarantee in order to decide termination.
             * @return
             */
            virtual bool requiresGuarantee(SolverGuarantee const& guarantee) const = 0;
        };
        
        template<typename ValueType>
        class NoTerminationCondition : public TerminationCondition<ValueType> {
        public:
            virtual bool terminateNow(std::function<ValueType(uint64_t const&)> const& valueGetter, SolverGuarantee const& guarantee = SolverGuarantee::None) const override;
            virtual bool requiresGuarantee(SolverGuarantee const& guarantee) const override;
        };
        
        template<typename ValueType>
        class TerminateIfFilteredSumExceedsThreshold : public TerminationCondition<ValueType> {
        public:
            TerminateIfFilteredSumExceedsThreshold(storm::storage::BitVector const& filter, ValueType const& threshold, bool strict);
            
            bool terminateNow(std::function<ValueType(uint64_t const&)> const& valueGetter, SolverGuarantee const& guarantee = SolverGuarantee::None) const override;
            virtual bool requiresGuarantee(SolverGuarantee const& guarantee) const override;

        protected:
            ValueType threshold;
            storm::storage::BitVector filter;
            bool strict;
        };
        
        template<typename ValueType>
        class TerminateIfFilteredExtremumExceedsThreshold : public TerminateIfFilteredSumExceedsThreshold<ValueType>{
        public:
            TerminateIfFilteredExtremumExceedsThreshold(storm::storage::BitVector const& filter, bool strict, ValueType const& threshold, bool useMinimum);
            
            bool terminateNow(std::function<ValueType(uint64_t const&)> const& valueGetter, SolverGuarantee const& guarantee = SolverGuarantee::None) const override;
            virtual bool requiresGuarantee(SolverGuarantee const& guarantee) const override;
            
        protected:
            bool useMinimum;
            mutable uint64_t cachedExtremumIndex;
        };
        
        template<typename ValueType>
        class TerminateIfFilteredExtremumBelowThreshold : public TerminateIfFilteredSumExceedsThreshold<ValueType>{
        public:
            TerminateIfFilteredExtremumBelowThreshold(storm::storage::BitVector const& filter, bool strict, ValueType const& threshold, bool useMinimum);
            
            bool terminateNow(std::function<ValueType(uint64_t const&)> const& valueGetter, SolverGuarantee const& guarantee = SolverGuarantee::None) const override;
            virtual bool requiresGuarantee(SolverGuarantee const& guarantee) const override;
            
        protected:
            bool useMinimum;
            mutable uint64_t cachedExtremumIndex;
        };
    }
}