Mercurial > hg > graal-jvmci-8

comparison graal/com.oracle.truffle.dsl.processor/src/com/oracle/truffle/dsl/processor/compiler/JDTCompiler.java @ 10597:79041ab43660

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Truffle-DSL: API-change: Renamed truffle.api.codegen to truffle.api.dsl for all projects and packages.
author Christian Humer <christian.humer@gmail.com>
date Mon, 01 Jul 2013 20:58:32 +0200
parents
children ae6b8ec920e2
comparison
equal deleted inserted replaced
10596:f43eb2f1bbbc 10597:79041ab43660
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.dsl.processor.compiler;
24
25 import javax.annotation.processing.*;
26 import javax.lang.model.element.*;
27
28 import com.oracle.truffle.dsl.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 =
52 * ((MethodBinding)(((ElementImpl)method)._binding)).sourceMethod(); int bodyStart =
53 * decl.bodyStart; 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 private static char[] getSource(Element element) throws Exception {
70 /*
71 * Binding binding = ((ElementImpl)element)._binding; char[] source = null; if (binding
72 * instanceof MethodBinding) { source = ((MethodBinding)
73 * binding).sourceMethod().compilationResult.getCompilationUnit().getContents(); } else if
74 * (binding instanceof SourceTypeBinding) { source =
75 * ((SourceTypeBinding)binding).scope.referenceContext
76 * .compilationResult.compilationUnit.getContents(); } return source;
77 */
78
79 Object binding = field(element, "_binding");
80 Class<?> methodBindingClass = Class.forName("org.eclipse.jdt.internal.compiler.lookup.MethodBinding");
81 Class<?> referenceBindingClass = Class.forName("org.eclipse.jdt.internal.compiler.lookup.SourceTypeBinding");
82
83 char[] source = null;
84 if (methodBindingClass.isAssignableFrom(binding.getClass())) {
85 Object sourceMethod = method(binding, "sourceMethod");
86 if (sourceMethod == null) {
87 return null;
88 }
89 source = (char[]) method(method(field(sourceMethod, "compilationResult"), "getCompilationUnit"), "getContents");
90 } else if (referenceBindingClass.isAssignableFrom(binding.getClass())) {
91 source = (char[]) method(field(field(field(field(binding, "scope"), "referenceContext"), "compilationResult"), "compilationUnit"), "getContents");
92 }
93 return source;
94 }
95
96 @Override
97 public String getHeaderComment(ProcessingEnvironment env, Element type) {
98 try {
99 char[] source = getSource(type);
100 if (source == null) {
101 return null;
102 }
103 return parseHeader(new String(source));
104 } catch (Exception e) {
105 return Utils.printException(e);
106 }
107 }
108
109 }