duke@1: /*
ohair@798:  * Copyright (c) 2006, 2010, Oracle and/or its affiliates. All rights reserved.
duke@1:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
duke@1:  *
duke@1:  * This code is free software; you can redistribute it and/or modify it
duke@1:  * under the terms of the GNU General Public License version 2 only, as
duke@1:  * published by the Free Software Foundation.
duke@1:  *
duke@1:  * This code is distributed in the hope that it will be useful, but WITHOUT
duke@1:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
duke@1:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
duke@1:  * version 2 for more details (a copy is included in the LICENSE file that
duke@1:  * accompanied this code).
duke@1:  *
duke@1:  * You should have received a copy of the GNU General Public License version
duke@1:  * 2 along with this work; if not, write to the Free Software Foundation,
duke@1:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
duke@1:  *
ohair@554:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@554:  * or visit www.oracle.com if you need additional information or have any
ohair@554:  * questions.
duke@1:  */
duke@1: 
duke@1: /*
duke@1:  * @test
duke@1:  * @bug 6392782
duke@1:  * @summary TreeScanner.visitImport returns null, not result of nested scan
duke@1:  */
duke@1: 
duke@1: import java.io.*;
duke@1: import java.util.*;
duke@1: import javax.tools.*;
duke@1: import com.sun.source.tree.*;
duke@1: import com.sun.source.util.*;
duke@1: import com.sun.tools.javac.api.*;
duke@1: 
duke@1: public class T6392782 {
duke@1:     public static void main(String... args) throws IOException {
duke@1:         String testSrc = System.getProperty("test.src", ".");
duke@1:         JavacTool tool = JavacTool.create();
duke@1:         StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null);
duke@1:         Iterable<? extends JavaFileObject> files =
duke@1:             fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrc, T6392782.class.getName()+".java")));
duke@1:         JavacTask task = tool.getTask(null, fm, null, null, null, files);
duke@1:         Iterable<? extends Tree> trees = task.parse();
duke@1:         TreeScanner<Integer,Void> scanner = new MyScanner();
duke@1:         check(scanner, 6, scanner.scan(trees, null));
duke@1: 
duke@1:         CountNodes nodeCounter = new CountNodes();
jjg@679:         // 359 nodes with the regular parser; 360 nodes with EndPosParser
jjg@679:         // We automatically switch to EndPosParser when calling JavacTask.parse()
jjg@679:         check(nodeCounter, 360, nodeCounter.scan(trees, null));
duke@1: 
duke@1:         CountIdentifiers idCounter = new CountIdentifiers();
jjg@679:         check(idCounter, 107, idCounter.scan(trees, null));
duke@1:     }
duke@1: 
duke@1:     private static void check(TreeScanner<?,?> scanner, int expect, int found) {
duke@1:         if (found != expect)
duke@1:             throw new AssertionError(scanner.getClass().getName() + ": expected: " + expect + " found: " + found);
duke@1:     }
duke@1: 
duke@1:     static class MyScanner extends TreeScanner<Integer,Void> {
duke@1:         @Override
duke@1:         public Integer visitImport(ImportTree tree, Void ignore) {
duke@1:             //System.err.println(tree);
duke@1:             return 1;
duke@1:         }
duke@1: 
duke@1:         @Override
duke@1:         public Integer reduce(Integer i1, Integer i2) {
duke@1:             return (i1 == null ? 0 : i1) + (i2 == null ? 0 : i2);
duke@1:         }
duke@1:     }
duke@1: 
duke@1:     static class CountNodes extends TreeScanner<Integer,Void> {
duke@1:         @Override
duke@1:         public Integer scan(Tree node, Void p) {
jjg@679:             if (node == null)
jjg@679:                 return 0;
duke@1:             Integer n = super.scan(node, p);
duke@1:             return (n == null ? 0 : n) + 1;
duke@1:         }
duke@1:         @Override
duke@1:         public Integer reduce(Integer r1, Integer r2) {
duke@1:             return (r1 == null ? 0 : r1) + (r2 == null ? 0 : r2);
duke@1:         }
duke@1:     }
duke@1: 
duke@1:     // example from TreeScanner javadoc
duke@1:     static class CountIdentifiers extends TreeScanner<Integer,Void> {
duke@1:         @Override
duke@1:         public Integer visitIdentifier(IdentifierTree node, Void p) {
duke@1:             return 1;
duke@1:         }
duke@1:         @Override
duke@1:         public Integer reduce(Integer r1, Integer r2) {
duke@1:             return (r1 == null ? 0 : r1) + (r2 == null ? 0 : r2);
duke@1:         }
duke@1:     }
duke@1: }