Mercurial > hg > graal-compiler
view graal/com.oracle.graal.printer/src/com/oracle/graal/printer/BinaryGraphPrinter.java @ 16479:558cf39c646b
moved toJavaName(JavaType type) from MetaUtil to be a default method in JavaType
author | Doug Simon <doug.simon@oracle.com> |
---|---|
date | Thu, 10 Jul 2014 22:33:53 +0200 |
parents | f9f7bd1a6b2c |
children | c62c1e0060cc |
line wrap: on
line source
/* * Copyright (c) 2011, 2014, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License version 2 only, as * published by the Free Software Foundation. * * This code is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License * version 2 for more details (a copy is included in the LICENSE file that * accompanied this code). * * You should have received a copy of the GNU General Public License version * 2 along with this work; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have any * questions. */ package com.oracle.graal.printer; import static com.oracle.graal.compiler.common.GraalOptions.*; import java.io.*; import java.nio.*; import java.nio.channels.*; import java.util.*; import java.util.Map.Entry; import java.util.function.*; import com.oracle.graal.api.meta.*; import com.oracle.graal.compiler.common.cfg.*; import com.oracle.graal.graph.*; import com.oracle.graal.graph.NodeClass.Position; import com.oracle.graal.nodes.*; import com.oracle.graal.nodes.cfg.*; import com.oracle.graal.phases.graph.*; import com.oracle.graal.phases.schedule.*; public class BinaryGraphPrinter implements GraphPrinter { private static final int CONSTANT_POOL_MAX_SIZE = 8000; private static final int BEGIN_GROUP = 0x00; private static final int BEGIN_GRAPH = 0x01; private static final int CLOSE_GROUP = 0x02; private static final int POOL_NEW = 0x00; private static final int POOL_STRING = 0x01; private static final int POOL_ENUM = 0x02; private static final int POOL_CLASS = 0x03; private static final int POOL_METHOD = 0x04; private static final int POOL_NULL = 0x05; private static final int POOL_NODE_CLASS = 0x06; private static final int POOL_FIELD = 0x07; private static final int POOL_SIGNATURE = 0x08; private static final int PROPERTY_POOL = 0x00; private static final int PROPERTY_INT = 0x01; private static final int PROPERTY_LONG = 0x02; private static final int PROPERTY_DOUBLE = 0x03; private static final int PROPERTY_FLOAT = 0x04; private static final int PROPERTY_TRUE = 0x05; private static final int PROPERTY_FALSE = 0x06; private static final int PROPERTY_ARRAY = 0x07; private static final int PROPERTY_SUBGRAPH = 0x08; private static final int KLASS = 0x00; private static final int ENUM_KLASS = 0x01; private static final class ConstantPool extends LinkedHashMap<Object, Character> { private final LinkedList<Character> availableIds; private char nextId; private static final long serialVersionUID = -2676889957907285681L; public ConstantPool() { super(50, 0.65f); availableIds = new LinkedList<>(); } @Override protected boolean removeEldestEntry(java.util.Map.Entry<Object, Character> eldest) { if (size() > CONSTANT_POOL_MAX_SIZE) { availableIds.addFirst(eldest.getValue()); return true; } return false; } private Character nextAvailableId() { if (!availableIds.isEmpty()) { return availableIds.removeFirst(); } return nextId++; } public char add(Object obj) { Character id = nextAvailableId(); put(obj, id); return id; } } private final ConstantPool constantPool; private final ByteBuffer buffer; private final WritableByteChannel channel; public BinaryGraphPrinter(WritableByteChannel channel) { constantPool = new ConstantPool(); buffer = ByteBuffer.allocateDirect(256 * 1024); this.channel = channel; } public void print(Graph graph, String title, SchedulePhase predefinedSchedule) throws IOException { writeByte(BEGIN_GRAPH); writePoolObject(title); writeGraph(graph, predefinedSchedule); flush(); } private void writeGraph(Graph graph) throws IOException { writeGraph(graph, null); } private void writeGraph(Graph graph, SchedulePhase predefinedSchedule) throws IOException { SchedulePhase schedule = predefinedSchedule; if (schedule == null) { try { schedule = new SchedulePhase(); schedule.apply((StructuredGraph) graph); } catch (Throwable t) { } } ControlFlowGraph cfg = schedule == null ? null : schedule.getCFG(); BlockMap<List<ScheduledNode>> blockToNodes = schedule == null ? null : schedule.getBlockToNodesMap(); List<Block> blocks = cfg == null ? null : cfg.getBlocks(); writeNodes(graph); writeBlocks(blocks, blockToNodes); } private void flush() throws IOException { buffer.flip(); channel.write(buffer); buffer.compact(); } private void ensureAvailable(int i) throws IOException { assert buffer.capacity() >= i : "Can not make " + i + " bytes available, buffer is too small"; while (buffer.remaining() < i) { flush(); } } private void writeByte(int b) throws IOException { ensureAvailable(1); buffer.put((byte) b); } private void writeInt(int b) throws IOException { ensureAvailable(4); buffer.putInt(b); } private void writeLong(long b) throws IOException { ensureAvailable(8); buffer.putLong(b); } private void writeDouble(double b) throws IOException { ensureAvailable(8); buffer.putDouble(b); } private void writeFloat(float b) throws IOException { ensureAvailable(4); buffer.putFloat(b); } private void writeShort(char b) throws IOException { ensureAvailable(2); buffer.putChar(b); } private void writeString(String str) throws IOException { writeInt(str.length()); int sizeInBytes = str.length() * 2; ensureAvailable(sizeInBytes); buffer.asCharBuffer().put(str); buffer.position(buffer.position() + sizeInBytes); } private void writeBytes(byte[] b) throws IOException { if (b == null) { writeInt(-1); } else { writeInt(b.length); ensureAvailable(b.length); buffer.put(b); } } private void writeInts(int[] b) throws IOException { if (b == null) { writeInt(-1); } else { writeInt(b.length); int sizeInBytes = b.length * 4; ensureAvailable(sizeInBytes); buffer.asIntBuffer().put(b); buffer.position(buffer.position() + sizeInBytes); } } private void writeDoubles(double[] b) throws IOException { if (b == null) { writeInt(-1); } else { writeInt(b.length); int sizeInBytes = b.length * 8; ensureAvailable(sizeInBytes); buffer.asDoubleBuffer().put(b); buffer.position(buffer.position() + sizeInBytes); } } private void writePoolObject(Object object) throws IOException { if (object == null) { writeByte(POOL_NULL); return; } Character id = constantPool.get(object); if (id == null) { addPoolEntry(object); } else { if (object instanceof Enum<?>) { writeByte(POOL_ENUM); } else if (object instanceof Class<?> || object instanceof JavaType) { writeByte(POOL_CLASS); } else if (object instanceof NodeClass) { writeByte(POOL_NODE_CLASS); } else if (object instanceof ResolvedJavaMethod) { writeByte(POOL_METHOD); } else if (object instanceof ResolvedJavaField) { writeByte(POOL_FIELD); } else if (object instanceof Signature) { writeByte(POOL_SIGNATURE); } else { writeByte(POOL_STRING); } writeShort(id.charValue()); } } private static String getClassName(Class<?> klass) { if (!klass.isArray()) { return klass.getName(); } return getClassName(klass.getComponentType()) + "[]"; } private void addPoolEntry(Object object) throws IOException { char index = constantPool.add(object); writeByte(POOL_NEW); writeShort(index); if (object instanceof Class<?>) { Class<?> klass = (Class<?>) object; writeByte(POOL_CLASS); writeString(getClassName(klass)); if (klass.isEnum()) { writeByte(ENUM_KLASS); Object[] enumConstants = klass.getEnumConstants(); writeInt(enumConstants.length); for (Object o : enumConstants) { writePoolObject(((Enum<?>) o).name()); } } else { writeByte(KLASS); } } else if (object instanceof Enum<?>) { writeByte(POOL_ENUM); writePoolObject(object.getClass()); writeInt(((Enum<?>) object).ordinal()); } else if (object instanceof JavaType) { JavaType type = (JavaType) object; writeByte(POOL_CLASS); writeString(type.toJavaName()); writeByte(KLASS); } else if (object instanceof NodeClass) { NodeClass nodeClass = (NodeClass) object; writeByte(POOL_NODE_CLASS); writeString(nodeClass.getJavaClass().getSimpleName()); writeString(nodeClass.getNameTemplate()); Collection<Position> directInputPositions = nodeClass.getFirstLevelInputPositions(); writeShort((char) directInputPositions.size()); for (Position pos : directInputPositions) { writeByte(pos.getSubIndex() == NodeClass.NOT_ITERABLE ? 0 : 1); writePoolObject(nodeClass.getName(pos)); writePoolObject(nodeClass.getInputType(pos)); } Collection<Position> directSuccessorPositions = nodeClass.getFirstLevelSuccessorPositions(); writeShort((char) directSuccessorPositions.size()); for (Position pos : directSuccessorPositions) { writeByte(pos.getSubIndex() == NodeClass.NOT_ITERABLE ? 0 : 1); writePoolObject(nodeClass.getName(pos)); } } else if (object instanceof ResolvedJavaMethod) { writeByte(POOL_METHOD); ResolvedJavaMethod method = ((ResolvedJavaMethod) object); writePoolObject(method.getDeclaringClass()); writePoolObject(method.getName()); writePoolObject(method.getSignature()); writeInt(method.getModifiers()); writeBytes(method.getCode()); } else if (object instanceof ResolvedJavaField) { writeByte(POOL_FIELD); ResolvedJavaField field = ((ResolvedJavaField) object); writePoolObject(field.getDeclaringClass()); writePoolObject(field.getName()); writePoolObject(field.getType().getName()); writeInt(field.getModifiers()); } else if (object instanceof Signature) { writeByte(POOL_SIGNATURE); Signature signature = ((Signature) object); int args = signature.getParameterCount(false); writeShort((char) args); for (int i = 0; i < args; i++) { writePoolObject(signature.getParameterType(i, null).getName()); } writePoolObject(signature.getReturnType(null).getName()); } else { writeByte(POOL_STRING); writeString(object.toString()); } } private void writePropertyObject(Object obj) throws IOException { if (obj instanceof Integer) { writeByte(PROPERTY_INT); writeInt(((Integer) obj).intValue()); } else if (obj instanceof Long) { writeByte(PROPERTY_LONG); writeLong(((Long) obj).longValue()); } else if (obj instanceof Double) { writeByte(PROPERTY_DOUBLE); writeDouble(((Double) obj).doubleValue()); } else if (obj instanceof Float) { writeByte(PROPERTY_FLOAT); writeFloat(((Float) obj).floatValue()); } else if (obj instanceof Boolean) { if (((Boolean) obj).booleanValue()) { writeByte(PROPERTY_TRUE); } else { writeByte(PROPERTY_FALSE); } } else if (obj instanceof Graph) { writeByte(PROPERTY_SUBGRAPH); writeGraph((Graph) obj); } else if (obj != null && obj.getClass().isArray()) { Class<?> componentType = obj.getClass().getComponentType(); if (componentType.isPrimitive()) { if (componentType == Double.TYPE) { writeByte(PROPERTY_ARRAY); writeByte(PROPERTY_DOUBLE); writeDoubles((double[]) obj); } else if (componentType == Integer.TYPE) { writeByte(PROPERTY_ARRAY); writeByte(PROPERTY_INT); writeInts((int[]) obj); } else { writeByte(PROPERTY_POOL); writePoolObject(obj); } } else { writeByte(PROPERTY_ARRAY); writeByte(PROPERTY_POOL); Object[] array = (Object[]) obj; writeInt(array.length); for (Object o : array) { writePoolObject(o); } } } else { writeByte(PROPERTY_POOL); writePoolObject(obj); } } @SuppressWarnings("deprecation") private static int getNodeId(Node node) { return node.getId(); } private void writeNodes(Graph graph) throws IOException { ToDoubleFunction<FixedNode> probabilities = null; if (PrintGraphProbabilities.getValue()) { try { probabilities = new FixedNodeProbabilityCache(); } catch (Throwable t) { } } Map<Object, Object> props = new HashMap<>(); writeInt(graph.getNodeCount()); for (Node node : graph.getNodes()) { NodeClass nodeClass = node.getNodeClass(); node.getDebugProperties(props); if (probabilities != null && node instanceof FixedNode) { props.put("probability", probabilities.applyAsDouble((FixedNode) node)); } writeInt(getNodeId(node)); writePoolObject(nodeClass); writeByte(node.predecessor() == null ? 0 : 1); // properties writeShort((char) props.size()); for (Entry<Object, Object> entry : props.entrySet()) { String key = entry.getKey().toString(); writePoolObject(key); writePropertyObject(entry.getValue()); } // inputs writeEdges(node, nodeClass.getFirstLevelInputPositions()); // successors writeEdges(node, nodeClass.getFirstLevelSuccessorPositions()); props.clear(); } } private void writeEdges(Node node, Collection<Position> positions) throws IOException { NodeClass nodeClass = node.getNodeClass(); for (Position pos : positions) { if (pos.getSubIndex() == NodeClass.NOT_ITERABLE) { Node edge = nodeClass.get(node, pos); writeNodeRef(edge); } else { NodeList<?> list = nodeClass.getNodeList(node, pos); if (list == null) { writeShort((char) 0); } else { int listSize = list.count(); assert listSize == ((char) listSize); writeShort((char) listSize); for (Node edge : list) { writeNodeRef(edge); } } } } } private void writeNodeRef(Node edge) throws IOException { if (edge != null) { writeInt(getNodeId(edge)); } else { writeInt(-1); } } private void writeBlocks(List<Block> blocks, BlockMap<List<ScheduledNode>> blockToNodes) throws IOException { if (blocks != null) { writeInt(blocks.size()); for (Block block : blocks) { List<ScheduledNode> nodes = blockToNodes.get(block); writeInt(block.getId()); writeInt(nodes.size()); for (Node node : nodes) { writeInt(getNodeId(node)); } writeInt(block.getSuccessors().size()); for (Block sux : block.getSuccessors()) { writeInt(sux.getId()); } } } else { writeInt(0); } } public void beginGroup(String name, String shortName, ResolvedJavaMethod method, int bci) throws IOException { writeByte(BEGIN_GROUP); writePoolObject(name); writePoolObject(shortName); writePoolObject(method); writeInt(bci); } public void endGroup() throws IOException { writeByte(CLOSE_GROUP); } @Override public void close() { try { flush(); channel.close(); } catch (IOException ex) { throw new Error(ex); } } }