EntityGraph.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2022 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
package org.turro.entities.graph;
 
import java.io.Writer;
import java.util.LinkedHashMap;
import java.util.Map;
import org.jgrapht.Graphs;
import org.jgrapht.graph.DefaultEdge;
import org.jgrapht.graph.SimpleGraph;
import org.jgrapht.nio.Attribute;
import org.jgrapht.nio.DefaultAttribute;
import org.turro.graph.GraphUtil;
import org.turro.reflection.Instances;
import org.turro.string.Strings;
 
public class EntityGraph extends SimpleGraph<EntityVertex, DefaultEdge> {
 
  /* Populate */
  
  private void addEdges() {
    Instances.fresh().byInterface(IVertexProvider.class, IVertexProvider.class)
            .forEach(p -> p.getEdges()
                    .forEach(e -> Graphs.addEdgeWithVertices(this, e.getSource(), e.getTarget())));
  }
 
  /* Utils */
  
  public static void DOT(Writer writer) {
    GraphUtil.<EntityVertex, DefaultEdge>toDOT(writer, load(), 
            fi -> Strings.identifier(fi.getIdentifier()),
            (fi) -> {
              Map<String, Attribute> map = new LinkedHashMap<>();
              map.put("color", DefaultAttribute.createAttribute(fi.getColor()));
              map.put("fontcolor", DefaultAttribute.createAttribute(fi.getFontColor()));
              map.put("label", DefaultAttribute.createAttribute(fi.getName()));
              return map;              
            });
  }
  
  /* Factory */
 
  public static EntityGraph load() {
    return new EntityGraph();
  }
  
  private EntityGraph() {
    super(DefaultEdge.class);
    addEdges();
  }
  
}