/**
 *  \file ComplexDouble.h
 *  Class for ComplexDouble built on top of Number class
 *
 **/
#ifndef SYMENGINE_COMPLEX_DOUBLE_H
#define SYMENGINE_COMPLEX_DOUBLE_H

#include <symengine/real_double.h>
#include <symengine/symengine_exception.h>

namespace SymEngine
{
//! Complex Double Class to hold std::complex<double> values
class ComplexDouble : public ComplexBase
{
public:
    std::complex<double> i;

public:
    IMPLEMENT_TYPEID(SYMENGINE_COMPLEX_DOUBLE)
    //! Constructor of ComplexDouble class
    explicit ComplexDouble(std::complex<double> i);
    //! \return size of the hash
    hash_t __hash__() const override;
    /*! Equality comparator
     * \param o - Object to be compared with
     * \return whether the 2 objects are equal
     * */
    bool __eq__(const Basic &o) const override;
    int compare(const Basic &o) const override;
    //! Get the real part of the complex number
    RCP<const Number> real_part() const override;
    //! Get the imaginary part of the complex number
    RCP<const Number> imaginary_part() const override;
    //! Get the conjugate of the complex number
    RCP<const Basic> conjugate() const override;
    //! \returns `false`
    // False is returned because complex cannot be compared with zero
    inline bool is_positive() const override
    {
        return false;
    }
    //! \returns `false`
    // False is returned because complex cannot be compared with zero
    inline bool is_negative() const override
    {
        return false;
    }
    //! \returns `true`
    inline bool is_complex() const override
    {
        return true;
    }
    //! \return self as a double
    inline std::complex<double> as_complex_double() const
    {
        return i;
    }
    //! \returns `false`
    // False is returned because std::complex<double> is not exact
    inline bool is_exact() const override
    {
        return false;
    }
    //! Get `Evaluate` singleton to evaluate numerically
    Evaluate &get_eval() const override;

    //! \return `true` if equal to `0`
    bool is_zero() const override
    {
        return i == 0.0;
    }
    //! \return `false`
    // A std::complex<double> is not exactly equal to `1`
    bool is_one() const override
    {
        return false;
    }
    //! \return `false`
    // A std::complex<double> is not exactly equal to `-1`
    bool is_minus_one() const override
    {
        return false;
    }

    /*! Add ComplexDoubles
     * \param other of type Integer
     * */
    RCP<const Number> addcomp(const Integer &other) const
    {
        return make_rcp<const ComplexDouble>(
            i + mp_get_d(other.as_integer_class()));
    }

    /*! Add ComplexDoubles
     * \param other of type Rational
     * */
    RCP<const Number> addcomp(const Rational &other) const
    {
        return make_rcp<const ComplexDouble>(
            i + mp_get_d(other.as_rational_class()));
    }

    /*! Add ComplexDoubles
     * \param other of type Complex
     * */
    RCP<const Number> addcomp(const Complex &other) const
    {
        return make_rcp<const ComplexDouble>(
            i
            + std::complex<double>(mp_get_d(other.real_),
                                   mp_get_d(other.imaginary_)));
    }

    /*! Add ComplexDoubles
     * \param other of type RealDouble
     * */
    RCP<const Number> addcomp(const RealDouble &other) const
    {
        return make_rcp<const ComplexDouble>(i + other.i);
    }

    /*! Add ComplexDoubles
     * \param other of type ComplexDouble
     * */
    RCP<const Number> addcomp(const ComplexDouble &other) const
    {
        return make_rcp<const ComplexDouble>(i + other.i);
    }

    //! Converts the param `other` appropriately and then calls `addcomp`
    RCP<const Number> add(const Number &other) const override
    {
        if (is_a<Rational>(other)) {
            return addcomp(down_cast<const Rational &>(other));
        } else if (is_a<Integer>(other)) {
            return addcomp(down_cast<const Integer &>(other));
        } else if (is_a<Complex>(other)) {
            return addcomp(down_cast<const Complex &>(other));
        } else if (is_a<RealDouble>(other)) {
            return addcomp(down_cast<const RealDouble &>(other));
        } else if (is_a<ComplexDouble>(other)) {
            return addcomp(down_cast<const ComplexDouble &>(other));
        } else {
            return other.add(*this);
        }
    }

    /*! Subtract ComplexDoubles
     * \param other of type Integer
     * */
    RCP<const Number> subcomp(const Integer &other) const
    {
        return make_rcp<const ComplexDouble>(
            i - mp_get_d(other.as_integer_class()));
    }

    /*! Subtract ComplexDoubles
     * \param other of type Rational
     * */
    RCP<const Number> subcomp(const Rational &other) const
    {
        return make_rcp<const ComplexDouble>(
            i - mp_get_d(other.as_rational_class()));
    }

    /*! Subtract ComplexDoubles
     * \param other of type Complex
     * */
    RCP<const Number> subcomp(const Complex &other) const
    {
        return make_rcp<const ComplexDouble>(
            i
            - std::complex<double>(mp_get_d(other.real_),
                                   mp_get_d(other.imaginary_)));
    }

    /*! Subtract ComplexDoubles
     * \param other of type RealDouble
     * */
    RCP<const Number> subcomp(const RealDouble &other) const
    {
        return make_rcp<const ComplexDouble>(i - other.i);
    }

    /*! Subtract ComplexDoubles
     * \param other of type ComplexDouble
     * */
    RCP<const Number> subcomp(const ComplexDouble &other) const
    {
        return make_rcp<const ComplexDouble>(i - other.i);
    }

    //! Converts the param `other` appropriately and then calls `subcomp`
    RCP<const Number> sub(const Number &other) const override
    {
        if (is_a<Rational>(other)) {
            return subcomp(down_cast<const Rational &>(other));
        } else if (is_a<Integer>(other)) {
            return subcomp(down_cast<const Integer &>(other));
        } else if (is_a<Complex>(other)) {
            return subcomp(down_cast<const Complex &>(other));
        } else if (is_a<RealDouble>(other)) {
            return subcomp(down_cast<const RealDouble &>(other));
        } else if (is_a<ComplexDouble>(other)) {
            return subcomp(down_cast<const ComplexDouble &>(other));
        } else {
            return other.rsub(*this);
        }
    }

    /*! Subtract ComplexDoubles
     * \param other of type Integer
     * */
    RCP<const Number> rsubcomp(const Integer &other) const
    {
        return make_rcp<const ComplexDouble>(mp_get_d(other.as_integer_class())
                                             - i);
    }

    /*! Subtract ComplexDoubles
     * \param other of type Rational
     * */
    RCP<const Number> rsubcomp(const Rational &other) const
    {
        return make_rcp<const ComplexDouble>(mp_get_d(other.as_rational_class())
                                             - i);
    }

    /*! Subtract ComplexDoubles
     * \param other of type Complex
     * */
    RCP<const Number> rsubcomp(const Complex &other) const
    {
        return make_rcp<const ComplexDouble>(
            -i
            + std::complex<double>(mp_get_d(other.real_),
                                   mp_get_d(other.imaginary_)));
    }

    /*! Subtract ComplexDoubles
     * \param other of type RealDouble
     * */
    RCP<const Number> rsubcomp(const RealDouble &other) const
    {
        return make_rcp<const ComplexDouble>(other.i - i);
    }

    //! Converts the param `other` appropriately and then calls `subcomp`
    RCP<const Number> rsub(const Number &other) const override
    {
        if (is_a<Rational>(other)) {
            return rsubcomp(down_cast<const Rational &>(other));
        } else if (is_a<Integer>(other)) {
            return rsubcomp(down_cast<const Integer &>(other));
        } else if (is_a<Complex>(other)) {
            return rsubcomp(down_cast<const Complex &>(other));
        } else if (is_a<RealDouble>(other)) {
            return rsubcomp(down_cast<const RealDouble &>(other));
        } else {
            throw NotImplementedError("Not Implemented");
        }
    }

    /*! Multiply ComplexDoubles
     * \param other of type Integer
     * */
    RCP<const Number> mulcomp(const Integer &other) const
    {
        return make_rcp<const ComplexDouble>(
            i * mp_get_d(other.as_integer_class()));
    }

    /*! Multiply ComplexDoubles
     * \param other of type Rational
     * */
    RCP<const Number> mulcomp(const Rational &other) const
    {
        return make_rcp<const ComplexDouble>(
            i * mp_get_d(other.as_rational_class()));
    }

    /*! Multiply ComplexDoubles
     * \param other of type Complex
     * */
    RCP<const Number> mulcomp(const Complex &other) const
    {
        return make_rcp<const ComplexDouble>(
            i
            * std::complex<double>(mp_get_d(other.real_),
                                   mp_get_d(other.imaginary_)));
    }

    /*! Multiply ComplexDoubles
     * \param other of type RealDouble
     * */
    RCP<const Number> mulcomp(const RealDouble &other) const
    {
        return make_rcp<const ComplexDouble>(i * other.i);
    }

    /*! Multiply ComplexDoubles
     * \param other of type ComplexDouble
     * */
    RCP<const Number> mulcomp(const ComplexDouble &other) const
    {
        return make_rcp<const ComplexDouble>(i * other.i);
    }

    //! Converts the param `other` appropriately and then calls `mulcomp`
    RCP<const Number> mul(const Number &other) const override
    {
        if (is_a<Rational>(other)) {
            return mulcomp(down_cast<const Rational &>(other));
        } else if (is_a<Integer>(other)) {
            return mulcomp(down_cast<const Integer &>(other));
        } else if (is_a<Complex>(other)) {
            return mulcomp(down_cast<const Complex &>(other));
        } else if (is_a<RealDouble>(other)) {
            return mulcomp(down_cast<const RealDouble &>(other));
        } else if (is_a<ComplexDouble>(other)) {
            return mulcomp(down_cast<const ComplexDouble &>(other));
        } else {
            return other.mul(*this);
        }
    }

    /*! Divide ComplexDoubles
     * \param other of type Integer
     * */
    RCP<const Number> divcomp(const Integer &other) const
    {
        return make_rcp<const ComplexDouble>(
            i / mp_get_d(other.as_integer_class()));
    }

    /*! Divide ComplexDoubles
     * \param other of type Rational
     * */
    RCP<const Number> divcomp(const Rational &other) const
    {
        return make_rcp<const ComplexDouble>(
            i / mp_get_d(other.as_rational_class()));
    }

    /*! Divide ComplexDoubles
     * \param other of type Complex
     * */
    RCP<const Number> divcomp(const Complex &other) const
    {
        return make_rcp<const ComplexDouble>(
            i
            / std::complex<double>(mp_get_d(other.real_),
                                   mp_get_d(other.imaginary_)));
    }

    /*! Divide ComplexDoubles
     * \param other of type RealDouble
     * */
    RCP<const Number> divcomp(const RealDouble &other) const
    {
        return make_rcp<const ComplexDouble>(i / other.i);
    }

    /*! Divide ComplexDoubles
     * \param other of type ComplexDouble
     * */
    RCP<const Number> divcomp(const ComplexDouble &other) const
    {
        return make_rcp<const ComplexDouble>(i / other.i);
    }

    //! Converts the param `other` appropriately and then calls `divcomp`
    RCP<const Number> div(const Number &other) const override
    {
        if (is_a<Rational>(other)) {
            return divcomp(down_cast<const Rational &>(other));
        } else if (is_a<Integer>(other)) {
            return divcomp(down_cast<const Integer &>(other));
        } else if (is_a<Complex>(other)) {
            return divcomp(down_cast<const Complex &>(other));
        } else if (is_a<RealDouble>(other)) {
            return divcomp(down_cast<const RealDouble &>(other));
        } else if (is_a<ComplexDouble>(other)) {
            return divcomp(down_cast<const ComplexDouble &>(other));
        } else {
            return other.rdiv(*this);
        }
    }

    /*! Divide ComplexDoubles
     * \param other of type Integer
     * */
    RCP<const Number> rdivcomp(const Integer &other) const
    {
        return make_rcp<const ComplexDouble>(mp_get_d(other.as_integer_class())
                                             / i);
    }

    /*! Divide ComplexDoubles
     * \param other of type Rational
     * */
    RCP<const Number> rdivcomp(const Rational &other) const
    {
        return make_rcp<const ComplexDouble>(mp_get_d(other.as_rational_class())
                                             / i);
    }

    /*! Divide ComplexDoubles
     * \param other of type Complex
     * */
    RCP<const Number> rdivcomp(const Complex &other) const
    {
        return make_rcp<const ComplexDouble>(
            std::complex<double>(mp_get_d(other.real_),
                                 mp_get_d(other.imaginary_))
            / i);
    }

    /*! Divide ComplexDoubles
     * \param other of type RealDouble
     * */
    RCP<const Number> rdivcomp(const RealDouble &other) const
    {
        return make_rcp<const ComplexDouble>(other.i / i);
    }

    //! Converts the param `other` appropriately and then calls `divcomp`
    RCP<const Number> rdiv(const Number &other) const override
    {
        if (is_a<Rational>(other)) {
            return rdivcomp(down_cast<const Rational &>(other));
        } else if (is_a<Integer>(other)) {
            return rdivcomp(down_cast<const Integer &>(other));
        } else if (is_a<Complex>(other)) {
            return rdivcomp(down_cast<const Complex &>(other));
        } else if (is_a<RealDouble>(other)) {
            return rdivcomp(down_cast<const RealDouble &>(other));
        } else {
            throw NotImplementedError("Not Implemented");
        }
    }

    /*! Raise ComplexDouble to power `other`
     * \param other of type Integer
     * */
    RCP<const Number> powcomp(const Integer &other) const
    {
        return make_rcp<const ComplexDouble>((std::complex<double>)std::pow(
            i, mp_get_d(other.as_integer_class())));
    }

    /*! Raise ComplexDouble to power `other`
     * \param other of type Rational
     * */
    RCP<const Number> powcomp(const Rational &other) const
    {
        return make_rcp<const ComplexDouble>((std::complex<double>)std::pow(
            i, mp_get_d(other.as_rational_class())));
    }

    /*! Raise ComplexDouble to power `other`
     * \param other of type Complex
     * */
    RCP<const Number> powcomp(const Complex &other) const
    {
        return make_rcp<const ComplexDouble>((std::complex<double>)std::pow(
            i, std::complex<double>(mp_get_d(other.real_),
                                    mp_get_d(other.imaginary_))));
    }
    /*! Raise ComplexDouble to power `other`
     * \param other of type RealDouble
     * */
    RCP<const Number> powcomp(const RealDouble &other) const
    {
        return make_rcp<const ComplexDouble>(
            (std::complex<double>)std::pow(i, other.i));
    }

    /*! Raise ComplexDouble to power `other`
     * \param other of type ComplexDouble
     * */
    RCP<const Number> powcomp(const ComplexDouble &other) const
    {
        return make_rcp<const ComplexDouble>(
            (std::complex<double>)std::pow(i, other.i));
    }

    //! Converts the param `other` appropriately and then calls `powcomp`
    RCP<const Number> pow(const Number &other) const override
    {
        if (is_a<Rational>(other)) {
            return powcomp(down_cast<const Rational &>(other));
        } else if (is_a<Integer>(other)) {
            return powcomp(down_cast<const Integer &>(other));
        } else if (is_a<Complex>(other)) {
            return powcomp(down_cast<const Complex &>(other));
        } else if (is_a<RealDouble>(other)) {
            return powcomp(down_cast<const RealDouble &>(other));
        } else if (is_a<ComplexDouble>(other)) {
            return powcomp(down_cast<const ComplexDouble &>(other));
        } else {
            return other.rpow(*this);
        }
    }

    /*! Raise `other` to power ComplexDouble
     * \param other of type Integer
     * */
    RCP<const Number> rpowcomp(const Integer &other) const
    {
        return make_rcp<const ComplexDouble>((std::complex<double>)std::pow(
            mp_get_d(other.as_integer_class()), i));
    }

    /*! Raise `other` to power ComplexDouble
     * \param other of type Rational
     * */
    RCP<const Number> rpowcomp(const Rational &other) const
    {
        return make_rcp<const ComplexDouble>((std::complex<double>)std::pow(
            mp_get_d(other.as_rational_class()), i));
    }

    /*! Raise `other` to power ComplexDouble
     * \param other of type Complex
     * */
    RCP<const Number> rpowcomp(const Complex &other) const
    {
        return make_rcp<const ComplexDouble>((std::complex<double>)std::pow(
            std::complex<double>(mp_get_d(other.real_),
                                 mp_get_d(other.imaginary_)),
            i));
    }

    /*! Raise `other` to power ComplexDouble
     * \param other of type RealDouble
     * */
    RCP<const Number> rpowcomp(const RealDouble &other) const
    {
        return make_rcp<const ComplexDouble>(
            (std::complex<double>)std::pow(other.i, i));
    }

    //! Converts the param `other` appropriately and then calls `powcomp`
    RCP<const Number> rpow(const Number &other) const override
    {
        if (is_a<Rational>(other)) {
            return rpowcomp(down_cast<const Rational &>(other));
        } else if (is_a<Integer>(other)) {
            return rpowcomp(down_cast<const Integer &>(other));
        } else if (is_a<Complex>(other)) {
            return rpowcomp(down_cast<const Complex &>(other));
        } else if (is_a<RealDouble>(other)) {
            return rpowcomp(down_cast<const RealDouble &>(other));
        } else {
            throw NotImplementedError("Not Implemented");
        }
    }
};

RCP<const ComplexDouble> complex_double(std::complex<double> x);
RCP<const ComplexDouble> complex_double(double real, double imag);

} // namespace SymEngine

#endif