org.atemsource.atem.utility.doc.dot.NodeBuilder Maven / Gradle / Ivy

package org.atemsource.atem.utility.doc.dot;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;


public class NodeBuilder
{

	private Set connections = new HashSet();

	private String id;

	private String label;

	private List records = new ArrayList();

	public NodeBuilder(String id, String label)
	{
		this.id = id;
		this.label = label;
	}

	public void create(StringBuilder builder)
	{
		builder.append(id);
		builder.append(" [label=\"{");
		builder.append(label);
		for (int index = 0; index < records.size(); index++)
		{
			if (index < records.size() - 1)
			{
				builder.append("\\l|");
				records.get(index).create(builder);
			}
		}
		builder.append("\\l}\"]\r\n");
	}

	public ConnectionBuilder createConnection()
	{
		ConnectionBuilder connectionBuilder = new ConnectionBuilder();
		connectionBuilder.source(id);
		connections.add(connectionBuilder);
		return connectionBuilder;
	}

	public RecordBuilder createRecord()
	{
		RecordBuilder recordBuilder = new RecordBuilder();
		records.add(recordBuilder);
		return recordBuilder;
	}

	public Set getConnections()
	{
		return connections;
	}

}