com.sqlapp.graphviz.Graph Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sqlapp-graphviz Show documentation
sqlapp graphviz library.
There is a newer version: 0.12.37
/**
 * Copyright (C) 2007-2017 Tatsuo Satoh 
 *
 * This file is part of sqlapp-graphviz.
 *
 * sqlapp-graphviz 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 3 of the License, or
 * (at your option) any later version.
 *
 * sqlapp-graphviz 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 sqlapp-graphviz.  If not, see .
 */

package com.sqlapp.graphviz;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.function.Consumer;

import com.sqlapp.util.CommonUtils;

import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.Getter;
import lombok.Setter;

@Data
@EqualsAndHashCode(callSuper=true)
public class Graph extends AbstractCommonElement{

	
	public Graph(String name){
		this.name=name;
	}
	
	private boolean directed;
	
	private boolean cluster;

	@Getter(value=lombok.AccessLevel.PUBLIC)
	@Setter(value=lombok.AccessLevel.PROTECTED)
	private GraphCollection parent;

	private String name=null;

	private Layout layout=null;
	
	@Getter(value=lombok.AccessLevel.PRIVATE)
	@Setter(value=lombok.AccessLevel.PRIVATE)
	private GraphCollection graphs=new GraphCollection(this);

	@Getter(value=lombok.AccessLevel.PRIVATE)
	@Setter(value=lombok.AccessLevel.PRIVATE)
	private NodeCollection nodes=new NodeCollection(this);
	
	@Getter(value=lombok.AccessLevel.PRIVATE)
	@Setter(value=lombok.AccessLevel.PRIVATE)
	private EdgeCollection edges=new EdgeCollection(this);

	@Getter(value=lombok.AccessLevel.PRIVATE)
	@Setter(value=lombok.AccessLevel.PRIVATE)
	private RankCollection ranks=new RankCollection(this);
	
	private List elements=new ArrayList<>();
	
	public String getName() {
		if (name==null){
			return null;
		}
		if (this.isCluster()){
			return "cluster_"+name;
		}
		return name;
	}

	protected Graph instance(){
		return this;
	}
	
	public Node addNode(String name){
		Node node=this.nodes.get(name);
		if (node!=null){
			return node;
		}
		node=new Node(name);
		return addNode(node);
	}

	public Graph addNode(Consumer c, String... names){
		List list=new ArrayList<>();
		for(String name:names){
			Node node=this.nodes.get(name);
			if (node==null){
				node=new Node(name);
				list.add(node);
			}
		}
		this.nodes.addAll(list);
		this.elements.addAll(list);
		if(c!=null){ 
			list.forEach(obj->{
				c.accept(obj);
			});
		}
		return instance();
	}
	
	public Graph addRank(RankType rankType, Node... nodes){
		if (CommonUtils.isEmpty(nodes)){
			return instance();
		}
		Rank rank=new Rank(rankType, nodes);
		this.ranks.add(rank);
		this.elements.add(rank);
		return instance();
	}

	public Graph addRank(RankType rankType, java.util.Collection nodes){
		if (CommonUtils.isEmpty(nodes)){
			return instance();
		}
		Rank rank=new Rank(rankType, nodes.toArray(new Node[0]));
		this.ranks.add(rank);
		this.elements.add(rank);
		return instance();
	}

	
	public Graph addRank(Node... nodes){
		return addRank(RankType.same, nodes);
	}

	public Graph addRank(RankType rankType, String... nodes){
		return addRank(rankType, getNodesInternal(nodes).toArray(new Node[0]));
	}

	public Graph addRank(String... nodes){
		return addRank(RankType.same, nodes);
	}

	private List getNodesInternal(String... nodes){
		List list=new ArrayList<>();
		if (CommonUtils.isEmpty(nodes)){
			return list;
		}
		for(String name:nodes){
			Node node=this.nodes.get(name);
			if (node==null){
				throw new IllegalArgumentException("Node["+name+"] not found.");
			}
			list.add(node);
		}
		return list;
	}
	
	public  T addNode(T node){
		nodes.add(node);
		this.elements.add(node);
		return node;
	}


	public Graph addNode(Consumer c, Node... args){
		List list=new ArrayList<>();
		for(Node node:args){
			Node nd=this.nodes.get(node.getName());
			if (nd!=null){
				node=nd;
			}
			list.add(node);
		}
		nodes.addAll(list);
		this.elements.addAll(list);
		if(c!=null){ 
			list.forEach(obj->{
				c.accept(obj);
			});
		}
		return instance();
	}

	public Graph addGraph(Consumer c, String... args){
		List list=new ArrayList<>();
		for(String arg:args){
			Graph graph=new Graph(arg);
			list.add(graph);
		}
		graphs.addAll(list);
		this.elements.addAll(list);
		if(c!=null){ 
			list.forEach(obj->{
				c.accept(obj);
			});
		}
		return instance();
	}

	public Graph addGraph(Graph graph){
		graphs.add(graph);
		this.elements.add(graph);
		return instance();
	}

	public Edge addEdge(Node from, Node to){
		Edge obj=new Edge(from, to);
		edges.add(obj);
		this.elements.add(obj);
		return obj;
	}

	public Node getNode(String name){
		Node node= this.nodes.get(name);
		if (node!=null){
			return node;
		}
		for(Graph graph:this.graphs){
			node=graph.getNode(name);
			if (node!=null){
				return node;
			}
		}
		return null;
	}

	public Graph addGraphSetting(Consumer c){
		GraphSetting obj=new GraphSetting(this);
		this.elements.add(obj);
		c.accept(obj);
		return instance();
	}

	public GraphSetting addGraphSetting(){
		GraphSetting obj=new GraphSetting(this);
		this.elements.add(obj);
		return obj;
	}
	
	public Graph addNodeSetting(Consumer c){
		NodeSetting obj=new NodeSetting(this);
		this.elements.add(obj);
		c.accept(obj);
		return instance();
	}

	public NodeSetting addNodeSetting(){
		NodeSetting obj=new NodeSetting(this);
		this.elements.add(obj);
		return obj;
	}

	public Graph addEdgeSetting(Consumer c){
		EdgeSetting obj=new EdgeSetting(this);
		this.elements.add(obj);
		c.accept(obj);
		return instance();
	}
	
	public EdgeSetting addEdgeSetting(){
		EdgeSetting obj=new EdgeSetting(this);
		this.elements.add(obj);
		return obj;
	}

	public Edge addEdge(NodePort... nodes){
		Edge obj=new Edge(nodes);
		edges.add(obj);
		this.elements.add(obj);
		return obj;
	}

	public Graph addEdge(Consumer c, Node... nodes){
		Edge obj=new Edge((Object[])nodes);
		edges.add(obj);
		this.elements.add(obj);
		if (c!=null){
			c.accept(obj);
		}
		return instance();
	}

	public Graph addEdge(Consumer c, NodePort... nodes){
		Edge obj=new Edge(nodes);
		edges.add(obj);
		this.elements.add(obj);
		if (c!=null){
			c.accept(obj);
		}
		return instance();
	}
	
	public Graph addEdge(Consumer c, String... nodePorts){
		for(String nodePort:nodePorts){
			String[] splits=nodePort.split("\\s*->\\s*");
			for(int i=0;i{
			builder.put(k, v);
		});
		for(AbstractGraphVizElement element:elements){
			if (element instanceof ToGraphStringBuilder){
				GraphStringBuilder child=((ToGraphStringBuilder)element).toGraphStringBuilder();
				if (element instanceof Edge){
					child.setIndentLevel(builder.getIndentLevel()+1);
					builder.put(child);
				} else{
					if (!child.isEmpty()){
						child.setIndentLevel(builder.getIndentLevel()+1);
						builder.put(child);
					}
				}
			} else{
				throw new RuntimeException("Invalid elemnt. element=["+element+"]");
			}
		}
		return builder;
	}
	
	protected Graph getRoot(){
		return getParent(this);
	}

	protected Graph getParent(Graph graph){
		if (graph==null){
			return this;
		}
		if (graph.parent==null){
			return graph;
		}
		return getParent(graph.parent.getParent());
	}

}