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

annotate Lib/compiler/pycodegen.py @ 28646:193b8aec00ce legacy-trunk

Changes from Jonathan Riehl to allow his pgen extension (PEP 269) to work. This includes some more code that used to be part of pgen in the main parser; I'm okay with that. I'll see if the Windows build needs work next.

author	Guido van Rossum <guido@python.org>
date	Thu, 17 Apr 2003 14:55:42 +0000
parents	2c67169f6e13
children	caa8c120e2cb

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
11793 d2ab9553f871 support for arglists with implicit tuple unpacks Jeremy Hylton <jeremy@alum.mit.edu> parents: 11789 diff changeset	6 import types
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	7 from cStringIO import StringIO
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	8
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	9 from compiler import ast, parse, walk, syntax
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	10 from compiler import pyassem, misc, future, symbols
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	11 from compiler.consts import SC_LOCAL, SC_GLOBAL, SC_FREE, SC_CELL
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	12 from compiler.consts import CO_VARARGS, CO_VARKEYWORDS, CO_NEWLOCALS,\
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	13 CO_NESTED, CO_GENERATOR, CO_GENERATOR_ALLOWED, CO_FUTURE_DIVISION
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()
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	207 self.curStack = 0
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	208 self.maxStack = 0
14362 a665bdc4922f patch by Neil Schemenauer to improve (fix?) line number generation Jeremy Hylton <jeremy@alum.mit.edu> parents: 14359 diff changeset	209 self.last_lineno = None
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	210 self._setupGraphDelegation()
19936 6944f5b9de03 support true division Jeremy Hylton <jeremy@alum.mit.edu> parents: 19923 diff changeset	211 self._div_op = "BINARY_DIVIDE"
11793 d2ab9553f871 support for arglists with implicit tuple unpacks Jeremy Hylton <jeremy@alum.mit.edu> parents: 11789 diff changeset	212
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	213 # 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	214 futures = self.get_module().futures
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	215 for feature in futures:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	216 if feature == "division":
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	217 self.graph.setFlag(CO_FUTURE_DIVISION)
19936 6944f5b9de03 support true division Jeremy Hylton <jeremy@alum.mit.edu> parents: 19923 diff changeset	218 self._div_op = "BINARY_TRUE_DIVIDE"
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	219 elif feature == "generators":
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	220 self.graph.setFlag(CO_GENERATOR_ALLOWED)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	221
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	222 def initClass(self):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	223 """This method is called once for each class"""
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	224
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	225 def checkClass(self):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	226 """Verify that class is constructed correctly"""
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	227 try:
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	228 assert hasattr(self, 'graph')
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	229 assert getattr(self, 'NameFinder')
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	230 assert getattr(self, 'FunctionGen')
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	231 assert getattr(self, 'ClassGen')
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	232 except AssertionError, msg:
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	233 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	234 raise AssertionError, intro
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	235
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	236 def _setupGraphDelegation(self):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	237 self.emit = self.graph.emit
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	238 self.newBlock = self.graph.newBlock
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	239 self.startBlock = self.graph.startBlock
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	240 self.nextBlock = self.graph.nextBlock
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	241 self.setDocstring = self.graph.setDocstring
11793 d2ab9553f871 support for arglists with implicit tuple unpacks Jeremy Hylton <jeremy@alum.mit.edu> parents: 11789 diff changeset	242
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	243 def getCode(self):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	244 """Return a code object"""
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	245 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	246
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	247 def mangle(self, name):
a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	248 if self.class_name is not None:
a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	249 return misc.mangle(name, self.class_name)
a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	250 else:
a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	251 return name
a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	252
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	253 def parseSymbols(self, tree):
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	254 s = symbols.SymbolVisitor()
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	255 walk(tree, s)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	256 return s.scopes
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	257
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	258 def get_module(self):
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	259 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	260
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	261 # 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	262
11727 1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	263 def isLocalName(self, name):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	264 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	265
11743 c978a23a4dab add namespace attr to CodeGenerator, can be either MODULE_NAMESPACE or Jeremy Hylton <jeremy@alum.mit.edu> parents: 11734 diff changeset	266 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	267 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	268
c978a23a4dab add namespace attr to CodeGenerator, can be either MODULE_NAMESPACE or Jeremy Hylton <jeremy@alum.mit.edu> parents: 11734 diff changeset	269 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	270 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	271
c978a23a4dab add namespace attr to CodeGenerator, can be either MODULE_NAMESPACE or Jeremy Hylton <jeremy@alum.mit.edu> parents: 11734 diff changeset	272 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	273 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	274
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	275 def _nameOp(self, prefix, name):
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	276 name = self.mangle(name)
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	277 scope = self.scope.check_name(name)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	278 if scope == SC_LOCAL:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	279 if not self.optimized:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	280 self.emit(prefix + '_NAME', name)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	281 else:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	282 self.emit(prefix + '_FAST', name)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	283 elif scope == SC_GLOBAL:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	284 if not self.optimized:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	285 self.emit(prefix + '_NAME', name)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	286 else:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	287 self.emit(prefix + '_GLOBAL', name)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	288 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	289 self.emit(prefix + '_DEREF', name)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	290 else:
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	291 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	292 (name, scope)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	293
17459 e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	294 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	295 """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	296
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	297 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	298 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	299 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	300 """
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	301 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	302 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	303 else:
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 + '_NAME', name)
e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	305
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	306 # 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	307 # 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	308 # 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	309 # 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	310
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	311 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	312 """Emit SET_LINENO if necessary.
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	313
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	314 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	315 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	316 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	317
c58149c0670a replace most calls to emit 'SET_LINENO' will call to method set_lineno Jeremy Hylton <jeremy@alum.mit.edu> parents: 12563 diff changeset	318 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	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 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	321 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	322 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	323 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	324 """
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 lineno = getattr(node, 'lineno', None)
17453 f3b6ac128de9 pyassem.py: Jeremy Hylton <jeremy@alum.mit.edu> parents: 17449 diff changeset	326 if lineno is not None and (lineno != self.last_lineno
f3b6ac128de9 pyassem.py: Jeremy Hylton <jeremy@alum.mit.edu> parents: 17449 diff changeset	327 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	328 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	329 self.last_lineno = lineno
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	330 return True
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	331 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	332
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	333 # 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	334 # 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	335 # specialized code generators to use.
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	336
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	337 NameFinder = LocalNameFinder
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	338 FunctionGen = None
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	339 ClassGen = None
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	340
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	341 def visitModule(self, node):
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	342 self.scopes = self.parseSymbols(node)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	343 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	344 self.emit('SET_LINENO', 0)
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	345 if node.doc:
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	346 self.emit('LOAD_CONST', node.doc)
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	347 self.storeName('__doc__')
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	348 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	349 self.locals.push(lnf.getLocals())
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	350 self.visit(node.node)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	351 self.emit('LOAD_CONST', None)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	352 self.emit('RETURN_VALUE')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	353
21522 cb55f15b2730 Merge of the release22 branch changes back into the trunk. Barry Warsaw <barry@python.org> parents: 21059 diff changeset	354 def visitExpression(self, node):
cb55f15b2730 Merge of the release22 branch changes back into the trunk. Barry Warsaw <barry@python.org> parents: 21059 diff changeset	355 self.set_lineno(node)
cb55f15b2730 Merge of the release22 branch changes back into the trunk. Barry Warsaw <barry@python.org> parents: 21059 diff changeset	356 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	357 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	358 self.visit(node.node)
cb55f15b2730 Merge of the release22 branch changes back into the trunk. Barry Warsaw <barry@python.org> parents: 21059 diff changeset	359 self.emit('RETURN_VALUE')
cb55f15b2730 Merge of the release22 branch changes back into the trunk. Barry Warsaw <barry@python.org> parents: 21059 diff changeset	360
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	361 def visitFunction(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	362 self._visitFuncOrLambda(node, isLambda=0)
17410 f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	363 if node.doc:
f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	364 self.setDocstring(node.doc)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	365 self.storeName(node.name)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	366
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	367 def visitLambda(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	368 self._visitFuncOrLambda(node, isLambda=1)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	369
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	370 def _visitFuncOrLambda(self, node, isLambda=0):
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	371 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	372 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	373 walk(node.code, gen)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	374 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	375 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	376 for default in node.defaults:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	377 self.visit(default)
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	378 frees = gen.scope.get_free_vars()
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	379 if frees:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	380 for name in frees:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	381 self.emit('LOAD_CLOSURE', name)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	382 self.emit('LOAD_CONST', gen)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	383 self.emit('MAKE_CLOSURE', len(node.defaults))
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	384 else:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	385 self.emit('LOAD_CONST', gen)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	386 self.emit('MAKE_FUNCTION', len(node.defaults))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	387
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	388 def visitClass(self, node):
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	389 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	390 self.get_module())
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	391 walk(node.code, gen)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	392 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	393 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	394 self.emit('LOAD_CONST', node.name)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	395 for base in node.bases:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	396 self.visit(base)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	397 self.emit('BUILD_TUPLE', len(node.bases))
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	398 frees = gen.scope.get_free_vars()
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	399 for name in frees:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	400 self.emit('LOAD_CLOSURE', name)
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	401 self.emit('LOAD_CONST', gen)
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	402 if frees:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	403 self.emit('MAKE_CLOSURE', 0)
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	404 else:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	405 self.emit('MAKE_FUNCTION', 0)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	406 self.emit('CALL_FUNCTION', 0)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	407 self.emit('BUILD_CLASS')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	408 self.storeName(node.name)
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 # The rest are standard visitor methods
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	411
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	412 # The next few implement control-flow statements
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	413
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	414 def visitIf(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	415 end = self.newBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	416 numtests = len(node.tests)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	417 for i in range(numtests):
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	418 test, suite = node.tests[i]
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	419 if is_constant_false(test):
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	420 # 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	421 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	422 self.set_lineno(test)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	423 self.visit(test)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	424 nextTest = self.newBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	425 self.emit('JUMP_IF_FALSE', nextTest)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	426 self.nextBlock()
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 self.visit(suite)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	429 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	430 self.startBlock(nextTest)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	431 self.emit('POP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	432 if node.else_:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	433 self.visit(node.else_)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	434 self.nextBlock(end)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	435
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	436 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	437 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	438
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	439 loop = self.newBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	440 else_ = self.newBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	441
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	442 after = self.newBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	443 self.emit('SETUP_LOOP', after)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	444
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	445 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	446 self.setups.push((LOOP, loop))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	447
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	448 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	449 self.visit(node.test)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	450 self.emit('JUMP_IF_FALSE', else_ or after)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	451
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	452 self.nextBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	453 self.emit('POP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	454 self.visit(node.body)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	455 self.emit('JUMP_ABSOLUTE', loop)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	456
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	457 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	458 self.emit('POP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	459 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	460 self.setups.pop()
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	461 if node.else_:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	462 self.visit(node.else_)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	463 self.nextBlock(after)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	464
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	465 def visitFor(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	466 start = self.newBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	467 anchor = self.newBlock()
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	468 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	469 self.setups.push((LOOP, start))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	470
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	471 self.set_lineno(node)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	472 self.emit('SETUP_LOOP', after)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	473 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	474 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	475
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	476 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	477 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	478 self.emit('FOR_ITER', anchor)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	479 self.visit(node.assign)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	480 self.visit(node.body)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	481 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	482 self.nextBlock(anchor)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	483 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	484 self.setups.pop()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	485 if node.else_:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	486 self.visit(node.else_)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	487 self.nextBlock(after)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	488
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	489 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	490 if not self.setups:
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	491 raise SyntaxError, "'break' outside loop (%s, %d)" % \
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	492 (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	493 self.set_lineno(node)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	494 self.emit('BREAK_LOOP')
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	495
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	496 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	497 if not self.setups:
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	498 raise SyntaxError, "'continue' outside loop (%s, %d)" % \
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	499 (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	500 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	501 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	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 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	504 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	505 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	506 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	507 # 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	508 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	509 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	510 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	511 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	512 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	513 break
725762ca2774 Track the block stack more reasonably in order to handle continue in Jeremy Hylton <jeremy@alum.mit.edu> parents: 19567 diff changeset	514 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	515 raise SyntaxError, "'continue' outside loop (%s, %d)" % \
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	516 (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	517 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	518 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	519 elif kind == END_FINALLY:
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	520 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	521 raise SyntaxError, msg % (node.filename, node.lineno)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	522
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	523 def visitTest(self, node, jump):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	524 end = self.newBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	525 for child in node.nodes[:-1]:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	526 self.visit(child)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	527 self.emit(jump, end)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	528 self.nextBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	529 self.emit('POP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	530 self.visit(node.nodes[-1])
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	531 self.nextBlock(end)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	532
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	533 def visitAnd(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	534 self.visitTest(node, 'JUMP_IF_FALSE')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	535
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	536 def visitOr(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	537 self.visitTest(node, 'JUMP_IF_TRUE')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	538
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	539 def visitCompare(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	540 self.visit(node.expr)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	541 cleanup = self.newBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	542 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	543 self.visit(code)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	544 self.emit('DUP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	545 self.emit('ROT_THREE')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	546 self.emit('COMPARE_OP', op)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	547 self.emit('JUMP_IF_FALSE', cleanup)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	548 self.nextBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	549 self.emit('POP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	550 # now do the last comparison
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	551 if node.ops:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	552 op, code = node.ops[-1]
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	553 self.visit(code)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	554 self.emit('COMPARE_OP', op)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	555 if len(node.ops) > 1:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	556 end = self.newBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	557 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	558 self.startBlock(cleanup)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	559 self.emit('ROT_TWO')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	560 self.emit('POP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	561 self.nextBlock(end)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	562
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	563 # list comprehensions
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	564 __list_count = 0
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	565
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	566 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	567 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	568 # setup list
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	569 append = "$append%d" % self.__list_count
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	570 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	571 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	572 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	573 self.emit('LOAD_ATTR', 'append')
17459 e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	574 self._implicitNameOp('STORE', append)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	575
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	576 stack = []
17453 f3b6ac128de9 pyassem.py: Jeremy Hylton <jeremy@alum.mit.edu> parents: 17449 diff changeset	577 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	578 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	579 cont = None
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	580 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	581 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	582 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	583 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	584 stack.insert(0, (start, cont, anchor))
17453 f3b6ac128de9 pyassem.py: Jeremy Hylton <jeremy@alum.mit.edu> parents: 17449 diff changeset	585
17459 e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	586 self._implicitNameOp('LOAD', append)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	587 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	588 self.emit('CALL_FUNCTION', 1)
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	589 self.emit('POP_TOP')
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	590
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	591 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	592 if cont:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	593 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	594 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	595 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	596 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	597 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	598 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	599 self.startBlock(anchor)
17459 e846e960e7f6 Use new _implicitNameOp() to generate name op code for list comprehensions. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17454 diff changeset	600 self._implicitNameOp('DELETE', append)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	601
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	602 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	603
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	604 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	605 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	606 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	607
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	608 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	609 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	610 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	611 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	612 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	613 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	614 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	615 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	616
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	617 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	618 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	619 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	620 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	621 self.newBlock()
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	622 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	623
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	624 # exception related
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	625
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	626 def visitAssert(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	627 # 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	628 # transformation of the AST before this stage
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	629 end = 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	630 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	631 # XXX __debug__ and AssertionError appear to be special cases
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	632 # -- they are always loaded as globals even if there are local
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	633 # names. I guess this is a sort of renaming op.
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	634 self.emit('LOAD_GLOBAL', '__debug__')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	635 self.emit('JUMP_IF_FALSE', end)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	636 self.nextBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	637 self.emit('POP_TOP')
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	638 self.visit(node.test)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	639 self.emit('JUMP_IF_TRUE', end)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	640 self.nextBlock()
15379 7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	641 self.emit('POP_TOP')
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	642 self.emit('LOAD_GLOBAL', 'AssertionError')
15379 7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	643 if node.fail:
7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	644 self.visit(node.fail)
7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	645 self.emit('RAISE_VARARGS', 2)
7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	646 else:
7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	647 self.emit('RAISE_VARARGS', 1)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	648 self.nextBlock(end)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	649 self.emit('POP_TOP')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	650
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	651 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	652 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	653 n = 0
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	654 if node.expr1:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	655 self.visit(node.expr1)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	656 n = n + 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	657 if node.expr2:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	658 self.visit(node.expr2)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	659 n = n + 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	660 if node.expr3:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	661 self.visit(node.expr3)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	662 n = n + 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	663 self.emit('RAISE_VARARGS', n)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	664
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	665 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	666 body = self.newBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	667 handlers = self.newBlock()
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	668 end = self.newBlock()
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	669 if node.else_:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	670 lElse = self.newBlock()
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	671 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	672 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	673 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	674 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	675 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	676 self.setups.push((EXCEPT, body))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	677 self.visit(node.body)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	678 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	679 self.setups.pop()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	680 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	681 self.startBlock(handlers)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	682
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	683 last = len(node.handlers) - 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	684 for i in range(len(node.handlers)):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	685 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	686 self.set_lineno(expr)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	687 if expr:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	688 self.emit('DUP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	689 self.visit(expr)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	690 self.emit('COMPARE_OP', 'exception match')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	691 next = self.newBlock()
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	692 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	693 self.nextBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	694 self.emit('POP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	695 self.emit('POP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	696 if target:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	697 self.visit(target)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	698 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	699 self.emit('POP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	700 self.emit('POP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	701 self.visit(body)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	702 self.emit('JUMP_FORWARD', end)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	703 if expr:
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	704 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	705 else:
7a82f84290f7 Change the graph structure to contain the code generator object for Jeremy Hylton <jeremy@alum.mit.edu> parents: 15205 diff changeset	706 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	707 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	708 self.emit('POP_TOP')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	709 self.emit('END_FINALLY')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	710 if node.else_:
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	711 self.nextBlock(lElse)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	712 self.visit(node.else_)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	713 self.nextBlock(end)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	714
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	715 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	716 body = self.newBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	717 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	718 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	719 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	720 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	721 self.setups.push((TRY_FINALLY, body))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	722 self.visit(node.body)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	723 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	724 self.setups.pop()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	725 self.emit('LOAD_CONST', None)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	726 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	727 self.setups.push((END_FINALLY, final))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	728 self.visit(node.final)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	729 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	730 self.setups.pop()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	731
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	732 # misc
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	733
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	734 def visitDiscard(self, node):
17412 3704be3db302 [finishing fix from earlier checkins] Jeremy Hylton <jeremy@alum.mit.edu> parents: 17410 diff changeset	735 self.set_lineno(node)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	736 self.visit(node.expr)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	737 self.emit('POP_TOP')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	738
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	739 def visitConst(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	740 self.emit('LOAD_CONST', node.value)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	741
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	742 def visitKeyword(self, node):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	743 self.emit('LOAD_CONST', node.name)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	744 self.visit(node.expr)
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	745
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	746 def visitGlobal(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	747 # no code to generate
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	748 pass
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	749
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	750 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	751 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	752 self.loadName(node.name)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	753
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	754 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	755 self.set_lineno(node)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	756
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	757 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	758 self.set_lineno(node)
14359 35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	759 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	760 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	761 self.emit('LOAD_CONST', None)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	762 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	763 mod = name.split(".")[0]
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	764 self.storeName(alias or mod)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	765
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	766 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	767 self.set_lineno(node)
14359 35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	768 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	769 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	770 self.emit('LOAD_CONST', tuple(fromlist))
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	771 self.emit('IMPORT_NAME', node.modname)
14359 35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	772 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	773 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	774 if name == '*':
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	775 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	776 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	777 # 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	778 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	779 return
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	780 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	781 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	782 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	783 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	784 else:
100beebde9df Fix SF bug #116263: support for from .. import * Jeremy Hylton <jeremy@alum.mit.edu> parents: 14362 diff changeset	785 self.emit('IMPORT_FROM', name)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	786 self.emit('POP_TOP')
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	787
14359 35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	788 def _resolveDots(self, name):
23710 44b9977d4848 Remove uses of string module and stat.ST_MODE Neal Norwitz <nnorwitz@gmail.com> parents: 21958 diff changeset	789 elts = name.split(".")
14359 35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	790 if len(elts) == 1:
35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	791 return
35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	792 for elt in elts[1:]:
35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	793 self.emit('LOAD_ATTR', elt)
35763be75ddb Update magic number. Jeremy Hylton <jeremy@alum.mit.edu> parents: 13941 diff changeset	794
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	795 def visitGetattr(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	796 self.visit(node.expr)
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	797 self.emit('LOAD_ATTR', self.mangle(node.attrname))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	798
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	799 # next five implement assignments
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	800
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	801 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	802 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	803 self.visit(node.expr)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	804 dups = len(node.nodes) - 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	805 for i in range(len(node.nodes)):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	806 elt = node.nodes[i]
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	807 if i < dups:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	808 self.emit('DUP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	809 if isinstance(elt, ast.Node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	810 self.visit(elt)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	811
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	812 def visitAssName(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	813 if node.flags == 'OP_ASSIGN':
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	814 self.storeName(node.name)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	815 elif node.flags == 'OP_DELETE':
19561 81168ef4cb65 Support // and //= Jeremy Hylton <jeremy@alum.mit.edu> parents: 19534 diff changeset	816 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	817 self.delName(node.name)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	818 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	819 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	820
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	821 def visitAssAttr(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	822 self.visit(node.expr)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	823 if node.flags == 'OP_ASSIGN':
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	824 self.emit('STORE_ATTR', self.mangle(node.attrname))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	825 elif node.flags == 'OP_DELETE':
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	826 self.emit('DELETE_ATTR', self.mangle(node.attrname))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	827 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	828 print "warning: unexpected flags:", node.flags
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	829 print node
11751 3f1da05ced55 add support for Lambda nodes Jeremy Hylton <jeremy@alum.mit.edu> parents: 11748 diff changeset	830
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	831 def _visitAssSequence(self, node, op='UNPACK_SEQUENCE'):
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	832 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	833 self.emit(op, len(node.nodes))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	834 for child in node.nodes:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	835 self.visit(child)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	836
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	837 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	838 visitAssTuple = _visitAssSequence
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	839 visitAssList = _visitAssSequence
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	840 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	841 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	842 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	843
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	844 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	845 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	846
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	847 # augmented assignment
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	848
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	849 def visitAugAssign(self, node):
19519 99e953c6709c emit SET_LINENO for augmented assignments Jeremy Hylton <jeremy@alum.mit.edu> parents: 19518 diff changeset	850 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	851 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	852 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	853 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	854 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	855 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	856
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	857 _augmented_opcode = {
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	858 '+=' : 'INPLACE_ADD',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	859 '-=' : 'INPLACE_SUBTRACT',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	860 '*=' : 'INPLACE_MULTIPLY',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	861 '/=' : 'INPLACE_DIVIDE',
19561 81168ef4cb65 Support // and //= Jeremy Hylton <jeremy@alum.mit.edu> parents: 19534 diff changeset	862 '//=': '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	863 '%=' : 'INPLACE_MODULO',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	864 '**=': 'INPLACE_POWER',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	865 '>>=': 'INPLACE_RSHIFT',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	866 '<<=': 'INPLACE_LSHIFT',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	867 '&=' : 'INPLACE_AND',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	868 '^=' : 'INPLACE_XOR',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	869 '\|=' : 'INPLACE_OR',
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	870 }
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	871
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	872 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	873 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	874 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	875 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	876 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	877
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	878 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	879 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	880 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	881 self.emit('DUP_TOP')
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	882 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	883 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	884 self.emit('ROT_TWO')
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	885 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	886
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	887 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	888 if mode == "load":
17413 bad2daeb7440 typo Jeremy Hylton <jeremy@alum.mit.edu> parents: 17412 diff changeset	889 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	890 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	891 slice = 0
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	892 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	893 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	894 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	895 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	896 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	897 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	898 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	899 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	900 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	901 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	902 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	903
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	904 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	905 if len(node.subs) > 1:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	906 raise SyntaxError, "augmented assignment to tuple is not possible"
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	907 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	908 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	909 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	910 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	911 self.emit('STORE_SUBSCR')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	912
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	913 def visitExec(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	914 self.visit(node.expr)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	915 if node.locals is None:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	916 self.emit('LOAD_CONST', None)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	917 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	918 self.visit(node.locals)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	919 if node.globals is None:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	920 self.emit('DUP_TOP')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	921 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	922 self.visit(node.globals)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	923 self.emit('EXEC_STMT')
11751 3f1da05ced55 add support for Lambda nodes Jeremy Hylton <jeremy@alum.mit.edu> parents: 11748 diff changeset	924
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	925 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	926 pos = 0
dffbd921eeed change MODULE_NAMESPACE/FUNCTION_NAMESPACE stuff to have a single flag Jeremy Hylton <jeremy@alum.mit.edu> parents: 11751 diff changeset	927 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	928 self.set_lineno(node)
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	929 self.visit(node.node)
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	930 for arg in node.args:
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	931 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	932 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	933 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	934 else:
dffbd921eeed change MODULE_NAMESPACE/FUNCTION_NAMESPACE stuff to have a single flag Jeremy Hylton <jeremy@alum.mit.edu> parents: 11751 diff changeset	935 pos = pos + 1
12563 b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	936 if node.star_args is not None:
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	937 self.visit(node.star_args)
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	938 if node.dstar_args is not None:
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	939 self.visit(node.dstar_args)
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	940 have_star = node.star_args is not None
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	941 have_dstar = node.dstar_args is not None
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	942 opcode = callfunc_opcode_info[have_star, have_dstar]
b87d75798d81 patches from Mark Hammond Jeremy Hylton <jeremy@alum.mit.edu> parents: 12009 diff changeset	943 self.emit(opcode, kw << 8 \| pos)
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	944
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	945 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	946 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	947 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	948 self.visit(node.dest)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	949 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	950 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	951 self.emit('DUP_TOP')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	952 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	953 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	954 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	955 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	956 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	957 self.emit('PRINT_ITEM')
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	958 if node.dest and not newline:
fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	959 self.emit('POP_TOP')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	960
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	961 def visitPrintnl(self, node):
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	962 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	963 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	964 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	965 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	966 self.emit('PRINT_NEWLINE')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	967
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	968 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	969 self.set_lineno(node)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	970 self.visit(node.value)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	971 self.emit('RETURN_VALUE')
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	972
19923 cb35d0f704de Add code generator for yield stmt Jeremy Hylton <jeremy@alum.mit.edu> parents: 19920 diff changeset	973 def visitYield(self, node):
cb35d0f704de Add code generator for yield stmt Jeremy Hylton <jeremy@alum.mit.edu> parents: 19920 diff changeset	974 self.set_lineno(node)
cb35d0f704de Add code generator for yield stmt Jeremy Hylton <jeremy@alum.mit.edu> parents: 19920 diff changeset	975 self.visit(node.value)
26979 784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	976 self.emit('YIELD_VALUE')
19923 cb35d0f704de Add code generator for yield stmt Jeremy Hylton <jeremy@alum.mit.edu> parents: 19920 diff changeset	977
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	978 # slice and subscript stuff
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	979
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	980 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	981 # aug_flag is used by visitAugSlice
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	982 self.visit(node.expr)
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	983 slice = 0
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	984 if node.lower:
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	985 self.visit(node.lower)
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	986 slice = slice \| 1
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	987 if node.upper:
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	988 self.visit(node.upper)
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	989 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	990 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	991 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	992 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	993 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	994 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	995 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	996 self.emit('DUP_TOPX', 2)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	997 if node.flags == 'OP_APPLY':
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	998 self.emit('SLICE+%d' % slice)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	999 elif node.flags == 'OP_ASSIGN':
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1000 self.emit('STORE_SLICE+%d' % slice)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1001 elif node.flags == 'OP_DELETE':
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1002 self.emit('DELETE_SLICE+%d' % slice)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1003 else:
11789 860524182b9c finish first impl of code generator Jeremy Hylton <jeremy@alum.mit.edu> parents: 11766 diff changeset	1004 print "weird slice", node.flags
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1005 raise
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1006
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1007 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	1008 self.visit(node.expr)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1009 for sub in node.subs:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1010 self.visit(sub)
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1011 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	1012 self.emit('DUP_TOPX', 2)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1013 if len(node.subs) > 1:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1014 self.emit('BUILD_TUPLE', len(node.subs))
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1015 if node.flags == 'OP_APPLY':
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1016 self.emit('BINARY_SUBSCR')
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1017 elif node.flags == 'OP_ASSIGN':
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1018 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	1019 elif node.flags == 'OP_DELETE':
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1020 self.emit('DELETE_SUBSCR')
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1021
11927 79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1022 # binary ops
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1023
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1024 def binaryOp(self, node, op):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1025 self.visit(node.left)
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1026 self.visit(node.right)
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1027 self.emit(op)
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1028
d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1029 def visitAdd(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1030 return self.binaryOp(node, 'BINARY_ADD')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1031
d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1032 def visitSub(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1033 return self.binaryOp(node, 'BINARY_SUBTRACT')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1034
d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1035 def visitMul(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1036 return self.binaryOp(node, 'BINARY_MULTIPLY')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1037
d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1038 def visitDiv(self, node):
19936 6944f5b9de03 support true division Jeremy Hylton <jeremy@alum.mit.edu> parents: 19923 diff changeset	1039 return self.binaryOp(node, self._div_op)
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1040
19561 81168ef4cb65 Support // and //= Jeremy Hylton <jeremy@alum.mit.edu> parents: 19534 diff changeset	1041 def visitFloorDiv(self, node):
81168ef4cb65 Support // and //= Jeremy Hylton <jeremy@alum.mit.edu> parents: 19534 diff changeset	1042 return self.binaryOp(node, 'BINARY_FLOOR_DIVIDE')
81168ef4cb65 Support // and //= Jeremy Hylton <jeremy@alum.mit.edu> parents: 19534 diff changeset	1043
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1044 def visitMod(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1045 return self.binaryOp(node, 'BINARY_MODULO')
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1046
11761 62fa8ec8595f looks like everything is working except for try/except (pystone Jeremy Hylton <jeremy@alum.mit.edu> parents: 11755 diff changeset	1047 def visitPower(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1048 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	1049
62fa8ec8595f looks like everything is working except for try/except (pystone Jeremy Hylton <jeremy@alum.mit.edu> parents: 11755 diff changeset	1050 def visitLeftShift(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1051 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	1052
62fa8ec8595f looks like everything is working except for try/except (pystone Jeremy Hylton <jeremy@alum.mit.edu> parents: 11755 diff changeset	1053 def visitRightShift(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1054 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	1055
11927 79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1056 # unary ops
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1057
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1058 def unaryOp(self, node, op):
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1059 self.visit(node.expr)
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1060 self.emit(op)
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1061
11761 62fa8ec8595f looks like everything is working except for try/except (pystone Jeremy Hylton <jeremy@alum.mit.edu> parents: 11755 diff changeset	1062 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	1063 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	1064
11727 1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1065 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	1066 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	1067
1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1068 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	1069 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	1070
1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1071 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	1072 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	1073
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1074 def visitNot(self, node):
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1075 return self.unaryOp(node, 'UNARY_NOT')
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1076
11727 1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1077 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	1078 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	1079
11927 79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1080 # bit ops
79b6a6a882d9 rename compile.py to pycodegen.py Jeremy Hylton <jeremy@alum.mit.edu> parents: 11808 diff changeset	1081
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1082 def bitOp(self, nodes, op):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1083 self.visit(nodes[0])
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1084 for node in nodes[1:]:
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1085 self.visit(node)
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1086 self.emit(op)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1087
57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1088 def visitBitand(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1089 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	1090
57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1091 def visitBitor(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1092 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	1093
57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1094 def visitBitxor(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1095 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	1096
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1097 # 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	1098
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1099 def visitEllipsis(self, node):
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1100 self.emit('LOAD_CONST', Ellipsis)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1101
11727 1e952d933abb add optional verbose arg to walk function. it overrides the global Jeremy Hylton <jeremy@alum.mit.edu> parents: 11724 diff changeset	1102 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	1103 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	1104 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	1105 self.visit(elt)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1106 self.emit('BUILD_TUPLE', len(node.nodes))
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1107
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1108 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	1109 self.set_lineno(node)
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1110 for elt in node.nodes:
6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1111 self.visit(elt)
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1112 self.emit('BUILD_LIST', len(node.nodes))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1113
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1114 def visitSliceobj(self, node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1115 for child in node.nodes:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1116 self.visit(child)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1117 self.emit('BUILD_SLICE', len(node.nodes))
11734 6a99ff069457 add ExampleASTVisitor: Jeremy Hylton <jeremy@alum.mit.edu> parents: 11727 diff changeset	1118
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1119 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	1120 self.set_lineno(node)
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1121 self.emit('BUILD_MAP', 0)
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1122 for k, v in node.items:
57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1123 self.emit('DUP_TOP')
26670 1312297887a3 Fixing bug Gustavo Niemeyer <gustavo@niemeyer.net> parents: 23710 diff changeset	1124 self.visit(k)
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1125 self.visit(v)
26670 1312297887a3 Fixing bug Gustavo Niemeyer <gustavo@niemeyer.net> parents: 23710 diff changeset	1126 self.emit('ROT_THREE')
11806 57167731d124 satisfy the tabnanny Jeremy Hylton <jeremy@alum.mit.edu> parents: 11793 diff changeset	1127 self.emit('STORE_SUBSCR')
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1128
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1129 class NestedScopeMixin:
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1130 """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	1131 def initClass(self):
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1132 self.__class__.NameFinder = LocalNameFinder
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1133 self.__class__.FunctionGen = FunctionCodeGenerator
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1134 self.__class__.ClassGen = ClassCodeGenerator
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1135
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1136 class ModuleCodeGenerator(NestedScopeMixin, CodeGenerator):
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1137 __super_init = CodeGenerator.__init__
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1138
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1139 scopes = None
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1140
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1141 def __init__(self, tree):
aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1142 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	1143 self.futures = future.find_futures(tree)
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1144 self.__super_init()
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1145 walk(tree, self)
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1146
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1147 def get_module(self):
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1148 return self
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1149
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1150 class ExpressionCodeGenerator(NestedScopeMixin, CodeGenerator):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1151 __super_init = CodeGenerator.__init__
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1152
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1153 scopes = None
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1154 futures = ()
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1155
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1156 def __init__(self, tree):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1157 self.graph = pyassem.PyFlowGraph("<expression>", tree.filename)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1158 self.__super_init()
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1159 walk(tree, self)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1160
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1161 def get_module(self):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1162 return self
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1163
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1164 class InteractiveCodeGenerator(NestedScopeMixin, CodeGenerator):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1165
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1166 __super_init = CodeGenerator.__init__
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1167
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1168 scopes = None
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1169 futures = ()
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1170
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1171 def __init__(self, tree):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1172 self.graph = pyassem.PyFlowGraph("<interactive>", tree.filename)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1173 self.__super_init()
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1174 self.set_lineno(tree)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1175 walk(tree, self)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1176 self.emit('RETURN_VALUE')
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1177
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1178 def get_module(self):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1179 return self
21958 5ed782bdbb2a Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 21522 diff changeset	1180
19946 41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1181 def visitDiscard(self, node):
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1182 # 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	1183 # name.
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1184 self.visit(node.expr)
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1185 self.emit('PRINT_EXPR')
41799b70a96c API change: Jeremy Hylton <jeremy@alum.mit.edu> parents: 19943 diff changeset	1186
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1187 class AbstractFunctionCode:
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1188 optimized = 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1189 lambdaCount = 0
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1190
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1191 def __init__(self, func, scopes, isLambda, class_name, mod):
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	1192 self.class_name = class_name
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1193 self.module = mod
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1194 if isLambda:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1195 klass = FunctionCodeGenerator
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1196 name = "<lambda.%d>" % klass.lambdaCount
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1197 klass.lambdaCount = klass.lambdaCount + 1
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1198 else:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1199 name = func.name
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1200 args, hasTupleArg = generateArgList(func.argnames)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1201 self.graph = pyassem.PyFlowGraph(name, func.filename, args,
a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1202 optimized=1)
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1203 self.isLambda = isLambda
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1204 self.super_init()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1205
17410 f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	1206 if not isLambda and func.doc:
f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	1207 self.setDocstring(func.doc)
f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	1208
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	1209 lnf = walk(func.code, self.NameFinder(args), verbose=0)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1210 self.locals.push(lnf.getLocals())
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1211 if func.varargs:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1212 self.graph.setFlag(CO_VARARGS)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1213 if func.kwargs:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1214 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	1215 self.set_lineno(func)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1216 if hasTupleArg:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1217 self.generateArgUnpack(func.argnames)
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1218
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1219 def get_module(self):
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1220 return self.module
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1221
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1222 def finish(self):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1223 self.graph.startExitBlock()
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1224 if not self.isLambda:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1225 self.emit('LOAD_CONST', None)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1226 self.emit('RETURN_VALUE')
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1227
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1228 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	1229 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	1230 arg = args[i]
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1231 if type(arg) == types.TupleType:
17449 51c8ce1a2e25 Revise handling of tuple arguments so that the variables names match Jeremy Hylton <jeremy@alum.mit.edu> parents: 17438 diff changeset	1232 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	1233 self.unpackSequence(arg)
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1234
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1235 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	1236 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	1237 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	1238 else:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1239 self.emit('UNPACK_TUPLE', len(tup))
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1240 for elt in tup:
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1241 if type(elt) == types.TupleType:
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1242 self.unpackSequence(elt)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1243 else:
17438 41a423294cb5 Fix unpackSequence() to use _nameOp() rather than LOAD_FAST Jeremy Hylton <jeremy@alum.mit.edu> parents: 17437 diff changeset	1244 self._nameOp('STORE', elt)
11705 d10c108d3fd7 checking in initial weekend's work Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1245
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1246 unpackTuple = unpackSequence
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1247
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1248 class FunctionCodeGenerator(NestedScopeMixin, AbstractFunctionCode,
20547 a12ccdbce8da Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 19946 diff changeset	1249 CodeGenerator):
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1250 super_init = CodeGenerator.__init__ # call be other init
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1251 scopes = None
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1252
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1253 __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	1254
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1255 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	1256 self.scopes = scopes
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1257 self.scope = scopes[func]
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1258 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	1259 self.graph.setFreeVars(self.scope.get_free_vars())
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1260 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	1261 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	1262 self.graph.setFlag(CO_GENERATOR)
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1263
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1264 class AbstractClassCode:
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1265
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1266 def __init__(self, klass, scopes, module):
19522 a0847b7a8902 Handle private names Jeremy Hylton <jeremy@alum.mit.edu> parents: 19519 diff changeset	1267 self.class_name = klass.name
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1268 self.module = module
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1269 self.graph = pyassem.PyFlowGraph(klass.name, klass.filename,
19561 81168ef4cb65 Support // and //= Jeremy Hylton <jeremy@alum.mit.edu> parents: 19534 diff changeset	1270 optimized=0, klass=1)
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1271 self.super_init()
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	1272 lnf = walk(klass.code, self.NameFinder(), verbose=0)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1273 self.locals.push(lnf.getLocals())
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1274 self.graph.setFlag(CO_NEWLOCALS)
17410 f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	1275 if klass.doc:
f967489ee3b6 Generate docstrings. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17346 diff changeset	1276 self.setDocstring(klass.doc)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1277
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1278 def get_module(self):
63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1279 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	1280
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1281 def finish(self):
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1282 self.graph.startExitBlock()
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1283 self.emit('LOAD_LOCALS')
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1284 self.emit('RETURN_VALUE')
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1285
19920 63930159ca3c Various sundry changes for 2.2 compatibility Jeremy Hylton <jeremy@alum.mit.edu> parents: 19619 diff changeset	1286 class ClassCodeGenerator(NestedScopeMixin, AbstractClassCode, CodeGenerator):
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1287 super_init = CodeGenerator.__init__
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1288 scopes = None
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1289
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1290 __super_init = AbstractClassCode.__init__
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1291
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1292 def __init__(self, klass, scopes, module):
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1293 self.scopes = scopes
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1294 self.scope = scopes[klass]
19939 aaca4d3cf62b Handle more syntax errors. Jeremy Hylton <jeremy@alum.mit.edu> parents: 19936 diff changeset	1295 self.__super_init(klass, scopes, module)
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1296 self.graph.setFreeVars(self.scope.get_free_vars())
98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1297 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	1298 self.set_lineno(klass)
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	1299 self.emit("LOAD_GLOBAL", "__name__")
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	1300 self.storeName("__module__")
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	1301 if klass.doc:
784e7fdf8c52 SF patch [ 597919 ] compiler package and SET_LINENO Jeremy Hylton <jeremy@alum.mit.edu> parents: 26670 diff changeset	1302 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	1303 self.storeName('__doc__')
17436 98f69245bb69 Preliminary support for nested scopes Jeremy Hylton <jeremy@alum.mit.edu> parents: 17413 diff changeset	1304
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1305 def generateArgList(arglist):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1306 """Generate an arg list marking TupleArgs"""
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1307 args = []
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1308 extra = []
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1309 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	1310 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	1311 elt = arglist[i]
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1312 if type(elt) == types.StringType:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1313 args.append(elt)
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1314 elif type(elt) == types.TupleType:
17449 51c8ce1a2e25 Revise handling of tuple arguments so that the variables names match Jeremy Hylton <jeremy@alum.mit.edu> parents: 17438 diff changeset	1315 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	1316 extra.extend(misc.flatten(elt))
13941 d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1317 count = count + 1
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1318 else:
d56dd1530ce1 Bring Tools/compiler almost up to date. Specifically: Thomas Wouters <thomas@python.org> parents: 13864 diff changeset	1319 raise ValueError, "unexpect argument type:", elt
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1320 return args + extra, count
11755 57fb3501ab22 split compile.py into two files Jeremy Hylton <jeremy@alum.mit.edu> parents: 11754 diff changeset	1321
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1322 def findOp(node):
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1323 """Find the op (DELETE, LOAD, STORE) in an AssTuple tree"""
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1324 v = OpFinder()
19518 fbb35268fed4 Many changes -- bug fixes and sundry improvements Jeremy Hylton <jeremy@alum.mit.edu> parents: 17460 diff changeset	1325 walk(node, v, verbose=0)
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1326 return v.op
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1327
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	1328 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	1329 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	1330 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	1331 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	1332 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	1333 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	1334 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	1335 raise ValueError, "mixed ops in stmt"
17454 d91577783335 Add support for visitAssAttr to findOp(). Jeremy Hylton <jeremy@alum.mit.edu> parents: 17453 diff changeset	1336 visitAssAttr = visitAssName
20925 ee1437f5de03 Fix SF bug #479186: compiler generates bad code for "del" Jeremy Hylton <jeremy@alum.mit.edu> parents: 20547 diff changeset	1337 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	1338
15205 04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1339 class Delegator:
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1340 """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	1341
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1342 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	1343 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	1344 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	1345 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	1346 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	1347 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	1348 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	1349 """
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1350 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	1351 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	1352
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1353 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	1354 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	1355
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1356 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	1357 pass
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1358
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1359 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	1360 pass
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1361
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1362 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	1363 pass
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1364
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1365 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	1366 pass
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1367
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1368 wrapper = {
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1369 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	1370 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	1371 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	1372 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	1373 }
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1374
04e76e08e10f Now supports entire Python 2.0 language and still supports Python Jeremy Hylton <jeremy@alum.mit.edu> parents: 15185 diff changeset	1375 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	1376 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	1377
12009 127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1378 if __name__ == "__main__":
127f70f59b02 complete rewrite Jeremy Hylton <jeremy@alum.mit.edu> parents: 11927 diff changeset	1379 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	1380 compileFile(file)

Mercurial > cpython-lru_cache-weakref

annotate Lib/compiler/pycodegen.py @ 28646:193b8aec00ce legacy-trunk