#ifndef PYTHONIC_NUMPY_ATLEAST2D_HPP
#define PYTHONIC_NUMPY_ATLEAST2D_HPP

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

#include "pythonic/types/ndarray.hpp"

PYTHONIC_NS_BEGIN

namespace numpy
{
  template <class T>
  std::enable_if_t<types::is_dtype<T>::value,
                   types::ndarray<T, types::pshape<std::integral_constant<long, 1>,
                                                   std::integral_constant<long, 1>>>>
  atleast_2d(T t)
  {
    return {types::pshape<std::integral_constant<long, 1>, std::integral_constant<long, 1>>(), t};
  }

  template <class T>
          auto atleast_2d(T const &t) -> std::enable_if_t < (!types::is_dtype<T>::value) &&
      T::value<2, types::ndarray<typename T::dtype,
                                 types::pshape<std::integral_constant<long, 1>,
                                               std::tuple_element_t<0, typename T::shape_t>>>>
  {
    return t.reshape(types::pshape<std::integral_constant<long, 1>,
                                   std::tuple_element_t<0, typename T::shape_t>>(
        std::integral_constant<long, 1>(), t.template shape<0>()));
  }

  template <class T>
  auto atleast_2d(T &&t)
      -> std::enable_if_t<(!types::is_dtype<std::remove_cv_t<std::remove_reference_t<T>>>::value) &&
                              std::decay_t<T>::value >= 2,
                          decltype(std::forward<T>(t))>
  {
    return std::forward<T>(t);
  }
} // namespace numpy
PYTHONIC_NS_END

#endif
