Skip to content
Snippets Groups Projects
Commit 7ad846b1 authored by Sören Henning's avatar Sören Henning
Browse files

made DotBuilder ready for clusters

parent b24049fc
No related branches found
No related tags found
1 merge request!17Get impletemented stages and Java 8
......@@ -16,6 +16,11 @@ public class GraphTransformer extends AbstractTransformer<String> {
dotBuilder = new DotBuilder(graph.getName());
}
public GraphTransformer(final Graph graph, final boolean isCluster) {
super(graph);
dotBuilder = new DotBuilder("cluster_" + graph.getName(), "subgraph");
}
@Override
protected void mapVertex(final Vertex vertex) {
if (vertex.hasChildGraph()) {
......
......@@ -15,27 +15,40 @@ import com.google.common.base.Joiner;
public class DotBuilder {
private final String start;
private final String end = "}";
private final String end;
private final StringBuilder body = new StringBuilder();
private Map<String, String> defaultNodeProperties;
private Map<String, String> defaultEdgeProperties;
private Map<String, String> defaultProperties;
private static final String START_BRACKET = "{";
private static final String END_BRACKET = "}";
private static final String DEFAULT_GRAPH_NAME = "G";
private static final String DEFAULT_GRAPH_TYPE = "digraph";
public DotBuilder() {
this("G");
this(DEFAULT_GRAPH_NAME);
}
public DotBuilder(final String graphName) {
this(graphName, DEFAULT_GRAPH_TYPE);
}
public DotBuilder(final String name) {
start = "digraph " + name + " {\n";
// TODO graphType has to be one of "graph", "digraph" or "subgraph" so maybe use an enum
public DotBuilder(final String graphName, final String graphType) {
start = graphType + " " + graphName + " " + START_BRACKET + "\n";
end = END_BRACKET;
}
// TODO Deprecated
public DotBuilder(final String name, final Map<String, String> defaultNodeProperties, final Map<String, String> defaultEdgeProperties) {
this(name);
this.defaultNodeProperties = defaultNodeProperties;
this.defaultEdgeProperties = defaultEdgeProperties;
}
// TODO Deprecated
public DotBuilder(final String name, final Map<String, String> defaultNodeProperties, final Map<String, String> defaultEdgeProperties,
final Map<String, String> defaultProperties) {
this(name, defaultNodeProperties, defaultEdgeProperties);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment