diff --git a/MemberNodes.py b/MemberNodes.py index 87caedc71cbef61faf8310a21af5f7db7c4f67ba..872a646b5a717ddb17e541ec23873152b29f2d87 100644 --- a/MemberNodes.py +++ b/MemberNodes.py @@ -139,6 +139,13 @@ class MethodNode(ASTNode): if not self.body: return + # check no use of this in static method + if 'static' in self.mods: + names = getNameNodes(self.body) + for n in names: + if 'this' in n.name or (n.pointToThis and n.prefixLink.__class__.__name__ == ['MethodNode', 'FieldNode'] and 'static' not in n.prefixLink.mods): + raise Exception("ERROR: Cannot use non-static member {} in static method {} in class {}".format(n.name, self.name, self.typeName)) + # With method body returnNodes = getASTNode(["ReturnNode"], self.body) @@ -177,3 +184,18 @@ def getForwardRefNames(node): result.extend(getForwardRefNames(c)) return result + +# Input: AST Node +# Output: A list of names to be check +def getNameNodes(node): + if not node: + return [] + + if node.__class__.__name__ == 'NameNode': + return [node] + + result = [] + for c in node.children: + result.extend(getNameNodes(c)) + + return result diff --git a/NameNode.py b/NameNode.py index 74dee52affac06112b49cf865d245470603733e0..f9d03bdecc870558021c5482892e72dc9f4ae27e 100644 --- a/NameNode.py +++ b/NameNode.py @@ -34,6 +34,7 @@ class NameNode(ASTNode): self.children = [] self.myType = None # will become TypeStruct to tell us what the whole is/returns self.shouldBeStatic = False + self.pointToThis = False self.name = getParseTreeNodes(["ID", "COMPID"], parseTree)[0].lex @@ -170,6 +171,7 @@ class NameNode(ASTNode): def disambigName(self): # Checking if a1 is "this" if self.checkThis(): + self.pointToThis = True return # Checking if a1 is length @@ -182,6 +184,7 @@ class NameNode(ASTNode): # Checking if a1 is in contains set if self.checkContains(): + self.pointToThis = True return