KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > netbeans > modules > refactoring > java > plugins > FindUsagesVisitor


1 /*
2  * The contents of this file are subject to the terms of the Common Development
3  * and Distribution License (the License). You may not use this file except in
4  * compliance with the License.
5  *
6  * You can obtain a copy of the License at http://www.netbeans.org/cddl.html
7  * or http://www.netbeans.org/cddl.txt.
8  *
9  * When distributing Covered Code, include this CDDL Header Notice in each file
10  * and include the License file at http://www.netbeans.org/cddl.txt.
11  * If applicable, add the following below the CDDL Header, with the fields
12  * enclosed by brackets [] replaced by your own identifying information:
13  * "Portions Copyrighted [year] [name of copyright owner]"
14  *
15  * The Original Software is NetBeans. The Initial Developer of the Original
16  * Software is Sun Microsystems, Inc. Portions Copyright 1997-2006 Sun
17  * Microsystems, Inc. All Rights Reserved.
18  */
19
20 package org.netbeans.modules.refactoring.java.plugins;
21
22 import com.sun.source.tree.*;
23 import com.sun.source.util.TreePath;
24 import javax.lang.model.element.*;
25 import org.netbeans.api.java.source.WorkingCopy;
26
27 /**
28  *
29  * @author Jan Becicka
30  */
31 public class FindUsagesVisitor extends SearchVisitor {
32
33     public FindUsagesVisitor(WorkingCopy workingCopy) {
34         super(workingCopy);
35     }
36
37     @Override JavaDoc
38     public Tree visitIdentifier(IdentifierTree node, Element p) {
39         addIfMatch(getCurrentPath(), node, p);
40         return super.visitIdentifier(node, p);
41     }
42
43     @Override JavaDoc
44     public Tree visitMemberSelect(MemberSelectTree node, Element p) {
45         addIfMatch(getCurrentPath(), node,p);
46         return super.visitMemberSelect(node, p);
47     }
48     
49     @Override JavaDoc
50     public Tree visitNewClass(NewClassTree node, Element p) {
51         addIfMatch(getCurrentPath(), node,p);
52         return super.visitNewClass(node, p);
53     }
54     
55     private void addIfMatch(TreePath path, Tree tree, Element elementToFind) {
56         if (workingCopy.getTreeUtilities().isSynthetic(path))
57             return ;
58         Element el = workingCopy.getTrees().getElement(path);
59         if (el==null)
60             return;
61         
62         if (elementToFind.getKind() == ElementKind.METHOD && el.getKind() == ElementKind.METHOD) {
63             if (el.equals(elementToFind) || workingCopy.getElements().overrides(((ExecutableElement) el), (ExecutableElement) elementToFind, (TypeElement) elementToFind.getEnclosingElement())) {
64                 addUsage(getCurrentPath());
65             }
66         } else if (el.equals(elementToFind)) {
67                 addUsage(getCurrentPath());
68         }
69     }
70 }
71
Popular Tags