#include "storm/solver/AbstractEquationSolver.h"

#include "storm/adapters/RationalNumberAdapter.h"
#include "storm/adapters/RationalFunctionAdapter.h"

#include "storm/utility/macros.h"
#include "storm/exceptions/UnmetRequirementException.h"

namespace storm {
    namespace solver {
        
        template<typename ValueType>
        void AbstractEquationSolver<ValueType>::setTerminationCondition(std::unique_ptr<TerminationCondition<ValueType>> terminationCondition) {
            this->terminationCondition = std::move(terminationCondition);
        }
        
        template<typename ValueType>
        void AbstractEquationSolver<ValueType>::resetTerminationCondition() {
            this->terminationCondition = nullptr;
        }
        
        template<typename ValueType>
        bool AbstractEquationSolver<ValueType>::hasCustomTerminationCondition() const {
            return static_cast<bool>(this->terminationCondition);
        }
        
        template<typename ValueType>
        TerminationCondition<ValueType> const& AbstractEquationSolver<ValueType>::getTerminationCondition() const {
            return *terminationCondition;
        }
        
        template<typename ValueType>
        bool AbstractEquationSolver<ValueType>::terminateNow(std::vector<ValueType> const& values, SolverGuarantee const& guarantee) const {
            if (!this->hasCustomTerminationCondition()) {
                return false;
            }
            
            return this->getTerminationCondition().terminateNow(values, guarantee);
        }
        
        template<typename ValueType>
        bool AbstractEquationSolver<ValueType>::hasRelevantValues() const {
            return static_cast<bool>(relevantValues);
        }
        
        template<typename ValueType>
        storm::storage::BitVector const& AbstractEquationSolver<ValueType>::getRelevantValues()const {
            return relevantValues.get();
        }
        
        template<typename ValueType>
        void AbstractEquationSolver<ValueType>::setRelevantValues(storm::storage::BitVector&& relevantValues) {
            this->relevantValues = std::move(relevantValues);
        }
        
        template<typename ValueType>
        void AbstractEquationSolver<ValueType>::clearRelevantValues() {
            relevantValues = boost::none;
        }
        
        template<typename ValueType>
        bool AbstractEquationSolver<ValueType>::hasLowerBound(BoundType const& type) const {
            if (type == BoundType::Any) {
                return static_cast<bool>(lowerBound) || static_cast<bool>(lowerBounds);
            } else if (type == BoundType::Global) {
                return static_cast<bool>(lowerBound);
            } else if (type == BoundType::Local) {
                return static_cast<bool>(lowerBounds);
            }
            return false;
        }
        
        template<typename ValueType>
        bool AbstractEquationSolver<ValueType>::hasUpperBound(BoundType const& type) const {
            if (type == BoundType::Any) {
                return static_cast<bool>(upperBound) || static_cast<bool>(upperBounds);
            } else if (type == BoundType::Global) {
                return static_cast<bool>(upperBound);
            } else if (type == BoundType::Local) {
                return static_cast<bool>(upperBounds);
            }
            return false;
        }
        
        template<typename ValueType>
        void AbstractEquationSolver<ValueType>::setLowerBound(ValueType const& value) {
            lowerBound = value;
        }
        
        template<typename ValueType>
        void AbstractEquationSolver<ValueType>::setUpperBound(ValueType const& value) {
            upperBound = value;
        }
        
        template<typename ValueType>
        void AbstractEquationSolver<ValueType>::setBounds(ValueType const& lower, ValueType const& upper) {
            setLowerBound(lower);
            setUpperBound(upper);
        }
        
        template<typename ValueType>
        ValueType const& AbstractEquationSolver<ValueType>::getLowerBound() const {
            return lowerBound.get();
        }
        
        template<typename ValueType>
        ValueType const& AbstractEquationSolver<ValueType>::getUpperBound() const {
            return upperBound.get();
        }
        
        template<typename ValueType>
        std::vector<ValueType> const& AbstractEquationSolver<ValueType>::getLowerBounds() const {
            return lowerBounds.get();
        }
        
        template<typename ValueType>
        std::vector<ValueType> const& AbstractEquationSolver<ValueType>::getUpperBounds() const {
            return upperBounds.get();
        }
        
        template<typename ValueType>
        void AbstractEquationSolver<ValueType>::setLowerBounds(std::vector<ValueType> const& values) {
            lowerBounds = values;
        }
        
        template<typename ValueType>
        void AbstractEquationSolver<ValueType>::setUpperBounds(std::vector<ValueType> const& values) {
            upperBounds = values;
        }
        
        template<typename ValueType>
        void AbstractEquationSolver<ValueType>::setUpperBounds(std::vector<ValueType>&& values) {
            upperBounds = std::move(values);
        }
        
        template<typename ValueType>
        void AbstractEquationSolver<ValueType>::setBounds(std::vector<ValueType> const& lower, std::vector<ValueType> const& upper) {
            setLowerBounds(lower);
            setUpperBounds(upper);
        }
        
        template<typename ValueType>
        void AbstractEquationSolver<ValueType>::createLowerBoundsVector(std::vector<ValueType>& lowerBoundsVector) const {
            if (this->hasLowerBound(BoundType::Local)) {
                lowerBoundsVector = this->getLowerBounds();
            } else {
                STORM_LOG_THROW(this->hasLowerBound(BoundType::Global), storm::exceptions::UnmetRequirementException, "Cannot create lower bounds vector without lower bound.");
                for (auto& e : lowerBoundsVector) {
                    e = this->getLowerBound();
                }
            }
        }
        
        template<typename ValueType>
        void AbstractEquationSolver<ValueType>::createUpperBoundsVector(std::unique_ptr<std::vector<ValueType>>& upperBoundsVector, uint64_t length) const {
            STORM_LOG_ASSERT(this->hasUpperBound(), "Expecting upper bound(s).");
            if (!upperBoundsVector) {
                if (this->hasUpperBound(BoundType::Local)) {
                    STORM_LOG_ASSERT(length == this->getUpperBounds().size(), "Mismatching sizes.");
                    upperBoundsVector = std::make_unique<std::vector<ValueType>>(this->getUpperBounds());
                } else {
                    upperBoundsVector = std::make_unique<std::vector<ValueType>>(length, this->getUpperBound());
                }
            } else {
                if (this->hasUpperBound(BoundType::Global)) {
                    for (auto& e : *upperBoundsVector) {
                        e = this->getUpperBound();
                    }
                } else {
                    auto upperBoundsIt = this->getUpperBounds().begin();
                    for (auto& e : *upperBoundsVector) {
                        e = *upperBoundsIt;
                        ++upperBoundsIt;
                    }
                }
            }
        }
        
        template class AbstractEquationSolver<double>;
        
#ifdef STORM_HAVE_CARL
        template class AbstractEquationSolver<storm::RationalNumber>;
        template class AbstractEquationSolver<storm::RationalFunction>;
#endif
        
    }
}