/* Copyright 2008-2010 Gephi Authors : Mathieu Bastian <mathieu.bastian@gephi.org> Website : http://www.gephi.org This file is part of Gephi. DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. Copyright 2011 Gephi Consortium. All rights reserved. The contents of this file are subject to the terms of either the GNU General Public License Version 3 only ("GPL") or the Common Development and Distribution License("CDDL") (collectively, the "License"). You may not use this file except in compliance with the License. You can obtain a copy of the License at http://gephi.org/about/legal/license-notice/ or /cddl-1.0.txt and /gpl-3.0.txt. See the License for the specific language governing permissions and limitations under the License. When distributing the software, include this License Header Notice in each file and include the License files at /cddl-1.0.txt and /gpl-3.0.txt. If applicable, add the following below the License Header, with the fields enclosed by brackets [] replaced by your own identifying information: "Portions Copyrighted [year] [name of copyright owner]" If you wish your version of this file to be governed by only the CDDL or only the GPL Version 3, indicate your decision by adding "[Contributor] elects to include this software in this distribution under the [CDDL or GPL Version 3] license." If you do not indicate a single choice of license, a recipient has the option to distribute your version of this file under either the CDDL, the GPL Version 3 or to extend the choice of license to its licensees as provided above. However, if you add GPL Version 3 code and therefore, elected the GPL Version 3 license, then the option applies only if the new code is made subject to such option by the copyright holder. Contributor(s): Portions Copyrighted 2011 Gephi Consortium. */ package org.gephi.io.exporter.plugin; import java.io.Writer; import java.util.ArrayList; import java.util.List; import java.util.logging.Level; import java.util.logging.Logger; import org.gephi.graph.api.AttributeUtils; import org.gephi.graph.api.Column; import org.gephi.graph.api.Edge; import org.gephi.graph.api.EdgeIterable; import org.gephi.graph.api.Graph; import org.gephi.graph.api.GraphController; import org.gephi.graph.api.GraphModel; import org.gephi.graph.api.Node; import org.gephi.graph.api.NodeIterable; import org.gephi.graph.api.types.TimestampMap; import org.gephi.io.exporter.api.FileType; import org.gephi.io.exporter.spi.CharacterExporter; import org.gephi.io.exporter.spi.GraphExporter; import org.gephi.project.api.Workspace; import org.gephi.utils.longtask.spi.LongTask; import org.gephi.utils.progress.Progress; import org.gephi.utils.progress.ProgressTicket; import org.openide.util.Lookup; import org.openide.util.NbBundle; /** * * @author Mathieu Bastian */ public class ExporterGDF implements GraphExporter, CharacterExporter, LongTask { private Workspace workspace; private boolean exportVisible; private boolean cancel = false; private ProgressTicket progressTicket; //Settings private boolean normalize = false; private boolean simpleQuotes = false; private boolean useQuotes = true; private boolean exportColors = true; private boolean exportPosition = true; private boolean exportAttributes = true; private boolean exportDynamicWeight = true; private boolean exportVisibility = false; //Settings Helper private float minSize; private float maxSize; private float minX; private float maxX; private float minY; private float maxY; private boolean edgeLabels; private boolean edgeColors; //Columns private NodeColumnsGDF[] defaultNodeColumnsGDFs; private EdgeColumnsGDF[] defaultEdgeColumnsGDFs; //Buffer private Writer writer; @Override public boolean execute() { GraphController graphController = Lookup.getDefault().lookup(GraphController.class); GraphModel graphModel = graphController.getGraphModel(workspace); Graph graph = exportVisible ? graphModel.getGraphVisible() : graphModel.getGraph(); graph.readLock(); try { exportData(graph, graphModel); } catch (Exception e) { Logger.getLogger(ExporterGDF.class.getName()).log(Level.SEVERE, null, e); } finally { graph.readUnlock(); Progress.finish(progressTicket); } return !cancel; } private void exportData(Graph graph, GraphModel graphModel) throws Exception { Progress.start(progressTicket); defaultNodeColumns(graph); defaultEdgeColumns(graph); Column[] nodeColumns = attributesNodeColumns(graphModel); Column[] edgeColumns = attributesEdgeColumns(graphModel); StringBuilder stringBuilder = new StringBuilder(); //Node intro stringBuilder.append("nodedef> name VARCHAR,"); //Default Node columns title for (NodeColumnsGDF c : defaultNodeColumnsGDFs) { if (c.isEnable()) { stringBuilder.append(c.getTitle()); stringBuilder.append(" "); stringBuilder.append(c.getType().toString().toUpperCase()); if (c.getDefaultValue() != null) { stringBuilder.append(" default "); stringBuilder.append(c.getDefaultValue().toString()); } stringBuilder.append(","); } } //Attributes Node columns for (Column c : nodeColumns) { if (!c.isProperty()) { stringBuilder.append(c.getTitle()); stringBuilder.append(" "); DataTypeGDF dataTypeGDF = getDataTypeGDF(c.getTypeClass()); stringBuilder.append(dataTypeGDF.toString().toUpperCase()); if (c.getDefaultValue() != null) { stringBuilder.append(" default "); stringBuilder.append(c.getDefaultValue().toString()); } stringBuilder.append(","); } } //Remove last coma stringBuilder.setLength(stringBuilder.length() - 1); stringBuilder.append("\n"); //Options if (normalize) { calculateMinMax(graph); } //Calculate progress units count int max = graph.getNodeCount() + graph.getEdgeCount(); Progress.switchToDeterminate(progressTicket, max); //Node lines NodeIterable itr = graph.getNodes(); for (Node node : itr) { if (cancel) { itr.doBreak(); return; } //Id stringBuilder.append(node.getId()); stringBuilder.append(","); //Default columns for (NodeColumnsGDF c : defaultNodeColumnsGDFs) { if (c.isEnable()) { c.writeData(stringBuilder, node); stringBuilder.append(","); } } //Attributes columns for (Column c : nodeColumns) { if (!c.isProperty()) { Object val = node.getAttribute(c, graph.getView()); if (val != null) { if (c.getTypeClass().equals(String.class) || c.getTypeClass().equals(String[].class)) { String quote = !useQuotes ? "" : simpleQuotes ? "'" : "\""; stringBuilder.append(quote); stringBuilder.append(val.toString()); stringBuilder.append(quote); } else { stringBuilder.append(val.toString()); } } stringBuilder.append(","); } } //Remove last coma stringBuilder.setLength(stringBuilder.length() - 1); stringBuilder.append("\n"); Progress.progress(progressTicket); } //Edge intro stringBuilder.append("edgedef> node1,node2,"); //Edge settings helper for (Edge e : graph.getEdges()) { edgeColors = edgeColors || e.alpha() != 0; edgeLabels = edgeLabels || (e.getLabel() != null && !e.getLabel().isEmpty()); } //Edge columns title for (EdgeColumnsGDF c : defaultEdgeColumnsGDFs) { if (c.isEnable()) { stringBuilder.append(c.getTitle()); stringBuilder.append(" "); stringBuilder.append(c.getType().toString().toUpperCase()); if (c.getDefaultValue() != null) { stringBuilder.append(" default "); stringBuilder.append(c.getDefaultValue().toString()); } stringBuilder.append(","); } } //Attributes Edge columns for (Column c : edgeColumns) { if (!c.isProperty()) { stringBuilder.append(c.getTitle()); stringBuilder.append(" "); DataTypeGDF dataTypeGDF = getDataTypeGDF(c.getTypeClass()); stringBuilder.append(dataTypeGDF.toString().toUpperCase()); if (c.getDefaultValue() != null) { stringBuilder.append(" default "); stringBuilder.append(c.getDefaultValue().toString()); } stringBuilder.append(","); } } //Remove last coma stringBuilder.setLength(stringBuilder.length() - 1); stringBuilder.append("\n"); //Edge lines EdgeIterable itrEdges = graph.getEdges(); for (Edge edge : itrEdges) { if (cancel) { itrEdges.doBreak(); return; } //Source & Target stringBuilder.append(edge.getSource().getId()); stringBuilder.append(","); stringBuilder.append(edge.getTarget().getId()); stringBuilder.append(","); //Default columns for (EdgeColumnsGDF c : defaultEdgeColumnsGDFs) { if (c.isEnable()) { c.writeData(stringBuilder, edge); stringBuilder.append(","); } } //Attributes columns for (Column c : edgeColumns) { if (!c.isProperty()) { Object val = edge.getAttribute(c, graph.getView()); if (val != null) { if (c.getTypeClass().equals(String.class) || c.getTypeClass().equals(String[].class)) { String quote = !useQuotes ? "" : simpleQuotes ? "'" : "\""; stringBuilder.append(quote); stringBuilder.append(val.toString()); stringBuilder.append(quote); } else { stringBuilder.append(val.toString()); } } stringBuilder.append(","); } } //Remove last coma stringBuilder.setLength(stringBuilder.length() - 1); stringBuilder.append("\n"); Progress.progress(progressTicket); } //Write StringBuilder if (!cancel) { writer.append(stringBuilder); } Progress.finish(progressTicket); } private Column[] attributesNodeColumns(GraphModel graphModel) { List<Column> cols = new ArrayList<>(); if (exportAttributes && graphModel != null) { for (Column column : graphModel.getNodeTable()) { if (!column.isProperty() && !isNodeDefaultColumn(column.getId())) { cols.add(column); } } } return cols.toArray(new Column[0]); } private Column[] attributesEdgeColumns(GraphModel graphModel) { List<Column> cols = new ArrayList<>(); if (exportAttributes && graphModel != null) { for (Column column : graphModel.getEdgeTable()) { if (!column.isProperty() && !isEdgeDefaultColumn(column.getId())) { cols.add(column); } } } return cols.toArray(new Column[0]); } private boolean isNodeDefaultColumn(String id) { for (NodeColumnsGDF c : defaultNodeColumnsGDFs) { if (c.title.equalsIgnoreCase(id)) { return true; } } return false; } private boolean isEdgeDefaultColumn(String id) { for (EdgeColumnsGDF c : defaultEdgeColumnsGDFs) { if (c.title.equalsIgnoreCase(id)) { return true; } } return false; } private void defaultNodeColumns(Graph graph) { NodeColumnsGDF labelColumn = new NodeColumnsGDF("label") { @Override public boolean isEnable() { return true; } @Override public void writeData(StringBuilder builder, Node node) { String label = node.getLabel(); if (label != null) { String quote = !useQuotes ? "" : simpleQuotes ? "'" : "\""; builder.append(quote); builder.append(label); builder.append(quote); } } }; NodeColumnsGDF visibleColumn = new NodeColumnsGDF("visible", DataTypeGDF.BOOLEAN) { @Override public boolean isEnable() { return exportVisibility; } @Override public void writeData(StringBuilder builder, Node node) { builder.append(true); } }; NodeColumnsGDF labelVisibleColumn = new NodeColumnsGDF("labelvisible", DataTypeGDF.BOOLEAN) { @Override public boolean isEnable() { return exportVisibility; } @Override public void writeData(StringBuilder builder, Node node) { builder.append(node.getTextProperties().isVisible()); } }; NodeColumnsGDF widthColumn = new NodeColumnsGDF("width", DataTypeGDF.DOUBLE) { @Override public boolean isEnable() { return exportPosition; } @Override public void writeData(StringBuilder builder, Node node) { float size = node.size(); if (normalize) { size = (size - minSize) / (maxSize - minSize); } builder.append(size); } }; NodeColumnsGDF heightColumn = new NodeColumnsGDF("height", DataTypeGDF.DOUBLE) { @Override public boolean isEnable() { return exportPosition; } @Override public void writeData(StringBuilder builder, Node node) { float size = node.size(); if (normalize) { size = (size - minSize) / (maxSize - minSize); } builder.append(size); } }; NodeColumnsGDF xColumn = new NodeColumnsGDF("x", DataTypeGDF.DOUBLE) { @Override public boolean isEnable() { return exportPosition; } @Override public void writeData(StringBuilder builder, Node node) { float x = node.x(); if (normalize && x != 0.0) { x = (x - minX) / (maxX - minX); } builder.append(x); } }; NodeColumnsGDF yColumn = new NodeColumnsGDF("y", DataTypeGDF.DOUBLE) { @Override public boolean isEnable() { return exportPosition; } @Override public void writeData(StringBuilder builder, Node node) { float y = node.y(); if (normalize && y != 0.0) { y = (y - minY) / (maxY - minY); } builder.append(y); } }; NodeColumnsGDF colorColumn = new NodeColumnsGDF("color") { @Override public boolean isEnable() { return exportColors; } @Override public void writeData(StringBuilder builder, Node node) { String quote = "'"; builder.append(quote); builder.append((int) (node.r() * 255f)); builder.append(","); builder.append((int) (node.g() * 255f)); builder.append(","); builder.append((int) (node.b() * 255f)); builder.append(quote); } }; NodeColumnsGDF fixedColumn = new NodeColumnsGDF("fixed", DataTypeGDF.BOOLEAN) { @Override public boolean isEnable() { return exportVisibility; } @Override public void writeData(StringBuilder builder, Node node) { builder.append(node.isFixed()); } }; NodeColumnsGDF styleColumn = new NodeColumnsGDF("style", DataTypeGDF.INT) { @Override public boolean isEnable() { return false; } @Override public void writeData(StringBuilder builder, Node node) { } }; defaultNodeColumnsGDFs = new NodeColumnsGDF[10]; defaultNodeColumnsGDFs[0] = labelColumn; defaultNodeColumnsGDFs[1] = visibleColumn; defaultNodeColumnsGDFs[2] = labelVisibleColumn; defaultNodeColumnsGDFs[3] = widthColumn; defaultNodeColumnsGDFs[4] = heightColumn; defaultNodeColumnsGDFs[5] = xColumn; defaultNodeColumnsGDFs[6] = yColumn; defaultNodeColumnsGDFs[7] = colorColumn; defaultNodeColumnsGDFs[8] = fixedColumn; defaultNodeColumnsGDFs[9] = styleColumn; } private void defaultEdgeColumns(final Graph graph) { EdgeColumnsGDF labelColumn = new EdgeColumnsGDF("label") { @Override public boolean isEnable() { return edgeLabels; } @Override public void writeData(StringBuilder builder, Edge edge) { String label = edge.getLabel(); if (label != null) { String quote = !useQuotes ? "" : simpleQuotes ? "'" : "\""; builder.append(quote); builder.append(label); builder.append(quote); } } }; EdgeColumnsGDF weightColumn = new EdgeColumnsGDF("weight", DataTypeGDF.DOUBLE) { @Override public boolean isEnable() { return true; } @Override public void writeData(StringBuilder builder, Edge edge) { double weight; if (exportDynamicWeight) { weight = edge.getWeight(graph.getView()); } else { weight = edge.getWeight(); } builder.append(weight); } }; EdgeColumnsGDF directedColumn = new EdgeColumnsGDF("directed", DataTypeGDF.BOOLEAN) { @Override public boolean isEnable() { return true; } @Override public void writeData(StringBuilder builder, Edge edge) { builder.append(graph.isDirected(edge)); } }; EdgeColumnsGDF colorColumn = new EdgeColumnsGDF("color") { @Override public boolean isEnable() { return exportColors && edgeColors; } @Override public void writeData(StringBuilder builder, Edge edge) { if (edge.alpha() != 0) { String quote = "'"; builder.append(quote); builder.append((int) (edge.r() * 255f)); builder.append(","); builder.append((int) (edge.g() * 255f)); builder.append(","); builder.append((int) (edge.b() * 255f)); builder.append(quote); } } }; EdgeColumnsGDF visibleColumn = new EdgeColumnsGDF("visible", DataTypeGDF.BOOLEAN) { @Override public boolean isEnable() { return exportVisibility; } @Override public void writeData(StringBuilder builder, Edge edge) { builder.append(true); } }; EdgeColumnsGDF labelVisibleColumn = new EdgeColumnsGDF("labelvisible", DataTypeGDF.BOOLEAN) { @Override public boolean isEnable() { return exportVisibility; } @Override public void writeData(StringBuilder builder, Edge edge) { builder.append(edge.getTextProperties().isVisible()); } }; EdgeColumnsGDF edgeIdColumn = new EdgeColumnsGDF("id", DataTypeGDF.VARCHAR) { @Override public boolean isEnable() { return false; } @Override public void writeData(StringBuilder builder, Edge edge) { } }; defaultEdgeColumnsGDFs = new EdgeColumnsGDF[7]; defaultEdgeColumnsGDFs[0] = edgeIdColumn; defaultEdgeColumnsGDFs[1] = labelColumn; defaultEdgeColumnsGDFs[2] = weightColumn; defaultEdgeColumnsGDFs[3] = directedColumn; defaultEdgeColumnsGDFs[4] = colorColumn; defaultEdgeColumnsGDFs[5] = visibleColumn; defaultEdgeColumnsGDFs[6] = labelVisibleColumn; } private void calculateMinMax(Graph graph) { minX = Float.POSITIVE_INFINITY; maxX = Float.NEGATIVE_INFINITY; minY = Float.POSITIVE_INFINITY; maxY = Float.NEGATIVE_INFINITY; minSize = Float.POSITIVE_INFINITY; maxSize = Float.NEGATIVE_INFINITY; for (Node node : graph.getNodes()) { minX = Math.min(minX, node.x()); maxX = Math.max(maxX, node.x()); minY = Math.min(minY, node.y()); maxY = Math.max(maxY, node.y()); minSize = Math.min(minSize, node.size()); maxSize = Math.max(maxSize, node.size()); } } @Override public boolean cancel() { cancel = true; return true; } @Override public void setProgressTicket(ProgressTicket progressTicket) { this.progressTicket = progressTicket; } public String getName() { return NbBundle.getMessage(getClass(), "ExporterGDF_name"); } public FileType[] getFileTypes() { FileType ft = new FileType(".gdf", NbBundle.getMessage(getClass(), "fileType_GDF_Name")); return new FileType[]{ft}; } public void setExportAttributes(boolean exportAttributes) { this.exportAttributes = exportAttributes; } public void setExportColors(boolean exportColors) { this.exportColors = exportColors; } public void setExportPosition(boolean exportPosition) { this.exportPosition = exportPosition; } public void setNormalize(boolean normalize) { this.normalize = normalize; } public void setSimpleQuotes(boolean simpleQuotes) { this.simpleQuotes = simpleQuotes; } public boolean isExportAttributes() { return exportAttributes; } public boolean isExportColors() { return exportColors; } public boolean isExportPosition() { return exportPosition; } public boolean isNormalize() { return normalize; } public boolean isSimpleQuotes() { return simpleQuotes; } public boolean isUseQuotes() { return useQuotes; } public boolean isExportVisibility() { return exportVisibility; } public void setExportVisibility(boolean exportVisibility) { this.exportVisibility = exportVisibility; } public void setUseQuotes(boolean useQuotes) { this.useQuotes = useQuotes; } private DataTypeGDF getDataTypeGDF(Class type) { if (AttributeUtils.isDynamicType(type)) { type = AttributeUtils.getStaticType((Class<? extends TimestampMap>) type); } if (type.equals(Boolean.class)) { return DataTypeGDF.BOOLEAN; } else if (type.equals(Double.class)) { return DataTypeGDF.DOUBLE; } else if (type.equals(Float.class)) { return DataTypeGDF.FLOAT; } else if (type.equals(Integer.class)) { return DataTypeGDF.INTEGER; } else if (type.equals(Long.class)) { return DataTypeGDF.INTEGER; } else if (type.equals(Short.class)) { return DataTypeGDF.INTEGER; } else if (type.equals(Byte.class)) { return DataTypeGDF.TINYINT; } else if (type.equals(String.class)) { return DataTypeGDF.VARCHAR; } else { return DataTypeGDF.VARCHAR; } } private enum DataTypeGDF { VARCHAR, BOOL, BOOLEAN, INTEGER, TINYINT, INT, DOUBLE, FLOAT } private abstract class NodeColumnsGDF { protected final String title; protected final DataTypeGDF type; protected final Object defaultValue; public NodeColumnsGDF(String title) { this(title, DataTypeGDF.VARCHAR); } public NodeColumnsGDF(String title, DataTypeGDF type) { this(title, type, null); } public NodeColumnsGDF(String title, DataTypeGDF type, Object defaultValue) { this.title = title; this.type = type; this.defaultValue = defaultValue; } public abstract boolean isEnable(); public abstract void writeData(StringBuilder builder, Node node); public String getTitle() { return title; } public DataTypeGDF getType() { return type; } public Object getDefaultValue() { return defaultValue; } } private abstract class EdgeColumnsGDF { protected final String title; protected final DataTypeGDF type; protected final Object defaultValue; public EdgeColumnsGDF(String title) { this(title, DataTypeGDF.VARCHAR); } public EdgeColumnsGDF(String title, DataTypeGDF type) { this(title, type, null); } public EdgeColumnsGDF(String title, DataTypeGDF type, Object defaultValue) { this.title = title; this.type = type; this.defaultValue = defaultValue; } public abstract boolean isEnable(); public abstract void writeData(StringBuilder builder, Edge edge); public String getTitle() { return title; } public DataTypeGDF getType() { return type; } public Object getDefaultValue() { return defaultValue; } } @Override public boolean isExportVisible() { return exportVisible; } @Override public void setExportVisible(boolean exportVisible) { this.exportVisible = exportVisible; } @Override public void setWriter(Writer writer) { this.writer = writer; } @Override public Workspace getWorkspace() { return workspace; } @Override public void setWorkspace(Workspace workspace) { this.workspace = workspace; } }