org.jgraph.pad.GPConverter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of freak-core Show documentation
Core library of the Free Evolutionary Algorithm Toolkit
There is a newer version: 0.4.7
/*
 * @(#)GPConverter.java	1.2 11/11/02
 *
 * Copyright (C) 2001 Gaudenz Alder
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 *
 */

package org.jgraph.pad;

import org.jgraph.graph.GraphModel;

import java.awt.*;
import java.awt.image.BufferedImage;
import java.text.DateFormat;
import java.util.Date;
import java.util.Hashtable;
import java.util.Iterator;

public class GPConverter {

	//
	// Image Converter
	//

	// Create a buffered image of the specified graph.
	public static BufferedImage toImage(GPGraph graph) {
		Object[] cells = graph.getRoots();

		if (cells.length > 0) {
			Rectangle bounds = graph.getCellBounds(cells);
			graph.toScreen(bounds);

			// Create a Buffered Image
			Dimension d = bounds.getSize();
			BufferedImage img =
				new BufferedImage(
					d.width + 10,
					d.height + 10,
					BufferedImage.TYPE_INT_RGB);
			Graphics2D graphics = img.createGraphics();
			graphics.setColor(graph.getBackground());
			graphics.fillRect(0, 0, img.getWidth(), img.getHeight());
			graphics.translate(-bounds.x + 5, -bounds.y + 5);

			Object[] selection = graph.getSelectionCells();
			boolean gridVisible = graph.isGridVisible();
			graph.setGridVisible(false);
			graph.clearSelection();

			graph.paint(graphics);

			graph.setSelectionCells(selection);
			graph.setGridVisible(gridVisible);

			return img;
		}
		return null;
	}

	//
	// GXL Converter
	//

	static transient Hashtable hash;

	// Create a GXL-representation for the specified cells.
	public static String toGXL(GPGraph graph, Object[] cells) {
		hash = new Hashtable();
		String gxl = "";

		// Create external keys for nodes
		for (int i = 0; i < cells.length; i++)
			if (graph.isVertex(cells[i]))
				hash.put(cells[i], new Integer(hash.size()));

		// Convert Nodes
		Iterator it = hash.keySet().iterator();
		while (it.hasNext()) {
			Object node = it.next();
			gxl += vertexGXL(graph, hash.get(node), node);
		}

		// Convert Edges
		int edges = 0;
		for (int i = 0; i < cells.length; i++)
			if (graph.isEdge(cells[i]))
				gxl += edgeGXL(graph, new Integer(edges++), cells[i]);

		// Close main tags
		gxl += "\n";
		return gxl;
	}

	public static String vertexGXL(GPGraph graph, Object id, Object vertex) {
		String label = graph.convertValueToString(vertex);
		return "\n\t"
			+ "\n\t\t"
			+ "\n\t\t\t"
			+ label
			+ ""
			+ "\n\t\t"
			+ "\n\t";
	}

	public static String edgeGXL(GPGraph graph, Object id, Object edge) {
		GraphModel model = graph.getModel();
		String from = "";
		if (model.getSource(edge) != null) {
			Object source = hash.get(model.getParent(model.getSource(edge)));
			if (source != null)
				from = "node" + source.toString();
		}
		String to = "";
		if (model.getTarget(edge) != null) {
			Object target = hash.get(model.getParent(model.getTarget(edge)));
			if (target != null)
				to = "node" + target.toString();
		}
		if (from != null && to != null) {
			String label = graph.convertValueToString(edge);
			return "\n\t"
				+ "\n\t\t"
				+ "\n\t\t\t"
				+ label
				+ ""
				+ "\n\t\t"
				+ "\n\t";
		} else
			return "";
	}



	// Graphviz converter ===============================================
	/* sample syntax:
			digraph G {
				subgraph cluster0 {
					node [style=filled,color=white];
					style=filled;
					color=lightgrey;
					a0 -> a1 -> a2 -> a3;
					label = "process #1";
				}
				subgraph cluster1 {
					node [style=filled];
					b0 -> b1 -> b2 -> b3;
					label = "process #2";
					color=blue
				}
				start -> a0;
				start -> b0;
				a1 -> b3;
				b2 -> a3;
				a3 -> a0;
				a3 -> end;
				b3 -> end;
				start [shape=Mdiamond];
				end [shape=Msquare];
			}
			*/
	public static String toGraphviz(GPGraph graph, Object[] cells) {
		hash = new Hashtable();
		DateFormat dateformat = DateFormat.getDateTimeInstance(DateFormat.LONG, DateFormat.LONG);
		String date = dateformat.format(new Date());
		StringBuffer gv = new StringBuffer("/* Graphviz file generated by " +
			"JGraph - " + date + " */" +
			"\n\ndigraph G {");

		// Create external keys for nodes
		for (int i = 0; i < cells.length; i++)
			if (graph.isVertex(cells[i]))
				hash.put(cells[i], new Integer(hash.size()));

		// Process Nodes
		Iterator it = hash.keySet().iterator();
		while (it.hasNext()) {
			Object node = it.next();
			gv.append(vertexGraphviz(graph, hash.get(node), node));
		}

		// Process Edges
		int edges = 0;
		for (int i = 0; i < cells.length; i++)
			if (graph.isEdge(cells[i]))
				gv.append(edgeGraphviz(graph, new Integer(edges++), cells[i]));

		// Close main tags
		gv.append("\n}");
		return gv.toString();
	}


	private static String vertexGraphviz(GPGraph graph, Object id, Object vertex) {
		if (id==null) return "";
		String label = graph.convertValueToString(vertex);
		if (label==null) label = "";
		return "\n\t" +
		       id.toString() +
			   " [label=\"" + label + "\", " +
			   "shape=\"box\"];";
	}

	private static String edgeGraphviz(GPGraph graph, Object id, Object edge) {
		GraphModel model = graph.getModel();
		String from = null;

		Object es = model.getSource(edge);
		if (es != null) {
			Object ps = hash.get(model.getParent(es));
			if (ps != null)
				from = ps.toString();
//			if (es != null)
// debug		from = graph.convertValueToString(model.getParent(es));

			String to = null;
			Object et = model.getTarget(edge);
			if (et != null) {
				Object tp = hash.get(model.getParent(et));
				if (tp != null)
					to = tp.toString();
// debug		if (et != null)
//					to = graph.convertValueToString(model.getParent(et));
			}

			if (from != null && to != null)
				return "\n\t" + from + " -> " + to + ";";
		}
		return "";
	}
}