signature.edgecolored.EdgeColoredGraph Maven / Gradle / Ivy

Go to download
package signature.edgecolored;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

/**
 * A very simple graph class - the equivalent of a client library class.
 * 
 * @author maclean
 *
 */
public class EdgeColoredGraph {
    
    public class Edge implements Comparable {
        
        public int a;
        
        public int b;
        
        public String edgeLabel;
        
        public Edge(int a, int b, String edgeLabel) {
            if (a < b) {
                this.a = a;
                this.b = b;
            } else {
                this.a = b;
                this.b = a;
            }
            this.edgeLabel = edgeLabel;
        }
        
        public int compareTo(Edge other) {
            if (this.a < other.a || (this.a == other.a && this.b < other.b)) {
                return -1;
            } else {
                if (this.a == other.a && this.b == other.b) {
                    return 0;
                } else {
                    return 1;
                }
            }
        }
        
        public String toString() {
            return this.a + "-" + this.b + "(" + this.edgeLabel + ")";
        }

    }
    
    public List edges;
    
    public int maxVertexIndex;
    
    public String name;
    
    public EdgeColoredGraph() {
        this.edges = new ArrayList();
    }
    
    public EdgeColoredGraph(EdgeColoredGraph graph, int[] permutation) {
        this();
        for (Edge e : graph.edges) {
            makeEdge(permutation[e.a], permutation[e.b], e.edgeLabel);
        }
    }
    
    public Edge getEdge(int a, int b) {
        for (Edge edge : edges) {
            if ((edge.a == a && edge.b == b) || (edge.a == b && edge.b == a)) {
                return edge;
            }
        }
        return null;
    }

    public void makeEdge(int a, int b, String edgeLabel) {
        if (a > maxVertexIndex) maxVertexIndex = a;
        if (b > maxVertexIndex) maxVertexIndex = b;
        this.edges.add(new Edge(a, b, edgeLabel));
    }
    
    public int getVertexCount() {
        return this.maxVertexIndex + 1;
    }
    
    public boolean isConnected(int i, int j) {
        for (Edge e : edges) {
            if ((e.a == i && e.b == j) || (e.b == i && e.a == j)) {
                return true;
            }
        }
        return false;
    }
    
    public int[] getConnected(int vertexIndex) {
        List connected = new ArrayList();
        for (Edge edge : this.edges) {
            if (edge.a == vertexIndex) {
                connected.add(edge.b);
            } else if (edge.b == vertexIndex) {
                connected.add(edge.a);
            } else {
                continue;
            }
        }
        int[] connectedArray = new int[connected.size()];
        int i = 0;
        for (int connectedVertexIndex : connected) {
            connectedArray[i] = connectedVertexIndex;
            i++;
        }
        return connectedArray;
    }
    
    public int degree(int vertexIndex) {
        int degreeCount = 0;
        for (Edge e : edges) {
            if (e.a == vertexIndex || e.b == vertexIndex) {
                degreeCount++;
            }
        }
        return degreeCount;
    }

    public String toString() {
        Collections.sort(edges);
        return edges.toString();
    }
    
}