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
|
// -*- c++ -*-
%module Tensor
%{
#define SWIG_FILE_WITH_INIT
#include "Tensor.h"
%}
// Get the NumPy typemaps
%include "../numpy.i"
%init %{
import_array();
%}
%define %apply_numpy_typemaps(TYPE)
%apply (TYPE IN_ARRAY3[ANY][ANY][ANY]) {(TYPE tensor[ANY][ANY][ANY])};
%apply (TYPE* IN_ARRAY3, int DIM1, int DIM2, int DIM3)
{(TYPE* tensor, int rows, int cols, int num)};
%apply (int DIM1, int DIM2, int DIM3, TYPE* IN_ARRAY3)
{(int rows, int cols, int num, TYPE* tensor)};
%apply (TYPE INPLACE_ARRAY3[ANY][ANY][ANY]) {(TYPE array[3][3][3])};
%apply (TYPE* INPLACE_ARRAY3, int DIM1, int DIM2, int DIM3)
{(TYPE* array, int rows, int cols, int num)};
%apply (int DIM1, int DIM2, int DIM3, TYPE* INPLACE_ARRAY3)
{(int rows, int cols, int num, TYPE* array)};
%apply (TYPE ARGOUT_ARRAY3[ANY][ANY][ANY]) {(TYPE lower[2][2][2])};
%apply (TYPE ARGOUT_ARRAY3[ANY][ANY][ANY]) {(TYPE upper[2][2][2])};
%enddef /* %apply_numpy_typemaps() macro */
%apply_numpy_typemaps(signed char )
%apply_numpy_typemaps(unsigned char )
%apply_numpy_typemaps(short )
%apply_numpy_typemaps(unsigned short )
%apply_numpy_typemaps(int )
%apply_numpy_typemaps(unsigned int )
%apply_numpy_typemaps(long )
%apply_numpy_typemaps(unsigned long )
%apply_numpy_typemaps(long long )
%apply_numpy_typemaps(unsigned long long)
%apply_numpy_typemaps(float )
%apply_numpy_typemaps(double )
// Include the header file to be wrapped
%include "Tensor.h"
|
