contrib/python/pythran/pythran/pythonic/numpy/flip.hpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

#ifndef PYTHONIC_NUMPY_FLIP_HPP
#define PYTHONIC_NUMPY_FLIP_HPP

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

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

PYTHONIC_NS_BEGIN

namespace numpy
{
  namespace details
  {
    template <class E, class S, size_t... I>
    auto flip(E const &expr, S const &slices, utils::index_sequence<I...>)
        -> decltype(expr(slices[I]...))
    {
      return expr(slices[I]...);
    }
  }

  template <class E>
  auto flip(E const &expr, long axis)
      -> decltype(details::flip(expr, std::array<types::slice, E::value>{},
                                utils::make_index_sequence<E::value>{}))
  {
    std::array<types::slice, E::value> slices;
    slices[axis].step = -1;
    return details::flip(expr, slices, utils::make_index_sequence<E::value>{});
  }
}
PYTHONIC_NS_END

#endif