1 /*

     2  * Copyright 2006 Sun Microsystems, Inc.  All Rights Reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.

     8  *

     9  * This code is distributed in the hope that it will be useful, but WITHOUT

    10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    12  * version 2 for more details (a copy is included in the LICENSE file that

    13  * accompanied this code).

    14  *

    15  * You should have received a copy of the GNU General Public License version

    16  * 2 along with this work; if not, write to the Free Software Foundation,

    17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    18  *

    19  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

    20  * CA 95054 USA or visit www.sun.com if you need additional information or

    21  * have any questions.

    22  */

    23 

    24 /*

    25  * @test

    26  * @bug 6412669

    27  * @summary Should be able to get SourcePositions from 269 world

    28  */

    29 

    30 import java.io.*;

    31 import java.util.*;

    32 import javax.annotation.*;

    33 import javax.annotation.processing.*;

    34 import javax.lang.model.*;

    35 import javax.lang.model.element.*;

    36 import javax.tools.*;

    37 import com.sun.source.util.*;

    38 import com.sun.tools.javac.api.*;

    39 

    40 @SupportedAnnotationTypes("*")

    41 @SupportedSourceVersion(SourceVersion.RELEASE_6)

    42 public class T6412669 extends AbstractProcessor {

    43     public static void main(String... args) throws IOException {

    44         String testSrc = System.getProperty("test.src", ".");

    45         String testClasses = System.getProperty("test.classes", ".");

    46 

    47         JavacTool tool = JavacTool.create();

    48         StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null);

    49         fm.setLocation(StandardLocation.CLASS_PATH, Arrays.asList(new File(testClasses)));

    50         Iterable<? extends JavaFileObject> files =

    51             fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrc, T6412669.class.getName()+".java")));

    52         String[] opts = { "-proc:only", "-processor", T6412669.class.getName(),

    53                           "-classpath", new File(testClasses).getPath() };

    54         JavacTask task = tool.getTask(null, fm, null, Arrays.asList(opts), null, files);

    55         if (!task.call())

    56             throw new AssertionError("test failed");

    57     }

    58 

    59     public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {

    60         Trees trees = Trees.instance(processingEnv);

    61         SourcePositions sp = trees.getSourcePositions();

    62         Messager m = processingEnv.getMessager();

    63         for (TypeElement anno: annotations) {

    64             for (Element e: roundEnv.getElementsAnnotatedWith(anno)) {

    65                 TreePath p = trees.getPath(e);

    66                 long start = sp.getStartPosition(p.getCompilationUnit(), p.getLeaf());

    67                 long end = sp.getEndPosition(p.getCompilationUnit(), p.getLeaf());

    68                 Diagnostic.Kind k = (start > 0 && end > 0 && start < end

    69                                      ? Diagnostic.Kind.NOTE : Diagnostic.Kind.ERROR);

    70                 m.printMessage(k, "test [" + start + "," + end + "]", e);

    71             }

    72         }

    73         return true;

    74     }

    75 }