KickJava   Java API By Example, From Geeks To Geeks.

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


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 javax.lang.model.element.*;
24 import org.netbeans.api.java.source.WorkingCopy;
25
26 /**
27  *
28  * @author Jan Becicka
29  */

30 public class DeleteTransformer extends SearchVisitor {
31
32     public DeleteTransformer(WorkingCopy workingCopy) {
33         super(workingCopy);
34     }
35
36     @Override JavaDoc
37     public Tree visitMethod(MethodTree tree, Element p) {
38         deleteDeclIfMatch(tree, p);
39         return super.visitMethod(tree, p);
40     }
41
42     public Tree visitClass(ClassTree tree, Element p) {
43         deleteDeclIfMatch(tree, p);
44         return super.visitClass(tree, p);
45     }
46
47     public Tree visitVariable(VariableTree tree, Element p) {
48         deleteDeclIfMatch(tree, p);
49         return super.visitVariable(tree, p);
50     }
51     
52     private void deleteDeclIfMatch(Tree tree, Element elementToFind) {
53         if (workingCopy.getTreeUtilities().isSynthetic(getCurrentPath()))
54             return ;
55         
56         Element el = workingCopy.getTrees().getElement(getCurrentPath());
57         if (elementToFind.equals(el)) {
58             Tree parent = getCurrentPath().getParentPath().getLeaf();
59             Tree newOne = null;
60             if (parent.getKind() == Tree.Kind.CLASS) {
61                 newOne = make.removeClassMember((ClassTree) parent, tree);
62             } else if (parent.getKind() == Tree.Kind.COMPILATION_UNIT) {
63                 newOne = make.removeCompUnitTypeDecl((CompilationUnitTree) parent, tree);
64             }
65             if (newOne!=null) {
66                 workingCopy.rewrite(parent,newOne);
67             }
68             addUsage(getCurrentPath());
69         }
70     }
71 }
72
Popular Tags