|
307
|
1 from .astnodes import Constant, Variable, Function, BaseType, Symbol
|
|
354
|
2 from .astnodes import ArrayType, StructureType, DefinedType, PointerType
|
|
|
3 from .astnodes import StructField
|
|
150
|
4
|
|
288
|
5
|
|
150
|
6 class Scope:
|
|
306
|
7 """ A scope contains all symbols in a scope. It also has a parent scope,
|
|
|
8 when looking for a symbol, also the parent scopes are checked. """
|
|
217
|
9 def __init__(self, parent=None):
|
|
|
10 self.symbols = {}
|
|
|
11 self.parent = parent
|
|
|
12
|
|
|
13 def __iter__(self):
|
|
272
|
14 # Iterate in a deterministic manner:
|
|
|
15 return iter(self.Constants + self.Variables + self.Functions)
|
|
217
|
16
|
|
|
17 @property
|
|
|
18 def Syms(self):
|
|
394
|
19 """ Get all the symbols defined in this scope """
|
|
217
|
20 syms = self.symbols.values()
|
|
|
21 return sorted(syms, key=lambda v: v.name)
|
|
|
22
|
|
|
23 @property
|
|
389
|
24 def Types(self):
|
|
|
25 return [s for s in self.Syms if isinstance(s, DefinedType)]
|
|
|
26
|
|
|
27 @property
|
|
217
|
28 def Constants(self):
|
|
306
|
29 return [s for s in self.Syms if type(s) is Constant]
|
|
217
|
30
|
|
|
31 @property
|
|
|
32 def Variables(self):
|
|
306
|
33 return [s for s in self.Syms if isinstance(s, Variable)]
|
|
217
|
34
|
|
|
35 @property
|
|
|
36 def Functions(self):
|
|
306
|
37 return [s for s in self.Syms if type(s) is Function]
|
|
217
|
38
|
|
|
39 def getSymbol(self, name):
|
|
272
|
40 if name in self.symbols:
|
|
|
41 return self.symbols[name]
|
|
|
42 # Look for symbol:
|
|
306
|
43 elif self.parent:
|
|
272
|
44 return self.parent.getSymbol(name)
|
|
306
|
45 else:
|
|
|
46 raise KeyError(name)
|
|
|
47
|
|
|
48 def __getitem__(self, key):
|
|
|
49 return self.getSymbol(key)
|
|
217
|
50
|
|
|
51 def hasSymbol(self, name):
|
|
272
|
52 if name in self.symbols:
|
|
|
53 return True
|
|
306
|
54 elif self.parent:
|
|
272
|
55 return self.parent.hasSymbol(name)
|
|
306
|
56 else:
|
|
|
57 return False
|
|
|
58
|
|
|
59 def __contains__(self, name):
|
|
|
60 return self.hasSymbol(name)
|
|
217
|
61
|
|
|
62 def addSymbol(self, sym):
|
|
306
|
63 assert sym.name not in self.symbols
|
|
307
|
64 assert isinstance(sym, Symbol)
|
|
272
|
65 self.symbols[sym.name] = sym
|
|
150
|
66
|
|
217
|
67 def __repr__(self):
|
|
|
68 return 'Scope with {} symbols'.format(len(self.symbols))
|
|
|
69
|
|
272
|
70
|
|
306
|
71 def createTopScope(target):
|
|
|
72 scope = Scope()
|
|
288
|
73 for tn in ['u64', 'u32', 'u16', 'u8']:
|
|
306
|
74 scope.addSymbol(BaseType(tn))
|
|
|
75 # buildin types:
|
|
|
76 intType = BaseType('int')
|
|
|
77 intType.bytesize = target.byte_sizes['int']
|
|
|
78 scope.addSymbol(intType)
|
|
|
79 scope.addSymbol(BaseType('double'))
|
|
|
80 scope.addSymbol(BaseType('void'))
|
|
|
81 scope.addSymbol(BaseType('bool'))
|
|
354
|
82 byteType = BaseType('byte')
|
|
|
83 byteType.bytesize = target.byte_sizes['byte']
|
|
|
84 scope.addSymbol(byteType)
|
|
|
85
|
|
|
86 # Construct string type from others:
|
|
|
87 ln = StructField('len', intType)
|
|
|
88 txt = StructField('txt', ArrayType(byteType, 0))
|
|
389
|
89 strType = DefinedType('string', PointerType(StructureType([ln, txt])),
|
|
|
90 None)
|
|
354
|
91 scope.addSymbol(strType)
|
|
306
|
92 return scope
|
