duke@1: /*
duke@1:  * @test  /nodynamiccopyright/
duke@1:  * @bug 6406771
duke@1:  * @summary CompilationUnitTree needs access to a line map
duke@1:  */
duke@1: 
duke@1: // WARNING: White-space and layout is important in this file, especially tab characters.
duke@1: 
duke@1: import java.io.*;
duke@1: import java.util.*;
duke@1: import javax.annotation.processing.*;
duke@1: import javax.lang.model.*;
duke@1: import javax.lang.model.element.*;
duke@1: import javax.tools.*;
duke@1: import com.sun.tools.javac.api.*;
duke@1: import com.sun.source.tree.*;
duke@1: import com.sun.source.util.*;
duke@1: import com.sun.tools.javac.tree.JCTree;
duke@1: 
duke@1: @SupportedSourceVersion(SourceVersion.RELEASE_6)
duke@1: @SupportedAnnotationTypes("*")
duke@1: public class T6406771 extends AbstractProcessor {
duke@1:     String[] tests = {
duke@1:         "line:24",
duke@1:         "line:25",
duke@1:         "line:26", "line:26",
duke@1: //       1         2         3         4         5         6
duke@1: //3456789012345678901234567890123456789012345678901234567890
duke@1:       "col:7", "col:16", "col:26",                 // this line uses spaces
duke@1:         "col:9",        "col:25",       "col:41",  // this line uses tabs
duke@1:                    "col:20",              "col:43" // this line uses a mixture
duke@1:     };
duke@1: 
duke@1:     // White-space after this point does not matter
duke@1: 
duke@1:     public static void main(String[] args) {
duke@1:         String self = T6406771.class.getName();
duke@1:         String testSrc = System.getProperty("test.src");
duke@1:         String testClasses = System.getProperty("test.classes");
duke@1: 
duke@1:         JavacTool tool = JavacTool.create();
duke@1:         StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null);
duke@1:         JavaFileObject f = fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrc, self+".java"))).iterator().next();
duke@1: 
duke@1:         List<String> opts = Arrays.asList("-d", ".", "-processorpath", testClasses, "-processor", self, "-proc:only");
duke@1: 
duke@1:         JavacTask task = tool.getTask(null, fm, null, opts, null, Arrays.asList(f));
duke@1: 
duke@1:         if (!task.call())
duke@1:             throw new AssertionError("failed");
duke@1:     }
duke@1: 
duke@1:     public boolean process(Set<? extends TypeElement> elems, RoundEnvironment rEnv) {
duke@1:         final String LINE = "line" + ':';   // avoid matching this string
duke@1:         final String COLUMN  = "col" + ':';
duke@1:         final Messager messager = processingEnv.getMessager();
duke@1:         final Trees trees = Trees.instance(processingEnv);
duke@1: 
duke@1:         TreeScanner<Void,LineMap> s = new  TreeScanner<Void,LineMap>() {
duke@1:             public Void visitLiteral(LiteralTree tree, LineMap lineMap) {
duke@1:                 if (tree.getKind() == Tree.Kind.STRING_LITERAL) {
duke@1:                     String s = (String) tree.getValue();
duke@1:                     int pos = ((JCTree) tree).pos; // can't get through public api, bug 6412669 filed
duke@1:                     String prefix;
duke@1:                     long found;
duke@1:                     if (s.startsWith(LINE)) {
duke@1:                         prefix = LINE;
duke@1:                         found = lineMap.getLineNumber(pos);
duke@1:                     }
duke@1:                     else if (s.startsWith(COLUMN)) {
duke@1:                         prefix = COLUMN;
duke@1:                         found = lineMap.getColumnNumber(pos);
duke@1:                     }
duke@1:                     else
duke@1:                         return null;
duke@1:                     int expect = Integer.parseInt(s.substring(prefix.length()));
duke@1:                     if (expect != found) {
duke@1:                         messager.printMessage(Diagnostic.Kind.ERROR,
duke@1:                                               "Error: " + prefix + " pos=" + pos
duke@1:                                               + " expect=" + expect + " found=" + found);
duke@1:                     }
duke@1:                 }
duke@1:                 return null;
duke@1:             }
duke@1:         };
duke@1: 
duke@1:         for (Element e: rEnv.getRootElements()) {
duke@1:             System.err.println(e);
duke@1:             Tree t = trees.getTree(e);
duke@1:             TreePath p = trees.getPath(e);
duke@1:             s.scan(p.getLeaf(), p.getCompilationUnit().getLineMap());
duke@1: 
duke@1:         }
duke@1: 
duke@1:         return true;
duke@1:     }
duke@1: 
duke@1: }