summaryrefslogtreecommitdiffstats
path: root/contrib/python/pythran/pythran/pythonic/include/numpy/interp.hpp
blob: 320bff15c2cf8bbbfa730c9f2c39dbb257e5e3a2 (plain) (blame)
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
#ifndef PYTHONIC_INCLUDE_NUMPY_INTERP_HPP
#define PYTHONIC_INCLUDE_NUMPY_INTERP_HPP

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

PYTHONIC_NS_BEGIN
namespace numpy
{
  template <class T>
  using interp_out_type =
      typename std::conditional<types::is_complex<typename T::dtype>::value,
                                std::complex<double>, double>::type;

  // None,None,None
  template <class T1, class T2, class T3>
  typename std::enable_if<
      !std::is_arithmetic<T1>::value,
      types::ndarray<interp_out_type<T3>, types::pshape<long>>>::type
  interp(T1 x, T2 xp, T3 fp, types::none_type left = types::none_type{},
         types::none_type right = types::none_type{},
         types::none_type period = types::none_type{});

  // left None None
  template <class T1, class T2, class T3, typename t1>
  typename std::enable_if<
      !std::is_arithmetic<T1>::value,
      types::ndarray<interp_out_type<T3>, types::pshape<long>>>::type
  interp(T1 x, T2 xp, T3 fp, t1 left,
         types::none_type right = types::none_type{},
         types::none_type period = types::none_type{});

  // None right None
  template <class T1, class T2, class T3, typename t1>
  typename std::enable_if<
      !std::is_arithmetic<T1>::value,
      types::ndarray<interp_out_type<T3>, types::pshape<long>>>::type
  interp(T1 x, T2 xp, T3 fp, types::none_type left, t1 right,
         types::none_type period = types::none_type{});
  // None None period
  template <class T1, class T2, class T3, typename t1>
  typename std::enable_if<
      !std::is_arithmetic<T1>::value,
      types::ndarray<interp_out_type<T3>, types::pshape<long>>>::type
  interp(T1 x, T2 xp, T3 fp, types::none_type left, types::none_type right,
         t1 period);

  // left right None
  template <class T1, class T2, class T3, typename t1, typename t2>
  typename std::enable_if<
      !std::is_arithmetic<T1>::value,
      types::ndarray<interp_out_type<T3>, types::pshape<long>>>::type
  interp(T1 x, T2 xp, T3 fp, t1 left, t2 right,
         types::none_type period = types::none_type{});

  ////////////////////////// NUMERIC TYPES for x.
  template <class T1, class T2, class T3>
  typename std::enable_if<std::is_arithmetic<T1>::value,
                          interp_out_type<T3>>::type
  interp(T1 x, T2 xp, T3 fp, types::none_type left = types::none_type{},
         types::none_type right = types::none_type{},
         types::none_type period = types::none_type{});

  // left None None
  template <class T1, class T2, class T3, typename t1>
  typename std::enable_if<std::is_arithmetic<T1>::value,
                          interp_out_type<T3>>::type
  interp(T1 x, T2 xp, T3 fp, t1 left,
         types::none_type right = types::none_type{},
         types::none_type period = types::none_type{});

  // None right None
  template <class T1, class T2, class T3, typename t1>
  typename std::enable_if<std::is_arithmetic<T1>::value,
                          interp_out_type<T3>>::type
  interp(T1 x, T2 xp, T3 fp, types::none_type left, t1 right,
         types::none_type period = types::none_type{});

  // None None period
  template <class T1, class T2, class T3, typename t1>
  typename std::enable_if<std::is_arithmetic<T1>::value,
                          interp_out_type<T3>>::type
  interp(T1 x, T2 xp, T3 fp, types::none_type left, types::none_type right,
         t1 period);

  // left right None
  template <class T1, class T2, class T3, typename t1, typename t2>
  typename std::enable_if<std::is_arithmetic<T1>::value,
                          interp_out_type<T3>>::type
  interp(T1 x, T2 xp, T3 fp, t1 left, t2 right,
         types::none_type period = types::none_type{});

  NUMPY_EXPR_TO_NDARRAY0_DECL(interp);
  DEFINE_FUNCTOR(pythonic::numpy, interp);
} // namespace numpy
PYTHONIC_NS_END

#endif