01 /**
02 * BSD-style license; for more info see http://pmd.sourceforge.net/license.html
03 */
04 package net.sourceforge.pmd.lang.java.symboltable;
05
06 import java.util.ArrayList;
07 import java.util.HashMap;
08 import java.util.List;
09 import java.util.Map;
10
11 import net.sourceforge.pmd.lang.ast.Node;
12 import net.sourceforge.pmd.lang.java.ast.ASTConstructorDeclaration;
13 import net.sourceforge.pmd.lang.java.ast.ASTName;
14 import net.sourceforge.pmd.util.Applier;
15
16 public class MethodScope extends AbstractScope {
17
18 protected Map<VariableNameDeclaration, List<NameOccurrence>> variableNames = new HashMap<VariableNameDeclaration, List<NameOccurrence>>();
19 private Node node;
20
21 public MethodScope(Node node) {
22 this.node = node;
23 }
24
25 public MethodScope getEnclosingMethodScope() {
26 return this;
27 }
28
29 public Map<VariableNameDeclaration, List<NameOccurrence>> getVariableDeclarations() {
30 VariableUsageFinderFunction f = new VariableUsageFinderFunction(variableNames);
31 Applier.apply(f, variableNames.keySet().iterator());
32 return f.getUsed();
33 }
34
35 public NameDeclaration addVariableNameOccurrence(NameOccurrence occurrence) {
36 NameDeclaration decl = findVariableHere(occurrence);
37 if (decl != null && !occurrence.isThisOrSuper()) {
38 variableNames.get(decl).add(occurrence);
39 Node n = occurrence.getLocation();
40 if (n instanceof ASTName) {
41 ((ASTName) n).setNameDeclaration(decl);
42 } // TODO what to do with PrimarySuffix case?
43 }
44 return decl;
45 }
46
47 public void addDeclaration(VariableNameDeclaration variableDecl) {
48 if (variableNames.containsKey(variableDecl)) {
49 throw new RuntimeException("Variable " + variableDecl + " is already in the symbol table");
50 }
51 variableNames.put(variableDecl, new ArrayList<NameOccurrence>());
52 }
53
54 public NameDeclaration findVariableHere(NameOccurrence occurrence) {
55 if (occurrence.isThisOrSuper() || occurrence.isMethodOrConstructorInvocation()) {
56 return null;
57 }
58 ImageFinderFunction finder = new ImageFinderFunction(occurrence.getImage());
59 Applier.apply(finder, variableNames.keySet().iterator());
60 return finder.getDecl();
61 }
62
63 public String getName() {
64 if (node instanceof ASTConstructorDeclaration) {
65 return this.getEnclosingClassScope().getClassName();
66 }
67 return node.jjtGetChild(1).getImage();
68 }
69
70 public String toString() {
71 return "MethodScope:" + glomNames(variableNames.keySet());
72 }
73 }
|