TotalGraph.java

// GraphTea Project: http://github.com/graphtheorysoftware/GraphTea
// Copyright (C) 2012 Graph Theory Software Foundation: http://GraphTheorySoftware.com
// Copyright (C) 2008 Mathematical Science Department of Sharif University of Technology
// Distributed under the terms of the GNU Lesser General Public License (LGPL): http://www.gnu.org/licenses/

package graphtea.extensions.actions;

import graphtea.graph.graph.Edge;
import graphtea.graph.graph.GPoint;
import graphtea.graph.graph.GraphModel;
import graphtea.graph.graph.Vertex;
import graphtea.platform.parameter.Parameter;
import graphtea.platform.parameter.Parametrizable;
import graphtea.plugins.main.GraphData;
import graphtea.plugins.main.extension.GraphActionExtension;


/**
 * Creates a line graph from the current graph and shows it in a new tab
 *
 * @author Mohammad Ali Rostami
 * @author Azin Azadi
 */
public class TotalGraph implements GraphActionExtension, Parametrizable {
    @Parameter
    public int k = 2;

    public void action(GraphData graphData) {

        GraphModel g1 = graphData.getGraph();
        GraphModel g2 = new GraphModel(false);
        g2.setIsEdgesCurved(true);

        for(Vertex v : g1.getVertexArray()) {
            Vertex tmp = new Vertex();
            tmp.setLocation(v.getLocation());
            g2.addVertex(tmp);
        }

        for(Edge e : g1.getEdges()) {
            GPoint v1 = e.source.getLocation();
            GPoint v2 = e.target.getLocation();
            double dis = v1.distance(v2);
            GPoint v3 = GPoint.sub(v2, v1);
            v3 = GPoint.div(v3, k+1);

            Vertex src = g2.getVertex(e.source.getId());
            for (int i = 0; i < k; i++) {
                Vertex tmp = new Vertex();
                tmp.getProp().obj=e;
                GPoint v4 = new GPoint(v3);
                v4.multiply(i + 1);
                v4.add(v1);
                tmp.setLocation(v4);
                g2.addVertex(tmp);
                g2.addEdge(new Edge(src,tmp));
                src=tmp;
            }
            g2.addEdge(new Edge(src, g2.getVertex(e.target.getId())));
            Edge ee = new Edge(g2.getVertex(e.source.getId()),
                    g2.getVertex(e.target.getId()));
            GPoint vector = GPoint.sub(g2.getVertex(e.target.getId()).getLocation(),
                    g2.getVertex(e.source.getId()).getLocation());
            vector= GPoint.div(vector,vector.norm());
            GPoint ppVec = new GPoint(-vector.y,vector.x);
            ppVec.multiply(30.0);
            ee.setCurveControlPoint(ppVec);
            g2.addEdge(ee);

        }

        for(Vertex v1 : g2.getVertexArray()) {
            for (Vertex v2 : g2.getVertexArray()) {
                if(v1.getProp().obj!=null) {
                    if (v2.getProp().obj != null) {
                        Edge e1 = (Edge) v1.getProp().obj;
                        Edge e2 = (Edge) v2.getProp().obj;
                        if(edgeConnects(e1,e2)) {
                            Edge ee = new Edge(v1,v2);
                            GPoint vector =
                                    GPoint.sub(v1.getLocation(),v2.getLocation());
                            vector= GPoint.div(vector,vector.norm());
                            GPoint ppVec = new GPoint(-vector.y,vector.x);
                            ppVec.multiply(30.0);
                            ee.setCurveControlPoint(ppVec);
                            g2.addEdge(ee);
                        }
                    }
                }
            }
        }


       graphData.core.showGraph(g2);
    }

    public boolean edgeConnects(Edge e1,Edge e2) {
        if(e1.source.getId() == e2.source.getId()) return true;
        if(e1.target.getId() == e2.source.getId()) return true;
        if(e1.source.getId() == e2.target.getId()) return true;
        if(e1.target.getId() == e2.target.getId()) return true;
        return false;
    }

    public String getName() {
        return "Total Graph";
    }

    public String getDescription() {
        return "Total Graph";
    }

    @Override
    public String checkParameters() {
        return null;
    }

    @Override
    public String getCategory() {
        return "Transformations";
    }
}