|
148
|
1 from . import astnodes
|
|
150
|
2 from .scope import Scope, topScope
|
|
148
|
3
|
|
|
4 class Semantics:
|
|
|
5 """ This class constructs the AST from parser input """
|
|
|
6 def __init__(self, diag):
|
|
|
7 self.diag = diag
|
|
150
|
8 def addSymbol(self, s):
|
|
|
9 if self.curScope.hasSymbol(s.name):
|
|
|
10 msg = 'Redefinition of {0}'.format(s.name)
|
|
152
|
11 self.diag.error(msg, s.loc)
|
|
150
|
12 else:
|
|
|
13 self.curScope.addSymbol(s)
|
|
148
|
14 def handlePackage(self, name, loc):
|
|
|
15 self.mod = astnodes.Package(name)
|
|
|
16 self.mod.loc = loc
|
|
150
|
17 self.mod.scope = self.curScope = Scope(topScope)
|
|
|
18 def actOnVarDef(self, name, loc, t, ival):
|
|
|
19 s = astnodes.Variable(name, t)
|
|
|
20 s.loc = loc
|
|
|
21 self.addSymbol(s)
|
|
163
|
22 def actOnConstDef(self, name, loc, t, val):
|
|
|
23 s = astnodes.Constant(name, t, val)
|
|
|
24 s.loc = loc
|
|
|
25 self.addSymbol(s)
|
|
150
|
26 def actOnFuncDef1(self, name, loc):
|
|
|
27 self.curFunc = astnodes.Function(name)
|
|
|
28 self.curFunc.loc = loc
|
|
|
29 self.addSymbol(self.curFunc)
|
|
|
30 self.curScope = self.curFunc.scope = Scope(self.curScope)
|
|
|
31 def actOnParameter(self, name, loc, t):
|
|
|
32 p = astnodes.Variable(name, t)
|
|
|
33 p.loc = loc
|
|
|
34 p.parameter = True
|
|
|
35 self.addSymbol(p)
|
|
|
36 return p
|
|
|
37 def actOnFuncDef2(self, parameters, returntype, body):
|
|
|
38 self.curFunc.body = body
|
|
167
|
39 paramtypes = [p.typ for p in parameters]
|
|
|
40 self.curFunc.typ = astnodes.FunctionType(paramtypes, returntype)
|
|
150
|
41 self.curFunc = None
|
|
|
42 self.curScope = self.curScope.parent
|
|
|
43 def actOnType(self, tok):
|
|
|
44 # Try to lookup type, in case of failure return void
|
|
|
45 pass
|
|
|
46 def actOnDesignator(self, tname, loc):
|
|
|
47 d = astnodes.Designator(tname)
|
|
|
48 d.scope = self.curScope
|
|
|
49 d.loc = loc
|
|
|
50 return d
|
|
149
|
51 def actOnBinop(self, lhs, op, rhs, loc):
|
|
|
52 bo = astnodes.Binop(lhs, op, rhs)
|
|
|
53 bo.loc = loc
|
|
|
54 return bo
|
|
|
55 def actOnNumber(self, num, loc):
|
|
163
|
56 n = astnodes.Literal(num)
|
|
149
|
57 n.loc = loc
|
|
|
58 return n
|
|
165
|
59 def actOnVariableUse(self, d, loc):
|
|
164
|
60 vu = astnodes.VariableUse(d)
|
|
165
|
61 vu.loc = loc
|
|
164
|
62 return vu
|
|
|
63 def actOnAssignment(self, lval, rval, loc):
|
|
|
64 a = astnodes.Assignment(lval, rval)
|
|
|
65 a.loc = loc
|
|
|
66 return a
|
|
167
|
67 def actOnFunctionCall(self, func, args, loc):
|
|
|
68 fc = astnodes.FunctionCall(func, args)
|
|
|
69 fc.loc = loc
|
|
|
70 return fc
|
|
165
|
71 def actOnIfStatement(self, cond, yes, no, loc):
|
|
|
72 i = astnodes.IfStatement(cond, yes, no)
|
|
|
73 i.loc = loc
|
|
|
74 return i
|
