Mercurial > hg > orthanc
annotate Resources/EmbedResources.py @ 324:64925c94825c
api improvement
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Tue, 08 Jan 2013 09:29:56 +0100 |
parents | 052dede32761 |
children | 71d30b9ce801 |
rev | line source |
---|---|
143 | 1 # Orthanc - A Lightweight, RESTful DICOM Store |
2 # Copyright (C) 2012 Medical Physics Department, CHU of Liege, | |
3 # Belgium | |
4 # | |
5 # This program is free software: you can redistribute it and/or | |
6 # modify it under the terms of the GNU General Public License as | |
7 # published by the Free Software Foundation, either version 3 of the | |
8 # License, or (at your option) any later version. | |
9 # | |
10 # In addition, as a special exception, the copyright holders of this | |
11 # program give permission to link the code of its release with the | |
12 # OpenSSL project's "OpenSSL" library (or with modified versions of it | |
13 # that use the same license as the "OpenSSL" library), and distribute | |
14 # the linked executables. You must obey the GNU General Public License | |
15 # in all respects for all of the code used other than "OpenSSL". If you | |
16 # modify file(s) with this exception, you may extend this exception to | |
17 # your version of the file(s), but you are not obligated to do so. If | |
18 # you do not wish to do so, delete this exception statement from your | |
19 # version. If you delete this exception statement from all source files | |
20 # in the program, then also delete it here. | |
21 # | |
22 # This program is distributed in the hope that it will be useful, but | |
23 # WITHOUT ANY WARRANTY; without even the implied warranty of | |
24 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
25 # General Public License for more details. | |
26 # | |
27 # You should have received a copy of the GNU General Public License | |
28 # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
29 | |
30 | |
0 | 31 import sys |
32 import os | |
33 import os.path | |
34 import pprint | |
129 | 35 import re |
0 | 36 |
37 if len(sys.argv) < 2 or len(sys.argv) % 2 != 0: | |
38 print ('Usage:') | |
39 print ('python %s <TargetBaseFilename> [ <Name> <Source> ]*' % sys.argv[0]) | |
40 exit(-1) | |
41 | |
42 | |
43 try: | |
44 # Make sure the destination directory exists | |
45 os.makedirs(os.path.normpath(os.path.join(sys.argv[1], '..'))) | |
46 except: | |
47 pass | |
48 | |
49 | |
50 ##################################################################### | |
51 ## Read each resource file | |
52 ##################################################################### | |
53 | |
129 | 54 def CheckNoUpcase(s): |
55 if re.search('[A-Z]', s) != None: | |
56 raise Exception("Path in a directory with an upcase letter: %s" % s) | |
57 | |
0 | 58 resources = {} |
59 | |
60 counter = 0 | |
61 i = 2 | |
62 while i < len(sys.argv): | |
63 resourceName = sys.argv[i].upper() | |
64 pathName = sys.argv[i + 1] | |
65 | |
66 if not os.path.exists(pathName): | |
67 raise Exception("Non existing path: %s" % pathName) | |
68 | |
69 if resourceName in resources: | |
70 raise Exception("Twice the same resource: " + resourceName) | |
71 | |
72 if os.path.isdir(pathName): | |
73 # The resource is a directory: Recursively explore its files | |
74 content = {} | |
75 for root, dirs, files in os.walk(pathName): | |
76 base = os.path.relpath(root, pathName) | |
77 for f in files: | |
78 if f.find('~') == -1: # Ignore Emacs backup files | |
79 if base == '.': | |
129 | 80 r = f |
0 | 81 else: |
129 | 82 r = os.path.join(base, f) |
0 | 83 |
129 | 84 CheckNoUpcase(r) |
20 | 85 r = '/' + r.replace('\\', '/') |
0 | 86 if r in content: |
87 raise Exception("Twice the same filename (check case): " + r) | |
88 | |
89 content[r] = { | |
90 'Filename' : os.path.join(root, f), | |
91 'Index' : counter | |
92 } | |
93 counter += 1 | |
94 | |
95 resources[resourceName] = { | |
96 'Type' : 'Directory', | |
97 'Files' : content | |
98 } | |
99 | |
100 elif os.path.isfile(pathName): | |
101 resources[resourceName] = { | |
102 'Type' : 'File', | |
103 'Index' : counter, | |
104 'Filename' : pathName | |
105 } | |
106 counter += 1 | |
107 | |
108 else: | |
109 raise Exception("Not a regular file, nor a directory: " + pathName) | |
110 | |
111 i += 2 | |
112 | |
113 #pprint.pprint(resources) | |
114 | |
115 | |
116 ##################################################################### | |
117 ## Write .h header | |
118 ##################################################################### | |
119 | |
120 header = open(sys.argv[1] + '.h', 'w') | |
121 | |
122 header.write(""" | |
123 #pragma once | |
124 | |
125 #include <string> | |
323
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
126 #include <list> |
0 | 127 |
64
71c4a4abe90b
renaming of resources
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
55
diff
changeset
|
128 namespace Orthanc |
0 | 129 { |
130 namespace EmbeddedResources | |
131 { | |
132 enum FileResourceId | |
133 { | |
134 """) | |
135 | |
136 isFirst = True | |
137 for name in resources: | |
138 if resources[name]['Type'] == 'File': | |
139 if isFirst: | |
140 isFirst = False | |
141 else: | |
142 header.write(',\n') | |
143 header.write(' %s' % name) | |
144 | |
145 header.write(""" | |
146 }; | |
147 | |
148 enum DirectoryResourceId | |
149 { | |
150 """) | |
151 | |
152 isFirst = True | |
153 for name in resources: | |
154 if resources[name]['Type'] == 'Directory': | |
155 if isFirst: | |
156 isFirst = False | |
157 else: | |
158 header.write(',\n') | |
159 header.write(' %s' % name) | |
160 | |
161 header.write(""" | |
162 }; | |
163 | |
164 const void* GetFileResourceBuffer(FileResourceId id); | |
165 size_t GetFileResourceSize(FileResourceId id); | |
166 void GetFileResource(std::string& result, FileResourceId id); | |
167 | |
168 const void* GetDirectoryResourceBuffer(DirectoryResourceId id, const char* path); | |
169 size_t GetDirectoryResourceSize(DirectoryResourceId id, const char* path); | |
170 void GetDirectoryResource(std::string& result, DirectoryResourceId id, const char* path); | |
323
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
171 |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
172 void ListResources(std::list<std::string>& result, DirectoryResourceId id); |
0 | 173 } |
174 } | |
175 """) | |
176 header.close() | |
177 | |
178 | |
179 | |
180 ##################################################################### | |
181 ## Write the resource content in the .cpp source | |
182 ##################################################################### | |
183 | |
55 | 184 PYTHON_MAJOR_VERSION = sys.version_info[0] |
185 | |
0 | 186 def WriteResource(cpp, item): |
187 cpp.write(' static const uint8_t resource%dBuffer[] = {' % item['Index']) | |
188 | |
189 f = open(item['Filename'], "rb") | |
190 content = f.read() | |
191 f.close() | |
192 | |
193 # http://stackoverflow.com/a/1035360 | |
194 pos = 0 | |
195 for b in content: | |
55 | 196 if PYTHON_MAJOR_VERSION == 2: |
197 c = ord(b[0]) | |
198 else: | |
199 c = b | |
200 | |
0 | 201 if pos > 0: |
202 cpp.write(', ') | |
203 | |
204 if (pos % 16) == 0: | |
205 cpp.write('\n ') | |
206 | |
55 | 207 if c < 0: |
0 | 208 raise Exception("Internal error") |
209 | |
55 | 210 cpp.write("0x%02x" % c) |
0 | 211 pos += 1 |
212 | |
213 cpp.write(' };\n') | |
214 cpp.write(' static const size_t resource%dSize = %d;\n' % (item['Index'], pos)) | |
215 | |
216 | |
217 cpp = open(sys.argv[1] + '.cpp', 'w') | |
218 | |
219 cpp.write(""" | |
220 #include "%s.h" | |
64
71c4a4abe90b
renaming of resources
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
55
diff
changeset
|
221 #include "%s/Core/OrthancException.h" |
0 | 222 |
223 #include <stdint.h> | |
224 #include <string.h> | |
225 | |
64
71c4a4abe90b
renaming of resources
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
55
diff
changeset
|
226 namespace Orthanc |
0 | 227 { |
228 namespace EmbeddedResources | |
229 { | |
230 """ % (os.path.basename(sys.argv[1]), | |
231 os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))) | |
232 | |
233 | |
234 for name in resources: | |
235 if resources[name]['Type'] == 'File': | |
236 WriteResource(cpp, resources[name]) | |
237 else: | |
238 for f in resources[name]['Files']: | |
239 WriteResource(cpp, resources[name]['Files'][f]) | |
240 | |
241 | |
242 | |
243 ##################################################################### | |
244 ## Write the accessors to the file resources in .cpp | |
245 ##################################################################### | |
246 | |
247 cpp.write(""" | |
248 const void* GetFileResourceBuffer(FileResourceId id) | |
249 { | |
250 switch (id) | |
251 { | |
252 """) | |
253 for name in resources: | |
254 if resources[name]['Type'] == 'File': | |
255 cpp.write(' case %s:\n' % name) | |
256 cpp.write(' return resource%dBuffer;\n' % resources[name]['Index']) | |
257 | |
258 cpp.write(""" | |
259 default: | |
64
71c4a4abe90b
renaming of resources
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
55
diff
changeset
|
260 throw OrthancException(ErrorCode_ParameterOutOfRange); |
0 | 261 } |
262 } | |
263 | |
264 size_t GetFileResourceSize(FileResourceId id) | |
265 { | |
266 switch (id) | |
267 { | |
268 """) | |
269 | |
270 for name in resources: | |
271 if resources[name]['Type'] == 'File': | |
272 cpp.write(' case %s:\n' % name) | |
273 cpp.write(' return resource%dSize;\n' % resources[name]['Index']) | |
274 | |
275 cpp.write(""" | |
276 default: | |
64
71c4a4abe90b
renaming of resources
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
55
diff
changeset
|
277 throw OrthancException(ErrorCode_ParameterOutOfRange); |
0 | 278 } |
279 } | |
280 """) | |
281 | |
282 | |
283 | |
284 ##################################################################### | |
285 ## Write the accessors to the directory resources in .cpp | |
286 ##################################################################### | |
287 | |
288 cpp.write(""" | |
289 const void* GetDirectoryResourceBuffer(DirectoryResourceId id, const char* path) | |
290 { | |
291 switch (id) | |
292 { | |
293 """) | |
294 | |
295 for name in resources: | |
296 if resources[name]['Type'] == 'Directory': | |
297 cpp.write(' case %s:\n' % name) | |
298 isFirst = True | |
299 for path in resources[name]['Files']: | |
300 cpp.write(' if (!strcmp(path, "%s"))\n' % path) | |
301 cpp.write(' return resource%dBuffer;\n' % resources[name]['Files'][path]['Index']) | |
64
71c4a4abe90b
renaming of resources
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
55
diff
changeset
|
302 cpp.write(' throw OrthancException("Unknown path in a directory resource");\n\n') |
0 | 303 |
304 cpp.write(""" default: | |
64
71c4a4abe90b
renaming of resources
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
55
diff
changeset
|
305 throw OrthancException(ErrorCode_ParameterOutOfRange); |
0 | 306 } |
307 } | |
308 | |
309 size_t GetDirectoryResourceSize(DirectoryResourceId id, const char* path) | |
310 { | |
311 switch (id) | |
312 { | |
313 """) | |
314 | |
315 for name in resources: | |
316 if resources[name]['Type'] == 'Directory': | |
317 cpp.write(' case %s:\n' % name) | |
318 isFirst = True | |
319 for path in resources[name]['Files']: | |
320 cpp.write(' if (!strcmp(path, "%s"))\n' % path) | |
321 cpp.write(' return resource%dSize;\n' % resources[name]['Files'][path]['Index']) | |
64
71c4a4abe90b
renaming of resources
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
55
diff
changeset
|
322 cpp.write(' throw OrthancException("Unknown path in a directory resource");\n\n') |
0 | 323 |
324 cpp.write(""" default: | |
64
71c4a4abe90b
renaming of resources
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
55
diff
changeset
|
325 throw OrthancException(ErrorCode_ParameterOutOfRange); |
0 | 326 } |
327 } | |
328 """) | |
329 | |
330 | |
331 | |
332 | |
333 ##################################################################### | |
323
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
334 ## List the resources in a directory |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
335 ##################################################################### |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
336 |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
337 cpp.write(""" |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
338 void ListResources(std::list<std::string>& result, DirectoryResourceId id) |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
339 { |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
340 result.clear(); |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
341 |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
342 switch (id) |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
343 { |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
344 """) |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
345 |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
346 for name in resources: |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
347 if resources[name]['Type'] == 'Directory': |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
348 cpp.write(' case %s:\n' % name) |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
349 for path in sorted(resources[name]['Files']): |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
350 cpp.write(' result.push_back("%s");\n' % path) |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
351 cpp.write(' break;\n\n') |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
352 |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
353 cpp.write(""" default: |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
354 throw OrthancException(ErrorCode_ParameterOutOfRange); |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
355 } |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
356 } |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
357 """) |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
358 |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
359 |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
360 |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
361 |
052dede32761
list files in resource directories
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
143
diff
changeset
|
362 ##################################################################### |
0 | 363 ## Write the convenience wrappers in .cpp |
364 ##################################################################### | |
365 | |
366 cpp.write(""" | |
367 void GetFileResource(std::string& result, FileResourceId id) | |
368 { | |
369 size_t size = GetFileResourceSize(id); | |
370 result.resize(size); | |
371 if (size > 0) | |
372 memcpy(&result[0], GetFileResourceBuffer(id), size); | |
373 } | |
374 | |
375 void GetDirectoryResource(std::string& result, DirectoryResourceId id, const char* path) | |
376 { | |
377 size_t size = GetDirectoryResourceSize(id, path); | |
378 result.resize(size); | |
379 if (size > 0) | |
380 memcpy(&result[0], GetDirectoryResourceBuffer(id, path), size); | |
381 } | |
382 } | |
383 } | |
384 """) | |
385 cpp.close() |