Mercurial > hg > orthanc-python
annotate CodeAnalysis/ParseOrthancSDK.py @ 63:32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Thu, 10 Jun 2021 18:19:27 +0200 |
parents | 23f3099bed47 |
children | 091fb1903bfc |
rev | line source |
---|---|
0 | 1 #!/usr/bin/env python |
2 | |
3 ## | |
4 ## Python plugin for Orthanc | |
56
23f3099bed47
upgrade to year 2021
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
31
diff
changeset
|
5 ## Copyright (C) 2020-2021 Osimis S.A., Belgium |
0 | 6 ## |
7 ## This program is free software: you can redistribute it and/or | |
8 ## modify it under the terms of the GNU Affero General Public License | |
9 ## as published by the Free Software Foundation, either version 3 of | |
10 ## the License, or (at your option) any later version. | |
11 ## | |
12 ## This program is distributed in the hope that it will be useful, but | |
13 ## WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 ## Affero General Public License for more details. | |
16 ## | |
17 ## You should have received a copy of the GNU Affero General Public License | |
18 ## along with this program. If not, see <http://www.gnu.org/licenses/>. | |
19 ## | |
20 | |
21 | |
22 import argparse | |
23 import clang.cindex | |
24 import os | |
25 import pprint | |
26 import pystache | |
27 import sys | |
28 | |
29 | |
30 ROOT = os.path.dirname(os.path.realpath(sys.argv[0])) | |
31 | |
32 | |
33 ## | |
63
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
34 ## Configuration of the custom primitives that are manually |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
35 ## implemented (not autogenerated) |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
36 ## |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
37 |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
38 CUSTOM_FUNCTIONS = { |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
39 'OrthancPluginCreateDicom', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
40 'OrthancPluginFreeMemoryBuffer', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
41 'OrthancPluginFreeString', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
42 'OrthancPluginGetFindQueryTag', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
43 'OrthancPluginRegisterFindCallback', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
44 'OrthancPluginRegisterIncomingHttpRequestFilter', # Implemented through v2 |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
45 'OrthancPluginRegisterIncomingHttpRequestFilter2', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
46 'OrthancPluginRegisterMoveCallback', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
47 'OrthancPluginRegisterOnChangeCallback', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
48 'OrthancPluginRegisterOnStoredInstanceCallback', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
49 'OrthancPluginRegisterRestCallback', # Implemented using OrthancPlugins::RegisterRestCallback |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
50 'OrthancPluginRegisterRestCallbackNoLock', # Implemented using OrthancPlugins::RegisterRestCallback |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
51 } |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
52 |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
53 CUSTOM_METHODS = [ |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
54 { |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
55 'class_name' : 'OrthancPluginFindQuery', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
56 'method_name' : 'GetFindQueryTagGroup', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
57 'implementation' : 'GetFindQueryTagGroup', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
58 'sdk_function' : 'OrthancPluginGetFindQueryTag', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
59 }, |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
60 { |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
61 'class_name' : 'OrthancPluginFindQuery', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
62 'method_name' : 'GetFindQueryTagElement', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
63 'implementation' : 'GetFindQueryTagElement', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
64 'sdk_function' : 'OrthancPluginGetFindQueryTag', |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
65 }, |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
66 ] |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
67 |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
68 |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
69 ## |
0 | 70 ## Parse the command-line arguments |
71 ## | |
72 | |
73 parser = argparse.ArgumentParser(description = 'Parse the Orthanc SDK.') | |
74 parser.add_argument('--libclang', | |
75 default = 'libclang-4.0.so.1', | |
76 help = 'manually provides the path to the libclang shared library') | |
77 parser.add_argument('--source', | |
78 default = os.path.join(os.path.dirname(__file__), | |
63
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
79 '../Resources/Orthanc/Sdk-1.8.1/orthanc/OrthancCPlugin.h'), |
0 | 80 help = 'Input C++ file') |
81 parser.add_argument('--target', | |
82 default = os.path.join(os.path.dirname(__file__), | |
83 '../Sources/Autogenerated'), | |
84 help = 'Target folder') | |
85 | |
86 args = parser.parse_args() | |
87 | |
88 | |
89 | |
90 if len(args.libclang) != 0: | |
91 clang.cindex.Config.set_library_file(args.libclang) | |
92 | |
93 index = clang.cindex.Index.create() | |
94 | |
95 tu = index.parse(args.source, [ ]) | |
96 | |
97 TARGET = os.path.realpath(args.target) | |
98 | |
99 | |
100 | |
101 def ToUpperCase(name): | |
102 s = '' | |
103 for i in range(len(name)): | |
104 if name[i].isupper(): | |
105 if len(s) == 0: | |
106 s += name[i] | |
107 elif name[i - 1].islower(): | |
108 s += '_' + name[i] | |
109 elif (i + 1 < len(name) and | |
110 name[i - 1].islower() and | |
111 name[i + 1].isupper()): | |
112 s += '_' + name[i] | |
113 else: | |
114 s += name[i] | |
115 else: | |
116 s += name[i].upper() | |
117 return s | |
118 | |
119 | |
120 | |
121 with open(os.path.join(ROOT, 'Enumeration.mustache'), 'r') as f: | |
122 TEMPLATE = f.read() | |
123 | |
124 | |
125 classes = {} | |
126 enumerations = {} | |
127 globalFunctions = [] | |
2 | 128 countAllFunctions = 0 |
129 countSupportedFunctions = 0 | |
0 | 130 |
131 def IsSourceStringType(t): | |
132 return (t.kind == clang.cindex.TypeKind.POINTER and | |
133 t.get_pointee().kind == clang.cindex.TypeKind.CHAR_S and | |
134 t.get_pointee().is_const_qualified()) | |
135 | |
136 def IsTargetStaticStringType(t): | |
137 return (t.kind == clang.cindex.TypeKind.POINTER and | |
138 t.get_pointee().kind == clang.cindex.TypeKind.CHAR_S and | |
139 t.get_pointee().is_const_qualified()) | |
140 | |
141 def IsTargetDynamicStringType(t): | |
142 return (t.kind == clang.cindex.TypeKind.POINTER and | |
143 t.get_pointee().kind == clang.cindex.TypeKind.CHAR_S and | |
144 not t.get_pointee().is_const_qualified()) | |
145 | |
146 def IsIntegerType(t): | |
147 return (t.kind == clang.cindex.TypeKind.INT or | |
148 t.spelling in [ 'int8_t', 'int16_t', 'int32_t', 'int64_t', | |
149 'uint8_t', 'uint16_t', 'uint32_t', 'uint64_t']) | |
150 | |
151 def IsFloatType(t): | |
152 return t.kind == clang.cindex.TypeKind.FLOAT | |
153 | |
154 def IsEnumerationType(t): | |
155 return (t.kind == clang.cindex.TypeKind.TYPEDEF and | |
156 t.spelling in enumerations) | |
157 | |
158 def IsTargetMemoryBufferType(t): | |
159 return (t.kind == clang.cindex.TypeKind.POINTER and | |
160 not t.get_pointee().is_const_qualified() and | |
161 t.get_pointee().spelling == 'OrthancPluginMemoryBuffer') | |
162 | |
163 def IsSourceMemoryBufferType(t): | |
164 return (t.kind == clang.cindex.TypeKind.POINTER and | |
165 t.get_pointee().kind == clang.cindex.TypeKind.VOID and | |
166 t.get_pointee().is_const_qualified()) | |
167 | |
168 def IsClassType(t): | |
169 return (t.kind == clang.cindex.TypeKind.POINTER and | |
170 ((t.get_pointee().is_const_qualified() and | |
171 t.get_pointee().spelling.startswith('const ') and | |
172 t.get_pointee().spelling[len('const '):] in classes) or | |
173 (not t.get_pointee().is_const_qualified() and | |
174 t.get_pointee().spelling in classes))) | |
175 | |
176 def IsSimpleSourceType(t): | |
177 return (IsSourceStringType(t) or | |
178 IsFloatType(t) or | |
179 IsIntegerType(t) or | |
180 IsEnumerationType(t) or | |
181 IsSourceMemoryBufferType(t)) | |
182 | |
183 def IsVoidType(t): | |
184 return t.kind == clang.cindex.TypeKind.VOID | |
185 | |
186 def IsSupportedTargetType(t): | |
187 return (IsVoidType(t) or | |
188 IsIntegerType(t) or | |
189 IsEnumerationType(t) or | |
190 # Constructor of a class | |
191 (t.kind == clang.cindex.TypeKind.POINTER and | |
192 not t.get_pointee().is_const_qualified() and | |
193 t.get_pointee().spelling in classes) or | |
194 # "const char*" or "char*" outputs | |
195 (t.kind == clang.cindex.TypeKind.POINTER and | |
196 #not t.get_pointee().is_const_qualified() and | |
197 t.get_pointee().kind == clang.cindex.TypeKind.CHAR_S)) | |
198 | |
199 def IsBytesArgument(args, index): | |
200 return (index + 1 < len(args) and | |
201 args[index].type.kind == clang.cindex.TypeKind.POINTER and | |
202 args[index].type.get_pointee().kind == clang.cindex.TypeKind.VOID and | |
203 args[index].type.get_pointee().is_const_qualified() and | |
204 args[index + 1].type.spelling == 'uint32_t') | |
205 | |
206 def CheckOnlySupportedArguments(args): | |
207 j = 0 | |
208 while j < len(args): | |
209 if IsBytesArgument(args, j): | |
210 j += 2 | |
211 elif IsSimpleSourceType(args[j].type): | |
212 j += 1 | |
213 else: | |
214 return False | |
215 return True | |
216 | |
217 | |
218 ORTHANC_TO_PYTHON_NUMERIC_TYPES = { | |
219 # https://docs.python.org/3/c-api/arg.html#numbers | |
220 'int' : { | |
221 'type' : 'int', | |
222 'format' : 'i', | |
223 }, | |
224 'uint8_t' : { | |
225 'type' : 'unsigned char', | |
226 'format' : 'b', | |
227 }, | |
228 'int32_t' : { | |
229 'type' : 'long int', | |
230 'format' : 'l', | |
231 }, | |
232 'uint16_t' : { | |
233 'type' : 'unsigned short', | |
234 'format' : 'H', | |
235 }, | |
236 'uint32_t' : { | |
237 'type' : 'unsigned long', | |
238 'format' : 'k', | |
239 }, | |
240 'uint64_t' : { | |
241 'type' : 'unsigned long long', | |
242 'format' : 'K', | |
243 }, | |
244 'float' : { | |
245 'type' : 'float', | |
246 'format' : 'f', | |
247 } | |
248 } | |
249 | |
250 | |
251 def GenerateFunctionBodyTemplate(cFunction, result_type, args): | |
252 if not cFunction.startswith('OrthancPlugin'): | |
253 raise Exception() | |
254 | |
255 func = { | |
256 'c_function' : cFunction, | |
257 'short_name' : cFunction[len('OrthancPlugin'):], | |
258 'args' : [], | |
259 } | |
260 | |
261 if IsIntegerType(result_type): | |
262 func['return_long'] = True | |
263 elif IsTargetDynamicStringType(result_type): | |
264 func['return_dynamic_string'] = True | |
265 elif IsTargetStaticStringType(result_type): | |
266 func['return_static_string'] = True | |
267 elif IsVoidType(result_type): | |
268 func['return_void'] = True | |
269 elif result_type.spelling == 'OrthancPluginErrorCode': | |
270 func['return_error'] = True | |
271 elif IsClassType(result_type): | |
272 func['return_object'] = result_type.get_pointee().spelling | |
273 elif IsTargetMemoryBufferType(result_type): | |
274 func['return_bytes'] = True | |
275 elif IsEnumerationType(result_type): | |
276 func['return_enumeration'] = result_type.spelling | |
277 else: | |
278 raise Exception('Not supported: %s' % result_type.spelling) | |
279 | |
280 i = 0 | |
281 while i < len(args): | |
282 a = { | |
283 'name' : 'arg%d' % i, | |
284 } | |
285 | |
286 if (IsIntegerType(args[i].type) or | |
287 IsFloatType(args[i].type)): | |
288 t = ORTHANC_TO_PYTHON_NUMERIC_TYPES[args[i].type.spelling] | |
289 a['python_type'] = t['type'] | |
290 a['python_format'] = t['format'] | |
291 a['initialization'] = ' = 0' | |
292 a['orthanc_cast'] = 'arg%d' % i | |
293 func['args'].append(a) | |
294 elif IsSourceStringType(args[i].type): | |
295 a['python_type'] = 'const char*' | |
296 a['python_format'] = 's' | |
297 a['initialization'] = ' = NULL' | |
298 a['orthanc_cast'] = 'arg%d' % i | |
299 func['args'].append(a) | |
300 elif IsEnumerationType(args[i].type): | |
301 a['python_type'] = 'long int' | |
302 a['python_format'] = 'l' | |
303 a['initialization'] = ' = 0' | |
304 a['orthanc_cast'] = 'static_cast<%s>(arg%d)' % (args[i].type.spelling, i) | |
305 func['args'].append(a) | |
306 elif IsBytesArgument(args, i): | |
307 a['python_type'] = 'Py_buffer' | |
308 # In theory, one should use "y*" (this is the recommended | |
309 # way to accept binary data). However, this is not | |
310 # available in Python 2.7 | |
311 a['python_format'] = 's*' | |
312 a['orthanc_cast'] = 'arg%d.buf, arg%d.len' % (i, i) | |
313 a['release'] = 'PyBuffer_Release(&arg%d);' % i | |
314 func['args'].append(a) | |
315 i += 1 | |
316 elif IsSourceMemoryBufferType(args[i].type): | |
317 a['python_type'] = 'Py_buffer' | |
318 a['python_format'] = 's*' | |
319 a['orthanc_cast'] = 'arg%d.buf' % i | |
320 a['release'] = 'PyBuffer_Release(&arg%d);' % i | |
321 func['args'].append(a) | |
322 else: | |
323 raise Exception('Not supported: %s, %s' % (cFunction, args[i].spelling)) | |
324 | |
325 i += 1 | |
326 | |
327 func['tuple_format'] = '"%s", %s' % ( | |
328 ''.join(map(lambda x: x['python_format'], func['args'])), | |
329 ', '.join(map(lambda x: '&' + x['name'], func['args']))) | |
330 | |
331 if len(func['args']) > 0: | |
332 func['count_args'] = len(func['args']) | |
333 func['has_args'] = True | |
334 func['call_args'] = ', ' + ', '.join(map(lambda x: x['orthanc_cast'], func['args'])) | |
335 | |
336 return func | |
337 | |
338 | |
339 for node in tu.cursor.get_children(): | |
340 if node.kind == clang.cindex.CursorKind.ENUM_DECL: | |
341 if node.type.spelling.startswith('OrthancPlugin'): | |
342 name = node.type.spelling | |
343 | |
344 values = [] | |
345 for item in node.get_children(): | |
346 if (item.kind == clang.cindex.CursorKind.ENUM_CONSTANT_DECL and | |
347 item.spelling.startswith(name + '_')): | |
348 values.append({ | |
349 'key' : ToUpperCase(item.spelling[len(name)+1:]), | |
350 'value' : item.enum_value | |
351 }) | |
352 | |
353 path = 'sdk_%s.impl.h' % name | |
354 shortName = name[len('OrthancPlugin'):] | |
355 | |
356 with open(os.path.join(TARGET, path), 'w') as f: | |
357 f.write(pystache.render(TEMPLATE, { | |
358 'name' : name, | |
359 'short_name' : shortName, | |
360 'values' : values, | |
361 })) | |
362 | |
363 enumerations[name] = { | |
364 'name' : name, | |
365 'path' : path, | |
366 } | |
367 | |
368 elif node.kind == clang.cindex.CursorKind.FUNCTION_DECL: | |
369 if node.spelling.startswith('OrthancPlugin'): | |
370 #if node.spelling != 'OrthancPluginWorklistGetDicomQuery': | |
371 # continue | |
372 shortName = node.spelling[len('OrthancPlugin'):] | |
373 | |
374 # Check that the first argument is the Orthanc context | |
375 args = list(filter(lambda x: x.kind == clang.cindex.CursorKind.PARM_DECL, | |
376 node.get_children())) | |
377 | |
378 if (len(args) == 0 or | |
379 args[0].type.kind != clang.cindex.TypeKind.POINTER or | |
380 args[0].type.get_pointee().spelling != 'OrthancPluginContext'): | |
381 print('Not in the Orthanc SDK: %s()' % node.spelling) | |
382 continue | |
383 | |
384 # Discard the context from the arguments | |
2 | 385 countAllFunctions += 1 |
0 | 386 args = args[1:] |
387 | |
388 if not IsSupportedTargetType(node.result_type): | |
389 print('*** UNSUPPORTED OUTPUT: %s' % node.spelling) | |
390 | |
391 elif (len(args) == 1 and | |
392 IsClassType(args[0].type) and | |
393 node.spelling.startswith('OrthancPluginFree')): | |
394 print('Destructor: %s' % node.spelling) | |
395 className = args[0].type.get_pointee().spelling | |
396 classes[className]['destructor'] = node.spelling | |
2 | 397 countSupportedFunctions += 1 |
0 | 398 |
399 elif CheckOnlySupportedArguments(args): | |
400 if IsClassType(node.result_type): | |
401 print('Constructor: %s' % node.spelling) | |
402 else: | |
403 print('Simple global function: %s => %s' % (node.spelling, node.result_type.spelling)) | |
404 | |
405 body = GenerateFunctionBodyTemplate(node.spelling, node.result_type, args) | |
406 globalFunctions.append(body) | |
2 | 407 countSupportedFunctions += 1 |
0 | 408 |
409 elif (len(args) >= 2 and | |
410 IsTargetMemoryBufferType(args[0].type) and | |
411 CheckOnlySupportedArguments(args[1:])): | |
412 print('Simple global function, returning bytes: %s' % node.spelling) | |
413 | |
414 body = GenerateFunctionBodyTemplate(node.spelling, args[0].type, args[1:]) | |
415 globalFunctions.append(body) | |
2 | 416 countSupportedFunctions += 1 |
0 | 417 |
418 elif (IsClassType(args[0].type) and | |
419 CheckOnlySupportedArguments(args[1:])): | |
420 className = args[0].type.get_pointee().spelling | |
421 | |
63
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
422 print('Simple method of class %s: %s' % (className, node.spelling)) |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
423 if node.spelling in CUSTOM_FUNCTIONS: |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
424 raise Exception('Cannot overwrite an autogenerated method: %s()' % node.spelling) |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
425 |
0 | 426 if className.startswith('const '): |
427 className = className[len('const '):] | |
428 | |
429 method = GenerateFunctionBodyTemplate(node.spelling, node.result_type, args[1:]) | |
430 method['self'] = ', self->object_' | |
431 classes[className]['methods'].append(method) | |
2 | 432 countSupportedFunctions += 1 |
0 | 433 |
434 elif (len(args) >= 2 and | |
435 IsTargetMemoryBufferType(args[0].type) and | |
436 IsClassType(args[1].type) and | |
437 CheckOnlySupportedArguments(args[2:])): | |
31 | 438 className = args[1].type.get_pointee().spelling |
439 | |
63
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
440 print('Simple method of class %s, returning bytes: %s' % (className, node.spelling)) |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
441 if node.spelling in CUSTOM_FUNCTIONS: |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
442 raise Exception('Cannot overwrite an autogenerated method: %s()' % node.spelling) |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
443 |
31 | 444 if className.startswith('const '): |
445 className = className[len('const '):] | |
446 | |
0 | 447 method = GenerateFunctionBodyTemplate(node.spelling, args[0].type, args[2:]) |
448 method['self'] = ', self->object_' | |
449 classes[className]['methods'].append(method) | |
2 | 450 countSupportedFunctions += 1 |
63
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
451 |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
452 elif node.spelling in CUSTOM_FUNCTIONS: |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
453 print('Ignoring custom function that is manually implemented: %s()' % node.spelling) |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
454 countSupportedFunctions += 1 |
0 | 455 |
456 else: | |
457 print('*** UNSUPPORTED INPUT: %s' % node.spelling) | |
458 | |
459 | |
460 | |
461 elif node.kind == clang.cindex.CursorKind.STRUCT_DECL: | |
462 if (node.spelling.startswith('_OrthancPlugin') and | |
463 node.spelling.endswith('_t') and | |
464 node.spelling != '_OrthancPluginContext_t'): | |
465 name = node.spelling[len('_') : -len('_t')] | |
466 classes[name] = { | |
467 'class_name' : name, | |
468 'short_name' : name[len('OrthancPlugin'):], | |
63
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
469 'methods' : [ ], |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
470 'custom_methods' : [ ], |
0 | 471 } |
472 | |
473 | |
474 | |
475 | |
476 partials = {} | |
477 | |
478 with open(os.path.join(ROOT, 'FunctionBody.mustache'), 'r') as f: | |
479 partials['function_body'] = f.read() | |
480 | |
481 renderer = pystache.Renderer( | |
482 escape = lambda u: u, # No escaping | |
483 partials = partials, | |
484 ) | |
485 | |
486 with open(os.path.join(ROOT, 'Class.mustache'), 'r') as f: | |
487 template = f.read() | |
488 | |
63
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
489 for method in CUSTOM_METHODS: |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
490 classes[method['class_name']]['custom_methods'].append(method) |
32de70a1e4c7
New functions from the SDK wrapped in Python: CreateDicom, RegisterFindCallback, RegisterMoveCallback
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
56
diff
changeset
|
491 |
0 | 492 for (key, value) in classes.items(): |
493 with open(os.path.join(TARGET, 'sdk_%s.impl.h' % value['class_name']), 'w') as h: | |
494 h.write(renderer.render(template, value)) | |
495 | |
496 | |
497 def FlattenDictionary(source): | |
498 result = [] | |
499 for (key, value) in source.items(): | |
500 result.append(value) | |
501 return result | |
502 | |
503 | |
504 with open(os.path.join(ROOT, 'GlobalFunctions.mustache'), 'r') as f: | |
505 with open(os.path.join(TARGET, 'sdk_GlobalFunctions.impl.h'), 'w') as h: | |
506 h.write(renderer.render(f.read(), { | |
507 'global_functions' : globalFunctions, | |
508 })) | |
509 | |
510 with open(os.path.join(ROOT, 'sdk.cpp.mustache'), 'r') as f: | |
511 with open(os.path.join(TARGET, 'sdk.cpp'), 'w') as h: | |
512 h.write(renderer.render(f.read(), { | |
513 'classes' : FlattenDictionary(classes), | |
514 'enumerations' : FlattenDictionary(enumerations), | |
515 'global_functions' : globalFunctions, | |
516 })) | |
517 | |
518 with open(os.path.join(ROOT, 'sdk.h.mustache'), 'r') as f: | |
519 with open(os.path.join(TARGET, 'sdk.h'), 'w') as h: | |
520 h.write(renderer.render(f.read(), { | |
521 'classes' : FlattenDictionary(classes), | |
522 })) | |
2 | 523 |
524 | |
525 print('') | |
526 print('Total functions in the SDK: %d' % countAllFunctions) | |
527 print('Total supported functions: %d' % countSupportedFunctions) | |
528 print('Coverage: %.0f%%' % (float(countSupportedFunctions) / | |
529 float(countAllFunctions) * 100.0)) |