FunnelGraphDOT.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.web.funnel.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.SimpleDirectedGraph;
import org.jgrapht.nio.Attribute;
import org.jgrapht.nio.DefaultAttribute;
import org.turro.elephant.db.ElephantPU;
import org.turro.elephant.entities.web.WebGoal;
import org.turro.elephant.entities.web.WebItem;
import org.turro.graph.GraphUtil;
import org.turro.jpa.Dao;
import org.turro.sql.SqlClause;
import org.turro.string.Strings;
import org.turro.util.Cached;
 
public class FunnelGraphDOT extends SimpleDirectedGraph<FunnelVertexDOT, DefaultEdge> {
 
  /* Initialization */
  
  private void addVertices() {
    SqlClause.select("wi").from("WebItem wi")
            .orderBy("ordering")
            .dao(dao.get())
            .resultList(WebItem.class)
            .forEach(item -> {
              addVertex(new FunnelVertexDOT(item));
            });
    SqlClause.select("wg").from("WebGoal wg")
            .orderBy("ordering")
            .dao(dao.get())
            .resultList(WebGoal.class)
            .forEach(item -> {
              addVertex(new FunnelVertexDOT(item));
            });
  }
  
  private void addEdges() {
    vertexSet().stream().sorted().forEach(vertex -> {
      vertex.getNexts().forEach(item -> {
        FunnelVertexDOT vitem = new FunnelVertexDOT(item);
        if(containsVertex(vitem)) {
          Graphs.addEdgeWithVertices(this, vertex, vitem);
        }
      });
    });
  }
  
  /* Utils */
  
  public static void DOT(Writer writer) {
    GraphUtil.<FunnelVertexDOT, 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("#FFFFFF"));
              map.put("label", DefaultAttribute.createAttribute(fi.getCaption()));
              return map;              
            });
  }
  
  /* Factory */
 
  public static FunnelGraphDOT load() {
    return new FunnelGraphDOT();
  }
  
  private FunnelGraphDOT() {
    super(DefaultEdge.class);
    addVertices();
    addEdges();
  }
 
  /* Dao */
  
  protected final Cached<Dao> dao = Cached.instance(() -> new ElephantPU());
  
}