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
|
#ifndef PYTHONIC_NUMPY_APPEND_HPP
#define PYTHONIC_NUMPY_APPEND_HPP
#include "pythonic/include/numpy/append.hpp"
#include "pythonic/utils/functor.hpp"
#include "pythonic/types/ndarray.hpp"
#include "pythonic/numpy/asarray.hpp"
PYTHONIC_NS_BEGIN
namespace numpy
{
template <class T, class pS, class F>
typename std::enable_if<
!types::is_dtype<F>::value,
types::ndarray<
typename __combined<T, typename types::dtype_of<F>::type>::type,
types::pshape<long>>>::type
append(types::ndarray<T, pS> const &nto, F const &data)
{
auto ndata = numpy::functor::asarray{}(data);
long nsize = nto.flat_size() + ndata.flat_size();
types::ndarray<
typename __combined<T, typename types::dtype_of<F>::type>::type,
types::pshape<long>>
out(types::pshape<long>(nsize), builtins::None);
auto out_back = std::copy(nto.fbegin(), nto.fend(), out.fbegin());
std::copy(ndata.fbegin(), ndata.fend(), out_back);
return out;
}
template <class T, class pS, class F>
typename std::enable_if<
types::is_dtype<F>::value,
types::ndarray<
typename __combined<T, typename types::dtype_of<F>::type>::type,
types::pshape<long>>>::type
append(types::ndarray<T, pS> const &nto, F const &data)
{
long nsize = nto.flat_size() + 1;
types::ndarray<
typename __combined<T, typename types::dtype_of<F>::type>::type,
types::pshape<long>>
out(types::pshape<long>(nsize), builtins::None);
auto out_back = std::copy(nto.fbegin(), nto.fend(), out.fbegin());
*out_back = data;
return out;
}
template <class T, class F>
types::ndarray<typename __combined<typename types::dtype_of<T>::type,
typename types::dtype_of<F>::type>::type,
types::pshape<long>>
append(T const &to, F const &data)
{
return append(numpy::functor::asarray{}(to), data);
}
}
PYTHONIC_NS_END
#endif
|
