blob: 9f855283eb9a2d662e50063155621c6344b670b0 [file] [log] [blame]
Benjamin Petersonf51d36a2011-12-29 12:07:21 -06001#! /usr/bin/env python
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002"""Generate C code from an ASDL description."""
3
4# TO DO
5# handle fields that have a type but no name
6
Benjamin Peterson5a3f49b2011-08-11 14:42:28 -05007import os, sys
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00008
9import asdl
10
11TABSIZE = 8
12MAX_COL = 80
13
14def get_c_type(name):
15 """Return a string for the C name of the type.
16
17 This function special cases the default types provided by asdl:
Neal Norwitz3591bbe2007-02-26 19:04:49 +000018 identifier, string, int.
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000019 """
20 # XXX ack! need to figure out where Id is useful and where string
21 if isinstance(name, asdl.Id):
22 name = name.value
23 if name in asdl.builtin_types:
24 return name
25 else:
26 return "%s_ty" % name
27
28def reflow_lines(s, depth):
29 """Reflow the line s indented depth tabs.
30
31 Return a sequence of lines where no line extends beyond MAX_COL
32 when properly indented. The first line is properly indented based
33 exclusively on depth * TABSIZE. All following lines -- these are
34 the reflowed lines generated by this function -- start at the same
35 column as the first character beyond the opening { in the first
36 line.
37 """
38 size = MAX_COL - depth * TABSIZE
39 if len(s) < size:
40 return [s]
41
42 lines = []
43 cur = s
44 padding = ""
45 while len(cur) > size:
46 i = cur.rfind(' ', 0, size)
47 # XXX this should be fixed for real
48 if i == -1 and 'GeneratorExp' in cur:
49 i = size + 3
Brett Cannonf0365512006-08-25 04:06:31 +000050 assert i != -1, "Impossible line %d to reflow: %r" % (size, s)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000051 lines.append(padding + cur[:i])
52 if len(lines) == 1:
53 # find new size based on brace
54 j = cur.find('{', 0, i)
55 if j >= 0:
56 j += 2 # account for the brace and the space after it
57 size -= j
58 padding = " " * j
59 else:
60 j = cur.find('(', 0, i)
61 if j >= 0:
62 j += 1 # account for the paren (no space after it)
63 size -= j
64 padding = " " * j
65 cur = cur[i+1:]
66 else:
67 lines.append(padding + cur)
68 return lines
69
70def is_simple(sum):
71 """Return True if a sum is a simple.
72
73 A sum is simple if its types have no fields, e.g.
74 unaryop = Invert | Not | UAdd | USub
75 """
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000076 for t in sum.types:
77 if t.fields:
78 return False
79 return True
80
Martin v. Löwis618dc5e2008-03-30 20:03:44 +000081
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000082class EmitVisitor(asdl.VisitorBase):
83 """Visit that emits lines"""
84
85 def __init__(self, file):
86 self.file = file
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +020087 self.identifiers = set()
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000088 super(EmitVisitor, self).__init__()
89
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +020090 def emit_identifier(self, name):
91 name = str(name)
92 if name in self.identifiers:
93 return
Meador Inge70d86bd2011-10-22 14:06:50 -050094 self.emit("_Py_IDENTIFIER(%s);" % name, 0)
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +020095 self.identifiers.add(name)
96
Benjamin Peterson87c8d872009-06-11 22:54:11 +000097 def emit(self, s, depth, reflow=True):
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000098 # XXX reflow long lines?
99 if reflow:
100 lines = reflow_lines(s, depth)
101 else:
102 lines = [s]
103 for line in lines:
104 line = (" " * TABSIZE * depth) + line + "\n"
105 self.file.write(line)
106
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000107
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000108class TypeDefVisitor(EmitVisitor):
109 def visitModule(self, mod):
110 for dfn in mod.dfns:
111 self.visit(dfn)
112
113 def visitType(self, type, depth=0):
114 self.visit(type.value, type.name, depth)
115
116 def visitSum(self, sum, name, depth):
117 if is_simple(sum):
118 self.simple_sum(sum, name, depth)
119 else:
120 self.sum_with_constructors(sum, name, depth)
121
122 def simple_sum(self, sum, name, depth):
123 enum = []
124 for i in range(len(sum.types)):
125 type = sum.types[i]
126 enum.append("%s=%d" % (type.name, i + 1))
127 enums = ", ".join(enum)
128 ctype = get_c_type(name)
129 s = "typedef enum _%s { %s } %s;" % (name, enums, ctype)
130 self.emit(s, depth)
131 self.emit("", depth)
132
133 def sum_with_constructors(self, sum, name, depth):
134 ctype = get_c_type(name)
135 s = "typedef struct _%(name)s *%(ctype)s;" % locals()
136 self.emit(s, depth)
137 self.emit("", depth)
138
139 def visitProduct(self, product, name, depth):
140 ctype = get_c_type(name)
141 s = "typedef struct _%(name)s *%(ctype)s;" % locals()
142 self.emit(s, depth)
143 self.emit("", depth)
144
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000145
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000146class StructVisitor(EmitVisitor):
147 """Visitor to generate typdefs for AST."""
148
149 def visitModule(self, mod):
150 for dfn in mod.dfns:
151 self.visit(dfn)
152
153 def visitType(self, type, depth=0):
154 self.visit(type.value, type.name, depth)
155
156 def visitSum(self, sum, name, depth):
157 if not is_simple(sum):
158 self.sum_with_constructors(sum, name, depth)
159
160 def sum_with_constructors(self, sum, name, depth):
161 def emit(s, depth=depth):
162 self.emit(s % sys._getframe(1).f_locals, depth)
163 enum = []
164 for i in range(len(sum.types)):
165 type = sum.types[i]
166 enum.append("%s_kind=%d" % (type.name, i + 1))
167
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000168 emit("enum _%(name)s_kind {" + ", ".join(enum) + "};")
169
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000170 emit("struct _%(name)s {")
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000171 emit("enum _%(name)s_kind kind;", depth + 1)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000172 emit("union {", depth + 1)
173 for t in sum.types:
174 self.visit(t, depth + 2)
175 emit("} v;", depth + 1)
176 for field in sum.attributes:
177 # rudimentary attribute handling
178 type = str(field.type)
179 assert type in asdl.builtin_types, type
180 emit("%s %s;" % (type, field.name), depth + 1);
181 emit("};")
182 emit("")
183
184 def visitConstructor(self, cons, depth):
185 if cons.fields:
186 self.emit("struct {", depth)
187 for f in cons.fields:
188 self.visit(f, depth + 1)
189 self.emit("} %s;" % cons.name, depth)
190 self.emit("", depth)
191 else:
192 # XXX not sure what I want here, nothing is probably fine
193 pass
194
195 def visitField(self, field, depth):
196 # XXX need to lookup field.type, because it might be something
197 # like a builtin...
198 ctype = get_c_type(field.type)
199 name = field.name
200 if field.seq:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000201 if field.type.value in ('cmpop',):
202 self.emit("asdl_int_seq *%(name)s;" % locals(), depth)
203 else:
204 self.emit("asdl_seq *%(name)s;" % locals(), depth)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000205 else:
206 self.emit("%(ctype)s %(name)s;" % locals(), depth)
207
208 def visitProduct(self, product, name, depth):
209 self.emit("struct _%(name)s {" % locals(), depth)
210 for f in product.fields:
211 self.visit(f, depth + 1)
Benjamin Petersoncda75be2013-03-18 10:48:58 -0700212 for field in product.attributes:
213 # rudimentary attribute handling
214 type = str(field.type)
215 assert type in asdl.builtin_types, type
216 self.emit("%s %s;" % (type, field.name), depth + 1);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000217 self.emit("};", depth)
218 self.emit("", depth)
219
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000220
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000221class PrototypeVisitor(EmitVisitor):
222 """Generate function prototypes for the .h file"""
223
224 def visitModule(self, mod):
225 for dfn in mod.dfns:
226 self.visit(dfn)
227
228 def visitType(self, type):
229 self.visit(type.value, type.name)
230
231 def visitSum(self, sum, name):
232 if is_simple(sum):
233 pass # XXX
234 else:
235 for t in sum.types:
236 self.visit(t, name, sum.attributes)
237
238 def get_args(self, fields):
239 """Return list of C argument into, one for each field.
240
241 Argument info is 3-tuple of a C type, variable name, and flag
242 that is true if type can be NULL.
243 """
244 args = []
245 unnamed = {}
246 for f in fields:
247 if f.name is None:
248 name = f.type
249 c = unnamed[name] = unnamed.get(name, 0) + 1
250 if c > 1:
251 name = "name%d" % (c - 1)
252 else:
253 name = f.name
254 # XXX should extend get_c_type() to handle this
255 if f.seq:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000256 if f.type.value in ('cmpop',):
257 ctype = "asdl_int_seq *"
258 else:
259 ctype = "asdl_seq *"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000260 else:
261 ctype = get_c_type(f.type)
262 args.append((ctype, name, f.opt or f.seq))
263 return args
264
265 def visitConstructor(self, cons, type, attrs):
266 args = self.get_args(cons.fields)
267 attrs = self.get_args(attrs)
268 ctype = get_c_type(type)
269 self.emit_function(cons.name, ctype, args, attrs)
270
Benjamin Peterson87c8d872009-06-11 22:54:11 +0000271 def emit_function(self, name, ctype, args, attrs, union=True):
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000272 args = args + attrs
273 if args:
274 argstr = ", ".join(["%s %s" % (atype, aname)
275 for atype, aname, opt in args])
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000276 argstr += ", PyArena *arena"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000277 else:
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000278 argstr = "PyArena *arena"
Thomas Woutersb2137042007-02-01 18:02:27 +0000279 margs = "a0"
280 for i in range(1, len(args)+1):
281 margs += ", a%d" % i
282 self.emit("#define %s(%s) _Py_%s(%s)" % (name, margs, name, margs), 0,
Benjamin Peterson87c8d872009-06-11 22:54:11 +0000283 reflow=False)
284 self.emit("%s _Py_%s(%s);" % (ctype, name, argstr), False)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000285
286 def visitProduct(self, prod, name):
287 self.emit_function(name, get_c_type(name),
Benjamin Peterson87c8d872009-06-11 22:54:11 +0000288 self.get_args(prod.fields), [], union=False)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000289
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000290
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000291class FunctionVisitor(PrototypeVisitor):
292 """Visitor to generate constructor functions for AST."""
293
Benjamin Peterson87c8d872009-06-11 22:54:11 +0000294 def emit_function(self, name, ctype, args, attrs, union=True):
295 def emit(s, depth=0, reflow=True):
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000296 self.emit(s, depth, reflow)
297 argstr = ", ".join(["%s %s" % (atype, aname)
298 for atype, aname, opt in args + attrs])
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000299 if argstr:
300 argstr += ", PyArena *arena"
301 else:
302 argstr = "PyArena *arena"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000303 self.emit("%s" % ctype, 0)
304 emit("%s(%s)" % (name, argstr))
305 emit("{")
306 emit("%s p;" % ctype, 1)
307 for argtype, argname, opt in args:
Neal Norwitz3591bbe2007-02-26 19:04:49 +0000308 if not opt and argtype != "int":
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000309 emit("if (!%s) {" % argname, 1)
310 emit("PyErr_SetString(PyExc_ValueError,", 2)
311 msg = "field %s is required for %s" % (argname, name)
312 emit(' "%s");' % msg,
Benjamin Peterson87c8d872009-06-11 22:54:11 +0000313 2, reflow=False)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000314 emit('return NULL;', 2)
315 emit('}', 1)
316
Neal Norwitzadb69fc2005-12-17 20:54:49 +0000317 emit("p = (%s)PyArena_Malloc(arena, sizeof(*p));" % ctype, 1);
Thomas Woutersa44f3a32007-02-26 18:20:15 +0000318 emit("if (!p)", 1)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000319 emit("return NULL;", 2)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000320 if union:
321 self.emit_body_union(name, args, attrs)
322 else:
323 self.emit_body_struct(name, args, attrs)
324 emit("return p;", 1)
325 emit("}")
326 emit("")
327
328 def emit_body_union(self, name, args, attrs):
Benjamin Peterson87c8d872009-06-11 22:54:11 +0000329 def emit(s, depth=0, reflow=True):
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000330 self.emit(s, depth, reflow)
331 emit("p->kind = %s_kind;" % name, 1)
332 for argtype, argname, opt in args:
333 emit("p->v.%s.%s = %s;" % (name, argname, argname), 1)
334 for argtype, argname, opt in attrs:
335 emit("p->%s = %s;" % (argname, argname), 1)
336
337 def emit_body_struct(self, name, args, attrs):
Benjamin Peterson87c8d872009-06-11 22:54:11 +0000338 def emit(s, depth=0, reflow=True):
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000339 self.emit(s, depth, reflow)
340 for argtype, argname, opt in args:
341 emit("p->%s = %s;" % (argname, argname), 1)
342 assert not attrs
343
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000344
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000345class PickleVisitor(EmitVisitor):
346
347 def visitModule(self, mod):
348 for dfn in mod.dfns:
349 self.visit(dfn)
350
351 def visitType(self, type):
352 self.visit(type.value, type.name)
353
354 def visitSum(self, sum, name):
355 pass
356
357 def visitProduct(self, sum, name):
358 pass
359
360 def visitConstructor(self, cons, name):
361 pass
362
363 def visitField(self, sum):
364 pass
365
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000366
367class Obj2ModPrototypeVisitor(PickleVisitor):
368 def visitProduct(self, prod, name):
369 code = "static int obj2ast_%s(PyObject* obj, %s* out, PyArena* arena);"
370 self.emit(code % (name, get_c_type(name)), 0)
371
372 visitSum = visitProduct
373
374
375class Obj2ModVisitor(PickleVisitor):
376 def funcHeader(self, name):
377 ctype = get_c_type(name)
378 self.emit("int", 0)
379 self.emit("obj2ast_%s(PyObject* obj, %s* out, PyArena* arena)" % (name, ctype), 0)
380 self.emit("{", 0)
Benjamin Peterson0e9e98e2010-11-20 02:01:45 +0000381 self.emit("int isinstance;", 1)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000382 self.emit("", 0)
383
Benjamin Peterson5b066812010-11-20 01:38:49 +0000384 def sumTrailer(self, name, add_label=False):
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000385 self.emit("", 0)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000386 # there's really nothing more we can do if this fails ...
Benjamin Peterson5b066812010-11-20 01:38:49 +0000387 error = "expected some sort of %s, but got %%R" % name
388 format = "PyErr_Format(PyExc_TypeError, \"%s\", obj);"
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000389 self.emit(format % error, 1, reflow=False)
Benjamin Peterson5b066812010-11-20 01:38:49 +0000390 if add_label:
391 self.emit("failed:", 1)
Benjamin Peterson0a4dae52010-11-21 15:12:34 +0000392 self.emit("Py_XDECREF(tmp);", 1)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000393 self.emit("return 1;", 1)
394 self.emit("}", 0)
395 self.emit("", 0)
396
397 def simpleSum(self, sum, name):
398 self.funcHeader(name)
399 for t in sum.types:
Benjamin Peterson97dd9872009-12-13 01:23:39 +0000400 line = ("isinstance = PyObject_IsInstance(obj, "
401 "(PyObject *)%s_type);")
402 self.emit(line % (t.name,), 1)
403 self.emit("if (isinstance == -1) {", 1)
404 self.emit("return 1;", 2)
405 self.emit("}", 1)
406 self.emit("if (isinstance) {", 1)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000407 self.emit("*out = %s;" % t.name, 2)
408 self.emit("return 0;", 2)
409 self.emit("}", 1)
410 self.sumTrailer(name)
411
412 def buildArgs(self, fields):
413 return ", ".join(fields + ["arena"])
414
415 def complexSum(self, sum, name):
416 self.funcHeader(name)
Benjamin Petersond8f65972010-11-20 04:31:07 +0000417 self.emit("PyObject *tmp = NULL;", 1)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000418 for a in sum.attributes:
419 self.visitAttributeDeclaration(a, name, sum=sum)
420 self.emit("", 0)
421 # XXX: should we only do this for 'expr'?
422 self.emit("if (obj == Py_None) {", 1)
423 self.emit("*out = NULL;", 2)
424 self.emit("return 0;", 2)
425 self.emit("}", 1)
426 for a in sum.attributes:
427 self.visitField(a, name, sum=sum, depth=1)
428 for t in sum.types:
Benjamin Peterson97dd9872009-12-13 01:23:39 +0000429 line = "isinstance = PyObject_IsInstance(obj, (PyObject*)%s_type);"
430 self.emit(line % (t.name,), 1)
431 self.emit("if (isinstance == -1) {", 1)
432 self.emit("return 1;", 2)
433 self.emit("}", 1)
434 self.emit("if (isinstance) {", 1)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000435 for f in t.fields:
436 self.visitFieldDeclaration(f, t.name, sum=sum, depth=2)
437 self.emit("", 0)
438 for f in t.fields:
439 self.visitField(f, t.name, sum=sum, depth=2)
440 args = [f.name.value for f in t.fields] + [a.name.value for a in sum.attributes]
441 self.emit("*out = %s(%s);" % (t.name, self.buildArgs(args)), 2)
442 self.emit("if (*out == NULL) goto failed;", 2)
443 self.emit("return 0;", 2)
444 self.emit("}", 1)
Benjamin Peterson5b066812010-11-20 01:38:49 +0000445 self.sumTrailer(name, True)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000446
447 def visitAttributeDeclaration(self, a, name, sum=sum):
448 ctype = get_c_type(a.type)
449 self.emit("%s %s;" % (ctype, a.name), 1)
450
451 def visitSum(self, sum, name):
452 if is_simple(sum):
453 self.simpleSum(sum, name)
454 else:
455 self.complexSum(sum, name)
456
457 def visitProduct(self, prod, name):
458 ctype = get_c_type(name)
459 self.emit("int", 0)
460 self.emit("obj2ast_%s(PyObject* obj, %s* out, PyArena* arena)" % (name, ctype), 0)
461 self.emit("{", 0)
462 self.emit("PyObject* tmp = NULL;", 1)
463 for f in prod.fields:
464 self.visitFieldDeclaration(f, name, prod=prod, depth=1)
465 self.emit("", 0)
466 for f in prod.fields:
467 self.visitField(f, name, prod=prod, depth=1)
468 args = [f.name.value for f in prod.fields]
469 self.emit("*out = %s(%s);" % (name, self.buildArgs(args)), 1)
470 self.emit("return 0;", 1)
471 self.emit("failed:", 0)
472 self.emit("Py_XDECREF(tmp);", 1)
473 self.emit("return 1;", 1)
474 self.emit("}", 0)
475 self.emit("", 0)
476
477 def visitFieldDeclaration(self, field, name, sum=None, prod=None, depth=0):
478 ctype = get_c_type(field.type)
479 if field.seq:
480 if self.isSimpleType(field):
481 self.emit("asdl_int_seq* %s;" % field.name, depth)
482 else:
483 self.emit("asdl_seq* %s;" % field.name, depth)
484 else:
485 ctype = get_c_type(field.type)
486 self.emit("%s %s;" % (ctype, field.name), depth)
487
488 def isSimpleSum(self, field):
489 # XXX can the members of this list be determined automatically?
490 return field.type.value in ('expr_context', 'boolop', 'operator',
491 'unaryop', 'cmpop')
492
493 def isNumeric(self, field):
494 return get_c_type(field.type) in ("int", "bool")
495
496 def isSimpleType(self, field):
497 return self.isSimpleSum(field) or self.isNumeric(field)
498
499 def visitField(self, field, name, sum=None, prod=None, depth=0):
500 ctype = get_c_type(field.type)
Benjamin Petersoncda75be2013-03-18 10:48:58 -0700501 if field.opt:
Benjamin Petersonb72406b2013-03-18 23:24:41 -0700502 check = "exists_not_none(obj, &PyId_%s)" % (field.name,)
Benjamin Petersoncda75be2013-03-18 10:48:58 -0700503 else:
Benjamin Petersonb72406b2013-03-18 23:24:41 -0700504 check = "_PyObject_HasAttrId(obj, &PyId_%s)" % (field.name,)
505 self.emit("if (%s) {" % (check,), depth, reflow=False)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000506 self.emit("int res;", depth+1)
507 if field.seq:
508 self.emit("Py_ssize_t len;", depth+1)
509 self.emit("Py_ssize_t i;", depth+1)
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200510 self.emit("tmp = _PyObject_GetAttrId(obj, &PyId_%s);" % field.name, depth+1)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000511 self.emit("if (tmp == NULL) goto failed;", depth+1)
512 if field.seq:
513 self.emit("if (!PyList_Check(tmp)) {", depth+1)
514 self.emit("PyErr_Format(PyExc_TypeError, \"%s field \\\"%s\\\" must "
515 "be a list, not a %%.200s\", tmp->ob_type->tp_name);" %
516 (name, field.name),
517 depth+2, reflow=False)
518 self.emit("goto failed;", depth+2)
519 self.emit("}", depth+1)
520 self.emit("len = PyList_GET_SIZE(tmp);", depth+1)
521 if self.isSimpleType(field):
522 self.emit("%s = asdl_int_seq_new(len, arena);" % field.name, depth+1)
523 else:
524 self.emit("%s = asdl_seq_new(len, arena);" % field.name, depth+1)
525 self.emit("if (%s == NULL) goto failed;" % field.name, depth+1)
526 self.emit("for (i = 0; i < len; i++) {", depth+1)
527 self.emit("%s value;" % ctype, depth+2)
528 self.emit("res = obj2ast_%s(PyList_GET_ITEM(tmp, i), &value, arena);" %
529 field.type, depth+2, reflow=False)
530 self.emit("if (res != 0) goto failed;", depth+2)
531 self.emit("asdl_seq_SET(%s, i, value);" % field.name, depth+2)
532 self.emit("}", depth+1)
533 else:
534 self.emit("res = obj2ast_%s(tmp, &%s, arena);" %
535 (field.type, field.name), depth+1)
536 self.emit("if (res != 0) goto failed;", depth+1)
537
538 self.emit("Py_XDECREF(tmp);", depth+1)
539 self.emit("tmp = NULL;", depth+1)
540 self.emit("} else {", depth)
541 if not field.opt:
542 message = "required field \\\"%s\\\" missing from %s" % (field.name, name)
543 format = "PyErr_SetString(PyExc_TypeError, \"%s\");"
544 self.emit(format % message, depth+1, reflow=False)
545 self.emit("return 1;", depth+1)
546 else:
547 if self.isNumeric(field):
548 self.emit("%s = 0;" % field.name, depth+1)
549 elif not self.isSimpleType(field):
550 self.emit("%s = NULL;" % field.name, depth+1)
551 else:
552 raise TypeError("could not determine the default value for %s" % field.name)
553 self.emit("}", depth)
554
555
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000556class MarshalPrototypeVisitor(PickleVisitor):
557
558 def prototype(self, sum, name):
559 ctype = get_c_type(name)
Neal Norwitz6576bd82005-11-13 18:41:28 +0000560 self.emit("static int marshal_write_%s(PyObject **, int *, %s);"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000561 % (name, ctype), 0)
562
563 visitProduct = visitSum = prototype
564
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000565
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000566class PyTypesDeclareVisitor(PickleVisitor):
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000567
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000568 def visitProduct(self, prod, name):
Neal Norwitz53d960c2006-02-28 22:47:29 +0000569 self.emit("static PyTypeObject *%s_type;" % name, 0)
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000570 self.emit("static PyObject* ast2obj_%s(void*);" % name, 0)
Benjamin Petersoncda75be2013-03-18 10:48:58 -0700571 if prod.attributes:
572 for a in prod.attributes:
573 self.emit_identifier(a.name)
574 self.emit("static char *%s_attributes[] = {" % name, 0)
575 for a in prod.attributes:
576 self.emit('"%s",' % a.name, 1)
577 self.emit("};", 0)
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000578 if prod.fields:
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200579 for f in prod.fields:
580 self.emit_identifier(f.name)
Neal Norwitz53d960c2006-02-28 22:47:29 +0000581 self.emit("static char *%s_fields[]={" % name,0)
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000582 for f in prod.fields:
583 self.emit('"%s",' % f.name, 1)
584 self.emit("};", 0)
Tim Peters710ab3b2006-02-28 18:30:36 +0000585
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000586 def visitSum(self, sum, name):
Neal Norwitz53d960c2006-02-28 22:47:29 +0000587 self.emit("static PyTypeObject *%s_type;" % name, 0)
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000588 if sum.attributes:
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200589 for a in sum.attributes:
590 self.emit_identifier(a.name)
Neal Norwitz53d960c2006-02-28 22:47:29 +0000591 self.emit("static char *%s_attributes[] = {" % name, 0)
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000592 for a in sum.attributes:
593 self.emit('"%s",' % a.name, 1)
594 self.emit("};", 0)
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000595 ptype = "void*"
596 if is_simple(sum):
597 ptype = get_c_type(name)
598 tnames = []
599 for t in sum.types:
600 tnames.append(str(t.name)+"_singleton")
601 tnames = ", *".join(tnames)
602 self.emit("static PyObject *%s;" % tnames, 0)
603 self.emit("static PyObject* ast2obj_%s(%s);" % (name, ptype), 0)
604 for t in sum.types:
605 self.visitConstructor(t, name)
Tim Peters710ab3b2006-02-28 18:30:36 +0000606
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000607 def visitConstructor(self, cons, name):
Neal Norwitz53d960c2006-02-28 22:47:29 +0000608 self.emit("static PyTypeObject *%s_type;" % cons.name, 0)
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000609 if cons.fields:
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200610 for t in cons.fields:
611 self.emit_identifier(t.name)
Neal Norwitz53d960c2006-02-28 22:47:29 +0000612 self.emit("static char *%s_fields[]={" % cons.name, 0)
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000613 for t in cons.fields:
614 self.emit('"%s",' % t.name, 1)
615 self.emit("};",0)
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000616
617class PyTypesVisitor(PickleVisitor):
618
619 def visitModule(self, mod):
620 self.emit("""
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700621typedef struct {
Victor Stinner45e50de2012-03-13 01:17:31 +0100622 PyObject_HEAD
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700623 PyObject *dict;
624} AST_object;
625
Benjamin Peterson1767e022012-03-14 21:50:29 -0500626static void
627ast_dealloc(AST_object *self)
628{
629 Py_CLEAR(self->dict);
Antoine Pitrou50750742012-07-08 12:43:32 +0200630 Py_TYPE(self)->tp_free(self);
Benjamin Peterson1767e022012-03-14 21:50:29 -0500631}
632
Neal Norwitz207c9f32008-03-31 04:42:11 +0000633static int
Benjamin Peterson81071762012-07-08 11:03:46 -0700634ast_traverse(AST_object *self, visitproc visit, void *arg)
635{
636 Py_VISIT(self->dict);
637 return 0;
638}
639
640static void
641ast_clear(AST_object *self)
642{
643 Py_CLEAR(self->dict);
644}
645
646static int
Neal Norwitz207c9f32008-03-31 04:42:11 +0000647ast_type_init(PyObject *self, PyObject *args, PyObject *kw)
648{
Meador Inge70d86bd2011-10-22 14:06:50 -0500649 _Py_IDENTIFIER(_fields);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000650 Py_ssize_t i, numfields = 0;
651 int res = -1;
652 PyObject *key, *value, *fields;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200653 fields = _PyObject_GetAttrId((PyObject*)Py_TYPE(self), &PyId__fields);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000654 if (!fields)
655 PyErr_Clear();
656 if (fields) {
657 numfields = PySequence_Size(fields);
658 if (numfields == -1)
659 goto cleanup;
660 }
661 res = 0; /* if no error occurs, this stays 0 to the end */
662 if (PyTuple_GET_SIZE(args) > 0) {
663 if (numfields != PyTuple_GET_SIZE(args)) {
664 PyErr_Format(PyExc_TypeError, "%.400s constructor takes %s"
Benjamin Peterson75080da2008-09-10 22:37:34 +0000665 "%zd positional argument%s",
Neal Norwitz207c9f32008-03-31 04:42:11 +0000666 Py_TYPE(self)->tp_name,
667 numfields == 0 ? "" : "either 0 or ",
668 numfields, numfields == 1 ? "" : "s");
669 res = -1;
670 goto cleanup;
671 }
672 for (i = 0; i < PyTuple_GET_SIZE(args); i++) {
673 /* cannot be reached when fields is NULL */
674 PyObject *name = PySequence_GetItem(fields, i);
675 if (!name) {
676 res = -1;
677 goto cleanup;
678 }
679 res = PyObject_SetAttr(self, name, PyTuple_GET_ITEM(args, i));
680 Py_DECREF(name);
681 if (res < 0)
682 goto cleanup;
683 }
684 }
685 if (kw) {
686 i = 0; /* needed by PyDict_Next */
687 while (PyDict_Next(kw, &i, &key, &value)) {
688 res = PyObject_SetAttr(self, key, value);
689 if (res < 0)
690 goto cleanup;
691 }
692 }
693 cleanup:
694 Py_XDECREF(fields);
695 return res;
696}
697
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000698/* Pickling support */
699static PyObject *
700ast_type_reduce(PyObject *self, PyObject *unused)
701{
702 PyObject *res;
Meador Inge70d86bd2011-10-22 14:06:50 -0500703 _Py_IDENTIFIER(__dict__);
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200704 PyObject *dict = _PyObject_GetAttrId(self, &PyId___dict__);
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000705 if (dict == NULL) {
706 if (PyErr_ExceptionMatches(PyExc_AttributeError))
707 PyErr_Clear();
708 else
709 return NULL;
710 }
711 if (dict) {
712 res = Py_BuildValue("O()O", Py_TYPE(self), dict);
713 Py_DECREF(dict);
714 return res;
715 }
716 return Py_BuildValue("O()", Py_TYPE(self));
717}
718
719static PyMethodDef ast_type_methods[] = {
720 {"__reduce__", ast_type_reduce, METH_NOARGS, NULL},
721 {NULL}
722};
723
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700724static PyGetSetDef ast_type_getsets[] = {
725 {"__dict__", PyObject_GenericGetDict, PyObject_GenericSetDict},
726 {NULL}
727};
728
Neal Norwitz207c9f32008-03-31 04:42:11 +0000729static PyTypeObject AST_type = {
730 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000731 "_ast.AST",
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700732 sizeof(AST_object),
Neal Norwitz207c9f32008-03-31 04:42:11 +0000733 0,
Benjamin Peterson1767e022012-03-14 21:50:29 -0500734 (destructor)ast_dealloc, /* tp_dealloc */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000735 0, /* tp_print */
736 0, /* tp_getattr */
737 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +0000738 0, /* tp_reserved */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000739 0, /* tp_repr */
740 0, /* tp_as_number */
741 0, /* tp_as_sequence */
742 0, /* tp_as_mapping */
743 0, /* tp_hash */
744 0, /* tp_call */
745 0, /* tp_str */
746 PyObject_GenericGetAttr, /* tp_getattro */
747 PyObject_GenericSetAttr, /* tp_setattro */
748 0, /* tp_as_buffer */
Benjamin Peterson81071762012-07-08 11:03:46 -0700749 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC, /* tp_flags */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000750 0, /* tp_doc */
Benjamin Peterson81071762012-07-08 11:03:46 -0700751 (traverseproc)ast_traverse, /* tp_traverse */
752 (inquiry)ast_clear, /* tp_clear */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000753 0, /* tp_richcompare */
754 0, /* tp_weaklistoffset */
755 0, /* tp_iter */
756 0, /* tp_iternext */
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000757 ast_type_methods, /* tp_methods */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000758 0, /* tp_members */
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700759 ast_type_getsets, /* tp_getset */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000760 0, /* tp_base */
761 0, /* tp_dict */
762 0, /* tp_descr_get */
763 0, /* tp_descr_set */
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700764 offsetof(AST_object, dict),/* tp_dictoffset */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000765 (initproc)ast_type_init, /* tp_init */
766 PyType_GenericAlloc, /* tp_alloc */
767 PyType_GenericNew, /* tp_new */
Benjamin Peterson81071762012-07-08 11:03:46 -0700768 PyObject_GC_Del, /* tp_free */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000769};
770
771
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000772static PyTypeObject* make_type(char *type, PyTypeObject* base, char**fields, int num_fields)
773{
774 PyObject *fnames, *result;
775 int i;
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000776 fnames = PyTuple_New(num_fields);
777 if (!fnames) return NULL;
778 for (i = 0; i < num_fields; i++) {
Neal Norwitze4dc3242007-08-25 01:33:49 +0000779 PyObject *field = PyUnicode_FromString(fields[i]);
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000780 if (!field) {
781 Py_DECREF(fnames);
782 return NULL;
783 }
784 PyTuple_SET_ITEM(fnames, i, field);
785 }
Benjamin Peterson5b927722010-06-07 22:24:18 +0000786 result = PyObject_CallFunction((PyObject*)&PyType_Type, "s(O){sOss}",
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000787 type, base, "_fields", fnames, "__module__", "_ast");
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000788 Py_DECREF(fnames);
789 return (PyTypeObject*)result;
790}
791
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000792static int add_attributes(PyTypeObject* type, char**attrs, int num_fields)
793{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000794 int i, result;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +0200795 _Py_IDENTIFIER(_attributes);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000796 PyObject *s, *l = PyTuple_New(num_fields);
Benjamin Peterson3e5cd1d2010-06-27 21:45:24 +0000797 if (!l)
798 return 0;
799 for (i = 0; i < num_fields; i++) {
Neal Norwitze4dc3242007-08-25 01:33:49 +0000800 s = PyUnicode_FromString(attrs[i]);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000801 if (!s) {
802 Py_DECREF(l);
803 return 0;
804 }
Neal Norwitz207c9f32008-03-31 04:42:11 +0000805 PyTuple_SET_ITEM(l, i, s);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000806 }
Martin v. Löwis1c67dd92011-10-14 15:16:45 +0200807 result = _PyObject_SetAttrId((PyObject*)type, &PyId__attributes, l) >= 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000808 Py_DECREF(l);
809 return result;
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000810}
811
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000812/* Conversion AST -> Python */
813
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000814static PyObject* ast2obj_list(asdl_seq *seq, PyObject* (*func)(void*))
815{
Benjamin Peterson77fa9372012-05-15 10:10:27 -0700816 Py_ssize_t i, n = asdl_seq_LEN(seq);
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000817 PyObject *result = PyList_New(n);
818 PyObject *value;
819 if (!result)
820 return NULL;
821 for (i = 0; i < n; i++) {
822 value = func(asdl_seq_GET(seq, i));
823 if (!value) {
824 Py_DECREF(result);
825 return NULL;
826 }
827 PyList_SET_ITEM(result, i, value);
828 }
829 return result;
830}
831
832static PyObject* ast2obj_object(void *o)
833{
834 if (!o)
835 o = Py_None;
836 Py_INCREF((PyObject*)o);
837 return (PyObject*)o;
838}
Benjamin Peterson442f2092012-12-06 17:41:04 -0500839#define ast2obj_singleton ast2obj_object
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000840#define ast2obj_identifier ast2obj_object
841#define ast2obj_string ast2obj_object
Benjamin Petersone2498412011-08-09 16:08:39 -0500842#define ast2obj_bytes ast2obj_object
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000843
Thomas Woutersa44f3a32007-02-26 18:20:15 +0000844static PyObject* ast2obj_int(long b)
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000845{
Christian Heimes2b1c5922007-12-02 22:43:00 +0000846 return PyLong_FromLong(b);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000847}
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000848
849/* Conversion Python -> AST */
850
Benjamin Peterson442f2092012-12-06 17:41:04 -0500851static int obj2ast_singleton(PyObject *obj, PyObject** out, PyArena* arena)
852{
853 if (obj != Py_None && obj != Py_True && obj != Py_False) {
854 PyErr_SetString(PyExc_ValueError,
855 "AST singleton must be True, False, or None");
856 return 1;
857 }
858 *out = obj;
859 return 0;
860}
861
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000862static int obj2ast_object(PyObject* obj, PyObject** out, PyArena* arena)
863{
864 if (obj == Py_None)
865 obj = NULL;
866 if (obj)
867 PyArena_AddPyObject(arena, obj);
868 Py_XINCREF(obj);
869 *out = obj;
870 return 0;
871}
872
Benjamin Peterson180e6352011-07-22 11:09:07 -0500873static int obj2ast_identifier(PyObject* obj, PyObject** out, PyArena* arena)
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500874{
Benjamin Peterson180e6352011-07-22 11:09:07 -0500875 if (!PyUnicode_CheckExact(obj) && obj != Py_None) {
876 PyErr_SetString(PyExc_TypeError, "AST identifier must be of type str");
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500877 return 1;
878 }
879 return obj2ast_object(obj, out, arena);
880}
881
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500882static int obj2ast_string(PyObject* obj, PyObject** out, PyArena* arena)
883{
Benjamin Peterson0224d4e2011-08-31 22:13:03 -0400884 if (!PyUnicode_CheckExact(obj) && !PyBytes_CheckExact(obj)) {
Benjamin Peterson180e6352011-07-22 11:09:07 -0500885 PyErr_SetString(PyExc_TypeError, "AST string must be of type str");
886 return 1;
887 }
888 return obj2ast_object(obj, out, arena);
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500889}
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000890
Benjamin Petersone2498412011-08-09 16:08:39 -0500891static int obj2ast_bytes(PyObject* obj, PyObject** out, PyArena* arena)
892{
893 if (!PyBytes_CheckExact(obj)) {
894 PyErr_SetString(PyExc_TypeError, "AST bytes must be of type bytes");
895 return 1;
896 }
897 return obj2ast_object(obj, out, arena);
898}
899
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000900static int obj2ast_int(PyObject* obj, int* out, PyArena* arena)
901{
902 int i;
903 if (!PyLong_Check(obj)) {
Amaury Forgeot d'Arc58e87612011-11-22 21:51:55 +0100904 PyErr_Format(PyExc_ValueError, "invalid integer value: %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000905 return 1;
906 }
907
908 i = (int)PyLong_AsLong(obj);
909 if (i == -1 && PyErr_Occurred())
910 return 1;
911 *out = i;
912 return 0;
913}
914
Benjamin Peterson1a6e0d02008-10-25 15:49:17 +0000915static int add_ast_fields(void)
Benjamin Peterson206e3072008-10-19 14:07:49 +0000916{
917 PyObject *empty_tuple, *d;
918 if (PyType_Ready(&AST_type) < 0)
919 return -1;
920 d = AST_type.tp_dict;
921 empty_tuple = PyTuple_New(0);
922 if (!empty_tuple ||
923 PyDict_SetItemString(d, "_fields", empty_tuple) < 0 ||
924 PyDict_SetItemString(d, "_attributes", empty_tuple) < 0) {
925 Py_XDECREF(empty_tuple);
926 return -1;
927 }
928 Py_DECREF(empty_tuple);
929 return 0;
930}
931
Benjamin Petersonb72406b2013-03-18 23:24:41 -0700932static int exists_not_none(PyObject *obj, _Py_Identifier *id)
933{
Benjamin Peterson7654ab92013-03-18 23:39:53 -0700934 int isnone;
Benjamin Petersonb72406b2013-03-18 23:24:41 -0700935 PyObject *attr = _PyObject_GetAttrId(obj, id);
936 if (!attr) {
937 PyErr_Clear();
938 return 0;
939 }
Benjamin Peterson7654ab92013-03-18 23:39:53 -0700940 isnone = attr == Py_None;
Benjamin Petersonb72406b2013-03-18 23:24:41 -0700941 Py_DECREF(attr);
942 return !isnone;
943}
944
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000945""", 0, reflow=False)
946
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000947 self.emit("static int init_types(void)",0)
948 self.emit("{", 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000949 self.emit("static int initialized;", 1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000950 self.emit("if (initialized) return 1;", 1)
Benjamin Peterson206e3072008-10-19 14:07:49 +0000951 self.emit("if (add_ast_fields() < 0) return 0;", 1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000952 for dfn in mod.dfns:
953 self.visit(dfn)
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000954 self.emit("initialized = 1;", 1)
955 self.emit("return 1;", 1);
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000956 self.emit("}", 0)
957
958 def visitProduct(self, prod, name):
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000959 if prod.fields:
960 fields = name.value+"_fields"
961 else:
962 fields = "NULL"
Neal Norwitz207c9f32008-03-31 04:42:11 +0000963 self.emit('%s_type = make_type("%s", &AST_type, %s, %d);' %
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000964 (name, name, fields, len(prod.fields)), 1)
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000965 self.emit("if (!%s_type) return 0;" % name, 1)
Benjamin Petersoncda75be2013-03-18 10:48:58 -0700966 if prod.attributes:
967 self.emit("if (!add_attributes(%s_type, %s_attributes, %d)) return 0;" %
968 (name, name, len(prod.attributes)), 1)
969 else:
970 self.emit("if (!add_attributes(%s_type, NULL, 0)) return 0;" % name, 1)
Tim Peters710ab3b2006-02-28 18:30:36 +0000971
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000972 def visitSum(self, sum, name):
Neal Norwitz207c9f32008-03-31 04:42:11 +0000973 self.emit('%s_type = make_type("%s", &AST_type, NULL, 0);' %
974 (name, name), 1)
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000975 self.emit("if (!%s_type) return 0;" % name, 1)
976 if sum.attributes:
Tim Peters710ab3b2006-02-28 18:30:36 +0000977 self.emit("if (!add_attributes(%s_type, %s_attributes, %d)) return 0;" %
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000978 (name, name, len(sum.attributes)), 1)
979 else:
980 self.emit("if (!add_attributes(%s_type, NULL, 0)) return 0;" % name, 1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000981 simple = is_simple(sum)
982 for t in sum.types:
983 self.visitConstructor(t, name, simple)
Tim Peters710ab3b2006-02-28 18:30:36 +0000984
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000985 def visitConstructor(self, cons, name, simple):
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000986 if cons.fields:
987 fields = cons.name.value+"_fields"
988 else:
989 fields = "NULL"
Tim Peters710ab3b2006-02-28 18:30:36 +0000990 self.emit('%s_type = make_type("%s", %s_type, %s, %d);' %
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000991 (cons.name, cons.name, name, fields, len(cons.fields)), 1)
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000992 self.emit("if (!%s_type) return 0;" % cons.name, 1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000993 if simple:
994 self.emit("%s_singleton = PyType_GenericNew(%s_type, NULL, NULL);" %
995 (cons.name, cons.name), 1)
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000996 self.emit("if (!%s_singleton) return 0;" % cons.name, 1)
Tim Peters710ab3b2006-02-28 18:30:36 +0000997
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000998
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000999class ASTModuleVisitor(PickleVisitor):
1000
1001 def visitModule(self, mod):
Martin v. Löwis1a214512008-06-11 05:26:20 +00001002 self.emit("static struct PyModuleDef _astmodule = {", 0)
1003 self.emit(' PyModuleDef_HEAD_INIT, "_ast"', 0)
1004 self.emit("};", 0)
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001005 self.emit("PyMODINIT_FUNC", 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00001006 self.emit("PyInit__ast(void)", 0)
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001007 self.emit("{", 0)
1008 self.emit("PyObject *m, *d;", 1)
Martin v. Löwis1a214512008-06-11 05:26:20 +00001009 self.emit("if (!init_types()) return NULL;", 1)
1010 self.emit('m = PyModule_Create(&_astmodule);', 1)
1011 self.emit("if (!m) return NULL;", 1)
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001012 self.emit("d = PyModule_GetDict(m);", 1)
Martin v. Löwis1a214512008-06-11 05:26:20 +00001013 self.emit('if (PyDict_SetItemString(d, "AST", (PyObject*)&AST_type) < 0) return NULL;', 1)
Benjamin Peterson8d89c2a2013-05-20 10:28:48 -07001014 self.emit('if (PyModule_AddIntMacro(m, PyCF_ONLY_AST) < 0)', 1)
Martin v. Löwis1a214512008-06-11 05:26:20 +00001015 self.emit("return NULL;", 2)
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001016 for dfn in mod.dfns:
1017 self.visit(dfn)
Martin v. Löwis1a214512008-06-11 05:26:20 +00001018 self.emit("return m;", 1)
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001019 self.emit("}", 0)
1020
1021 def visitProduct(self, prod, name):
1022 self.addObj(name)
Tim Peters710ab3b2006-02-28 18:30:36 +00001023
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001024 def visitSum(self, sum, name):
1025 self.addObj(name)
1026 for t in sum.types:
1027 self.visitConstructor(t, name)
Tim Peters710ab3b2006-02-28 18:30:36 +00001028
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001029 def visitConstructor(self, cons, name):
1030 self.addObj(cons.name)
Tim Peters710ab3b2006-02-28 18:30:36 +00001031
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001032 def addObj(self, name):
Martin v. Löwis1a214512008-06-11 05:26:20 +00001033 self.emit('if (PyDict_SetItemString(d, "%s", (PyObject*)%s_type) < 0) return NULL;' % (name, name), 1)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001034
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001035
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001036_SPECIALIZED_SEQUENCES = ('stmt', 'expr')
1037
1038def find_sequence(fields, doing_specialization):
1039 """Return True if any field uses a sequence."""
1040 for f in fields:
1041 if f.seq:
1042 if not doing_specialization:
1043 return True
1044 if str(f.type) not in _SPECIALIZED_SEQUENCES:
1045 return True
1046 return False
1047
1048def has_sequence(types, doing_specialization):
1049 for t in types:
1050 if find_sequence(t.fields, doing_specialization):
1051 return True
1052 return False
1053
1054
1055class StaticVisitor(PickleVisitor):
Neal Norwitz7b5a6042005-11-13 19:14:20 +00001056 CODE = '''Very simple, always emit this static code. Overide CODE'''
1057
1058 def visit(self, object):
1059 self.emit(self.CODE, 0, reflow=False)
1060
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001061
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001062class ObjVisitor(PickleVisitor):
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001063
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001064 def func_begin(self, name):
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001065 ctype = get_c_type(name)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001066 self.emit("PyObject*", 0)
1067 self.emit("ast2obj_%s(void* _o)" % (name), 0)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001068 self.emit("{", 0)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001069 self.emit("%s o = (%s)_o;" % (ctype, ctype), 1)
1070 self.emit("PyObject *result = NULL, *value = NULL;", 1)
1071 self.emit('if (!o) {', 1)
1072 self.emit("Py_INCREF(Py_None);", 2)
1073 self.emit('return Py_None;', 2)
1074 self.emit("}", 1)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001075 self.emit('', 0)
1076
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001077 def func_end(self):
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001078 self.emit("return result;", 1)
1079 self.emit("failed:", 0)
1080 self.emit("Py_XDECREF(value);", 1)
1081 self.emit("Py_XDECREF(result);", 1)
1082 self.emit("return NULL;", 1)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001083 self.emit("}", 0)
1084 self.emit("", 0)
1085
1086 def visitSum(self, sum, name):
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001087 if is_simple(sum):
1088 self.simpleSum(sum, name)
1089 return
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001090 self.func_begin(name)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001091 self.emit("switch (o->kind) {", 1)
1092 for i in range(len(sum.types)):
1093 t = sum.types[i]
1094 self.visitConstructor(t, i + 1, name)
1095 self.emit("}", 1)
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001096 for a in sum.attributes:
1097 self.emit("value = ast2obj_%s(o->%s);" % (a.type, a.name), 1)
1098 self.emit("if (!value) goto failed;", 1)
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02001099 self.emit('if (_PyObject_SetAttrId(result, &PyId_%s, value) < 0)' % a.name, 1)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00001100 self.emit('goto failed;', 2)
1101 self.emit('Py_DECREF(value);', 1)
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001102 self.func_end()
Tim Peters710ab3b2006-02-28 18:30:36 +00001103
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001104 def simpleSum(self, sum, name):
1105 self.emit("PyObject* ast2obj_%s(%s_ty o)" % (name, name), 0)
1106 self.emit("{", 0)
1107 self.emit("switch(o) {", 1)
1108 for t in sum.types:
1109 self.emit("case %s:" % t.name, 2)
1110 self.emit("Py_INCREF(%s_singleton);" % t.name, 3)
1111 self.emit("return %s_singleton;" % t.name, 3)
Ezio Melotticb2916a2012-09-30 22:41:37 +03001112 self.emit("default:", 2)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001113 self.emit('/* should never happen, but just in case ... */', 3)
1114 code = "PyErr_Format(PyExc_SystemError, \"unknown %s found\");" % name
1115 self.emit(code, 3, reflow=False)
1116 self.emit("return NULL;", 3)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001117 self.emit("}", 1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001118 self.emit("}", 0)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001119
1120 def visitProduct(self, prod, name):
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001121 self.func_begin(name)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001122 self.emit("result = PyType_GenericNew(%s_type, NULL, NULL);" % name, 1);
1123 self.emit("if (!result) return NULL;", 1)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001124 for field in prod.fields:
1125 self.visitField(field, name, 1, True)
Benjamin Petersoncda75be2013-03-18 10:48:58 -07001126 for a in prod.attributes:
1127 self.emit("value = ast2obj_%s(o->%s);" % (a.type, a.name), 1)
1128 self.emit("if (!value) goto failed;", 1)
1129 self.emit('if (_PyObject_SetAttrId(result, &PyId_%s, value) < 0)' % a.name, 1)
1130 self.emit('goto failed;', 2)
1131 self.emit('Py_DECREF(value);', 1)
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001132 self.func_end()
Tim Peters536cf992005-12-25 23:18:31 +00001133
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001134 def visitConstructor(self, cons, enum, name):
1135 self.emit("case %s_kind:" % cons.name, 1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001136 self.emit("result = PyType_GenericNew(%s_type, NULL, NULL);" % cons.name, 2);
1137 self.emit("if (!result) goto failed;", 2)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001138 for f in cons.fields:
1139 self.visitField(f, cons.name, 2, False)
1140 self.emit("break;", 2)
1141
1142 def visitField(self, field, name, depth, product):
1143 def emit(s, d):
1144 self.emit(s, depth + d)
1145 if product:
1146 value = "o->%s" % field.name
1147 else:
1148 value = "o->v.%s.%s" % (name, field.name)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001149 self.set(field, value, depth)
1150 emit("if (!value) goto failed;", 0)
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02001151 emit('if (_PyObject_SetAttrId(result, &PyId_%s, value) == -1)' % field.name, 0)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001152 emit("goto failed;", 1)
1153 emit("Py_DECREF(value);", 0)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001154
1155 def emitSeq(self, field, value, depth, emit):
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001156 emit("seq = %s;" % value, 0)
1157 emit("n = asdl_seq_LEN(seq);", 0)
1158 emit("value = PyList_New(n);", 0)
1159 emit("if (!value) goto failed;", 0)
1160 emit("for (i = 0; i < n; i++) {", 0)
1161 self.set("value", field, "asdl_seq_GET(seq, i)", depth + 1)
1162 emit("if (!value1) goto failed;", 1)
1163 emit("PyList_SET_ITEM(value, i, value1);", 1)
1164 emit("value1 = NULL;", 1)
1165 emit("}", 0)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001166
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001167 def set(self, field, value, depth):
1168 if field.seq:
Martin v. Löwisce1d5d22006-02-26 20:51:25 +00001169 # XXX should really check for is_simple, but that requires a symbol table
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001170 if field.type.value == "cmpop":
Martin v. Löwisce1d5d22006-02-26 20:51:25 +00001171 # While the sequence elements are stored as void*,
1172 # ast2obj_cmpop expects an enum
1173 self.emit("{", depth)
Benjamin Peterson77fa9372012-05-15 10:10:27 -07001174 self.emit("Py_ssize_t i, n = asdl_seq_LEN(%s);" % value, depth+1)
Martin v. Löwisce1d5d22006-02-26 20:51:25 +00001175 self.emit("value = PyList_New(n);", depth+1)
1176 self.emit("if (!value) goto failed;", depth+1)
1177 self.emit("for(i = 0; i < n; i++)", depth+1)
1178 # This cannot fail, so no need for error handling
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001179 self.emit("PyList_SET_ITEM(value, i, ast2obj_cmpop((cmpop_ty)asdl_seq_GET(%s, i)));" % value,
1180 depth+2, reflow=False)
Martin v. Löwisce1d5d22006-02-26 20:51:25 +00001181 self.emit("}", depth)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001182 else:
Martin v. Löwisce1d5d22006-02-26 20:51:25 +00001183 self.emit("value = ast2obj_list(%s, ast2obj_%s);" % (value, field.type), depth)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001184 else:
1185 ctype = get_c_type(field.type)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001186 self.emit("value = ast2obj_%s(%s);" % (field.type, value), depth, reflow=False)
Tim Peters536cf992005-12-25 23:18:31 +00001187
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001188
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001189class PartingShots(StaticVisitor):
1190
1191 CODE = """
1192PyObject* PyAST_mod2obj(mod_ty t)
1193{
1194 init_types();
1195 return ast2obj_mod(t);
1196}
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001197
Neal Norwitzdb4115f2008-03-31 04:20:05 +00001198/* mode is 0 for "exec", 1 for "eval" and 2 for "single" input */
1199mod_ty PyAST_obj2mod(PyObject* ast, PyArena* arena, int mode)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001200{
1201 mod_ty res;
Neal Norwitzdb4115f2008-03-31 04:20:05 +00001202 PyObject *req_type[] = {(PyObject*)Module_type, (PyObject*)Expression_type,
1203 (PyObject*)Interactive_type};
1204 char *req_name[] = {"Module", "Expression", "Interactive"};
Benjamin Peterson97dd9872009-12-13 01:23:39 +00001205 int isinstance;
Neal Norwitzdb4115f2008-03-31 04:20:05 +00001206 assert(0 <= mode && mode <= 2);
1207
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001208 init_types();
Neal Norwitzdb4115f2008-03-31 04:20:05 +00001209
Benjamin Peterson97dd9872009-12-13 01:23:39 +00001210 isinstance = PyObject_IsInstance(ast, req_type[mode]);
1211 if (isinstance == -1)
1212 return NULL;
1213 if (!isinstance) {
Neal Norwitzdb4115f2008-03-31 04:20:05 +00001214 PyErr_Format(PyExc_TypeError, "expected %s node, got %.400s",
1215 req_name[mode], Py_TYPE(ast)->tp_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001216 return NULL;
1217 }
1218 if (obj2ast_mod(ast, &res, arena) != 0)
1219 return NULL;
1220 else
1221 return res;
1222}
1223
1224int PyAST_Check(PyObject* obj)
1225{
1226 init_types();
Neal Norwitz207c9f32008-03-31 04:42:11 +00001227 return PyObject_IsInstance(obj, (PyObject*)&AST_type);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001228}
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001229"""
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001230
1231class ChainOfVisitors:
1232 def __init__(self, *visitors):
1233 self.visitors = visitors
1234
1235 def visit(self, object):
1236 for v in self.visitors:
1237 v.visit(object)
Neal Norwitz7b5a6042005-11-13 19:14:20 +00001238 v.emit("", 0)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001239
Guido van Rossum805365e2007-05-07 22:24:25 +00001240common_msg = "/* File automatically generated by %s. */\n\n"
Thomas Wouterscf297e42007-02-23 15:07:44 +00001241
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001242def main(srcfile):
Neal Norwitz897ff812005-12-11 21:18:22 +00001243 argv0 = sys.argv[0]
Armin Rigo6b1793f2005-12-14 18:05:14 +00001244 components = argv0.split(os.sep)
1245 argv0 = os.sep.join(components[-2:])
Thomas Wouterscf297e42007-02-23 15:07:44 +00001246 auto_gen_msg = common_msg % argv0
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001247 mod = asdl.parse(srcfile)
1248 if not asdl.check(mod):
1249 sys.exit(1)
1250 if INC_DIR:
1251 p = "%s/%s-ast.h" % (INC_DIR, mod.name)
Benjamin Peterson5a3f49b2011-08-11 14:42:28 -05001252 f = open(p, "w")
Guido van Rossum805365e2007-05-07 22:24:25 +00001253 f.write(auto_gen_msg)
1254 f.write('#include "asdl.h"\n\n')
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001255 c = ChainOfVisitors(TypeDefVisitor(f),
1256 StructVisitor(f),
1257 PrototypeVisitor(f),
1258 )
1259 c.visit(mod)
Guido van Rossum805365e2007-05-07 22:24:25 +00001260 f.write("PyObject* PyAST_mod2obj(mod_ty t);\n")
Neal Norwitzdb4115f2008-03-31 04:20:05 +00001261 f.write("mod_ty PyAST_obj2mod(PyObject* ast, PyArena* arena, int mode);\n")
1262 f.write("int PyAST_Check(PyObject* obj);\n")
Benjamin Peterson5a3f49b2011-08-11 14:42:28 -05001263 f.close()
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001264
1265 if SRC_DIR:
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001266 p = os.path.join(SRC_DIR, str(mod.name) + "-ast.c")
Guido van Rossum2af1d892007-11-23 21:26:49 +00001267 f = open(p, "w")
Guido van Rossum805365e2007-05-07 22:24:25 +00001268 f.write(auto_gen_msg)
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -07001269 f.write('#include <stddef.h>\n')
1270 f.write('\n')
Guido van Rossum805365e2007-05-07 22:24:25 +00001271 f.write('#include "Python.h"\n')
1272 f.write('#include "%s-ast.h"\n' % mod.name)
1273 f.write('\n')
Neal Norwitz207c9f32008-03-31 04:42:11 +00001274 f.write("static PyTypeObject AST_type;\n")
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001275 v = ChainOfVisitors(
1276 PyTypesDeclareVisitor(f),
1277 PyTypesVisitor(f),
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001278 Obj2ModPrototypeVisitor(f),
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001279 FunctionVisitor(f),
1280 ObjVisitor(f),
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001281 Obj2ModVisitor(f),
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001282 ASTModuleVisitor(f),
1283 PartingShots(f),
1284 )
1285 v.visit(mod)
1286 f.close()
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001287
1288if __name__ == "__main__":
1289 import sys
1290 import getopt
1291
1292 INC_DIR = ''
1293 SRC_DIR = ''
1294 opts, args = getopt.getopt(sys.argv[1:], "h:c:")
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001295 if len(opts) != 1:
Guido van Rossum805365e2007-05-07 22:24:25 +00001296 sys.stdout.write("Must specify exactly one output file\n")
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001297 sys.exit(1)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001298 for o, v in opts:
1299 if o == '-h':
1300 INC_DIR = v
1301 if o == '-c':
1302 SRC_DIR = v
1303 if len(args) != 1:
Guido van Rossum805365e2007-05-07 22:24:25 +00001304 sys.stdout.write("Must specify single input file\n")
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001305 sys.exit(1)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001306 main(args[0])