#ifndef PYTHONIC_NUMPY_ZEROS_HPP
#define PYTHONIC_NUMPY_ZEROS_HPP

#include "pythonic/include/numpy/zeros.hpp"

#include "pythonic/utils/functor.hpp"
#include "pythonic/types/ndarray.hpp"

PYTHONIC_NS_BEGIN

namespace numpy
{
  template <class pS, class dtype>
  types::ndarray<typename dtype::type, sutils::shape_t<pS>>
  zeros(pS const &shape, dtype d)
  {
    using T = typename dtype::type;
    // use calloc even if we have a non integer type. This looks ok on modern
    // architecture, although not really standard
    auto *buffer = (T *)calloc(sutils::sprod(shape), sizeof(T));
    return {buffer, (sutils::shape_t<pS>)shape, types::ownership::owned};
  }

  template <class dtype>
  types::ndarray<typename dtype::type, types::pshape<long>> zeros(long size,
                                                                  dtype d)
  {
    return zeros(types::pshape<long>(size), d);
  }

  template <long N, class dtype>
  types::ndarray<typename dtype::type,
                 types::pshape<std::integral_constant<long, N>>>
  zeros(std::integral_constant<long, N>, dtype d)
  {
    return zeros(types::pshape<std::integral_constant<long, N>>({}), d);
  }
}
PYTHONIC_NS_END

#endif
