]> Creatis software - gdcm.git/blob - gdcmPython/gdcm.i
Example/TestReadWriteReadCompare may now be called with command line parameters
[gdcm.git] / gdcmPython / gdcm.i
1 %module gdcm
2 %{
3 #include "gdcmCommon.h"
4 #include "gdcmDictEntry.h"
5 #include "gdcmDict.h"
6 #include "gdcmDictSet.h"
7 #include "gdcmDocEntrySet.h"
8 #include "gdcmSQItem.h"
9 #include "gdcmDocument.h"
10 #include "gdcmElementSet.h"
11 #include "gdcmHeader.h"
12 #include "gdcmHeaderHelper.h"
13 #include "gdcmFile.h"
14 #include "gdcmUtil.h"
15 #include "gdcmGlobal.h"
16 #include "gdcmObject.h"
17 #include "gdcmDicomDir.h"
18 #include "gdcmDicomDirElement.h"
19 #include "gdcmDicomDirMeta.h"
20 #include "gdcmDicomDirPatient.h"
21 #include "gdcmDicomDirStudy.h"
22 #include "gdcmDicomDirSerie.h"
23 #include "gdcmDicomDirImage.h"
24 #include "gdcmValEntry.h"
25
26 ////////////////////////////////////////////////////////////////////////////
27 // Utility functions on strings for removing leading and trailing spaces
28 void EatLeadingAndTrailingSpaces(std::string & s) {
29         while ( s.length() && (s[0] == ' ') )
30                 s.erase(0,1);
31         while ( s.length() && (s[s.length()-1] == ' ') )
32                 s.erase(s.length()-1, 1);
33 }
34
35 void gdcmPythonVoidFunc(void *arg)
36 {
37   PyObject *arglist, *result;
38   PyObject *func = (PyObject *)arg;
39
40   arglist = Py_BuildValue("()");
41
42   result = PyEval_CallObject(func, arglist);
43   Py_DECREF(arglist);
44
45   if (result)
46     {
47     Py_XDECREF(result);
48     }
49   else
50     {
51     if (PyErr_ExceptionMatches(PyExc_KeyboardInterrupt))
52       {
53       std::cerr << "Caught a Ctrl-C within python, exiting program.\n";
54       Py_Exit(1);
55       }
56     PyErr_Print();
57     }
58 }
59
60 void gdcmPythonVoidFuncArgDelete(void *arg)
61 {
62   PyObject *func = (PyObject *)arg;
63   if (func)
64     {
65     Py_DECREF(func);
66     }
67 }
68
69 %}
70 typedef  unsigned short guint16;
71 typedef  unsigned int guint32;
72
73 ////////////////////////////////////////////////////////////////////////////
74 %typemap(out) std::list<std::string> * {
75    PyObject* NewItem = (PyObject*)0;
76    PyObject* NewList = PyList_New(0); // The result of this typemap
77
78    for (std::list<std::string>::iterator NewString = ($1)->begin();
79         NewString != ($1)->end();
80         ++NewString)
81    {
82       NewItem = PyString_FromString(NewString->c_str());
83       PyList_Append( NewList, NewItem);
84    }
85    $result = NewList;
86 }
87
88 ////////////////////////////////////////////////////////////////////////////
89 // Convert a c++ hash table in a python native dictionary
90 %typemap(out) std::map<std::string, std::list<std::string> > * {
91    PyObject* NewDict = PyDict_New(); // The result of this typemap
92    PyObject* NewKey = (PyObject*)0;
93    PyObject* NewVal = (PyObject*)0;
94
95    for (std::map<std::string,
96         std::list<std::string> >::iterator tag = ($1)->begin();
97         tag != ($1)->end(); ++tag)
98    {
99       std::string first = tag->first;
100       // Do not publish entries whose keys is made of spaces
101       if (first.length() == 0)
102          continue;
103       NewKey = PyString_FromString(first.c_str());
104       PyObject* NewList = PyList_New(0);
105       for (std::list<std::string>::iterator Item = tag->second.begin();
106            Item != tag->second.end();
107            ++Item)
108       {
109          NewVal = PyString_FromString(Item->c_str());
110          PyList_Append( NewList, NewVal);
111       }
112       PyDict_SetItem( NewDict, NewKey, NewList);
113    }
114    $result = NewDict;
115 }
116
117 ////////////////////////////////////////////////////////////////////////////
118 // Convert a c++ hash table in a python native dictionary
119 %typemap(out) TagDocEntryHT & {
120    PyObject* NewDict = PyDict_New(); // The result of this typemap
121    std::string RawName;              // Element name as gotten from gdcm
122    PyObject* NewKey = (PyObject*)0;  // Associated name as python object
123    std::string RawValue;             // Element value as gotten from gdcm
124    PyObject* NewVal = (PyObject*)0;  // Associated value as python object
125
126    for (TagDocEntryHT::iterator tag = $1->begin(); tag != $1->end(); ++tag)
127    {
128       // The element name shall be the key:
129       RawName = tag->second->GetName();
130       // gdcm unrecognized (including not loaded because their size exceeds
131       // the user specified treshold) elements are exported with their
132       // TagKey as key.
133       if (RawName == "Unknown")
134          RawName = tag->second->GetKey();
135       NewKey = PyString_FromString(RawName.c_str());
136
137       // Element values are striped from leading/trailing spaces
138       if (gdcmValEntry* ValEntryPtr =
139                 dynamic_cast< gdcmValEntry* >(tag->second) )
140       {
141          RawValue = ValEntryPtr->GetValue();
142       }
143       else
144         continue; 
145       EatLeadingAndTrailingSpaces(RawValue);
146       NewVal = PyString_FromString(RawValue.c_str());
147       PyDict_SetItem( NewDict, NewKey, NewVal);
148    }
149    $result = NewDict;
150 }
151
152 ////////////////////////////////////////////////////////////////////////////
153 %typemap(out) ListDicomDirPatient & {
154         PyObject* NewItem = (PyObject*)0;
155         $result = PyList_New(0); // The result of this typemap
156
157         for (std::list<gdcmDicomDirPatient *>::iterator New = ($1)->begin();
158             New != ($1)->end(); ++New) {
159                 NewItem = SWIG_NewPointerObj(*New,SWIGTYPE_p_gdcmDicomDirPatient,1);
160                 PyList_Append($result, NewItem);
161         }
162 }
163
164 %typemap(out) ListDicomDirStudy & {
165         PyObject* NewItem = (PyObject*)0;
166         $result = PyList_New(0); // The result of this typemap
167
168         for (std::list<gdcmDicomDirStudy *>::iterator New = ($1)->begin();
169             New != ($1)->end(); ++New) {
170                 NewItem = SWIG_NewPointerObj(*New,SWIGTYPE_p_gdcmDicomDirStudy,1);
171                 PyList_Append($result, NewItem);
172         }
173 }
174
175 %typemap(out) ListDicomDirSerie & {
176         PyObject* NewItem = (PyObject*)0;
177         $result = PyList_New(0); // The result of this typemap
178
179         for (std::list<gdcmDicomDirSerie *>::iterator New = ($1)->begin();
180             New != ($1)->end(); ++New) {
181                 NewItem = SWIG_NewPointerObj(*New,SWIGTYPE_p_gdcmDicomDirSerie,1);
182                 PyList_Append($result, NewItem);
183         }
184 }
185
186 %typemap(out) ListDicomDirImage & {
187         PyObject* NewItem = (PyObject*)0;
188         $result = PyList_New(0); // The result of this typemap
189
190         for (std::list<gdcmDicomDirImage *>::iterator New = ($1)->begin();
191             New != ($1)->end(); ++New) {
192                 NewItem = SWIG_NewPointerObj(*New,SWIGTYPE_p_gdcmDicomDirImage,1);
193                 PyList_Append($result, NewItem);
194         }
195 }
196
197 ////////////////////////////////////////////////////////////////////////////
198 // Deals with function returning a C++ string.
199 %typemap(python, in) (gdcmMethod *,void * =NULL,gdcmMethod * =NULL) {
200         if($input!=Py_None)
201         {
202                 Py_INCREF($input);
203                 $1=gdcmPythonVoidFunc;
204                 $2=$input;
205                 $3=gdcmPythonVoidFuncArgDelete;
206         }
207         else
208         {
209                 $1=NULL;
210                 $2=NULL;
211                 $3=NULL;
212         }
213 }
214
215
216 ////////////////////////////////////////////////////////////////////////////
217 // Deals with function returning a C++ string.
218 %typemap(out) string, std::string  {
219     $result = PyString_FromString(($1).c_str());
220 }
221
222 %typemap(python, in) const std::string, std::string
223 {
224   $1 = PyString_AsString($input);
225 }
226
227 ////////////////////////////////////////////////////////////////////////////
228 // Warning: Order matters !
229 %include "gdcmCommon.h"
230 %include "gdcmDictEntry.h"
231 %include "gdcmDict.h"
232 %include "gdcmDictSet.h"
233 %include "gdcmDocEntrySet.h"
234 %include "gdcmElementSet.h"
235 %include "gdcmDocument.h"
236 %include "gdcmSQItem.h"
237 %include "gdcmHeader.h"
238 %include "gdcmHeaderHelper.h"
239 %include "gdcmFile.h"
240 %include "gdcmUtil.h"
241 %include "gdcmGlobal.h"
242 %include "gdcmObject.h"
243 %include "gdcmDicomDir.h"
244 %include "gdcmDicomDirElement.h"
245 %include "gdcmDicomDirMeta.h"
246 %include "gdcmDicomDirPatient.h"
247 %include "gdcmDicomDirStudy.h"
248 %include "gdcmDicomDirSerie.h"
249 %include "gdcmDicomDirImage.h"