-
Notifications
You must be signed in to change notification settings - Fork 0
/
Graph.cs
79 lines (63 loc) · 1.62 KB
/
Graph.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
using System.Collections.Generic;
public class Graph
{
private List<Vertex> vertices;
private HashSet<Edge> edges;
public int Order => vertices.Count;
public int Size => edges.Count;
public GraphRenderer Renderer;
public Vertex[] Vertices => vertices.ToArray();
public Graph()
{
vertices = new List<Vertex>();
edges = new HashSet<Edge>();
Renderer = new GraphRenderer(this);
}
public Vertex AddVertex(Vertex vertex)
{
vertices.Add(vertex);
return vertex;
}
public void RemoveVertex(Vertex vertex)
{
RemoveAllEdges(vertex);
vertices.Remove(vertex);
}
public void AddEdge(Vertex v1, Vertex v2)
{
edges.Add(v1.AddEdge(v2));
edges.Add(v2.AddEdge(v1));
}
public void RemoveEdge(Vertex v1, Vertex v2)
{
v1.RemoveEdge(v2);
v2.RemoveEdge(v1);
}
public void RemoveAllEdges(Vertex vertex)
{
foreach (Vertex connectedVertex in vertex.GetConnectedVertices())
{
RemoveEdge(vertex, connectedVertex);
}
}
public Vertex AddConnectedVertex(Vertex v1, Vertex v2)
{
AddVertex(v2);
AddEdge(v1, v2);
return v2;
}
}
public class Edge
{
private Vertex source;
private Vertex destination;
public EdgeRenderer Renderer;
public Vertex Source => source;
public Vertex Destination => destination;
public Edge(Vertex source, Vertex destination)
{
this.source = source;
this.destination = destination;
Renderer = new EdgeRenderer(this);
}
}