Package

Source Code of T6956462$MyVisitor

/*
* Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/

/*
* @test
* @bug 6956462
* @summary AssertionError exception throws in the Compiler Tree API in JDK 7.
*
* @build TestClass T6956462
* @run main T6956462
*/

import java.io.*;
import java.net.URISyntaxException;
import java.util.*;
import javax.tools.*;
import javax.tools.JavaCompiler.CompilationTask;
import com.sun.source.tree.*;
import com.sun.source.util.*;

public class T6956462 {
    public static void main(String[] args) throws Exception {
        JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
        if (compiler == null) {
            throw new RuntimeException("can't get javax.tools.JavaCompiler!");
        }
        StandardJavaFileManager fm = compiler.getStandardFileManager(null, null, null);
        List<File> files = new ArrayList<File>();
        files.add(new File(T6956462.class.getResource("TestClass.java").toURI()));
        final CompilationTask task = compiler.getTask(null, fm, null,
            null, null, fm.getJavaFileObjectsFromFiles(files));
        JavacTask javacTask = (JavacTask) task;
        for (CompilationUnitTree cu : javacTask.parse()) {
            cu.accept(new MyVisitor(javacTask), null);
        }
    }

    private static class MyVisitor extends SimpleTreeVisitor<Tree, Void> {
        private final Trees trees;
        private CompilationUnitTree file;

        private MyVisitor(JavacTask javac) {
            this.trees = Trees.instance(javac);
        }

        @Override
        public Tree visitCompilationUnit(CompilationUnitTree file, Void v) {
            this.file = file;
            for (Tree typeDecl : file.getTypeDecls()) {
                typeDecl.accept(this, v);
            }
            return null;
        }

        @Override
        public Tree visitImport(ImportTree imp, Void v) {
            return null;
        }

        @Override
        public Tree visitMethodInvocation(MethodInvocationTree invoke, Void v) {
            invoke.getMethodSelect().accept(this, v);
            return null;
        }

        @Override
        public Tree visitBlock(BlockTree block, Void v) {
            for (StatementTree stat : block.getStatements()) {
                stat.accept(this, v);
            }
            return null;
        }

        @Override
        public Tree visitClass(ClassTree clazz, Void v) {
            for (Tree member : clazz.getMembers()) {
                member.accept(this, v);
            }
            return null;
        }

        @Override
        public Tree visitIdentifier(IdentifierTree ident, Void v) {
            trees.getScope(trees.getPath(file, ident));
            return null;
        }

        @Override
        public Tree visitMethod(MethodTree method, Void v) {
            method.getBody().accept(this, v);
            return null;
        }

        @Override
        public Tree visitMemberSelect(MemberSelectTree select, Void v) {
            select.getExpression().accept(this, v);
            return null;
        }

        @Override
        public Tree visitVariable(VariableTree var, Void v) {
            var.getInitializer().accept(this, v);
            return null;
        }
    }
}
TOP

Related Classes of T6956462$MyVisitor

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.