blob: 615549a5da416f0bdbeff44799b331e28ff36eb5 [file] [log] [blame]
José Fonseca7ad40262009-09-30 17:17:12 +01001##########################################################################
José Fonseca95442442008-07-08 10:32:53 +09002#
José Fonseca7ad40262009-09-30 17:17:12 +01003# Copyright 2008-2009 VMware, Inc.
4# All Rights Reserved.
José Fonseca95442442008-07-08 10:32:53 +09005#
José Fonseca7ad40262009-09-30 17:17:12 +01006# Permission is hereby granted, free of charge, to any person obtaining a copy
7# of this software and associated documentation files (the "Software"), to deal
8# in the Software without restriction, including without limitation the rights
9# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10# copies of the Software, and to permit persons to whom the Software is
11# furnished to do so, subject to the following conditions:
José Fonseca95442442008-07-08 10:32:53 +090012#
José Fonseca7ad40262009-09-30 17:17:12 +010013# The above copyright notice and this permission notice shall be included in
14# all copies or substantial portions of the Software.
José Fonseca95442442008-07-08 10:32:53 +090015#
José Fonseca7ad40262009-09-30 17:17:12 +010016# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22# THE SOFTWARE.
José Fonseca95442442008-07-08 10:32:53 +090023#
José Fonseca7ad40262009-09-30 17:17:12 +010024##########################################################################/
José Fonseca95442442008-07-08 10:32:53 +090025
José Fonsecad626cf42008-07-07 07:43:16 +090026"""C basic types"""
27
José Fonseca8a56d142008-07-09 12:18:08 +090028
29import debug
30
31
32all_types = {}
33
José Fonseca501f2862010-11-19 20:41:18 +000034
35class Visitor:
36
37 def visit(self, type, *args, **kwargs):
38 return type.visit(self, *args, **kwargs)
39
40 __call__ = visit
41
42 def visit_void(self, type, *args, **kwargs):
43 raise NotImplementedError
44
45 def visit_literal(self, type, *args, **kwargs):
46 raise NotImplementedError
47
José Fonseca2defc982010-11-22 16:59:10 +000048 def visit_string(self, type, *args, **kwargs):
49 raise NotImplementedError
50
José Fonseca501f2862010-11-19 20:41:18 +000051 def visit_const(self, type, *args, **kwargs):
52 raise NotImplementedError
53
54 def visit_struct(self, type, *args, **kwargs):
55 raise NotImplementedError
56
57 def visit_array(self, type, *args, **kwargs):
58 raise NotImplementedError
59
José Fonseca885f2652010-11-20 11:22:25 +000060 def visit_blob(self, type, *args, **kwargs):
61 raise NotImplementedError
62
José Fonseca501f2862010-11-19 20:41:18 +000063 def visit_enum(self, type, *args, **kwargs):
64 raise NotImplementedError
65
66 def visit_bitmask(self, type, *args, **kwargs):
67 raise NotImplementedError
68
69 def visit_pointer(self, type, *args, **kwargs):
70 raise NotImplementedError
71
72 def visit_alias(self, type, *args, **kwargs):
73 raise NotImplementedError
74
75 def visit_opaque(self, type, *args, **kwargs):
76 raise NotImplementedError
77
78
José Fonsecac9edb832010-11-20 09:03:10 +000079class Rebuilder(Visitor):
80
81 def visit_void(self, void):
82 return void
83
84 def visit_literal(self, literal):
85 return literal
86
José Fonseca2defc982010-11-22 16:59:10 +000087 def visit_string(self, string):
88 return string
89
José Fonsecac9edb832010-11-20 09:03:10 +000090 def visit_const(self, const):
91 return Const(const.type)
92
93 def visit_struct(self, struct):
94 members = [self.visit(member) for member in struct.members]
95 return Struct(struct.name, members)
96
97 def visit_array(self, array):
98 type = self.visit(array.type)
99 return Array(type, array.length)
100
José Fonseca885f2652010-11-20 11:22:25 +0000101 def visit_blob(self, blob):
102 type = self.visit(blob.type)
103 return Blob(type, blob.size)
104
José Fonsecac9edb832010-11-20 09:03:10 +0000105 def visit_enum(self, enum):
106 return enum
107
108 def visit_bitmask(self, bitmask):
109 type = self.visit(bitmask.type)
110 return Bitmask(type, bitmask.values)
111
112 def visit_pointer(self, pointer):
113 type = self.visit(pointer.type)
114 return Pointer(type)
115
116 def visit_alias(self, alias):
117 type = self.visit(alias.type)
118 return Alias(alias.expr, type)
119
120 def visit_opaque(self, opaque):
121 return opaque
122
123
José Fonsecad626cf42008-07-07 07:43:16 +0900124class Type:
125
José Fonsecae753ce82009-07-22 18:13:52 +0100126 __seq = 0
127
128 def __init__(self, expr, id = ''):
129 self.expr = expr
130
131 for char in id:
132 assert char.isalnum() or char in '_ '
133
134 id = id.replace(' ', '_')
135
136 if id in all_types:
137 Type.__seq += 1
138 id += str(Type.__seq)
139
140 assert id not in all_types
141 all_types[id] = self
142
143 self.id = id
José Fonsecad626cf42008-07-07 07:43:16 +0900144
145 def __str__(self):
José Fonsecae753ce82009-07-22 18:13:52 +0100146 return self.expr
José Fonsecae9e1d062009-04-13 13:26:29 +0100147
José Fonseca501f2862010-11-19 20:41:18 +0000148 def visit(self, visitor, *args, **kwargs):
149 raise NotImplementedError
150
José Fonseca8a56d142008-07-09 12:18:08 +0900151 def decl(self):
152 pass
153
154 def impl(self):
155 pass
156
José Fonsecaa83fb242008-07-07 16:55:52 +0900157 def dump(self, instance):
158 raise NotImplementedError
159
José Fonsecad626cf42008-07-07 07:43:16 +0900160 def wrap_instance(self, instance):
José Fonseca8a56d142008-07-09 12:18:08 +0900161 pass
José Fonsecad626cf42008-07-07 07:43:16 +0900162
José Fonseca27cd25d2008-07-07 13:44:00 +0900163 def unwrap_instance(self, instance):
164 pass
165
José Fonsecad626cf42008-07-07 07:43:16 +0900166
José Fonseca8a56d142008-07-09 12:18:08 +0900167class _Void(Type):
José Fonsecad626cf42008-07-07 07:43:16 +0900168
169 def __init__(self):
170 Type.__init__(self, "void")
171
José Fonseca501f2862010-11-19 20:41:18 +0000172 def visit(self, visitor, *args, **kwargs):
173 return visitor.visit_void(self, *args, **kwargs)
174
José Fonseca8a56d142008-07-09 12:18:08 +0900175Void = _Void()
José Fonsecad626cf42008-07-07 07:43:16 +0900176
177
José Fonseca8a56d142008-07-09 12:18:08 +0900178class Concrete(Type):
José Fonsecad626cf42008-07-07 07:43:16 +0900179
José Fonseca8a56d142008-07-09 12:18:08 +0900180 def decl(self):
José Fonseca622af962009-09-27 19:13:58 +0100181 print 'static void Dump%s(const %s &value);' % (self.id, self.expr)
José Fonseca8a56d142008-07-09 12:18:08 +0900182
183 def impl(self):
José Fonseca622af962009-09-27 19:13:58 +0100184 print 'static void Dump%s(const %s &value) {' % (self.id, self.expr)
José Fonseca8a56d142008-07-09 12:18:08 +0900185 self._dump("value");
186 print '}'
187 print
188
189 def _dump(self, instance):
190 raise NotImplementedError
191
192 def dump(self, instance):
José Fonsecae753ce82009-07-22 18:13:52 +0100193 print ' Dump%s(%s);' % (self.id, instance)
José Fonseca8a56d142008-07-09 12:18:08 +0900194
195
José Fonseca51c1ef82010-11-15 16:09:14 +0000196class Literal(Concrete):
José Fonseca8a56d142008-07-09 12:18:08 +0900197
José Fonseca51c1ef82010-11-15 16:09:14 +0000198 def __init__(self, expr, format, base=10):
José Fonsecae753ce82009-07-22 18:13:52 +0100199 Concrete.__init__(self, expr)
José Fonsecad626cf42008-07-07 07:43:16 +0900200 self.format = format
201
José Fonseca501f2862010-11-19 20:41:18 +0000202 def visit(self, visitor, *args, **kwargs):
203 return visitor.visit_literal(self, *args, **kwargs)
204
José Fonseca8a56d142008-07-09 12:18:08 +0900205 def _dump(self, instance):
José Fonseca51c1ef82010-11-15 16:09:14 +0000206 print ' Log::Literal%s(%s);' % (self.format, instance)
José Fonsecaa83fb242008-07-07 16:55:52 +0900207
José Fonsecad626cf42008-07-07 07:43:16 +0900208
209class Const(Type):
210
211 def __init__(self, type):
José Fonsecae753ce82009-07-22 18:13:52 +0100212
José Fonsecaf6592d72010-11-21 12:44:41 +0000213 if type.expr.startswith("const "):
José Fonsecae753ce82009-07-22 18:13:52 +0100214 expr = type.expr + " const"
215 else:
216 expr = "const " + type.expr
217
218 Type.__init__(self, expr, 'C' + type.id)
219
José Fonsecad626cf42008-07-07 07:43:16 +0900220 self.type = type
221
José Fonseca501f2862010-11-19 20:41:18 +0000222 def visit(self, visitor, *args, **kwargs):
223 return visitor.visit_const(self, *args, **kwargs)
224
José Fonsecaa83fb242008-07-07 16:55:52 +0900225 def dump(self, instance):
226 self.type.dump(instance)
227
José Fonsecad626cf42008-07-07 07:43:16 +0900228
229class Pointer(Type):
230
231 def __init__(self, type):
José Fonsecae753ce82009-07-22 18:13:52 +0100232 Type.__init__(self, type.expr + " *", 'P' + type.id)
José Fonsecad626cf42008-07-07 07:43:16 +0900233 self.type = type
234
José Fonseca501f2862010-11-19 20:41:18 +0000235 def visit(self, visitor, *args, **kwargs):
236 return visitor.visit_pointer(self, *args, **kwargs)
237
José Fonsecaa83fb242008-07-07 16:55:52 +0900238 def dump(self, instance):
José Fonseca4a9c40c2008-07-07 18:04:53 +0900239 print ' if(%s) {' % instance
José Fonsecac7337452010-11-15 22:10:53 +0000240 print ' Log::BeginPointer("%s", (const void *)%s);' % (self.type, instance)
José Fonseca31524192010-11-22 21:05:26 +0000241 self.type.dump("*" + instance)
José Fonsecac7337452010-11-15 22:10:53 +0000242 print ' Log::EndPointer();'
José Fonseca4a9c40c2008-07-07 18:04:53 +0900243 print ' }'
244 print ' else'
José Fonseca51c1ef82010-11-15 16:09:14 +0000245 print ' Log::LiteralNull();'
José Fonsecad626cf42008-07-07 07:43:16 +0900246
247 def wrap_instance(self, instance):
248 self.type.wrap_instance("*" + instance)
249
José Fonseca27cd25d2008-07-07 13:44:00 +0900250 def unwrap_instance(self, instance):
251 self.type.wrap_instance("*" + instance)
252
José Fonsecad626cf42008-07-07 07:43:16 +0900253
José Fonsecab974caa2008-07-09 08:12:34 +0900254def ConstPointer(type):
255 return Pointer(Const(type))
256
257
José Fonseca8a56d142008-07-09 12:18:08 +0900258class Enum(Concrete):
José Fonsecad626cf42008-07-07 07:43:16 +0900259
260 def __init__(self, name, values):
José Fonseca8a56d142008-07-09 12:18:08 +0900261 Concrete.__init__(self, name)
José Fonsecad626cf42008-07-07 07:43:16 +0900262 self.values = values
José Fonsecaa83fb242008-07-07 16:55:52 +0900263
José Fonseca501f2862010-11-19 20:41:18 +0000264 def visit(self, visitor, *args, **kwargs):
265 return visitor.visit_enum(self, *args, **kwargs)
266
José Fonseca8a56d142008-07-09 12:18:08 +0900267 def _dump(self, instance):
José Fonsecaa83fb242008-07-07 16:55:52 +0900268 print ' switch(%s) {' % instance
269 for value in self.values:
270 print ' case %s:' % value
José Fonseca7e329022010-11-19 17:05:18 +0000271 print ' Log::LiteralNamedConstant("%s", %s);' % (value, value)
José Fonsecaa83fb242008-07-07 16:55:52 +0900272 print ' break;'
273 print ' default:'
José Fonseca51c1ef82010-11-15 16:09:14 +0000274 print ' Log::LiteralSInt(%s);' % instance
José Fonsecaa83fb242008-07-07 16:55:52 +0900275 print ' break;'
276 print ' }'
José Fonsecad626cf42008-07-07 07:43:16 +0900277
278
José Fonseca501f2862010-11-19 20:41:18 +0000279def FakeEnum(type, values):
280 return Enum(type.expr, values)
José Fonseca6edf23c2009-05-04 10:20:52 +0100281
282
José Fonseca501f2862010-11-19 20:41:18 +0000283class Bitmask(Concrete):
José Fonsecad626cf42008-07-07 07:43:16 +0900284
285 def __init__(self, type, values):
José Fonsecae753ce82009-07-22 18:13:52 +0100286 Concrete.__init__(self, type.expr)
José Fonsecaec61f312008-07-09 02:16:43 +0900287 self.type = type
José Fonsecad626cf42008-07-07 07:43:16 +0900288 self.values = values
289
José Fonseca501f2862010-11-19 20:41:18 +0000290 def visit(self, visitor, *args, **kwargs):
291 return visitor.visit_bitmask(self, *args, **kwargs)
292
José Fonseca8a56d142008-07-09 12:18:08 +0900293 def _dump(self, instance):
294 print ' %s l_Value = %s;' % (self.type, instance)
José Fonseca51c1ef82010-11-15 16:09:14 +0000295 print ' Log::BeginBitmask("%s");' % (self.type,)
José Fonsecaec61f312008-07-09 02:16:43 +0900296 for value in self.values:
José Fonseca8a56d142008-07-09 12:18:08 +0900297 print ' if((l_Value & %s) == %s) {' % (value, value)
José Fonseca7e329022010-11-19 17:05:18 +0000298 print ' Log::LiteralNamedConstant("%s", %s);' % (value, value)
José Fonseca8a56d142008-07-09 12:18:08 +0900299 print ' l_Value &= ~%s;' % value
300 print ' }'
José Fonseca51c1ef82010-11-15 16:09:14 +0000301 print ' if(l_Value) {'
José Fonsecaec61f312008-07-09 02:16:43 +0900302 self.type.dump("l_Value");
José Fonseca47e85e12009-05-04 11:05:11 +0100303 print ' }'
José Fonseca51c1ef82010-11-15 16:09:14 +0000304 print ' Log::EndBitmask();'
José Fonsecaec61f312008-07-09 02:16:43 +0900305
José Fonseca501f2862010-11-19 20:41:18 +0000306Flags = Bitmask
307
José Fonsecad626cf42008-07-07 07:43:16 +0900308
José Fonsecaccae31c2009-07-22 18:14:12 +0100309class Array(Type):
310
311 def __init__(self, type, length):
José Fonseca885f2652010-11-20 11:22:25 +0000312 Type.__init__(self, type.expr + " *")
José Fonsecaccae31c2009-07-22 18:14:12 +0100313 self.type = type
314 self.length = length
315
José Fonseca501f2862010-11-19 20:41:18 +0000316 def visit(self, visitor, *args, **kwargs):
317 return visitor.visit_array(self, *args, **kwargs)
318
José Fonsecaccae31c2009-07-22 18:14:12 +0100319 def dump(self, instance):
José Fonseca62ab02d2010-11-22 18:35:52 +0000320 print ' if(%s) {' % instance
José Fonsecaccae31c2009-07-22 18:14:12 +0100321 index = '__i' + self.type.id
José Fonseca62ab02d2010-11-22 18:35:52 +0000322 print ' Log::BeginArray("%s", %s);' % (self.type, self.length)
323 print ' for (int %s = 0; %s < %s; ++%s) {' % (index, index, self.length, index)
324 print ' Log::BeginElement("%s");' % (self.type,)
José Fonsecaccae31c2009-07-22 18:14:12 +0100325 self.type.dump('(%s)[%s]' % (instance, index))
José Fonseca62ab02d2010-11-22 18:35:52 +0000326 print ' Log::EndElement();'
327 print ' }'
328 print ' Log::EndArray();'
José Fonsecaccae31c2009-07-22 18:14:12 +0100329 print ' }'
José Fonseca62ab02d2010-11-22 18:35:52 +0000330 print ' else'
331 print ' Log::LiteralNull();'
José Fonsecaccae31c2009-07-22 18:14:12 +0100332
333 def wrap_instance(self, instance):
334 self.type.wrap_instance("*" + instance)
335
336 def unwrap_instance(self, instance):
337 self.type.wrap_instance("*" + instance)
338
339
José Fonseca885f2652010-11-20 11:22:25 +0000340class Blob(Type):
341
342 def __init__(self, type, size):
343 Type.__init__(self, type.expr + ' *')
344 self.type = type
345 self.size = size
346
347 def visit(self, visitor, *args, **kwargs):
348 return visitor.visit_blob(self, *args, **kwargs)
349
350 def dump(self, instance):
351 print ' Log::LiteralBlob(%s, %s);' % (instance, self.size)
352
353
José Fonseca8a56d142008-07-09 12:18:08 +0900354class Struct(Concrete):
José Fonsecad626cf42008-07-07 07:43:16 +0900355
356 def __init__(self, name, members):
José Fonseca8a56d142008-07-09 12:18:08 +0900357 Concrete.__init__(self, name)
José Fonseca51c1ef82010-11-15 16:09:14 +0000358 self.name = name
José Fonsecad626cf42008-07-07 07:43:16 +0900359 self.members = members
360
José Fonseca501f2862010-11-19 20:41:18 +0000361 def visit(self, visitor, *args, **kwargs):
362 return visitor.visit_struct(self, *args, **kwargs)
363
José Fonseca8a56d142008-07-09 12:18:08 +0900364 def _dump(self, instance):
José Fonseca51c1ef82010-11-15 16:09:14 +0000365 print ' Log::BeginStruct("%s");' % (self.name,)
José Fonsecaa83fb242008-07-07 16:55:52 +0900366 for type, name in self.members:
José Fonseca51c1ef82010-11-15 16:09:14 +0000367 print ' Log::BeginMember("%s", "%s");' % (type, name)
José Fonsecaa83fb242008-07-07 16:55:52 +0900368 type.dump('(%s).%s' % (instance, name))
José Fonseca51c1ef82010-11-15 16:09:14 +0000369 print ' Log::EndMember();'
370 print ' Log::EndStruct();'
José Fonsecaa83fb242008-07-07 16:55:52 +0900371
José Fonsecad626cf42008-07-07 07:43:16 +0900372
373class Alias(Type):
374
José Fonsecac9edb832010-11-20 09:03:10 +0000375 def __init__(self, expr, type):
376 Type.__init__(self, expr)
José Fonsecad626cf42008-07-07 07:43:16 +0900377 self.type = type
378
José Fonseca501f2862010-11-19 20:41:18 +0000379 def visit(self, visitor, *args, **kwargs):
380 return visitor.visit_alias(self, *args, **kwargs)
381
José Fonsecaa83fb242008-07-07 16:55:52 +0900382 def dump(self, instance):
383 self.type.dump(instance)
384
José Fonsecad626cf42008-07-07 07:43:16 +0900385
José Fonsecac9096f02010-11-22 13:02:26 +0000386def Out(type, name):
387 arg = Arg(type, name, output=True)
388 return arg
389
390
391class Arg:
392
393 def __init__(self, type, name, output=False):
394 self.type = type
395 self.name = name
396 self.output = output
397
398 def __str__(self):
399 return '%s %s' % (self.type, self.name)
José Fonseca83c9ac82010-01-28 14:45:36 +0000400
401
José Fonsecad626cf42008-07-07 07:43:16 +0900402class Function:
403
José Fonsecaee855d92010-11-22 17:14:47 +0000404 def __init__(self, type, name, args, call = '__stdcall', fail = None, sideeffects=True):
José Fonsecad626cf42008-07-07 07:43:16 +0900405 self.type = type
406 self.name = name
José Fonsecac9096f02010-11-22 13:02:26 +0000407
408 self.args = []
409 for arg in args:
410 if isinstance(arg, tuple):
411 arg_type, arg_name = arg
412 arg = Arg(arg_type, arg_name)
413 self.args.append(arg)
414
José Fonsecad626cf42008-07-07 07:43:16 +0900415 self.call = call
José Fonseca290c28c2009-04-23 15:20:29 +0100416 self.fail = fail
José Fonsecaee855d92010-11-22 17:14:47 +0000417 self.sideeffects = sideeffects
José Fonsecad626cf42008-07-07 07:43:16 +0900418
419 def prototype(self, name=None):
420 if name is not None:
421 name = name.strip()
422 else:
423 name = self.name
424 s = name
425 if self.call:
426 s = self.call + ' ' + s
427 if name.startswith('*'):
428 s = '(' + s + ')'
José Fonsecae753ce82009-07-22 18:13:52 +0100429 s = self.type.expr + ' ' + s
José Fonsecad626cf42008-07-07 07:43:16 +0900430 s += "("
431 if self.args:
José Fonsecac9096f02010-11-22 13:02:26 +0000432 s += ", ".join(["%s %s" % (arg.type, arg.name) for arg in self.args])
José Fonsecad626cf42008-07-07 07:43:16 +0900433 else:
434 s += "void"
435 s += ")"
436 return s
437
José Fonseca3c2c9292009-05-04 12:16:30 +0100438 def pointer_type(self):
439 return 'P' + self.name
440
441 def pointer_value(self):
442 return 'p' + self.name
443
444 def wrap_decl(self):
445 ptype = self.pointer_type()
446 pvalue = self.pointer_value()
447 print 'typedef ' + self.prototype('* %s' % ptype) + ';'
448 print 'static %s %s = NULL;' % (ptype, pvalue)
449 print
450
451 def get_true_pointer(self):
452 raise NotImplementedError
453
José Fonseca83178a02010-11-14 00:35:05 +0000454 def exit_impl(self):
José Fonseca6d633a92010-11-22 13:24:45 +0000455 print ' Log::Abort();'
José Fonseca83178a02010-11-14 00:35:05 +0000456
José Fonseca3c2c9292009-05-04 12:16:30 +0100457 def fail_impl(self):
458 if self.fail is not None:
José Fonseca243772e2009-06-25 13:56:57 +0100459 if self.type is Void:
460 assert self.fail == ''
461 print ' return;'
462 else:
463 assert self.fail != ''
464 print ' return %s;' % self.fail
José Fonseca3c2c9292009-05-04 12:16:30 +0100465 else:
José Fonseca83178a02010-11-14 00:35:05 +0000466 self.exit_impl()
José Fonseca3c2c9292009-05-04 12:16:30 +0100467
468 def wrap_impl(self):
469 pvalue = self.pointer_value()
470 print self.prototype() + ' {'
471 if self.type is Void:
472 result = ''
473 else:
474 print ' %s result;' % self.type
475 result = 'result = '
476 self.get_true_pointer()
477 print ' Log::BeginCall("%s");' % (self.name)
José Fonsecac9096f02010-11-22 13:02:26 +0000478 for arg in self.args:
479 if not arg.output:
480 arg.type.unwrap_instance(arg.name)
481 print ' Log::BeginArg("%s", "%s");' % (arg.type, arg.name)
482 arg.type.dump(arg.name)
José Fonseca3c2c9292009-05-04 12:16:30 +0100483 print ' Log::EndArg();'
José Fonsecac9096f02010-11-22 13:02:26 +0000484 print ' %s%s(%s);' % (result, pvalue, ', '.join([str(arg.name) for arg in self.args]))
485 for arg in self.args:
486 if arg.output:
487 print ' Log::BeginArg("%s", "%s");' % (arg.type, arg.name)
488 arg.type.dump(arg.name)
José Fonseca3c2c9292009-05-04 12:16:30 +0100489 print ' Log::EndArg();'
José Fonsecac9096f02010-11-22 13:02:26 +0000490 arg.type.wrap_instance(arg.name)
José Fonseca3c2c9292009-05-04 12:16:30 +0100491 if self.type is not Void:
492 print ' Log::BeginReturn("%s");' % self.type
493 self.type.dump("result")
494 print ' Log::EndReturn();'
495 self.type.wrap_instance('result')
496 print ' Log::EndCall();'
José Fonsecac77023d2009-05-04 12:53:50 +0100497 self.post_call_impl()
José Fonseca3c2c9292009-05-04 12:16:30 +0100498 if self.type is not Void:
499 print ' return result;'
500 print '}'
501 print
502
José Fonsecac77023d2009-05-04 12:53:50 +0100503 def post_call_impl(self):
504 pass
505
José Fonsecad626cf42008-07-07 07:43:16 +0900506
José Fonseca31524192010-11-22 21:05:26 +0000507def FunctionPointer(type, name, args, **kwargs):
508 # XXX
509 return Opaque(name)
510
511
José Fonsecad626cf42008-07-07 07:43:16 +0900512class Interface(Type):
513
514 def __init__(self, name, base=None):
515 Type.__init__(self, name)
José Fonsecae753ce82009-07-22 18:13:52 +0100516 self.name = name
José Fonsecad626cf42008-07-07 07:43:16 +0900517 self.base = base
518 self.methods = []
519
José Fonseca31524192010-11-22 21:05:26 +0000520 def dump(self, instance):
521 print ' Log::LiteralOpaque((const void *)%s);' % instance
522
José Fonsecad626cf42008-07-07 07:43:16 +0900523 def itermethods(self):
524 if self.base is not None:
525 for method in self.base.itermethods():
526 yield method
527 for method in self.methods:
528 yield method
529 raise StopIteration
530
531 def wrap_name(self):
José Fonsecae753ce82009-07-22 18:13:52 +0100532 return "Wrap" + self.expr
José Fonsecad626cf42008-07-07 07:43:16 +0900533
534 def wrap_pre_decl(self):
535 print "class %s;" % self.wrap_name()
536
537 def wrap_decl(self):
538 print "class %s : public %s " % (self.wrap_name(), self.name)
539 print "{"
540 print "public:"
541 print " %s(%s * pInstance);" % (self.wrap_name(), self.name)
542 print " virtual ~%s();" % self.wrap_name()
543 print
544 for method in self.itermethods():
545 print " " + method.prototype() + ";"
546 print
José Fonseca27cd25d2008-07-07 13:44:00 +0900547 #print "private:"
José Fonsecad626cf42008-07-07 07:43:16 +0900548 print " %s * m_pInstance;" % (self.name,)
549 print "};"
550 print
551
552 def wrap_impl(self):
553 print '%s::%s(%s * pInstance) {' % (self.wrap_name(), self.wrap_name(), self.name)
José Fonsecad626cf42008-07-07 07:43:16 +0900554 print ' m_pInstance = pInstance;'
555 print '}'
556 print
557 print '%s::~%s() {' % (self.wrap_name(), self.wrap_name())
José Fonsecad626cf42008-07-07 07:43:16 +0900558 print '}'
559 print
560 for method in self.itermethods():
561 print method.prototype(self.wrap_name() + '::' + method.name) + ' {'
562 if method.type is Void:
563 result = ''
564 else:
565 print ' %s result;' % method.type
566 result = 'result = '
José Fonseca22aec832008-07-09 09:38:45 +0900567 print ' Log::BeginCall("%s");' % (self.name + '::' + method.name)
José Fonseca3bccbb12008-07-10 02:00:31 +0900568 print ' Log::BeginArg("%s *", "this");' % self.name
José Fonsecac7337452010-11-15 22:10:53 +0000569 print ' Log::BeginPointer("%s", (const void *)m_pInstance);' % self.name
570 print ' Log::EndPointer();'
José Fonseca3bccbb12008-07-10 02:00:31 +0900571 print ' Log::EndArg();'
José Fonsecac9096f02010-11-22 13:02:26 +0000572 for arg in method.args:
573 if not arg.output:
574 arg.type.unwrap_instance(arg.name)
575 print ' Log::BeginArg("%s", "%s");' % (arg.type, arg.name)
576 arg.type.dump(arg.name)
José Fonseca3bccbb12008-07-10 02:00:31 +0900577 print ' Log::EndArg();'
José Fonsecac9096f02010-11-22 13:02:26 +0000578 print ' %sm_pInstance->%s(%s);' % (result, method.name, ', '.join([str(arg.name) for arg in method.args]))
579 for arg in method.args:
580 if arg.output:
581 print ' Log::BeginArg("%s", "%s");' % (arg.type, arg.name)
582 arg.type.dump(arg.name)
José Fonseca3bccbb12008-07-10 02:00:31 +0900583 print ' Log::EndArg();'
José Fonsecac9096f02010-11-22 13:02:26 +0000584 arg.type.wrap_instance(arg.name)
José Fonsecad626cf42008-07-07 07:43:16 +0900585 if method.type is not Void:
José Fonseca22aec832008-07-09 09:38:45 +0900586 print ' Log::BeginReturn("%s");' % method.type
José Fonseca4a9c40c2008-07-07 18:04:53 +0900587 method.type.dump("result")
José Fonseca22aec832008-07-09 09:38:45 +0900588 print ' Log::EndReturn();'
José Fonsecad626cf42008-07-07 07:43:16 +0900589 method.type.wrap_instance('result')
José Fonseca22aec832008-07-09 09:38:45 +0900590 print ' Log::EndCall();'
José Fonsecad626cf42008-07-07 07:43:16 +0900591 if method.name == 'QueryInterface':
592 print ' if(*ppvObj == m_pInstance)'
593 print ' *ppvObj = this;'
594 if method.name == 'Release':
595 assert method.type is not Void
596 print ' if(!result)'
597 print ' delete this;'
598 if method.type is not Void:
599 print ' return result;'
600 print '}'
601 print
602 print
603
604
605class Method(Function):
606
607 def __init__(self, type, name, args):
José Fonseca51c1ef82010-11-15 16:09:14 +0000608 Function.__init__(self, type, name, args, call = '__stdcall')
José Fonsecad626cf42008-07-07 07:43:16 +0900609
610
611towrap = []
612
613class WrapPointer(Pointer):
614
615 def __init__(self, type):
616 Pointer.__init__(self, type)
617 if type not in towrap:
618 towrap.append(type)
619
620 def wrap_instance(self, instance):
621 print " if(%s)" % instance
622 print " %s = new %s(%s);" % (instance, self.type.wrap_name(), instance)
623
José Fonseca27cd25d2008-07-07 13:44:00 +0900624 def unwrap_instance(self, instance):
625 print " if(%s)" % instance
626 print " %s = static_cast<%s *>(%s)->m_pInstance;" % (instance, self.type.wrap_name(), instance)
627
José Fonseca8a56d142008-07-09 12:18:08 +0900628
629class _String(Type):
630
631 def __init__(self):
José Fonsecae753ce82009-07-22 18:13:52 +0100632 Type.__init__(self, "char *")
José Fonseca8a56d142008-07-09 12:18:08 +0900633
José Fonseca501f2862010-11-19 20:41:18 +0000634 def visit(self, visitor, *args, **kwargs):
José Fonseca2defc982010-11-22 16:59:10 +0000635 return visitor.visit_string(self, *args, **kwargs)
José Fonseca501f2862010-11-19 20:41:18 +0000636
José Fonseca8a56d142008-07-09 12:18:08 +0900637 def dump(self, instance):
José Fonseca51c1ef82010-11-15 16:09:14 +0000638 print ' Log::LiteralString((const char *)%s);' % instance
José Fonseca8a56d142008-07-09 12:18:08 +0900639
640String = _String()
641
José Fonseca51c1ef82010-11-15 16:09:14 +0000642
José Fonsecaf6592d72010-11-21 12:44:41 +0000643class Opaque(Type):
644 '''Opaque pointer.'''
José Fonsecae54e4112009-06-25 13:56:18 +0100645
José Fonsecaf6592d72010-11-21 12:44:41 +0000646 def __init__(self, expr):
647 Type.__init__(self, expr)
José Fonsecae54e4112009-06-25 13:56:18 +0100648
José Fonseca501f2862010-11-19 20:41:18 +0000649 def visit(self, visitor, *args, **kwargs):
650 return visitor.visit_opaque(self, *args, **kwargs)
651
José Fonsecae54e4112009-06-25 13:56:18 +0100652 def dump(self, instance):
José Fonsecaf6592d72010-11-21 12:44:41 +0000653 print ' Log::LiteralOpaque((const void *)%s);' % instance
José Fonsecae54e4112009-06-25 13:56:18 +0100654
José Fonsecaf6592d72010-11-21 12:44:41 +0000655
656def OpaquePointer(type):
657 return Opaque(type.expr + ' *')
José Fonsecae54e4112009-06-25 13:56:18 +0100658
José Fonseca8a56d142008-07-09 12:18:08 +0900659
José Fonseca51c1ef82010-11-15 16:09:14 +0000660Bool = Literal("bool", "Bool")
661SChar = Literal("signed char", "SInt")
662UChar = Literal("unsigned char", "UInt")
663Short = Literal("short", "SInt")
664Int = Literal("int", "SInt")
665Long = Literal("long", "SInt")
666LongLong = Literal("long long", "SInt")
667UShort = Literal("unsigned short", "UInt")
668UInt = Literal("unsigned int", "UInt")
669ULong = Literal("unsigned long", "UInt")
José Fonseca28f034f2010-11-22 20:31:25 +0000670ULongLong = Literal("unsigned long long", "UInt")
José Fonseca51c1ef82010-11-15 16:09:14 +0000671Float = Literal("float", "Float")
672Double = Literal("double", "Float")
673SizeT = Literal("size_t", "UInt")
674WString = Literal("wchar_t *", "WString")
José Fonsecad626cf42008-07-07 07:43:16 +0900675
676
677def wrap():
José Fonseca8a56d142008-07-09 12:18:08 +0900678 for type in all_types.itervalues():
679 type.decl()
680 print
681 for type in all_types.itervalues():
682 type.impl()
683 print
José Fonsecad626cf42008-07-07 07:43:16 +0900684 for type in towrap:
685 type.wrap_pre_decl()
686 print
687 for type in towrap:
688 type.wrap_decl()
689 print
690 for type in towrap:
691 type.wrap_impl()
692 print