001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common
008: * Development and Distribution License("CDDL") (collectively, the
009: * "License"). You may not use this file except in compliance with the
010: * License. You can obtain a copy of the License at
011: * http://www.netbeans.org/cddl-gplv2.html
012: * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the
013: * specific language governing permissions and limitations under the
014: * License. When distributing the software, include this License Header
015: * Notice in each file and include the License file at
016: * nbbuild/licenses/CDDL-GPL-2-CP. Sun designates this
017: * particular file as subject to the "Classpath" exception as provided
018: * by Sun in the GPL Version 2 section of the License file that
019: * accompanied this code. If applicable, add the following below the
020: * License Header, with the fields enclosed by brackets [] replaced by
021: * your own identifying information:
022: * "Portions Copyrighted [year] [name of copyright owner]"
023: *
024: * Contributor(s):
025: *
026: * The Original Software is NetBeans. The Initial Developer of the Original
027: * Software is Sun Microsystems, Inc. Portions Copyright 1997-2006 Sun
028: * Microsystems, Inc. All Rights Reserved.
029: *
030: * If you wish your version of this file to be governed by only the CDDL
031: * or only the GPL Version 2, indicate your decision by adding
032: * "[Contributor] elects to include this software in this distribution
033: * under the [CDDL or GPL Version 2] license." If you do not indicate a
034: * single choice of license, a recipient has the option to distribute
035: * your version of this file under either the CDDL, the GPL Version 2 or
036: * to extend the choice of license to its licensees as provided above.
037: * However, if you add GPL Version 2 code and therefore, elected the GPL
038: * Version 2 license, then the option applies only if the new code is
039: * made subject to such option by the copyright holder.
040: */
041: package org.netbeans.modules.java.editor.semantic;
042:
043: import com.sun.source.tree.ClassTree;
044: import com.sun.source.tree.IdentifierTree;
045: import com.sun.source.tree.MemberSelectTree;
046: import com.sun.source.tree.MethodTree;
047: import com.sun.source.tree.Tree;
048: import com.sun.source.tree.VariableTree;
049: import com.sun.source.util.TreePath;
050: import java.util.HashSet;
051: import java.util.Set;
052: import java.util.Stack;
053: import javax.lang.model.element.Element;
054: import javax.swing.text.Document;
055: import org.netbeans.api.java.lexer.JavaTokenId;
056: import org.netbeans.api.java.source.CompilationInfo;
057: import org.netbeans.api.java.source.support.CancellableTreePathScanner;
058: import org.netbeans.api.lexer.Token;
059:
060: /**
061: *
062: * @author Jan Lahoda
063: */
064: public class FindLocalUsagesQuery extends
065: CancellableTreePathScanner<Void, Stack<Tree>> {
066:
067: private CompilationInfo info;
068: private Set<Token<JavaTokenId>> usages;
069: private Element toFind;
070: private Document doc;
071:
072: /** Creates a new instance of FindLocalUsagesQuery */
073: public FindLocalUsagesQuery() {
074: }
075:
076: public Set<Token<JavaTokenId>> findUsages(Element element,
077: CompilationInfo info, Document doc) {
078: this .info = info;
079: this .usages = new HashSet<Token<JavaTokenId>>();
080: this .toFind = element;
081: this .doc = doc;
082:
083: scan(info.getCompilationUnit(), null);
084: return usages;
085: }
086:
087: private void handlePotentialVariable(TreePath tree) {
088: Element el = info.getTrees().getElement(tree);
089:
090: if (toFind.equals(el)) {
091: Token<JavaTokenId> t = Utilities.getToken(info, doc, tree);
092:
093: if (t != null)
094: usages.add(t);
095: }
096: }
097:
098: @Override
099: public Void visitIdentifier(IdentifierTree tree, Stack<Tree> d) {
100: handlePotentialVariable(getCurrentPath());
101: super .visitIdentifier(tree, d);
102: return null;
103: }
104:
105: @Override
106: public Void visitMethod(MethodTree tree, Stack<Tree> d) {
107: handlePotentialVariable(getCurrentPath());
108: super .visitMethod(tree, d);
109: return null;
110: }
111:
112: @Override
113: public Void visitMemberSelect(MemberSelectTree node, Stack<Tree> p) {
114: handlePotentialVariable(getCurrentPath());
115: super .visitMemberSelect(node, p);
116: return null;
117: }
118:
119: @Override
120: public Void visitVariable(VariableTree tree, Stack<Tree> d) {
121: handlePotentialVariable(getCurrentPath());
122: super .visitVariable(tree, d);
123: return null;
124: }
125:
126: @Override
127: public Void visitClass(ClassTree tree, Stack<Tree> d) {
128: handlePotentialVariable(getCurrentPath());
129: super.visitClass(tree, d);
130: return null;
131: }
132: }
|