[swig] Interfacer "const MyClass& obj" avec SWIG

Bonjour,

Voici mon problème, j'ai une classe Matrix qui encapsule un tableau 1D et qui possède un constructeur comme :

Code:

---

1 2 3 4 5 6 7 8

template<typename T> class Matrix { ... public: Matrix(int nrow, int ncol, double* data); ... }

---

et des fonctions qui utilisent cette classe :

Code:

---

1 2	double quelque_chose(const Matrix& mat);

---

et j'aimerais créer un typemap pour pouvoir directement utiliser les tableaux numpy. En m'inspirant des typemaps de numpy je suis parvenu à :

Code:

---

1 2 3 4 5 6 7 8 9 10 11 12 13 14

%typemap(in, fragment="NumPy_Fragments") const Matrix< double >& (PyArrayObject* array=NULL, int is_new_object=0,Matrix<double> *new_mat=0 ) { npy_intp size[2] = { -1, -1 }; array = obj_to_array_contiguous_allow_conversion($input, NPY_DOUBLE, &is_new_object); if (!array || !require_dimensions(array, 2) || !require_size(array, size, 2)) SWIG_fail; new_mat = new Matrix<double>((size_t) array_size(array,0),(size_t) array_size(array,1), (double*) array_data(array)); $1 = new_mat; }

---

mais il y a deux problèmes (au moins):

1. la matrice créée n'est pas détruite (fuite de mémoire)
2. je suis obligé de créer un typemap par type de données

Toute solution, piste, conseil seront les bienvenus.

D'avance merci,

Alexis.M

J'ai en fait trouvé une partie de la solution tout seul. Il suffit d'utiliser %freearg (oui, je sais, c'est fait pour ça) et d'utiliser $argnum pour avoir le bon nom de variable :

Code:

---

1 2 3 4

%typemap(freearg) const Matrix< double >& { if (new_mat$argnum) delete new_mat$argnum; }

---

et ça a le résultat attendu avec des fontions de type :

Code:

---

1 2	double function(const Matrix< double >&m1, const Matrix< double >& m2);

---

Pour en finir avec le problème existe-t-il une macro qui permettrait de prendre en argument un type de donnée et retournerait son code numpy ?

Voilà...

Finalement, en copiant la manière de faire dans numpy.i, je suis arrivé au code suivant:

Code:

---

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

%define %Matrix_typemaps(DATA_TYPE,DATA_TYPECODE)   %typecheck(SWIG_TYPECHECK_DOUBLE_ARRAY, fragment="NumPy_Macros") const Matrix< DATA_TYPE >& { $1 = is_array($input) || PySequence_Check($input); }   %typemap(in, fragment="NumPy_Fragments") const Matrix<DATA_TYPE>& (PyArrayObject* array=NULL, int is_new_object=0,Matrix<DATA_TYPE>* new_mat=0 ) { npy_intp size[2] = { -1, -1 }; array = obj_to_array_contiguous_allow_conversion($input, DATA_TYPECODE, &is_new_object); if (!array || !require_dimensions(array, 2) || !require_size(array, size, 2)) SWIG_fail; new_mat = new Matrix<DATA_TYPE>((size_t) array_size(array,0),(size_t) array_size(array,1), (DATA_TYPE*) array_data(array)); $1 = new_mat; }     %typemap(freearg) const Matrix< DATA_TYPE >& { if (new_mat$argnum) delete new_mat$argnum; } %enddef // %Matrix_typemaps() // Concrete instanciations of the macro %Matrix_typemaps()   %Matrix_typemaps(signed char , NPY_BYTE ) %Matrix_typemaps(unsigned char , NPY_UBYTE ) %Matrix_typemaps(short , NPY_SHORT ) %Matrix_typemaps(unsigned short , NPY_USHORT ) %Matrix_typemaps(int , NPY_INT ) %Matrix_typemaps(unsigned int , NPY_UINT ) %Matrix_typemaps(long , NPY_LONG ) %Matrix_typemaps(unsigned long , NPY_ULONG ) %Matrix_typemaps(long long , NPY_LONGLONG ) %Matrix_typemaps(unsigned long long, NPY_ULONGLONG) %Matrix_typemaps(float , NPY_FLOAT ) %Matrix_typemaps(double , NPY_DOUBLE )

---

Ceci semble fonctionner quoique je je ne sache pas exactement ce que font toutes les fonctions. Quoiqu'il en soit je dois mériter un RTFM puisque j'ai répondu à mes propres questions...

Je me pose toutefois des questions concernant:

[/CODE]
%typecheck(SWIG_TYPECHECK_DOUBLE_ARRAY,
fragment="NumPy_Macros")
[/CODE]

qui semble vérifier qu'il s'agit d'un tableau de double. Ce code est celui trouvé dans numpy.i , que fait-il exactemement ?