cpython-lru_cache-weakref: Lib/compiler/pycodegen.py annotate

annotate Lib/compiler/pycodegen.py @ 58925:eecf1bef052d 2.6

Merged revisions 78207 via svnmerge from svn+ssh://pythondev@svn.python.org/python/trunk ........ r78207 | ezio.melotti | 2010-02-17 01:26:09 +0200 (Wed, 17 Feb 2010) | 1 line #7930: fix stripid ........

author	Ezio Melotti <ezio.melotti@gmail.com>
date	Tue, 16 Feb 2010 23:29:44 +0000
parents	c77e3119e267
children	751c6dd71b5e 94e79d78dff0

rev	line source
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1 import imp
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	2 import os
11710 38c4d61db7ea rapid evolution towards producing real .pyc files (even though I don't Jeremy Hylton <jeremy@alum.mit.edu> parents: 11705 diff changeset	3 import marshal
11724 71d038db0dd2 now produces valid pyc files for a least a trivial subset of the Jeremy Hylton <jeremy@alum.mit.edu> parents: 11710 diff changeset	4 import struct
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	5 import sys
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	6 from cStringIO import StringIO
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	7
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	8 from compiler import ast, parse, walk, syntax
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	9 from compiler import pyassem, misc, future, symbols
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	10 from compiler.consts import SC_LOCAL, SC_GLOBAL, SC_FREE, SC_CELL
36820 c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	11 from compiler.consts import (CO_VARARGS, CO_VARKEYWORDS, CO_NEWLOCALS,
36821 12bcd16c31ad Hopefully this removes all vestiges of CO_GENERATOR_ALLOWED that can Neal Norwitz <nnorwitz@gmail.com> parents: 36820 diff changeset	12 CO_NESTED, CO_GENERATOR, CO_FUTURE_DIVISION,
45897 c77e3119e267 Fixed compiler module so __future__ print_function is compilable. Eric Smith <eric@trueblade.com> parents: 45703 diff changeset	13 CO_FUTURE_ABSIMPORT, CO_FUTURE_WITH_STATEMENT, CO_FUTURE_PRINT_FUNCTION)
19619 5663b883ea5d Fix _convert_NAME() so that it doesn't store locals for class bodies. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19609 diff changeset	14 from compiler.pyassem import TupleArg
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	15
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	16 # XXX The version-specific code can go, since this code only works with 2.x.
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	17 # Do we have Python 1.x or Python 2.x?
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	18 try:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	19 VERSION = sys.version_info[0]
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	20 except AttributeError:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	21 VERSION = 1
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	22
12563 b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	23 callfunc_opcode_info = {
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	24 # (Have args, Have *args) : opcode
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	25 (0,0) : "CALL_FUNCTION",
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	26 (1,0) : "CALL_FUNCTION_VAR",
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	27 (0,1) : "CALL_FUNCTION_KW",
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	28 (1,1) : "CALL_FUNCTION_VAR_KW",
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	29 }
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	30
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	31 LOOP = 1
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	32 EXCEPT = 2
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	33 TRY_FINALLY = 3
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	34 END_FINALLY = 4
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	35
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	36 def compileFile(filename, display=0):
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	37 f = open(filename, 'U')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	38 buf = f.read()
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	39 f.close()
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	40 mod = Module(buf, filename)
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	41 try:
aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	42 mod.compile(display)
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	43 except SyntaxError:
19943 6773dcfe34a3 Last set of change to get regression tests to pass Jeremy Hylton <jeremy@alum.mit.edu> parents: 19939 diff changeset	44 raise
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	45 else:
aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	46 f = open(filename + "c", "wb")
aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	47 mod.dump(f)
aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	48 f.close()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	49
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	50 def compile(source, filename, mode, flags=None, dont_inherit=None):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	51 """Replacement for builtin compile() function"""
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	52 if flags is not None or dont_inherit is not None:
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	53 raise RuntimeError, "not implemented yet"
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	54
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	55 if mode == "single":
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	56 gen = Interactive(source, filename)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	57 elif mode == "exec":
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	58 gen = Module(source, filename)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	59 elif mode == "eval":
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	60 gen = Expression(source, filename)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	61 else:
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	62 raise ValueError("compile() 3rd arg must be 'exec' or "
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	63 "'eval' or 'single'")
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	64 gen.compile()
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	65 return gen.code
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	66
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	67 class AbstractCompileMode:
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	68
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	69 mode = None # defined by subclass
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	70
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	71 def __init__(self, source, filename):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	72 self.source = source
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	73 self.filename = filename
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	74 self.code = None
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	75
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	76 def _get_tree(self):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	77 tree = parse(self.source, self.mode)
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	78 misc.set_filename(self.filename, tree)
aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	79 syntax.check(tree)
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	80 return tree
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	81
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	82 def compile(self):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	83 pass # implemented by subclass
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	84
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	85 def getCode(self):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	86 return self.code
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	87
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	88 class Expression(AbstractCompileMode):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	89
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	90 mode = "eval"
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	91
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	92 def compile(self):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	93 tree = self._get_tree()
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	94 gen = ExpressionCodeGenerator(tree)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	95 self.code = gen.getCode()
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	96
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	97 class Interactive(AbstractCompileMode):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	98
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	99 mode = "single"
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	100
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	101 def compile(self):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	102 tree = self._get_tree()
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	103 gen = InteractiveCodeGenerator(tree)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	104 self.code = gen.getCode()
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	105
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	106 class Module(AbstractCompileMode):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	107
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	108 mode = "exec"
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	109
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	110 def compile(self, display=0):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	111 tree = self._get_tree()
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	112 gen = ModuleCodeGenerator(tree)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	113 if display:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	114 import pprint
17346 f8fdbeb73667 Add support for future statements Jeremy Hylton <jeremy@alum.mit.edu> parents: 15379 diff changeset	115 print pprint.pprint(tree)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	116 self.code = gen.getCode()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	117
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	118 def dump(self, f):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	119 f.write(self.getPycHeader())
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	120 marshal.dump(self.code, f)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	121
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	122 MAGIC = imp.get_magic()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	123
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	124 def getPycHeader(self):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	125 # compile.c uses marshal to write a long directly, with
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	126 # calling the interface that would also generate a 1-byte code
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	127 # to indicate the type of the value. simplest way to get the
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	128 # same effect is to call marshal and then skip the code.
23710 44b9977d4848 Remove uses of string module and stat.ST_MODE Neal Norwitz <nnorwitz@gmail.com> parents: 21958 diff changeset	129 mtime = os.path.getmtime(self.filename)
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	130 mtime = struct.pack('<i', mtime)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	131 return self.MAGIC + mtime
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	132
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	133 class LocalNameFinder:
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	134 """Find local names in scope"""
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	135 def __init__(self, names=()):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	136 self.names = misc.Set()
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	137 self.globals = misc.Set()
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	138 for name in names:
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	139 self.names.add(name)
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	140
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	141 # XXX list comprehensions and for loops
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	142
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	143 def getLocals(self):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	144 for elt in self.globals.elements():
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	145 if self.names.has_elt(elt):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	146 self.names.remove(elt)
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	147 return self.names
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	148
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	149 def visitDict(self, node):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	150 pass
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	151
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	152 def visitGlobal(self, node):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	153 for name in node.names:
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	154 self.globals.add(name)
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	155
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	156 def visitFunction(self, node):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	157 self.names.add(node.name)
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	158
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	159 def visitLambda(self, node):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	160 pass
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	161
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	162 def visitImport(self, node):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	163 for name, alias in node.names:
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	164 self.names.add(alias or name)
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	165
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	166 def visitFrom(self, node):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	167 for name, alias in node.names:
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	168 self.names.add(alias or name)
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	169
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	170 def visitClass(self, node):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	171 self.names.add(node.name)
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	172
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	173 def visitAssName(self, node):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	174 self.names.add(node.name)
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	175
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	176 def is_constant_false(node):
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	177 if isinstance(node, ast.Const):
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	178 if not node.value:
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	179 return 1
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	180 return 0
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	181
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	182 class CodeGenerator:
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	183 """Defines basic code generator for Python bytecode
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	184
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	185 This class is an abstract base class. Concrete subclasses must
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	186 define an __init__() that defines self.graph and then calls the
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	187 __init__() defined in this class.
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	188
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	189 The concrete class must also define the class attributes
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	190 NameFinder, FunctionGen, and ClassGen. These attributes can be
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	191 defined in the initClass() method, which is a hook for
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	192 initializing these methods after all the classes have been
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	193 defined.
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	194 """
11743 c978a23a4dab add namespace attr to CodeGenerator, can be either MODULE_NAMESPACE or Jeremy Hylton <jeremy@alum.mit.edu> parents: 11734 diff changeset	195
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	196 optimized = 0 # is namespace access optimized?
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	197 __initialized = None
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	198 class_name = None # provide default for instance variable
11751 3f1da05ced55 add support for Lambda nodes Jeremy Hylton <jeremy@alum.mit.edu> parents: 11748 diff changeset	199
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	200 def __init__(self):
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	201 if self.__initialized is None:
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	202 self.initClass()
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	203 self.__class__.__initialized = 1
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	204 self.checkClass()
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	205 self.locals = misc.Stack()
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	206 self.setups = misc.Stack()
14362 a665bdc4922f patch by Neil Schemenauer to improve (fix?) line number generation Jeremy Hylton <jeremy@alum.mit.edu> parents: 14359 diff changeset	207 self.last_lineno = None
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	208 self._setupGraphDelegation()
19936 6944f5b9de03 support true division Jeremy Hylton <jeremy@alum.mit.edu> parents: 19923 diff changeset	209 self._div_op = "BINARY_DIVIDE"
11793 d2ab9553f871 support for arglists with implicit tuple unpacks Jeremy Hylton <jeremy@alum.mit.edu> parents: 11789 diff changeset	210
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	211 # XXX set flags based on future features
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	212 futures = self.get_module().futures
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	213 for feature in futures:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	214 if feature == "division":
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	215 self.graph.setFlag(CO_FUTURE_DIVISION)
19936 6944f5b9de03 support true division Jeremy Hylton <jeremy@alum.mit.edu> parents: 19923 diff changeset	216 self._div_op = "BINARY_TRUE_DIVIDE"
36820 c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	217 elif feature == "absolute_import":
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	218 self.graph.setFlag(CO_FUTURE_ABSIMPORT)
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	219 elif feature == "with_statement":
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	220 self.graph.setFlag(CO_FUTURE_WITH_STATEMENT)
45897 c77e3119e267 Fixed compiler module so __future__ print_function is compilable. Eric Smith <eric@trueblade.com> parents: 45703 diff changeset	221 elif feature == "print_function":
c77e3119e267 Fixed compiler module so __future__ print_function is compilable. Eric Smith <eric@trueblade.com> parents: 45703 diff changeset	222 self.graph.setFlag(CO_FUTURE_PRINT_FUNCTION)
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	223
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	224 def initClass(self):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	225 """This method is called once for each class"""
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	226
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	227 def checkClass(self):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	228 """Verify that class is constructed correctly"""
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	229 try:
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	230 assert hasattr(self, 'graph')
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	231 assert getattr(self, 'NameFinder')
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	232 assert getattr(self, 'FunctionGen')
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	233 assert getattr(self, 'ClassGen')
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	234 except AssertionError, msg:
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	235 intro = "Bad class construction for %s" % self.__class__.__name__
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	236 raise AssertionError, intro
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	237
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	238 def _setupGraphDelegation(self):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	239 self.emit = self.graph.emit
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	240 self.newBlock = self.graph.newBlock
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	241 self.startBlock = self.graph.startBlock
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	242 self.nextBlock = self.graph.nextBlock
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	243 self.setDocstring = self.graph.setDocstring
11793 d2ab9553f871 support for arglists with implicit tuple unpacks Jeremy Hylton <jeremy@alum.mit.edu> parents: 11789 diff changeset	244
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	245 def getCode(self):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	246 """Return a code object"""
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	247 return self.graph.getCode()
11754 dffbd921eeed change MODULE_NAMESPACE/FUNCTION_NAMESPACE stuff to have a single flag Jeremy Hylton <jeremy@alum.mit.edu> parents: 11751 diff changeset	248
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	249 def mangle(self, name):
a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	250 if self.class_name is not None:
a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	251 return misc.mangle(name, self.class_name)
a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	252 else:
a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	253 return name
a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	254
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	255 def parseSymbols(self, tree):
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	256 s = symbols.SymbolVisitor()
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	257 walk(tree, s)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	258 return s.scopes
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	259
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	260 def get_module(self):
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	261 raise RuntimeError, "should be implemented by subclasses"
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	262
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	263 # Next five methods handle name access
11724 71d038db0dd2 now produces valid pyc files for a least a trivial subset of the Jeremy Hylton <jeremy@alum.mit.edu> parents: 11710 diff changeset	264
11727 1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	265 def isLocalName(self, name):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	266 return self.locals.top().has_elt(name)
11727 1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	267
11743 c978a23a4dab add namespace attr to CodeGenerator, can be either MODULE_NAMESPACE or Jeremy Hylton <jeremy@alum.mit.edu> parents: 11734 diff changeset	268 def storeName(self, name):
c978a23a4dab add namespace attr to CodeGenerator, can be either MODULE_NAMESPACE or Jeremy Hylton <jeremy@alum.mit.edu> parents: 11734 diff changeset	269 self._nameOp('STORE', name)
c978a23a4dab add namespace attr to CodeGenerator, can be either MODULE_NAMESPACE or Jeremy Hylton <jeremy@alum.mit.edu> parents: 11734 diff changeset	270
c978a23a4dab add namespace attr to CodeGenerator, can be either MODULE_NAMESPACE or Jeremy Hylton <jeremy@alum.mit.edu> parents: 11734 diff changeset	271 def loadName(self, name):
c978a23a4dab add namespace attr to CodeGenerator, can be either MODULE_NAMESPACE or Jeremy Hylton <jeremy@alum.mit.edu> parents: 11734 diff changeset	272 self._nameOp('LOAD', name)
c978a23a4dab add namespace attr to CodeGenerator, can be either MODULE_NAMESPACE or Jeremy Hylton <jeremy@alum.mit.edu> parents: 11734 diff changeset	273
c978a23a4dab add namespace attr to CodeGenerator, can be either MODULE_NAMESPACE or Jeremy Hylton <jeremy@alum.mit.edu> parents: 11734 diff changeset	274 def delName(self, name):
c978a23a4dab add namespace attr to CodeGenerator, can be either MODULE_NAMESPACE or Jeremy Hylton <jeremy@alum.mit.edu> parents: 11734 diff changeset	275 self._nameOp('DELETE', name)
c978a23a4dab add namespace attr to CodeGenerator, can be either MODULE_NAMESPACE or Jeremy Hylton <jeremy@alum.mit.edu> parents: 11734 diff changeset	276
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	277 def _nameOp(self, prefix, name):
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	278 name = self.mangle(name)
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	279 scope = self.scope.check_name(name)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	280 if scope == SC_LOCAL:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	281 if not self.optimized:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	282 self.emit(prefix + '_NAME', name)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	283 else:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	284 self.emit(prefix + '_FAST', name)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	285 elif scope == SC_GLOBAL:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	286 if not self.optimized:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	287 self.emit(prefix + '_NAME', name)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	288 else:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	289 self.emit(prefix + '_GLOBAL', name)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	290 elif scope == SC_FREE or scope == SC_CELL:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	291 self.emit(prefix + '_DEREF', name)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	292 else:
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	293 raise RuntimeError, "unsupported scope for var %s: %d" % \
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	294 (name, scope)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	295
17459 e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	296 def _implicitNameOp(self, prefix, name):
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	297 """Emit name ops for names generated implicitly by for loops
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	298
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	299 The interpreter generates names that start with a period or
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	300 dollar sign. The symbol table ignores these names because
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	301 they aren't present in the program text.
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	302 """
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	303 if self.optimized:
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	304 self.emit(prefix + '_FAST', name)
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	305 else:
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	306 self.emit(prefix + '_NAME', name)
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	307
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	308 # The set_lineno() function and the explicit emit() calls for
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	309 # SET_LINENO below are only used to generate the line number table.
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	310 # As of Python 2.3, the interpreter does not have a SET_LINENO
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	311 # instruction. pyassem treats SET_LINENO opcodes as a special case.
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	312
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	313 def set_lineno(self, node, force=False):
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	314 """Emit SET_LINENO if necessary.
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	315
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	316 The instruction is considered necessary if the node has a
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	317 lineno attribute and it is different than the last lineno
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	318 emitted.
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	319
c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	320 Returns true if SET_LINENO was emitted.
c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	321
c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	322 There are no rules for when an AST node should have a lineno
c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	323 attribute. The transformer and AST code need to be reviewed
c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	324 and a consistent policy implemented and documented. Until
c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	325 then, this method works around missing line numbers.
c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	326 """
c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	327 lineno = getattr(node, 'lineno', None)
17453 f3b6ac128de9 pyassem.py: Jeremy Hylton <jeremy@alum.mit.edu> parents: 17449 diff changeset	328 if lineno is not None and (lineno != self.last_lineno
f3b6ac128de9 pyassem.py: Jeremy Hylton <jeremy@alum.mit.edu> parents: 17449 diff changeset	329 or force):
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	330 self.emit('SET_LINENO', lineno)
14362 a665bdc4922f patch by Neil Schemenauer to improve (fix?) line number generation Jeremy Hylton <jeremy@alum.mit.edu> parents: 14359 diff changeset	331 self.last_lineno = lineno
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	332 return True
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	333 return False
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	334
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	335 # The first few visitor methods handle nodes that generator new
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	336 # code objects. They use class attributes to determine what
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	337 # specialized code generators to use.
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	338
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	339 NameFinder = LocalNameFinder
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	340 FunctionGen = None
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	341 ClassGen = None
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	342
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	343 def visitModule(self, node):
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	344 self.scopes = self.parseSymbols(node)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	345 self.scope = self.scopes[node]
17459 e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	346 self.emit('SET_LINENO', 0)
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	347 if node.doc:
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	348 self.emit('LOAD_CONST', node.doc)
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	349 self.storeName('__doc__')
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	350 lnf = walk(node.node, self.NameFinder(), verbose=0)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	351 self.locals.push(lnf.getLocals())
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	352 self.visit(node.node)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	353 self.emit('LOAD_CONST', None)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	354 self.emit('RETURN_VALUE')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	355
21522 cb55f15b2730 Merge of the release22 branch changes back into the trunk. Barry Warsaw <barry@python.org> parents: 21059 diff changeset	356 def visitExpression(self, node):
cb55f15b2730 Merge of the release22 branch changes back into the trunk. Barry Warsaw <barry@python.org> parents: 21059 diff changeset	357 self.set_lineno(node)
cb55f15b2730 Merge of the release22 branch changes back into the trunk. Barry Warsaw <barry@python.org> parents: 21059 diff changeset	358 self.scopes = self.parseSymbols(node)
cb55f15b2730 Merge of the release22 branch changes back into the trunk. Barry Warsaw <barry@python.org> parents: 21059 diff changeset	359 self.scope = self.scopes[node]
cb55f15b2730 Merge of the release22 branch changes back into the trunk. Barry Warsaw <barry@python.org> parents: 21059 diff changeset	360 self.visit(node.node)
cb55f15b2730 Merge of the release22 branch changes back into the trunk. Barry Warsaw <barry@python.org> parents: 21059 diff changeset	361 self.emit('RETURN_VALUE')
cb55f15b2730 Merge of the release22 branch changes back into the trunk. Barry Warsaw <barry@python.org> parents: 21059 diff changeset	362
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	363 def visitFunction(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	364 self._visitFuncOrLambda(node, isLambda=0)
17410 f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	365 if node.doc:
f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	366 self.setDocstring(node.doc)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	367 self.storeName(node.name)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	368
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	369 def visitLambda(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	370 self._visitFuncOrLambda(node, isLambda=1)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	371
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	372 def _visitFuncOrLambda(self, node, isLambda=0):
32888 c7e03cf8333e PEP-0318, @decorator-style. In Guido's words: Anthony Baxter <anthonybaxter@gmail.com> parents: 32528 diff changeset	373 if not isLambda and node.decorators:
33166 2e8d2bdd33de This is Mark Russell's patch: Michael W. Hudson <mwh@python.net> parents: 33020 diff changeset	374 for decorator in node.decorators.nodes:
32888 c7e03cf8333e PEP-0318, @decorator-style. In Guido's words: Anthony Baxter <anthonybaxter@gmail.com> parents: 32528 diff changeset	375 self.visit(decorator)
c7e03cf8333e PEP-0318, @decorator-style. In Guido's words: Anthony Baxter <anthonybaxter@gmail.com> parents: 32528 diff changeset	376 ndecorators = len(node.decorators.nodes)
c7e03cf8333e PEP-0318, @decorator-style. In Guido's words: Anthony Baxter <anthonybaxter@gmail.com> parents: 32528 diff changeset	377 else:
c7e03cf8333e PEP-0318, @decorator-style. In Guido's words: Anthony Baxter <anthonybaxter@gmail.com> parents: 32528 diff changeset	378 ndecorators = 0
33204 c18aca7e81cb Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 33166 diff changeset	379
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	380 gen = self.FunctionGen(node, self.scopes, isLambda,
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	381 self.class_name, self.get_module())
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	382 walk(node.code, gen)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	383 gen.finish()
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	384 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	385 for default in node.defaults:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	386 self.visit(default)
39370 eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	387 self._makeClosure(gen, len(node.defaults))
32888 c7e03cf8333e PEP-0318, @decorator-style. In Guido's words: Anthony Baxter <anthonybaxter@gmail.com> parents: 32528 diff changeset	388 for i in range(ndecorators):
c7e03cf8333e PEP-0318, @decorator-style. In Guido's words: Anthony Baxter <anthonybaxter@gmail.com> parents: 32528 diff changeset	389 self.emit('CALL_FUNCTION', 1)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	390
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	391 def visitClass(self, node):
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	392 gen = self.ClassGen(node, self.scopes,
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	393 self.get_module())
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	394 walk(node.code, gen)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	395 gen.finish()
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	396 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	397 self.emit('LOAD_CONST', node.name)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	398 for base in node.bases:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	399 self.visit(base)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	400 self.emit('BUILD_TUPLE', len(node.bases))
39370 eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	401 self._makeClosure(gen, 0)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	402 self.emit('CALL_FUNCTION', 0)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	403 self.emit('BUILD_CLASS')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	404 self.storeName(node.name)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	405
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	406 # The rest are standard visitor methods
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	407
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	408 # The next few implement control-flow statements
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	409
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	410 def visitIf(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	411 end = self.newBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	412 numtests = len(node.tests)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	413 for i in range(numtests):
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	414 test, suite = node.tests[i]
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	415 if is_constant_false(test):
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	416 # XXX will need to check generator stuff here
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	417 continue
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	418 self.set_lineno(test)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	419 self.visit(test)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	420 nextTest = self.newBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	421 self.emit('JUMP_IF_FALSE', nextTest)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	422 self.nextBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	423 self.emit('POP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	424 self.visit(suite)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	425 self.emit('JUMP_FORWARD', end)
15379 7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	426 self.startBlock(nextTest)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	427 self.emit('POP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	428 if node.else_:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	429 self.visit(node.else_)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	430 self.nextBlock(end)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	431
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	432 def visitWhile(self, node):
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	433 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	434
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	435 loop = self.newBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	436 else_ = self.newBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	437
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	438 after = self.newBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	439 self.emit('SETUP_LOOP', after)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	440
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	441 self.nextBlock(loop)
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	442 self.setups.push((LOOP, loop))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	443
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	444 self.set_lineno(node, force=True)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	445 self.visit(node.test)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	446 self.emit('JUMP_IF_FALSE', else_ or after)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	447
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	448 self.nextBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	449 self.emit('POP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	450 self.visit(node.body)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	451 self.emit('JUMP_ABSOLUTE', loop)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	452
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	453 self.startBlock(else_) # or just the POPs if not else clause
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	454 self.emit('POP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	455 self.emit('POP_BLOCK')
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	456 self.setups.pop()
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	457 if node.else_:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	458 self.visit(node.else_)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	459 self.nextBlock(after)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	460
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	461 def visitFor(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	462 start = self.newBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	463 anchor = self.newBlock()
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	464 after = self.newBlock()
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	465 self.setups.push((LOOP, start))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	466
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	467 self.set_lineno(node)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	468 self.emit('SETUP_LOOP', after)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	469 self.visit(node.list)
19531 deacd60272be Generate FOR_ITER-based loops instead of old FOR_LOOP-based loops Jeremy Hylton <jeremy@alum.mit.edu> parents: 19522 diff changeset	470 self.emit('GET_ITER')
deacd60272be Generate FOR_ITER-based loops instead of old FOR_LOOP-based loops Jeremy Hylton <jeremy@alum.mit.edu> parents: 19522 diff changeset	471
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	472 self.nextBlock(start)
17460 99c471ced170 Pop loop off the loop stack before handling the loop's else clause. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17459 diff changeset	473 self.set_lineno(node, force=1)
19531 deacd60272be Generate FOR_ITER-based loops instead of old FOR_LOOP-based loops Jeremy Hylton <jeremy@alum.mit.edu> parents: 19522 diff changeset	474 self.emit('FOR_ITER', anchor)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	475 self.visit(node.assign)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	476 self.visit(node.body)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	477 self.emit('JUMP_ABSOLUTE', start)
19534 de10151a7f63 Make sure the JUMP_ABSOLUTE and POP_BLOCK at the end of a for loop are Jeremy Hylton <jeremy@alum.mit.edu> parents: 19531 diff changeset	478 self.nextBlock(anchor)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	479 self.emit('POP_BLOCK')
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	480 self.setups.pop()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	481 if node.else_:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	482 self.visit(node.else_)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	483 self.nextBlock(after)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	484
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	485 def visitBreak(self, node):
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	486 if not self.setups:
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	487 raise SyntaxError, "'break' outside loop (%s, %d)" % \
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	488 (node.filename, node.lineno)
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	489 self.set_lineno(node)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	490 self.emit('BREAK_LOOP')
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	491
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	492 def visitContinue(self, node):
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	493 if not self.setups:
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	494 raise SyntaxError, "'continue' outside loop (%s, %d)" % \
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	495 (node.filename, node.lineno)
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	496 kind, block = self.setups.top()
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	497 if kind == LOOP:
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	498 self.set_lineno(node)
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	499 self.emit('JUMP_ABSOLUTE', block)
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	500 self.nextBlock()
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	501 elif kind == EXCEPT or kind == TRY_FINALLY:
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	502 self.set_lineno(node)
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	503 # find the block that starts the loop
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	504 top = len(self.setups)
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	505 while top > 0:
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	506 top = top - 1
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	507 kind, loop_block = self.setups[top]
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	508 if kind == LOOP:
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	509 break
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	510 if kind != LOOP:
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	511 raise SyntaxError, "'continue' outside loop (%s, %d)" % \
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	512 (node.filename, node.lineno)
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	513 self.emit('CONTINUE_LOOP', loop_block)
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	514 self.nextBlock()
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	515 elif kind == END_FINALLY:
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	516 msg = "'continue' not allowed inside 'finally' clause (%s, %d)"
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	517 raise SyntaxError, msg % (node.filename, node.lineno)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	518
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	519 def visitTest(self, node, jump):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	520 end = self.newBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	521 for child in node.nodes[:-1]:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	522 self.visit(child)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	523 self.emit(jump, end)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	524 self.nextBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	525 self.emit('POP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	526 self.visit(node.nodes[-1])
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	527 self.nextBlock(end)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	528
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	529 def visitAnd(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	530 self.visitTest(node, 'JUMP_IF_FALSE')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	531
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	532 def visitOr(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	533 self.visitTest(node, 'JUMP_IF_TRUE')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	534
36820 c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	535 def visitIfExp(self, node):
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	536 endblock = self.newBlock()
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	537 elseblock = self.newBlock()
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	538 self.visit(node.test)
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	539 self.emit('JUMP_IF_FALSE', elseblock)
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	540 self.emit('POP_TOP')
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	541 self.visit(node.then)
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	542 self.emit('JUMP_FORWARD', endblock)
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	543 self.nextBlock(elseblock)
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	544 self.emit('POP_TOP')
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	545 self.visit(node.else_)
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	546 self.nextBlock(endblock)
c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	547
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	548 def visitCompare(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	549 self.visit(node.expr)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	550 cleanup = self.newBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	551 for op, code in node.ops[:-1]:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	552 self.visit(code)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	553 self.emit('DUP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	554 self.emit('ROT_THREE')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	555 self.emit('COMPARE_OP', op)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	556 self.emit('JUMP_IF_FALSE', cleanup)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	557 self.nextBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	558 self.emit('POP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	559 # now do the last comparison
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	560 if node.ops:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	561 op, code = node.ops[-1]
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	562 self.visit(code)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	563 self.emit('COMPARE_OP', op)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	564 if len(node.ops) > 1:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	565 end = self.newBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	566 self.emit('JUMP_FORWARD', end)
15379 7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	567 self.startBlock(cleanup)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	568 self.emit('ROT_TWO')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	569 self.emit('POP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	570 self.nextBlock(end)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	571
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	572 # list comprehensions
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	573 __list_count = 0
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	574
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	575 def visitListComp(self, node):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	576 self.set_lineno(node)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	577 # setup list
40242 4d833fca3670 Bug #1586448: the compiler module now emits the same bytecode for Georg Brandl <georg@python.org> parents: 39532 diff changeset	578 tmpname = "$list%d" % self.__list_count
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	579 self.__list_count = self.__list_count + 1
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	580 self.emit('BUILD_LIST', 0)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	581 self.emit('DUP_TOP')
40242 4d833fca3670 Bug #1586448: the compiler module now emits the same bytecode for Georg Brandl <georg@python.org> parents: 39532 diff changeset	582 self._implicitNameOp('STORE', tmpname)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	583
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	584 stack = []
17453 f3b6ac128de9 pyassem.py: Jeremy Hylton <jeremy@alum.mit.edu> parents: 17449 diff changeset	585 for i, for_ in zip(range(len(node.quals)), node.quals):
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	586 start, anchor = self.visit(for_)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	587 cont = None
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	588 for if_ in for_.ifs:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	589 if cont is None:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	590 cont = self.newBlock()
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	591 self.visit(if_, cont)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	592 stack.insert(0, (start, cont, anchor))
17453 f3b6ac128de9 pyassem.py: Jeremy Hylton <jeremy@alum.mit.edu> parents: 17449 diff changeset	593
40242 4d833fca3670 Bug #1586448: the compiler module now emits the same bytecode for Georg Brandl <georg@python.org> parents: 39532 diff changeset	594 self._implicitNameOp('LOAD', tmpname)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	595 self.visit(node.expr)
40243 2a1eae6c0eb8 Remove trailing comma. Georg Brandl <georg@python.org> parents: 40242 diff changeset	596 self.emit('LIST_APPEND')
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	597
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	598 for start, cont, anchor in stack:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	599 if cont:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	600 skip_one = self.newBlock()
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	601 self.emit('JUMP_FORWARD', skip_one)
15379 7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	602 self.startBlock(cont)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	603 self.emit('POP_TOP')
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	604 self.nextBlock(skip_one)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	605 self.emit('JUMP_ABSOLUTE', start)
15379 7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	606 self.startBlock(anchor)
40242 4d833fca3670 Bug #1586448: the compiler module now emits the same bytecode for Georg Brandl <georg@python.org> parents: 39532 diff changeset	607 self._implicitNameOp('DELETE', tmpname)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	608
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	609 self.__list_count = self.__list_count - 1
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	610
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	611 def visitListCompFor(self, node):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	612 start = self.newBlock()
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	613 anchor = self.newBlock()
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	614
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	615 self.visit(node.list)
19619 5663b883ea5d Fix _convert_NAME() so that it doesn't store locals for class bodies. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19609 diff changeset	616 self.emit('GET_ITER')
17459 e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	617 self.nextBlock(start)
26983 2c67169f6e13 Replace all but one explicit emit('SET_LINENO') with call to set_lineno(). Jeremy Hylton <jeremy@alum.mit.edu> parents: 26979 diff changeset	618 self.set_lineno(node, force=True)
19619 5663b883ea5d Fix _convert_NAME() so that it doesn't store locals for class bodies. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19609 diff changeset	619 self.emit('FOR_ITER', anchor)
15379 7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	620 self.nextBlock()
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	621 self.visit(node.assign)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	622 return start, anchor
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	623
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	624 def visitListCompIf(self, node, branch):
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	625 self.set_lineno(node, force=True)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	626 self.visit(node.test)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	627 self.emit('JUMP_IF_FALSE', branch)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	628 self.newBlock()
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	629 self.emit('POP_TOP')
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	630
39370 eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	631 def _makeClosure(self, gen, args):
eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	632 frees = gen.scope.get_free_vars()
eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	633 if frees:
eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	634 for name in frees:
eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	635 self.emit('LOAD_CLOSURE', name)
eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	636 self.emit('BUILD_TUPLE', len(frees))
eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	637 self.emit('LOAD_CONST', gen)
eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	638 self.emit('MAKE_CLOSURE', args)
eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	639 else:
eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	640 self.emit('LOAD_CONST', gen)
eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	641 self.emit('MAKE_FUNCTION', args)
eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	642
32121 302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	643 def visitGenExpr(self, node):
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	644 gen = GenExprCodeGenerator(node, self.scopes, self.class_name,
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	645 self.get_module())
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	646 walk(node.code, gen)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	647 gen.finish()
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	648 self.set_lineno(node)
39370 eb23d096d694 Fix the 'compiler' package to generate correct code for MAKE_CLOSURE. Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 38084 diff changeset	649 self._makeClosure(gen, 0)
32121 302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	650 # precomputation of outmost iterable
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	651 self.visit(node.code.quals[0].iter)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	652 self.emit('GET_ITER')
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	653 self.emit('CALL_FUNCTION', 1)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	654
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	655 def visitGenExprInner(self, node):
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	656 self.set_lineno(node)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	657 # setup list
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	658
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	659 stack = []
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	660 for i, for_ in zip(range(len(node.quals)), node.quals):
39532 a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	661 start, anchor, end = self.visit(for_)
32121 302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	662 cont = None
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	663 for if_ in for_.ifs:
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	664 if cont is None:
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	665 cont = self.newBlock()
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	666 self.visit(if_, cont)
39532 a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	667 stack.insert(0, (start, cont, anchor, end))
32121 302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	668
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	669 self.visit(node.expr)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	670 self.emit('YIELD_VALUE')
39532 a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	671 self.emit('POP_TOP')
32121 302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	672
39532 a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	673 for start, cont, anchor, end in stack:
32121 302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	674 if cont:
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	675 skip_one = self.newBlock()
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	676 self.emit('JUMP_FORWARD', skip_one)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	677 self.startBlock(cont)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	678 self.emit('POP_TOP')
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	679 self.nextBlock(skip_one)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	680 self.emit('JUMP_ABSOLUTE', start)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	681 self.startBlock(anchor)
39532 a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	682 self.emit('POP_BLOCK')
a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	683 self.setups.pop()
a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	684 self.startBlock(end)
a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	685
32121 302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	686 self.emit('LOAD_CONST', None)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	687
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	688 def visitGenExprFor(self, node):
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	689 start = self.newBlock()
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	690 anchor = self.newBlock()
39532 a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	691 end = self.newBlock()
a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	692
a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	693 self.setups.push((LOOP, start))
a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	694 self.emit('SETUP_LOOP', end)
32528 67539cfcf09f Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 32121 diff changeset	695
32121 302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	696 if node.is_outmost:
39532 a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	697 self.loadName('.0')
32121 302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	698 else:
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	699 self.visit(node.iter)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	700 self.emit('GET_ITER')
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	701
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	702 self.nextBlock(start)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	703 self.set_lineno(node, force=True)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	704 self.emit('FOR_ITER', anchor)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	705 self.nextBlock()
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	706 self.visit(node.assign)
39532 a6afe0eda711 Fix a bug in the ``compiler`` package that caused invalid code to be Neil Schemenauer <nascheme@enme.ucalgary.ca> parents: 39370 diff changeset	707 return start, anchor, end
32121 302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	708
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	709 def visitGenExprIf(self, node, branch):
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	710 self.set_lineno(node, force=True)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	711 self.visit(node.test)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	712 self.emit('JUMP_IF_FALSE', branch)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	713 self.newBlock()
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	714 self.emit('POP_TOP')
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	715
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	716 # exception related
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	717
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	718 def visitAssert(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	719 # XXX would be interesting to implement this via a
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	720 # transformation of the AST before this stage
33020 ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	721 if __debug__:
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	722 end = self.newBlock()
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	723 self.set_lineno(node)
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	724 # XXX AssertionError appears to be special case -- it is always
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	725 # loaded as a global even if there is a local name. I guess this
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	726 # is a sort of renaming op.
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	727 self.nextBlock()
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	728 self.visit(node.test)
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	729 self.emit('JUMP_IF_TRUE', end)
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	730 self.nextBlock()
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	731 self.emit('POP_TOP')
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	732 self.emit('LOAD_GLOBAL', 'AssertionError')
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	733 if node.fail:
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	734 self.visit(node.fail)
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	735 self.emit('RAISE_VARARGS', 2)
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	736 else:
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	737 self.emit('RAISE_VARARGS', 1)
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	738 self.nextBlock(end)
ad818338f8c3 SF patch 836879. Jeremy Hylton <jeremy@alum.mit.edu> parents: 32949 diff changeset	739 self.emit('POP_TOP')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	740
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	741 def visitRaise(self, node):
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	742 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	743 n = 0
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	744 if node.expr1:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	745 self.visit(node.expr1)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	746 n = n + 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	747 if node.expr2:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	748 self.visit(node.expr2)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	749 n = n + 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	750 if node.expr3:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	751 self.visit(node.expr3)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	752 n = n + 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	753 self.emit('RAISE_VARARGS', n)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	754
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	755 def visitTryExcept(self, node):
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	756 body = self.newBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	757 handlers = self.newBlock()
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	758 end = self.newBlock()
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	759 if node.else_:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	760 lElse = self.newBlock()
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	761 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	762 lElse = end
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	763 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	764 self.emit('SETUP_EXCEPT', handlers)
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	765 self.nextBlock(body)
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	766 self.setups.push((EXCEPT, body))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	767 self.visit(node.body)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	768 self.emit('POP_BLOCK')
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	769 self.setups.pop()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	770 self.emit('JUMP_FORWARD', lElse)
15379 7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	771 self.startBlock(handlers)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	772
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	773 last = len(node.handlers) - 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	774 for i in range(len(node.handlers)):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	775 expr, target, body = node.handlers[i]
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	776 self.set_lineno(expr)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	777 if expr:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	778 self.emit('DUP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	779 self.visit(expr)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	780 self.emit('COMPARE_OP', 'exception match')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	781 next = self.newBlock()
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	782 self.emit('JUMP_IF_FALSE', next)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	783 self.nextBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	784 self.emit('POP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	785 self.emit('POP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	786 if target:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	787 self.visit(target)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	788 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	789 self.emit('POP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	790 self.emit('POP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	791 self.visit(body)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	792 self.emit('JUMP_FORWARD', end)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	793 if expr:
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	794 self.nextBlock(next)
15379 7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	795 else:
7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	796 self.nextBlock()
19531 deacd60272be Generate FOR_ITER-based loops instead of old FOR_LOOP-based loops Jeremy Hylton <jeremy@alum.mit.edu> parents: 19522 diff changeset	797 if expr: # XXX
deacd60272be Generate FOR_ITER-based loops instead of old FOR_LOOP-based loops Jeremy Hylton <jeremy@alum.mit.edu> parents: 19522 diff changeset	798 self.emit('POP_TOP')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	799 self.emit('END_FINALLY')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	800 if node.else_:
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	801 self.nextBlock(lElse)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	802 self.visit(node.else_)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	803 self.nextBlock(end)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	804
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	805 def visitTryFinally(self, node):
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	806 body = self.newBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	807 final = self.newBlock()
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	808 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	809 self.emit('SETUP_FINALLY', final)
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	810 self.nextBlock(body)
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	811 self.setups.push((TRY_FINALLY, body))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	812 self.visit(node.body)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	813 self.emit('POP_BLOCK')
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	814 self.setups.pop()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	815 self.emit('LOAD_CONST', None)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	816 self.nextBlock(final)
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	817 self.setups.push((END_FINALLY, final))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	818 self.visit(node.final)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	819 self.emit('END_FINALLY')
19573 725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	820 self.setups.pop()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	821
36706 b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	822 __with_count = 0
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	823
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	824 def visitWith(self, node):
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	825 body = self.newBlock()
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	826 final = self.newBlock()
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	827 valuevar = "$value%d" % self.__with_count
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	828 self.__with_count += 1
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	829 self.set_lineno(node)
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	830 self.visit(node.expr)
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	831 self.emit('DUP_TOP')
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	832 self.emit('LOAD_ATTR', '__exit__')
45703 343597c7d682 Speed up with statements by storing the __exit__ method on the stack instead of in a temp variable (bumps the magic number for pyc files) Nick Coghlan <ncoghlan@gmail.com> parents: 40719 diff changeset	833 self.emit('ROT_TWO')
36706 b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	834 self.emit('LOAD_ATTR', '__enter__')
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	835 self.emit('CALL_FUNCTION', 0)
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	836 if node.vars is None:
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	837 self.emit('POP_TOP')
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	838 else:
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	839 self._implicitNameOp('STORE', valuevar)
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	840 self.emit('SETUP_FINALLY', final)
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	841 self.nextBlock(body)
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	842 self.setups.push((TRY_FINALLY, body))
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	843 if node.vars is not None:
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	844 self._implicitNameOp('LOAD', valuevar)
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	845 self._implicitNameOp('DELETE', valuevar)
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	846 self.visit(node.vars)
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	847 self.visit(node.body)
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	848 self.emit('POP_BLOCK')
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	849 self.setups.pop()
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	850 self.emit('LOAD_CONST', None)
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	851 self.nextBlock(final)
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	852 self.setups.push((END_FINALLY, final))
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	853 self.emit('WITH_CLEANUP')
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	854 self.emit('END_FINALLY')
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	855 self.setups.pop()
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	856 self.__with_count -= 1
b6b1c44d9110 Update the compiler package to compile the with-statement. Guido van Rossum <guido@python.org> parents: 35946 diff changeset	857
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	858 # misc
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	859
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	860 def visitDiscard(self, node):
17412 3704be3db302 [finishing fix from earlier checkins] Jeremy Hylton <jeremy@alum.mit.edu> parents: 17410 diff changeset	861 self.set_lineno(node)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	862 self.visit(node.expr)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	863 self.emit('POP_TOP')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	864
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	865 def visitConst(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	866 self.emit('LOAD_CONST', node.value)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	867
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	868 def visitKeyword(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	869 self.emit('LOAD_CONST', node.name)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	870 self.visit(node.expr)
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	871
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	872 def visitGlobal(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	873 # no code to generate
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	874 pass
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	875
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	876 def visitName(self, node):
14362 a665bdc4922f patch by Neil Schemenauer to improve (fix?) line number generation Jeremy Hylton <jeremy@alum.mit.edu> parents: 14359 diff changeset	877 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	878 self.loadName(node.name)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	879
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	880 def visitPass(self, node):
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	881 self.set_lineno(node)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	882
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	883 def visitImport(self, node):
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	884 self.set_lineno(node)
36823 7a6b174ddef2 Fix compiler breakage related to absolute imports Neal Norwitz <nnorwitz@gmail.com> parents: 36821 diff changeset	885 level = 0 if self.graph.checkFlag(CO_FUTURE_ABSIMPORT) else -1
14359 35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	886 for name, alias in node.names:
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	887 if VERSION > 1:
36820 c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	888 self.emit('LOAD_CONST', level)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	889 self.emit('LOAD_CONST', None)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	890 self.emit('IMPORT_NAME', name)
23710 44b9977d4848 Remove uses of string module and stat.ST_MODE Neal Norwitz <nnorwitz@gmail.com> parents: 21958 diff changeset	891 mod = name.split(".")[0]
29602 caa8c120e2cb Jacob Hallen cornered me here at EuroPython and got me to look at Michael W. Hudson <mwh@python.net> parents: 26983 diff changeset	892 if alias:
caa8c120e2cb Jacob Hallen cornered me here at EuroPython and got me to look at Michael W. Hudson <mwh@python.net> parents: 26983 diff changeset	893 self._resolveDots(name)
caa8c120e2cb Jacob Hallen cornered me here at EuroPython and got me to look at Michael W. Hudson <mwh@python.net> parents: 26983 diff changeset	894 self.storeName(alias)
caa8c120e2cb Jacob Hallen cornered me here at EuroPython and got me to look at Michael W. Hudson <mwh@python.net> parents: 26983 diff changeset	895 else:
caa8c120e2cb Jacob Hallen cornered me here at EuroPython and got me to look at Michael W. Hudson <mwh@python.net> parents: 26983 diff changeset	896 self.storeName(mod)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	897
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	898 def visitFrom(self, node):
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	899 self.set_lineno(node)
36820 c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	900 level = node.level
36823 7a6b174ddef2 Fix compiler breakage related to absolute imports Neal Norwitz <nnorwitz@gmail.com> parents: 36821 diff changeset	901 if level == 0 and not self.graph.checkFlag(CO_FUTURE_ABSIMPORT):
36820 c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	902 level = -1
14359 35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	903 fromlist = map(lambda (name, alias): name, node.names)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	904 if VERSION > 1:
36820 c1c46210b4b7 Add support for absolute/relative imports and if/else expressions: Thomas Wouters <thomas@python.org> parents: 36706 diff changeset	905 self.emit('LOAD_CONST', level)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	906 self.emit('LOAD_CONST', tuple(fromlist))
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	907 self.emit('IMPORT_NAME', node.modname)
14359 35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	908 for name, alias in node.names:
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	909 if VERSION > 1:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	910 if name == '*':
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	911 self.namespace = 0
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	912 self.emit('IMPORT_STAR')
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	913 # There can only be one name w/ from ... import *
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	914 assert len(node.names) == 1
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	915 return
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	916 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	917 self.emit('IMPORT_FROM', name)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	918 self._resolveDots(name)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	919 self.storeName(alias or name)
15185 100beebde9df Fix SF bug #116263: support for from .. import * Jeremy Hylton <jeremy@alum.mit.edu> parents: 14362 diff changeset	920 else:
100beebde9df Fix SF bug #116263: support for from .. import * Jeremy Hylton <jeremy@alum.mit.edu> parents: 14362 diff changeset	921 self.emit('IMPORT_FROM', name)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	922 self.emit('POP_TOP')
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	923
14359 35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	924 def _resolveDots(self, name):
23710 44b9977d4848 Remove uses of string module and stat.ST_MODE Neal Norwitz <nnorwitz@gmail.com> parents: 21958 diff changeset	925 elts = name.split(".")
14359 35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	926 if len(elts) == 1:
35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	927 return
35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	928 for elt in elts[1:]:
35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	929 self.emit('LOAD_ATTR', elt)
35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	930
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	931 def visitGetattr(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	932 self.visit(node.expr)
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	933 self.emit('LOAD_ATTR', self.mangle(node.attrname))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	934
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	935 # next five implement assignments
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	936
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	937 def visitAssign(self, node):
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	938 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	939 self.visit(node.expr)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	940 dups = len(node.nodes) - 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	941 for i in range(len(node.nodes)):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	942 elt = node.nodes[i]
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	943 if i < dups:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	944 self.emit('DUP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	945 if isinstance(elt, ast.Node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	946 self.visit(elt)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	947
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	948 def visitAssName(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	949 if node.flags == 'OP_ASSIGN':
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	950 self.storeName(node.name)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	951 elif node.flags == 'OP_DELETE':
19561 81168ef4cb65 Support // and //= Jeremy Hylton <jeremy@alum.mit.edu> parents: 19534 diff changeset	952 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	953 self.delName(node.name)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	954 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	955 print "oops", node.flags
11754 dffbd921eeed change MODULE_NAMESPACE/FUNCTION_NAMESPACE stuff to have a single flag Jeremy Hylton <jeremy@alum.mit.edu> parents: 11751 diff changeset	956
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	957 def visitAssAttr(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	958 self.visit(node.expr)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	959 if node.flags == 'OP_ASSIGN':
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	960 self.emit('STORE_ATTR', self.mangle(node.attrname))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	961 elif node.flags == 'OP_DELETE':
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	962 self.emit('DELETE_ATTR', self.mangle(node.attrname))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	963 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	964 print "warning: unexpected flags:", node.flags
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	965 print node
11751 3f1da05ced55 add support for Lambda nodes Jeremy Hylton <jeremy@alum.mit.edu> parents: 11748 diff changeset	966
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	967 def _visitAssSequence(self, node, op='UNPACK_SEQUENCE'):
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	968 if findOp(node) != 'OP_DELETE':
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	969 self.emit(op, len(node.nodes))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	970 for child in node.nodes:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	971 self.visit(child)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	972
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	973 if VERSION > 1:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	974 visitAssTuple = _visitAssSequence
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	975 visitAssList = _visitAssSequence
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	976 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	977 def visitAssTuple(self, node):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	978 self._visitAssSequence(node, 'UNPACK_TUPLE')
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	979
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	980 def visitAssList(self, node):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	981 self._visitAssSequence(node, 'UNPACK_LIST')
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	982
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	983 # augmented assignment
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	984
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	985 def visitAugAssign(self, node):
19519 99e953c6709c emit SET_LINENO for augmented assignments Jeremy Hylton <jeremy@alum.mit.edu> parents: 19518 diff changeset	986 self.set_lineno(node)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	987 aug_node = wrap_aug(node.node)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	988 self.visit(aug_node, "load")
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	989 self.visit(node.expr)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	990 self.emit(self._augmented_opcode[node.op])
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	991 self.visit(aug_node, "store")
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	992
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	993 _augmented_opcode = {
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	994 '+=' : 'INPLACE_ADD',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	995 '-=' : 'INPLACE_SUBTRACT',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	996 '*=' : 'INPLACE_MULTIPLY',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	997 '/=' : 'INPLACE_DIVIDE',
19561 81168ef4cb65 Support // and //= Jeremy Hylton <jeremy@alum.mit.edu> parents: 19534 diff changeset	998 '//=': 'INPLACE_FLOOR_DIVIDE',
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	999 '%=' : 'INPLACE_MODULO',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1000 '**=': 'INPLACE_POWER',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1001 '>>=': 'INPLACE_RSHIFT',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1002 '<<=': 'INPLACE_LSHIFT',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1003 '&=' : 'INPLACE_AND',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1004 '^=' : 'INPLACE_XOR',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1005 '\|=' : 'INPLACE_OR',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1006 }
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1007
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1008 def visitAugName(self, node, mode):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1009 if mode == "load":
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1010 self.loadName(node.name)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1011 elif mode == "store":
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1012 self.storeName(node.name)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1013
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1014 def visitAugGetattr(self, node, mode):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1015 if mode == "load":
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1016 self.visit(node.expr)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1017 self.emit('DUP_TOP')
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	1018 self.emit('LOAD_ATTR', self.mangle(node.attrname))
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1019 elif mode == "store":
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1020 self.emit('ROT_TWO')
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	1021 self.emit('STORE_ATTR', self.mangle(node.attrname))
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1022
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1023 def visitAugSlice(self, node, mode):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1024 if mode == "load":
17413 bad2daeb7440 typo Jeremy Hylton <jeremy@alum.mit.edu> parents: 17412 diff changeset	1025 self.visitSlice(node, 1)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1026 elif mode == "store":
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1027 slice = 0
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1028 if node.lower:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1029 slice = slice \| 1
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1030 if node.upper:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1031 slice = slice \| 2
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1032 if slice == 0:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1033 self.emit('ROT_TWO')
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1034 elif slice == 3:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1035 self.emit('ROT_FOUR')
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1036 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1037 self.emit('ROT_THREE')
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1038 self.emit('STORE_SLICE+%d' % slice)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1039
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1040 def visitAugSubscript(self, node, mode):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1041 if mode == "load":
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1042 self.visitSubscript(node, 1)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1043 elif mode == "store":
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1044 self.emit('ROT_THREE')
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1045 self.emit('STORE_SUBSCR')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1046
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1047 def visitExec(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1048 self.visit(node.expr)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1049 if node.locals is None:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1050 self.emit('LOAD_CONST', None)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1051 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1052 self.visit(node.locals)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1053 if node.globals is None:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1054 self.emit('DUP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1055 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1056 self.visit(node.globals)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1057 self.emit('EXEC_STMT')
11751 3f1da05ced55 add support for Lambda nodes Jeremy Hylton <jeremy@alum.mit.edu> parents: 11748 diff changeset	1058
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1059 def visitCallFunc(self, node):
11754 dffbd921eeed change MODULE_NAMESPACE/FUNCTION_NAMESPACE stuff to have a single flag Jeremy Hylton <jeremy@alum.mit.edu> parents: 11751 diff changeset	1060 pos = 0
dffbd921eeed change MODULE_NAMESPACE/FUNCTION_NAMESPACE stuff to have a single flag Jeremy Hylton <jeremy@alum.mit.edu> parents: 11751 diff changeset	1061 kw = 0
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	1062 self.set_lineno(node)
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1063 self.visit(node.node)
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1064 for arg in node.args:
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1065 self.visit(arg)
11754 dffbd921eeed change MODULE_NAMESPACE/FUNCTION_NAMESPACE stuff to have a single flag Jeremy Hylton <jeremy@alum.mit.edu> parents: 11751 diff changeset	1066 if isinstance(arg, ast.Keyword):
dffbd921eeed change MODULE_NAMESPACE/FUNCTION_NAMESPACE stuff to have a single flag Jeremy Hylton <jeremy@alum.mit.edu> parents: 11751 diff changeset	1067 kw = kw + 1
dffbd921eeed change MODULE_NAMESPACE/FUNCTION_NAMESPACE stuff to have a single flag Jeremy Hylton <jeremy@alum.mit.edu> parents: 11751 diff changeset	1068 else:
dffbd921eeed change MODULE_NAMESPACE/FUNCTION_NAMESPACE stuff to have a single flag Jeremy Hylton <jeremy@alum.mit.edu> parents: 11751 diff changeset	1069 pos = pos + 1
12563 b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	1070 if node.star_args is not None:
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	1071 self.visit(node.star_args)
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	1072 if node.dstar_args is not None:
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	1073 self.visit(node.dstar_args)
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	1074 have_star = node.star_args is not None
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	1075 have_dstar = node.dstar_args is not None
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	1076 opcode = callfunc_opcode_info[have_star, have_dstar]
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	1077 self.emit(opcode, kw << 8 \| pos)
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1078
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	1079 def visitPrint(self, node, newline=0):
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	1080 self.set_lineno(node)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1081 if node.dest:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1082 self.visit(node.dest)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1083 for child in node.nodes:
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1084 if node.dest:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1085 self.emit('DUP_TOP')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1086 self.visit(child)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1087 if node.dest:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1088 self.emit('ROT_TWO')
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1089 self.emit('PRINT_ITEM_TO')
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1090 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1091 self.emit('PRINT_ITEM')
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	1092 if node.dest and not newline:
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	1093 self.emit('POP_TOP')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1094
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1095 def visitPrintnl(self, node):
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	1096 self.visitPrint(node, newline=1)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1097 if node.dest:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1098 self.emit('PRINT_NEWLINE_TO')
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1099 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1100 self.emit('PRINT_NEWLINE')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1101
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1102 def visitReturn(self, node):
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	1103 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1104 self.visit(node.value)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1105 self.emit('RETURN_VALUE')
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1106
19923 cb35d0f704de Add code generator for yield stmt Jeremy Hylton <jeremy@alum.mit.edu> parents: 19920 diff changeset	1107 def visitYield(self, node):
cb35d0f704de Add code generator for yield stmt Jeremy Hylton <jeremy@alum.mit.edu> parents: 19920 diff changeset	1108 self.set_lineno(node)
cb35d0f704de Add code generator for yield stmt Jeremy Hylton <jeremy@alum.mit.edu> parents: 19920 diff changeset	1109 self.visit(node.value)
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	1110 self.emit('YIELD_VALUE')
19923 cb35d0f704de Add code generator for yield stmt Jeremy Hylton <jeremy@alum.mit.edu> parents: 19920 diff changeset	1111
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1112 # slice and subscript stuff
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1113
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1114 def visitSlice(self, node, aug_flag=None):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1115 # aug_flag is used by visitAugSlice
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1116 self.visit(node.expr)
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1117 slice = 0
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1118 if node.lower:
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1119 self.visit(node.lower)
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1120 slice = slice \| 1
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1121 if node.upper:
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1122 self.visit(node.upper)
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1123 slice = slice \| 2
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1124 if aug_flag:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1125 if slice == 0:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1126 self.emit('DUP_TOP')
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1127 elif slice == 3:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1128 self.emit('DUP_TOPX', 3)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1129 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1130 self.emit('DUP_TOPX', 2)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1131 if node.flags == 'OP_APPLY':
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1132 self.emit('SLICE+%d' % slice)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1133 elif node.flags == 'OP_ASSIGN':
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1134 self.emit('STORE_SLICE+%d' % slice)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1135 elif node.flags == 'OP_DELETE':
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1136 self.emit('DELETE_SLICE+%d' % slice)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1137 else:
11789 860524182b9c finish first impl of code generator Jeremy Hylton <jeremy@alum.mit.edu> parents: 11766 diff changeset	1138 print "weird slice", node.flags
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1139 raise
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1140
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1141 def visitSubscript(self, node, aug_flag=None):
11727 1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1142 self.visit(node.expr)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1143 for sub in node.subs:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1144 self.visit(sub)
36980 c8ffc913d014 Teach the compiler module about augmented assignment to tuple subscripts Nick Coghlan <ncoghlan@gmail.com> parents: 36926 diff changeset	1145 if len(node.subs) > 1:
c8ffc913d014 Teach the compiler module about augmented assignment to tuple subscripts Nick Coghlan <ncoghlan@gmail.com> parents: 36926 diff changeset	1146 self.emit('BUILD_TUPLE', len(node.subs))
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1147 if aug_flag:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1148 self.emit('DUP_TOPX', 2)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1149 if node.flags == 'OP_APPLY':
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1150 self.emit('BINARY_SUBSCR')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1151 elif node.flags == 'OP_ASSIGN':
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1152 self.emit('STORE_SUBSCR')
11808 cf8fa92ebf3d the previous quick hack to fix def foo((x,y)) failed on some cases Jeremy Hylton <jeremy@alum.mit.edu> parents: 11806 diff changeset	1153 elif node.flags == 'OP_DELETE':
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1154 self.emit('DELETE_SUBSCR')
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1155
11927 79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1156 # binary ops
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1157
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1158 def binaryOp(self, node, op):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1159 self.visit(node.left)
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1160 self.visit(node.right)
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1161 self.emit(op)
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1162
d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1163 def visitAdd(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1164 return self.binaryOp(node, 'BINARY_ADD')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1165
d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1166 def visitSub(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1167 return self.binaryOp(node, 'BINARY_SUBTRACT')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1168
d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1169 def visitMul(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1170 return self.binaryOp(node, 'BINARY_MULTIPLY')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1171
d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1172 def visitDiv(self, node):
19936 6944f5b9de03 support true division Jeremy Hylton <jeremy@alum.mit.edu> parents: 19923 diff changeset	1173 return self.binaryOp(node, self._div_op)
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1174
19561 81168ef4cb65 Support // and //= Jeremy Hylton <jeremy@alum.mit.edu> parents: 19534 diff changeset	1175 def visitFloorDiv(self, node):
81168ef4cb65 Support // and //= Jeremy Hylton <jeremy@alum.mit.edu> parents: 19534 diff changeset	1176 return self.binaryOp(node, 'BINARY_FLOOR_DIVIDE')
81168ef4cb65 Support // and //= Jeremy Hylton <jeremy@alum.mit.edu> parents: 19534 diff changeset	1177
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1178 def visitMod(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1179 return self.binaryOp(node, 'BINARY_MODULO')
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1180
11761 62fa8ec8595f looks like everything is working except for try/except (pystone Jeremy Hylton <jeremy@alum.mit.edu> parents: 11755 diff changeset	1181 def visitPower(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1182 return self.binaryOp(node, 'BINARY_POWER')
11761 62fa8ec8595f looks like everything is working except for try/except (pystone Jeremy Hylton <jeremy@alum.mit.edu> parents: 11755 diff changeset	1183
62fa8ec8595f looks like everything is working except for try/except (pystone Jeremy Hylton <jeremy@alum.mit.edu> parents: 11755 diff changeset	1184 def visitLeftShift(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1185 return self.binaryOp(node, 'BINARY_LSHIFT')
11761 62fa8ec8595f looks like everything is working except for try/except (pystone Jeremy Hylton <jeremy@alum.mit.edu> parents: 11755 diff changeset	1186
62fa8ec8595f looks like everything is working except for try/except (pystone Jeremy Hylton <jeremy@alum.mit.edu> parents: 11755 diff changeset	1187 def visitRightShift(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1188 return self.binaryOp(node, 'BINARY_RSHIFT')
11761 62fa8ec8595f looks like everything is working except for try/except (pystone Jeremy Hylton <jeremy@alum.mit.edu> parents: 11755 diff changeset	1189
11927 79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1190 # unary ops
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1191
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1192 def unaryOp(self, node, op):
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1193 self.visit(node.expr)
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1194 self.emit(op)
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1195
11761 62fa8ec8595f looks like everything is working except for try/except (pystone Jeremy Hylton <jeremy@alum.mit.edu> parents: 11755 diff changeset	1196 def visitInvert(self, node):
62fa8ec8595f looks like everything is working except for try/except (pystone Jeremy Hylton <jeremy@alum.mit.edu> parents: 11755 diff changeset	1197 return self.unaryOp(node, 'UNARY_INVERT')
62fa8ec8595f looks like everything is working except for try/except (pystone Jeremy Hylton <jeremy@alum.mit.edu> parents: 11755 diff changeset	1198
11727 1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1199 def visitUnarySub(self, node):
1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1200 return self.unaryOp(node, 'UNARY_NEGATIVE')
1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1201
1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1202 def visitUnaryAdd(self, node):
1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1203 return self.unaryOp(node, 'UNARY_POSITIVE')
1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1204
1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1205 def visitUnaryInvert(self, node):
1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1206 return self.unaryOp(node, 'UNARY_INVERT')
1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1207
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1208 def visitNot(self, node):
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1209 return self.unaryOp(node, 'UNARY_NOT')
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1210
11727 1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1211 def visitBackquote(self, node):
1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1212 return self.unaryOp(node, 'UNARY_CONVERT')
1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1213
11927 79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1214 # bit ops
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1215
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1216 def bitOp(self, nodes, op):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1217 self.visit(nodes[0])
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1218 for node in nodes[1:]:
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1219 self.visit(node)
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1220 self.emit(op)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1221
57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1222 def visitBitand(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1223 return self.bitOp(node.nodes, 'BINARY_AND')
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1224
57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1225 def visitBitor(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1226 return self.bitOp(node.nodes, 'BINARY_OR')
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1227
57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1228 def visitBitxor(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1229 return self.bitOp(node.nodes, 'BINARY_XOR')
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1230
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1231 # object constructors
11727 1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1232
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1233 def visitEllipsis(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1234 self.emit('LOAD_CONST', Ellipsis)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1235
11727 1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1236 def visitTuple(self, node):
19567 b98e4346257b Generate SET_LINENO for list and tuple literals when the open paren Jeremy Hylton <jeremy@alum.mit.edu> parents: 19561 diff changeset	1237 self.set_lineno(node)
11727 1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1238 for elt in node.nodes:
1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1239 self.visit(elt)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1240 self.emit('BUILD_TUPLE', len(node.nodes))
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1241
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1242 def visitList(self, node):
19567 b98e4346257b Generate SET_LINENO for list and tuple literals when the open paren Jeremy Hylton <jeremy@alum.mit.edu> parents: 19561 diff changeset	1243 self.set_lineno(node)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1244 for elt in node.nodes:
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1245 self.visit(elt)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1246 self.emit('BUILD_LIST', len(node.nodes))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1247
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1248 def visitSliceobj(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1249 for child in node.nodes:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1250 self.visit(child)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1251 self.emit('BUILD_SLICE', len(node.nodes))
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1252
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1253 def visitDict(self, node):
26983 2c67169f6e13 Replace all but one explicit emit('SET_LINENO') with call to set_lineno(). Jeremy Hylton <jeremy@alum.mit.edu> parents: 26979 diff changeset	1254 self.set_lineno(node)
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1255 self.emit('BUILD_MAP', 0)
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1256 for k, v in node.items:
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1257 self.emit('DUP_TOP')
26670 1312297887a3 Fixing bug Gustavo Niemeyer <gustavo@niemeyer.net> parents: 23710 diff changeset	1258 self.visit(k)
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1259 self.visit(v)
26670 1312297887a3 Fixing bug Gustavo Niemeyer <gustavo@niemeyer.net> parents: 23710 diff changeset	1260 self.emit('ROT_THREE')
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1261 self.emit('STORE_SUBSCR')
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1262
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1263 class NestedScopeMixin:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1264 """Defines initClass() for nested scoping (Python 2.2-compatible)"""
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1265 def initClass(self):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1266 self.__class__.NameFinder = LocalNameFinder
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1267 self.__class__.FunctionGen = FunctionCodeGenerator
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1268 self.__class__.ClassGen = ClassCodeGenerator
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1269
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1270 class ModuleCodeGenerator(NestedScopeMixin, CodeGenerator):
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1271 __super_init = CodeGenerator.__init__
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1272
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1273 scopes = None
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1274
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1275 def __init__(self, tree):
aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1276 self.graph = pyassem.PyFlowGraph("<module>", tree.filename)
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1277 self.futures = future.find_futures(tree)
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1278 self.__super_init()
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1279 walk(tree, self)
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1280
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1281 def get_module(self):
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1282 return self
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1283
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1284 class ExpressionCodeGenerator(NestedScopeMixin, CodeGenerator):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1285 __super_init = CodeGenerator.__init__
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1286
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1287 scopes = None
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1288 futures = ()
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1289
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1290 def __init__(self, tree):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1291 self.graph = pyassem.PyFlowGraph("<expression>", tree.filename)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1292 self.__super_init()
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1293 walk(tree, self)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1294
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1295 def get_module(self):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1296 return self
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1297
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1298 class InteractiveCodeGenerator(NestedScopeMixin, CodeGenerator):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1299
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1300 __super_init = CodeGenerator.__init__
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1301
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1302 scopes = None
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1303 futures = ()
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1304
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1305 def __init__(self, tree):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1306 self.graph = pyassem.PyFlowGraph("<interactive>", tree.filename)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1307 self.__super_init()
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1308 self.set_lineno(tree)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1309 walk(tree, self)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1310 self.emit('RETURN_VALUE')
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1311
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1312 def get_module(self):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1313 return self
21958 5ed782bdbb2a Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 21522 diff changeset	1314
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1315 def visitDiscard(self, node):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1316 # XXX Discard means it's an expression. Perhaps this is a bad
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1317 # name.
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1318 self.visit(node.expr)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1319 self.emit('PRINT_EXPR')
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1320
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1321 class AbstractFunctionCode:
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1322 optimized = 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1323 lambdaCount = 0
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1324
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1325 def __init__(self, func, scopes, isLambda, class_name, mod):
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	1326 self.class_name = class_name
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1327 self.module = mod
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1328 if isLambda:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1329 klass = FunctionCodeGenerator
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1330 name = "<lambda.%d>" % klass.lambdaCount
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1331 klass.lambdaCount = klass.lambdaCount + 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1332 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1333 name = func.name
32121 302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1334
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1335 args, hasTupleArg = generateArgList(func.argnames)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1336 self.graph = pyassem.PyFlowGraph(name, func.filename, args,
a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1337 optimized=1)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1338 self.isLambda = isLambda
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1339 self.super_init()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1340
17410 f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	1341 if not isLambda and func.doc:
f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	1342 self.setDocstring(func.doc)
f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	1343
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	1344 lnf = walk(func.code, self.NameFinder(args), verbose=0)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1345 self.locals.push(lnf.getLocals())
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1346 if func.varargs:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1347 self.graph.setFlag(CO_VARARGS)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1348 if func.kwargs:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1349 self.graph.setFlag(CO_VARKEYWORDS)
13864 c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	1350 self.set_lineno(func)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1351 if hasTupleArg:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1352 self.generateArgUnpack(func.argnames)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1353
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1354 def get_module(self):
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1355 return self.module
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1356
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1357 def finish(self):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1358 self.graph.startExitBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1359 if not self.isLambda:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1360 self.emit('LOAD_CONST', None)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1361 self.emit('RETURN_VALUE')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1362
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1363 def generateArgUnpack(self, args):
17449 51c8ce1a2e25 Revise handling of tuple arguments so that the variables names match Jeremy Hylton <jeremy@alum.mit.edu> parents: 17438 diff changeset	1364 for i in range(len(args)):
51c8ce1a2e25 Revise handling of tuple arguments so that the variables names match Jeremy Hylton <jeremy@alum.mit.edu> parents: 17438 diff changeset	1365 arg = args[i]
35946 b2eb1589c22b No need for types, use isinstance Neal Norwitz <nnorwitz@gmail.com> parents: 33204 diff changeset	1366 if isinstance(arg, tuple):
17449 51c8ce1a2e25 Revise handling of tuple arguments so that the variables names match Jeremy Hylton <jeremy@alum.mit.edu> parents: 17438 diff changeset	1367 self.emit('LOAD_FAST', '.%d' % (i * 2))
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1368 self.unpackSequence(arg)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1369
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1370 def unpackSequence(self, tup):
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1371 if VERSION > 1:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1372 self.emit('UNPACK_SEQUENCE', len(tup))
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1373 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1374 self.emit('UNPACK_TUPLE', len(tup))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1375 for elt in tup:
35946 b2eb1589c22b No need for types, use isinstance Neal Norwitz <nnorwitz@gmail.com> parents: 33204 diff changeset	1376 if isinstance(elt, tuple):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1377 self.unpackSequence(elt)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1378 else:
17438 41a423294cb5 Fix unpackSequence() to use _nameOp() rather than LOAD_FAST Jeremy Hylton <jeremy@alum.mit.edu> parents: 17437 diff changeset	1379 self._nameOp('STORE', elt)
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1380
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1381 unpackTuple = unpackSequence
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1382
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1383 class FunctionCodeGenerator(NestedScopeMixin, AbstractFunctionCode,
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1384 CodeGenerator):
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1385 super_init = CodeGenerator.__init__ # call be other init
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1386 scopes = None
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1387
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1388 __super_init = AbstractFunctionCode.__init__
11710 38c4d61db7ea rapid evolution towards producing real .pyc files (even though I don't Jeremy Hylton <jeremy@alum.mit.edu> parents: 11705 diff changeset	1389
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1390 def __init__(self, func, scopes, isLambda, class_name, mod):
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1391 self.scopes = scopes
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1392 self.scope = scopes[func]
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1393 self.__super_init(func, scopes, isLambda, class_name, mod)
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1394 self.graph.setFreeVars(self.scope.get_free_vars())
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1395 self.graph.setCellVars(self.scope.get_cell_vars())
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	1396 if self.scope.generator is not None:
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	1397 self.graph.setFlag(CO_GENERATOR)
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1398
32121 302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1399 class GenExprCodeGenerator(NestedScopeMixin, AbstractFunctionCode,
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1400 CodeGenerator):
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1401 super_init = CodeGenerator.__init__ # call be other init
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1402 scopes = None
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1403
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1404 __super_init = AbstractFunctionCode.__init__
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1405
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1406 def __init__(self, gexp, scopes, class_name, mod):
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1407 self.scopes = scopes
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1408 self.scope = scopes[gexp]
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1409 self.__super_init(gexp, scopes, 1, class_name, mod)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1410 self.graph.setFreeVars(self.scope.get_free_vars())
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1411 self.graph.setCellVars(self.scope.get_cell_vars())
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1412 self.graph.setFlag(CO_GENERATOR)
302d399cfd37 SF patch #872326: Generator expression implementation Raymond Hettinger <python@rcn.com> parents: 31729 diff changeset	1413
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1414 class AbstractClassCode:
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1415
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1416 def __init__(self, klass, scopes, module):
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	1417 self.class_name = klass.name
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1418 self.module = module
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1419 self.graph = pyassem.PyFlowGraph(klass.name, klass.filename,
19561 81168ef4cb65 Support // and //= Jeremy Hylton <jeremy@alum.mit.edu> parents: 19534 diff changeset	1420 optimized=0, klass=1)
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1421 self.super_init()
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	1422 lnf = walk(klass.code, self.NameFinder(), verbose=0)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1423 self.locals.push(lnf.getLocals())
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1424 self.graph.setFlag(CO_NEWLOCALS)
17410 f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	1425 if klass.doc:
f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	1426 self.setDocstring(klass.doc)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1427
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1428 def get_module(self):
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1429 return self.module
17437 02d69a887038 Inside a class scope always use LOAD_NAME, STORE_NAME, DEL_NAME Jeremy Hylton <jeremy@alum.mit.edu> parents: 17436 diff changeset	1430
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1431 def finish(self):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1432 self.graph.startExitBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1433 self.emit('LOAD_LOCALS')
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1434 self.emit('RETURN_VALUE')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1435
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1436 class ClassCodeGenerator(NestedScopeMixin, AbstractClassCode, CodeGenerator):
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1437 super_init = CodeGenerator.__init__
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1438 scopes = None
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1439
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1440 __super_init = AbstractClassCode.__init__
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1441
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1442 def __init__(self, klass, scopes, module):
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1443 self.scopes = scopes
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1444 self.scope = scopes[klass]
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1445 self.__super_init(klass, scopes, module)
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1446 self.graph.setFreeVars(self.scope.get_free_vars())
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1447 self.graph.setCellVars(self.scope.get_cell_vars())
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	1448 self.set_lineno(klass)
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	1449 self.emit("LOAD_GLOBAL", "__name__")
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	1450 self.storeName("__module__")
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	1451 if klass.doc:
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	1452 self.emit("LOAD_CONST", klass.doc)
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	1453 self.storeName('__doc__')
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1454
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1455 def generateArgList(arglist):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1456 """Generate an arg list marking TupleArgs"""
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1457 args = []
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1458 extra = []
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1459 count = 0
17449 51c8ce1a2e25 Revise handling of tuple arguments so that the variables names match Jeremy Hylton <jeremy@alum.mit.edu> parents: 17438 diff changeset	1460 for i in range(len(arglist)):
51c8ce1a2e25 Revise handling of tuple arguments so that the variables names match Jeremy Hylton <jeremy@alum.mit.edu> parents: 17438 diff changeset	1461 elt = arglist[i]
35946 b2eb1589c22b No need for types, use isinstance Neal Norwitz <nnorwitz@gmail.com> parents: 33204 diff changeset	1462 if isinstance(elt, str):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1463 args.append(elt)
35946 b2eb1589c22b No need for types, use isinstance Neal Norwitz <nnorwitz@gmail.com> parents: 33204 diff changeset	1464 elif isinstance(elt, tuple):
17449 51c8ce1a2e25 Revise handling of tuple arguments so that the variables names match Jeremy Hylton <jeremy@alum.mit.edu> parents: 17438 diff changeset	1465 args.append(TupleArg(i * 2, elt))
51c8ce1a2e25 Revise handling of tuple arguments so that the variables names match Jeremy Hylton <jeremy@alum.mit.edu> parents: 17438 diff changeset	1466 extra.extend(misc.flatten(elt))
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1467 count = count + 1
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1468 else:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1469 raise ValueError, "unexpect argument type:", elt
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1470 return args + extra, count
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1471
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1472 def findOp(node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1473 """Find the op (DELETE, LOAD, STORE) in an AssTuple tree"""
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1474 v = OpFinder()
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	1475 walk(node, v, verbose=0)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1476 return v.op
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1477
11808 cf8fa92ebf3d the previous quick hack to fix def foo((x,y)) failed on some cases Jeremy Hylton <jeremy@alum.mit.edu> parents: 11806 diff changeset	1478 class OpFinder:
cf8fa92ebf3d the previous quick hack to fix def foo((x,y)) failed on some cases Jeremy Hylton <jeremy@alum.mit.edu> parents: 11806 diff changeset	1479 def __init__(self):
cf8fa92ebf3d the previous quick hack to fix def foo((x,y)) failed on some cases Jeremy Hylton <jeremy@alum.mit.edu> parents: 11806 diff changeset	1480 self.op = None
cf8fa92ebf3d the previous quick hack to fix def foo((x,y)) failed on some cases Jeremy Hylton <jeremy@alum.mit.edu> parents: 11806 diff changeset	1481 def visitAssName(self, node):
cf8fa92ebf3d the previous quick hack to fix def foo((x,y)) failed on some cases Jeremy Hylton <jeremy@alum.mit.edu> parents: 11806 diff changeset	1482 if self.op is None:
cf8fa92ebf3d the previous quick hack to fix def foo((x,y)) failed on some cases Jeremy Hylton <jeremy@alum.mit.edu> parents: 11806 diff changeset	1483 self.op = node.flags
cf8fa92ebf3d the previous quick hack to fix def foo((x,y)) failed on some cases Jeremy Hylton <jeremy@alum.mit.edu> parents: 11806 diff changeset	1484 elif self.op != node.flags:
cf8fa92ebf3d the previous quick hack to fix def foo((x,y)) failed on some cases Jeremy Hylton <jeremy@alum.mit.edu> parents: 11806 diff changeset	1485 raise ValueError, "mixed ops in stmt"
17454 d91577783335 Add support for visitAssAttr to findOp(). Jeremy Hylton <jeremy@alum.mit.edu> parents: 17453 diff changeset	1486 visitAssAttr = visitAssName
20925 ee1437f5de03 Fix SF bug #479186: compiler generates bad code for "del" Jeremy Hylton <jeremy@alum.mit.edu> parents: 20547 diff changeset	1487 visitSubscript = visitAssName
11808 cf8fa92ebf3d the previous quick hack to fix def foo((x,y)) failed on some cases Jeremy Hylton <jeremy@alum.mit.edu> parents: 11806 diff changeset	1488
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1489 class Delegator:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1490 """Base class to support delegation for augmented assignment nodes
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1491
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1492 To generator code for augmented assignments, we use the following
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1493 wrapper classes. In visitAugAssign, the left-hand expression node
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1494 is visited twice. The first time the visit uses the normal method
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1495 for that node . The second time the visit uses a different method
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1496 that generates the appropriate code to perform the assignment.
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1497 These delegator classes wrap the original AST nodes in order to
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1498 support the variant visit methods.
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1499 """
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1500 def __init__(self, obj):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1501 self.obj = obj
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1502
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1503 def __getattr__(self, attr):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1504 return getattr(self.obj, attr)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1505
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1506 class AugGetattr(Delegator):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1507 pass
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1508
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1509 class AugName(Delegator):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1510 pass
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1511
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1512 class AugSlice(Delegator):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1513 pass
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1514
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1515 class AugSubscript(Delegator):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1516 pass
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1517
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1518 wrapper = {
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1519 ast.Getattr: AugGetattr,
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1520 ast.Name: AugName,
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1521 ast.Slice: AugSlice,
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1522 ast.Subscript: AugSubscript,
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1523 }
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1524
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1525 def wrap_aug(node):
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1526 return wrapper[node.__class__](node)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1527
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1528 if __name__ == "__main__":
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1529 for file in sys.argv[1:]:
21059 e76306eea370 Fix [ #484645 ] little bug in pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 20925 diff changeset	1530 compileFile(file)

Mercurial > cpython-lru_cache-weakref

annotate Lib/compiler/pycodegen.py @ 58925:eecf1bef052d 2.6