51 lines
1.7 KiB
C++
51 lines
1.7 KiB
C++
// -*- 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"
|
|
|