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
|
// -*- c++ -*-
%module SuperTensor
%{
#define SWIG_FILE_WITH_INIT
#include "SuperTensor.h"
%}
// Get the NumPy typemaps
%include "../numpy.i"
%init %{
import_array();
%}
%define %apply_numpy_typemaps(TYPE)
%apply (TYPE IN_ARRAY4[ANY][ANY][ANY][ANY]) {(TYPE supertensor[ANY][ANY][ANY][ANY])};
%apply (TYPE* IN_ARRAY4, int DIM1, int DIM2, int DIM3, int DIM4)
{(TYPE* supertensor, int cubes, int slices, int rows, int cols)};
%apply (int DIM1, int DIM2, int DIM3, int DIM4, TYPE* IN_ARRAY4)
{(int cubes, int slices, int rows, int cols, TYPE* supertensor)};
%apply (TYPE INPLACE_ARRAY4[ANY][ANY][ANY][ANY]) {(TYPE array[3][3][3][3])};
%apply (TYPE* INPLACE_ARRAY4, int DIM1, int DIM2, int DIM3, int DIM4)
{(TYPE* array, int cubes, int slices, int rows, int cols)};
%apply (int DIM1, int DIM2, int DIM3, int DIM4, TYPE* INPLACE_ARRAY4)
{(int cubes, int slices, int rows, int cols, TYPE* array)};
%apply (TYPE ARGOUT_ARRAY4[ANY][ANY][ANY][ANY]) {(TYPE lower[2][2][2][2])};
%apply (TYPE ARGOUT_ARRAY4[ANY][ANY][ANY][ANY]) {(TYPE upper[2][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 "SuperTensor.h"
|
