Mercurial > hg > graal-compiler

comparison graal/com.oracle.truffle.api/src/com/oracle/truffle/api/nodes/serial/PostOrderSerializer.java @ 12388:96c1d057a5ed

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Truffle: Added experimental serialization API.
author Christian Humer <christian.humer@gmail.com>
date Wed, 02 Oct 2013 15:33:08 +0200
parents
children 915ebb306fcc
comparison
equal deleted inserted replaced
12387:aff825fef0fd 12388:96c1d057a5ed
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. Oracle designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Oracle in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 * or visit www.oracle.com if you need additional information or have any
23 * questions.
24 */
25 package com.oracle.truffle.api.nodes.serial;
26
27 import java.lang.reflect.*;
28 import java.nio.*;
29
30 import sun.misc.*;
31
32 import com.oracle.truffle.api.*;
33 import com.oracle.truffle.api.nodes.*;
34 import com.oracle.truffle.api.nodes.NodeUtil.NodeClass;
35 import com.oracle.truffle.api.nodes.NodeUtil.NodeField;
36 import com.oracle.truffle.api.nodes.NodeUtil.NodeFieldKind;
37
38 /**
39 * Experimental API. May change without notice.
40 */
41 public final class PostOrderSerializer {
42
43 private static final Unsafe unsafe = loadUnsafe();
44
45 private final SerializerConstantPool cp;
46
47 /**
48 * Constructs a new deserializer using a custom {@link SerializerConstantPool} implementation.
49 * For the {@link SerializerConstantPool} implementation at least the following methods must be
50 * implemented:
51 * <ul>
52 * <li>{@link SerializerConstantPool#putInt(int)}</li>
53 * <li>{@link SerializerConstantPool#putClass(Class)}</li>
54 * </ul>
55 */
56 public PostOrderSerializer(SerializerConstantPool cp) {
57 this.cp = cp;
58 }
59
60 /**
61 * Serializes the node AST and returns the serialized data as byte array.
62 *
63 * @param node the root node that represents the Truffle AST that should be serialized.
64 * @return a trimmed byte array that contains the serialized data.
65 *
66 * @throws UnsupportedConstantPoolTypeException thrown if a type is encountered that is not
67 * supported by the constant pool implementation.
68 */
69 public byte[] serialize(Node node) throws UnsupportedConstantPoolTypeException {
70 VariableLengthIntBuffer buffer = new VariableLengthIntBuffer(ByteBuffer.allocate(512));
71 serialize(buffer, node);
72 return buffer.getBytes();
73 }
74
75 private void serialize(VariableLengthIntBuffer buffer, Node node) throws UnsupportedConstantPoolTypeException {
76 if (node == null) {
77 buffer.put(VariableLengthIntBuffer.NULL);
78 return;
79 }
80 Class<? extends Node> nodeClass = node.getClass();
81
82 NodeField[] nodeFields = NodeClass.get(nodeClass).getFields();
83 serializeChildFields(buffer, node, nodeFields);
84 serializeChildrenFields(buffer, node, nodeFields);
85 buffer.put(cp.putClass(node.getClass()));
86 serializeDataFields(buffer, node, nodeFields);
87 }
88
89 private void serializeDataFields(VariableLengthIntBuffer buffer, Node node, NodeField[] nodeFields) throws UnsupportedConstantPoolTypeException {
90 for (int i = 0; i < nodeFields.length; i++) {
91 NodeField field = nodeFields[i];
92 if (field.getKind() == NodeFieldKind.DATA) {
93 Class<?> fieldClass = field.getType();
94 long offset = field.getOffset();
95 int cpi;
96
97 if (field.getType().isAssignableFrom(SourceSection.class)) {
98 continue;
99 }
100
101 if (fieldClass == int.class) {
102 cpi = cp.putInt(unsafe.getInt(node, offset));
103 } else if (fieldClass == long.class) {
104 cpi = cp.putLong(unsafe.getLong(node, offset));
105 } else if (fieldClass == float.class) {
106 cpi = cp.putFloat(unsafe.getFloat(node, offset));
107 } else if (fieldClass == double.class) {
108 cpi = cp.putDouble(unsafe.getDouble(node, offset));
109 } else if (fieldClass == byte.class) {
110 cpi = cp.putInt(unsafe.getByte(node, offset));
111 } else if (fieldClass == short.class) {
112 cpi = cp.putInt(unsafe.getShort(node, offset));
113 } else if (fieldClass == char.class) {
114 cpi = cp.putInt(unsafe.getChar(node, offset));
115 } else if (fieldClass == boolean.class) {
116 cpi = cp.putInt(unsafe.getBoolean(node, offset) ? 1 : 0);
117 } else {
118 Object value = unsafe.getObject(node, offset);
119 if (value == null) {
120 cpi = VariableLengthIntBuffer.NULL;
121 } else if (fieldClass == Integer.class) {
122 cpi = cp.putInt((Integer) value);
123 } else if (fieldClass == Long.class) {
124 cpi = cp.putLong((Long) value);
125 } else if (fieldClass == Float.class) {
126 cpi = cp.putFloat((Float) value);
127 } else if (fieldClass == Double.class) {
128 cpi = cp.putDouble((Double) value);
129 } else if (fieldClass == Byte.class) {
130 cpi = cp.putInt((Byte) value);
131 } else if (fieldClass == Short.class) {
132 cpi = cp.putInt((Short) value);
133 } else if (fieldClass == Character.class) {
134 cpi = cp.putInt((Character) value);
135 } else if (fieldClass == Boolean.class) {
136 cpi = cp.putInt((Boolean) value ? 1 : 0);
137 } else {
138 cpi = cp.putObject(fieldClass, value);
139 }
140 }
141
142 buffer.put(cpi);
143 }
144 }
145 }
146
147 private void serializeChildrenFields(VariableLengthIntBuffer buffer, Node nodeInstance, NodeField[] nodeFields) throws UnsupportedConstantPoolTypeException {
148 for (int i = 0; i < nodeFields.length; i++) {
149 NodeField field = nodeFields[i];
150 if (field.getKind() == NodeFieldKind.CHILDREN) {
151 Object childArrayObject = unsafe.getObject(nodeInstance, field.getOffset());
152 if (childArrayObject != null && !(childArrayObject instanceof Node[])) {
153 throw new AssertionError("Node children must be instanceof Node[]");
154 }
155
156 buffer.put(cp.putClass(field.getType()));
157
158 Node[] childArray = (Node[]) childArrayObject;
159 if (childArray == null) {
160 buffer.put(VariableLengthIntBuffer.NULL);
161 } else {
162 buffer.put(cp.putInt(childArray.length));
163
164 for (int j = 0; j < childArray.length; j++) {
165 serialize(buffer, childArray[j]);
166 }
167 }
168 }
169 }
170 }
171
172 private void serializeChildFields(VariableLengthIntBuffer buffer, Node nodeInstance, NodeField[] nodeFields) throws UnsupportedConstantPoolTypeException {
173 for (int i = 0; i < nodeFields.length; i++) {
174 NodeField field = nodeFields[i];
175 if (field.getKind() == NodeFieldKind.CHILD) {
176 Object childObject = unsafe.getObject(nodeInstance, field.getOffset());
177 if (childObject != null && !(childObject instanceof Node)) {
178 throw new AssertionError("Node children must be instanceof Node");
179 }
180 serialize(buffer, (Node) childObject);
181 }
182 }
183 }
184
185 private static Unsafe loadUnsafe() {
186 try {
187 return Unsafe.getUnsafe();
188 } catch (SecurityException e) {
189 }
190 try {
191 Field theUnsafeInstance = Unsafe.class.getDeclaredField("theUnsafe");
192 theUnsafeInstance.setAccessible(true);
193 return (Unsafe) theUnsafeInstance.get(Unsafe.class);
194 } catch (Exception e) {
195 throw new RuntimeException("exception while trying to get Unsafe.theUnsafe via reflection:", e);
196 }
197 }
198
199 }