from ..config cimport translate_exception
from libc.stdint cimport intptr_t
from libcpp cimport bool
from libcpp.map cimport map
from libcpp.string cimport string
from libcpp.vector cimport vector
from .type cimport type, property_t

cdef extern from 'dynd/array.hpp' namespace 'dynd::nd' nogil:

    cdef enum:
        read_access_flag
        write_access_flag
        immutable_access_flag
        readwrite_access_flags
        default_access_flags

    cdef cppclass array:

        array()
        array(int)
        array(type *, int)

        T as[T]() except +translate_exception
        type get_type()
        type get_dtype()
        type get_dtype(size_t)
        intptr_t get_ndim()
        intptr_t get_dim_size() except +translate_exception
        intptr_t get_dim_size(intptr_t) except +translate_exception

        array p(string) except +translate_exception

        char *data() const
        const char *cdata() const

        array view_scalars(type&) except +translate_exception

        @staticmethod
        array from_type_property(const property_t &) except +translate_exception

        bool is_null()

        void assign(array &) except +translate_exception
        array eval() except +translate_exception
        array cast(type) except +translate_exception
        array ucast(type, ssize_t) except +translate_exception

        # These should only be used with versions of Cython later than 0.23.
        # Otherwise the exception handler isn't properly applied and the
        # resulting uncaught C++ exceptions can crash the Python interpreter.
        array operator<(array&) except +translate_exception
        array operator<=(array&) except +translate_exception
        array operator==(array&) except +translate_exception
        array operator!=(array&) except +translate_exception
        array operator>=(array&) except +translate_exception
        array operator>(array&) except +translate_exception
        array operator+(array&) except +translate_exception
        array operator-(array&) except +translate_exception
        array operator*(array&) except +translate_exception
        array operator/(array&) except +translate_exception
        array operator()(...) except +translate_exception

    array empty(type &tp) except +translate_exception

    # These should be usable with Cython's operator overloading syntax, but
    # the exception handling doesn't work for overloaded operators in Cython
    # versions before 0.24, so, for now, this will have to do.
    array array_add "operator+"(array&, array&) except +translate_exception
    array array_subtract "operator-"(array&, array&) except +translate_exception
    array array_multiply "operator*"(array&, array&) except +translate_exception
    array array_divide "operator/"(array&, array&) except +translate_exception

    array dtyped_zeros(ssize_t, ssize_t*, type&) except +translate_exception
    array dtyped_ones(ssize_t, ssize_t*, type&) except +translate_exception
    array dtyped_empty(ssize_t, ssize_t*, type&) except +translate_exception

    array groupby(array&, array&, type) except +translate_exception
    array groupby(array&, array&) except +translate_exception