org.graphstream.stream.file.FileSinkGraphML Maven / Gradle / Ivy

/*
 * This file is part of GraphStream .
 * 
 * GraphStream is a library whose purpose is to handle static or dynamic
 * graph, create them from scratch, file or any source and display them.
 * 
 * This program is free software distributed under the terms of two licenses, the
 * CeCILL-C license that fits European law, and the GNU Lesser General Public
 * License. You can  use, modify and/ or redistribute the software under the terms
 * of the CeCILL-C license as circulated by CEA, CNRS and INRIA at the following
 * URL  or under the terms of the GNU LGPL as published by
 * the Free Software Foundation, either version 3 of the License, or (at your
 * option) any later version.
 * 
 * This program 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 program.  If not, see .
 * 
 * The fact that you are presently reading this means that you have had
 * knowledge of the CeCILL-C and LGPL licenses and that you accept their terms.
 */

/**
 * @since 2011-12-06
 * 
 * @author Guilhelm Savin 
 * @author Hans Schulz 
 * @author Hicham Brahimi 
 */
package org.graphstream.stream.file;

import java.io.EOFException;
import java.io.IOException;
import java.util.HashMap;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Consumer;

import org.graphstream.graph.Edge;
import org.graphstream.graph.Graph;
import org.graphstream.graph.Node;

public class FileSinkGraphML extends FileSinkBase {

	protected void outputEndOfFile() throws IOException {
		print("\n");
	}

	protected void outputHeader() throws IOException {
		print("\n");
		print("\n");
	}

	private void print(String format, Object... args) throws IOException {
		output.write(String.format(format, args));
	}

	@Override
	protected void exportGraph(Graph g) {
		final Consumer onException = Exception::printStackTrace;

		AtomicInteger attribute = new AtomicInteger(0);
		HashMap nodeAttributes = new HashMap<>();
		HashMap edgeAttributes = new HashMap<>();

		g.nodes().forEach(n -> {
			n.attributeKeys().forEach(k -> {
				if (!nodeAttributes.containsKey(k)) {
					Object value = n.getAttribute(k);
					String type;

					if (value == null)
						return;

					String id = String.format("attr%04X", attribute.getAndIncrement());

					if (value instanceof Boolean)
						type = "boolean";
					else if (value instanceof Long)
						type = "long";
					else if (value instanceof Integer)
						type = "int";
					else if (value instanceof Double)
						type = "double";
					else if (value instanceof Float)
						type = "float";
					else
						type = "string";

					nodeAttributes.put(k, id);

					try {
						print("\t\n", id,
								escapeXmlString(k), type);
					} catch (Exception ex) {
						onException.accept(ex);
					}
				}
			});
		});

		g.edges().forEach(n -> {
			n.attributeKeys().forEach(k -> {
				if (!edgeAttributes.containsKey(k)) {
					Object value = n.getAttribute(k);
					String type;

					if (value == null)
						return;

					String id = String.format("attr%04X", attribute.getAndIncrement());

					if (value instanceof Boolean)
						type = "boolean";
					else if (value instanceof Long)
						type = "long";
					else if (value instanceof Integer)
						type = "int";
					else if (value instanceof Double)
						type = "double";
					else if (value instanceof Float)
						type = "float";
					else
						type = "string";

					edgeAttributes.put(k, id);

					try {
						print("\t\n", id,
								escapeXmlString(k), type);
					} catch (Exception ex) {
						onException.accept(ex);
					}
				}
			});
		});

		try {
			print("\t\n", escapeXmlString(g.getId()));
		} catch (Exception e) {
			onException.accept(e);
		}

		g.nodes().forEach(n -> {
			try {
				print("\t\t\n", n.getId());

				n.attributeKeys().forEach(k -> {
					try {
						print("\t\t\t%s\n", nodeAttributes.get(k),
								escapeXmlString(n.getAttribute(k).toString()));
					} catch (IOException e) {
						onException.accept(e);
					}
				});

				print("\t\t\n");
			} catch (Exception ex) {
				onException.accept(ex);
			}
		});

		g.edges().forEach(e -> {
			try {
				print("\t\t\n", e.getId(),
						e.getSourceNode().getId(), e.getTargetNode().getId(), e.isDirected());

				e.attributeKeys().forEach(k -> {
					try {
						print("\t\t\t%s\n", edgeAttributes.get(k),
								escapeXmlString(e.getAttribute(k).toString()));
					} catch (IOException e1) {
						onException.accept(e1);
					}
				});

				print("\t\t\n");
			} catch (Exception ex) {
				onException.accept(ex);
			}
		});

		try {
			print("\t\n");
		} catch (Exception e) {
			onException.accept(e);
		}
	}

	public void edgeAttributeAdded(String sourceId, long timeId, String edgeId, String attribute, Object value) {
		throw new UnsupportedOperationException();
	}

	public void edgeAttributeChanged(String sourceId, long timeId, String edgeId, String attribute, Object oldValue,
			Object newValue) {
		throw new UnsupportedOperationException();
	}

	public void edgeAttributeRemoved(String sourceId, long timeId, String edgeId, String attribute) {
		throw new UnsupportedOperationException();
	}

	public void graphAttributeAdded(String sourceId, long timeId, String attribute, Object value) {
		throw new UnsupportedOperationException();
	}

	public void graphAttributeChanged(String sourceId, long timeId, String attribute, Object oldValue,
			Object newValue) {
		throw new UnsupportedOperationException();
	}

	public void graphAttributeRemoved(String sourceId, long timeId, String attribute) {
		throw new UnsupportedOperationException();
	}

	public void nodeAttributeAdded(String sourceId, long timeId, String nodeId, String attribute, Object value) {
		throw new UnsupportedOperationException();
	}

	public void nodeAttributeChanged(String sourceId, long timeId, String nodeId, String attribute, Object oldValue,
			Object newValue) {
		throw new UnsupportedOperationException();
	}

	public void nodeAttributeRemoved(String sourceId, long timeId, String nodeId, String attribute) {
		throw new UnsupportedOperationException();
	}

	public void edgeAdded(String sourceId, long timeId, String edgeId, String fromNodeId, String toNodeId,
			boolean directed) {
		throw new UnsupportedOperationException();
	}

	public void edgeRemoved(String sourceId, long timeId, String edgeId) {
		throw new UnsupportedOperationException();
	}

	public void graphCleared(String sourceId, long timeId) {
		throw new UnsupportedOperationException();
	}

	public void nodeAdded(String sourceId, long timeId, String nodeId) {
		throw new UnsupportedOperationException();
	}

	public void nodeRemoved(String sourceId, long timeId, String nodeId) {
		throw new UnsupportedOperationException();
	}

	public void stepBegins(String sourceId, long timeId, double step) {
		throw new UnsupportedOperationException();
	}

	private static String escapeXmlString(String string) {
		/*
		 * Thankfully, the unescaping part is done by the xml parser used in
		 * FileSourceGraphML
		 */
		return string.replace("&", "&").replace("<", "<").replace(">", ">").replace("\"", """)
				.replace("'", "'");
	}
}