test/tools/javac/T6406771.java

changeset 1
9a66ca7c79fa
child 495
fe17a9dbef03
equal deleted inserted replaced
-1:000000000000 1:9a66ca7c79fa
1 /*
2 * @test /nodynamiccopyright/
3 * @bug 6406771
4 * @summary CompilationUnitTree needs access to a line map
5 */
6
7 // WARNING: White-space and layout is important in this file, especially tab characters.
8
9 import java.io.*;
10 import java.util.*;
11 import javax.annotation.processing.*;
12 import javax.lang.model.*;
13 import javax.lang.model.element.*;
14 import javax.tools.*;
15 import com.sun.tools.javac.api.*;
16 import com.sun.source.tree.*;
17 import com.sun.source.util.*;
18 import com.sun.tools.javac.tree.JCTree;
19
20 @SupportedSourceVersion(SourceVersion.RELEASE_6)
21 @SupportedAnnotationTypes("*")
22 public class T6406771 extends AbstractProcessor {
23 String[] tests = {
24 "line:24",
25 "line:25",
26 "line:26", "line:26",
27 // 1 2 3 4 5 6
28 //3456789012345678901234567890123456789012345678901234567890
29 "col:7", "col:16", "col:26", // this line uses spaces
30 "col:9", "col:25", "col:41", // this line uses tabs
31 "col:20", "col:43" // this line uses a mixture
32 };
33
34 // White-space after this point does not matter
35
36 public static void main(String[] args) {
37 String self = T6406771.class.getName();
38 String testSrc = System.getProperty("test.src");
39 String testClasses = System.getProperty("test.classes");
40
41 JavacTool tool = JavacTool.create();
42 StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null);
43 JavaFileObject f = fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrc, self+".java"))).iterator().next();
44
45 List<String> opts = Arrays.asList("-d", ".", "-processorpath", testClasses, "-processor", self, "-proc:only");
46
47 JavacTask task = tool.getTask(null, fm, null, opts, null, Arrays.asList(f));
48
49 if (!task.call())
50 throw new AssertionError("failed");
51 }
52
53 public boolean process(Set<? extends TypeElement> elems, RoundEnvironment rEnv) {
54 final String LINE = "line" + ':'; // avoid matching this string
55 final String COLUMN = "col" + ':';
56 final Messager messager = processingEnv.getMessager();
57 final Trees trees = Trees.instance(processingEnv);
58
59 TreeScanner<Void,LineMap> s = new TreeScanner<Void,LineMap>() {
60 public Void visitLiteral(LiteralTree tree, LineMap lineMap) {
61 if (tree.getKind() == Tree.Kind.STRING_LITERAL) {
62 String s = (String) tree.getValue();
63 int pos = ((JCTree) tree).pos; // can't get through public api, bug 6412669 filed
64 String prefix;
65 long found;
66 if (s.startsWith(LINE)) {
67 prefix = LINE;
68 found = lineMap.getLineNumber(pos);
69 }
70 else if (s.startsWith(COLUMN)) {
71 prefix = COLUMN;
72 found = lineMap.getColumnNumber(pos);
73 }
74 else
75 return null;
76 int expect = Integer.parseInt(s.substring(prefix.length()));
77 if (expect != found) {
78 messager.printMessage(Diagnostic.Kind.ERROR,
79 "Error: " + prefix + " pos=" + pos
80 + " expect=" + expect + " found=" + found);
81 }
82 }
83 return null;
84 }
85 };
86
87 for (Element e: rEnv.getRootElements()) {
88 System.err.println(e);
89 Tree t = trees.getTree(e);
90 TreePath p = trees.getPath(e);
91 s.scan(p.getLeaf(), p.getCompilationUnit().getLineMap());
92
93 }
94
95 return true;
96 }
97
98 }

mercurial