1
2
3
4 package net.sourceforge.pmd.lang.java.symboltable;
5
6 import java.util.ArrayList;
7 import java.util.HashMap;
8 import java.util.List;
9 import java.util.Map;
10
11 import net.sourceforge.pmd.lang.ast.Node;
12 import net.sourceforge.pmd.lang.java.ast.ASTName;
13
14 public class LocalScope extends AbstractScope {
15
16 protected Map<VariableNameDeclaration, List<NameOccurrence>> variableNames = new HashMap<VariableNameDeclaration, List<NameOccurrence>>();
17
18 public NameDeclaration addVariableNameOccurrence(NameOccurrence occurrence) {
19 NameDeclaration decl = findVariableHere(occurrence);
20 if (decl != null && !occurrence.isThisOrSuper()) {
21 List<NameOccurrence> nameOccurrences = variableNames.get(decl);
22 nameOccurrences.add(occurrence);
23 Node n = occurrence.getLocation();
24 if (n instanceof ASTName) {
25 ((ASTName) n).setNameDeclaration(decl);
26 }
27 }
28 return decl;
29 }
30
31 public Map<VariableNameDeclaration, List<NameOccurrence>> getVariableDeclarations() {
32 VariableUsageFinderFunction f = new VariableUsageFinderFunction(variableNames);
33 Applier.apply(f, variableNames.keySet().iterator());
34 return f.getUsed();
35 }
36
37 public void addDeclaration(VariableNameDeclaration nameDecl) {
38 if (variableNames.containsKey(nameDecl)) {
39 throw new RuntimeException("Variable " + nameDecl + " is already in the symbol table");
40 }
41 variableNames.put(nameDecl, new ArrayList<NameOccurrence>());
42 }
43
44 public NameDeclaration findVariableHere(NameOccurrence occurrence) {
45 if (occurrence.isThisOrSuper() || occurrence.isMethodOrConstructorInvocation()) {
46 return null;
47 }
48 ImageFinderFunction finder = new ImageFinderFunction(occurrence.getImage());
49 Applier.apply(finder, variableNames.keySet().iterator());
50 return finder.getDecl();
51 }
52
53 public String toString() {
54 return "LocalScope:" + glomNames(variableNames.keySet());
55 }
56 }