Mercurial > hg > graal-compiler

comparison graal/com.oracle.truffle.codegen.processor/src/com/oracle/truffle/codegen/processor/compiler/JDTCompiler.java @ 7291:a748e4d44694

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Truffle API to specify type-specalized Node classes; annotation processor for automatic code generation of the type-specialized Node classes during the build process
author Christian Humer <christian.humer@gmail.com>
date Fri, 21 Dec 2012 10:44:31 -0800
parents
children 6343a09b2ec1
comparison
equal deleted inserted replaced
7290:a81db08fe930 7291:a748e4d44694
1 /*
2 * Copyright (c) 2012, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23 package com.oracle.truffle.codegen.processor.compiler;
24
25 import javax.annotation.processing.*;
26 import javax.lang.model.element.*;
27
28 import com.oracle.truffle.codegen.processor.*;
29
30 public class JDTCompiler extends AbstractCompiler {
31
32 public static boolean isValidElement(Element currentElement) {
33 try {
34 Class<?> elementClass = Class.forName("org.eclipse.jdt.internal.compiler.apt.model.ElementImpl");
35 return elementClass.isAssignableFrom(currentElement.getClass());
36 } catch (ClassNotFoundException e) {
37 return false;
38 }
39 }
40
41 @Override
42 public String getMethodBody(ProcessingEnvironment env, ExecutableElement method) {
43 try {
44
45 char[] source = getSource(method);
46 if (source == null) {
47 return null;
48 }
49
50 /*
51 AbstractMethodDeclaration decl = ((MethodBinding)(((ElementImpl)method)._binding)).sourceMethod();
52 int bodyStart = decl.bodyStart;
53 int bodyEnd = decl.bodyEnd;
54 */
55 Object decl = method(field(method, "_binding"), "sourceMethod");
56 int bodyStart = (int) field(decl, "bodyStart");
57 int bodyEnd = (int) field(decl, "bodyEnd");
58
59 int length = bodyEnd - bodyStart;
60 char[] target = new char[length];
61 System.arraycopy(source, bodyStart, target, 0, length);
62
63 return new String(target);
64 } catch (Exception e) {
65 return Utils.printException(e);
66 }
67 }
68
69
70 private static char[] getSource(Element element) throws Exception {
71 /*
72 Binding binding = ((ElementImpl)element)._binding;
73 char[] source = null;
74 if (binding instanceof MethodBinding) {
75 source = ((MethodBinding) binding).sourceMethod().compilationResult.getCompilationUnit().getContents();
76 } else if (binding instanceof SourceTypeBinding) {
77 source = ((SourceTypeBinding)binding).scope.referenceContext.compilationResult.compilationUnit.getContents();
78 }
79 return source;
80 */
81
82 Object binding = field(element, "_binding");
83 Class<?> methodBindingClass = Class.forName("org.eclipse.jdt.internal.compiler.lookup.MethodBinding");
84 Class<?> referenceBindingClass = Class.forName("org.eclipse.jdt.internal.compiler.lookup.SourceTypeBinding");
85
86 char[] source = null;
87 if (methodBindingClass.isAssignableFrom(binding.getClass())) {
88 Object sourceMethod = method(binding, "sourceMethod");
89 if (sourceMethod == null) {
90 return null;
91 }
92 source = (char[]) method(method(field(sourceMethod, "compilationResult"), "getCompilationUnit"), "getContents");
93 } else if (referenceBindingClass.isAssignableFrom(binding.getClass())) {
94 source = (char[]) method(field(field(field(field(binding, "scope"), "referenceContext"), "compilationResult"), "compilationUnit"), "getContents");
95 }
96 return source;
97 }
98
99
100 @Override
101 public String getHeaderComment(ProcessingEnvironment env, Element type) {
102 try {
103 char[] source = getSource(type);
104 if (source == null) {
105 return null;
106 }
107 return parseHeader(new String(source));
108 } catch (Exception e) {
109 return Utils.printException(e);
110 }
111 }
112
113 }