// Draw the graph and returns the selected Node if there's any.
private void DrawGraph(IGraphLayout graphLayout, Rect drawingArea, GraphSettings graphSettings)
{
// add border, except on right-hand side where the legend will provide necessary padding
drawingArea = new Rect(drawingArea.x + s_BorderSize,
drawingArea.y + s_BorderSize,
drawingArea.width - s_BorderSize * 2,
drawingArea.height - s_BorderSize * 2);
var b = new Bounds(Vector3.zero, Vector3.zero);
foreach (Vertex v in graphLayout.vertices)
{
b.Encapsulate(new Vector3(v.position.x, v.position.y, 0.0f));
}
// Increase b by maximum node size (since b is measured between node centers)
b.Expand(new Vector3(graphSettings.maximumNormalizedNodeSize, graphSettings.maximumNormalizedNodeSize, 0));
var scale = new Vector2(drawingArea.width / b.size.x, drawingArea.height / b.size.y);
var offset = new Vector2(-b.min.x, -b.min.y);
Vector2 nodeSize = ComputeNodeSize(scale, graphSettings);
GUI.BeginGroup(drawingArea);
foreach (var e in graphLayout.edges)
{
Vector2 v0 = ScaleVertex(e.source.position, offset, scale);
Vector2 v1 = ScaleVertex(e.destination.position, offset, scale);
Node node = e.source.node;
if (graphLayout.leftToRight)
{
DrawEdge(v1, v0, node.weight);
}
else
{
DrawEdge(v0, v1, node.weight);
}
}
Event currentEvent = Event.current;
bool oldSelectionFound = false;
Node newSelectedNode = null;
foreach (Vertex v in graphLayout.vertices)
{
Vector2 nodeCenter = ScaleVertex(v.position, offset, scale) - nodeSize / 2;
var nodeRect = new Rect(nodeCenter.x, nodeCenter.y, nodeSize.x, nodeSize.y);
bool clicked = false;
if (currentEvent.type == EventType.MouseUp && currentEvent.button == 0)
{
Vector2 mousePos = currentEvent.mousePosition;
if (nodeRect.Contains(mousePos))
{
clicked = true;
currentEvent.Use();
}
}
bool currentSelection = (m_SelectedNode != null) &&
v.node.content.Equals(m_SelectedNode.content); // Make sure to use Equals() and not == to call any overriden comparison operator in the content type.
DrawNode(nodeRect, v.node, currentSelection || clicked);
if (currentSelection)
{
// Previous selection still there.
oldSelectionFound = true;
}
else if (clicked)
{
// Just Selected a new node.
newSelectedNode = v.node;
}
}
if (newSelectedNode != null)
{
m_SelectedNode = newSelectedNode;
}
else if (!oldSelectionFound)
{
m_SelectedNode = null;
}
GUI.EndGroup();
}
private void DrawLegend(GraphSettings graphSettings, Rect legendArea)
{
EditorGUI.DrawRect(legendArea, s_LegendBackground);
// Add a border around legend area
legendArea.x += s_BorderSize;
legendArea.width -= s_BorderSize * 2;
legendArea.y += s_BorderSize;
legendArea.height -= s_BorderSize * 2;
GUILayout.BeginArea(legendArea);
GUILayout.BeginVertical();
if (graphSettings.showInspector)
{
GUILayout.Label("Inspector", m_SubTitleStyle);
if (m_SelectedNode != null)
{
using (var scrollView = new EditorGUILayout.ScrollViewScope(m_ScrollPos))
{
m_ScrollPos = scrollView.scrollPosition;
GUILayout.Label(m_SelectedNode.ToString(), m_InspectorStyle);
}
}
else
{
GUILayout.Label("Click on a node\nto display its details.");
}
}
GUILayout.FlexibleSpace();
if (graphSettings.showLegend)
{
GUILayout.Label("Legend", m_SubTitleStyle);
foreach (var pair in m_LegendForType)
{
DrawLegendEntry(pair.Value.color, pair.Value.label, false);
}
DrawLegendEntry(Color.gray, "Playing", true);
GUILayout.Space(20);
GUILayout.Label("Edge weight", m_SubTitleStyle);
GUILayout.BeginHorizontal();
GUILayout.Label("0");
GUILayout.FlexibleSpace();
GUILayout.Label("1");
GUILayout.EndHorizontal();
DrawEdgeWeightColorBar(legendArea.width);
GUILayout.Space(20);
}
GUILayout.EndVertical();
GUILayout.EndArea();
}
