private void DrawGraphEdge(AbstractGraphEdge edge)
{
if (graphSceneComponents.HasEdgeComponent(edge))
{
return;
}
EdgeComponent edgeComponent = new EdgeComponent(edge, graphScenePrefabs.InstantiateEdge());
graphSceneComponents.AddEdgeComponent(edgeComponent);
NodeComponent startNode = graphSceneComponents.GetNodeComponent(edgeComponent.GetGraphEdge().GetStartGraphNode().GetId());
NodeComponent endNode = graphSceneComponents.GetNodeComponent(edgeComponent.GetGraphEdge().GetEndGraphNode().GetId());
if (endNode == null)
{
return;
}
SpringJoint spring = endNode.GetVisualComponent().GetComponent <SpringJoint> ();
spring.connectedBody = startNode.GetVisualComponent().GetComponent <Rigidbody> ();
// checking if there is any other edge connecting the same nodes
long sameNodeConnections = 0;
graphSceneComponents.AcceptEdge(existingEdgeComponent => {
AbstractGraphEdge existingEdge = existingEdgeComponent.GetGraphEdge();
AbstractGraphNode existingStartNode = existingEdge.GetStartGraphNode();
AbstractGraphNode existingEndNode = existingEdge.GetEndGraphNode();
if (
existingStartNode.GetId() == startNode.GetGraphNode().GetId() && existingEndNode.GetId() == endNode.GetGraphNode().GetId() ||
existingStartNode.GetId() == endNode.GetGraphNode().GetId() && existingEndNode.GetId() == startNode.GetGraphNode().GetId()
)
{
sameNodeConnections = sameNodeConnections + 1;
}
});
if (sameNodeConnections > 1)
{
edgeComponent.MultiEdge = true;
}
}
public void AddEdgeComponent(EdgeComponent edgeComponent)
{
AbstractGraphEdge edge = edgeComponent.GetGraphEdge();
NodeComponent edgeStartNode = GetNodeComponent(edge.GetStartGraphNode().GetId());
if (edgeStartNode == null)
{
Debug.Log("Unable to find start node component.");
return;
}
NodeComponent edgeEndNode = GetNodeComponent(edge.GetEndGraphNode().GetId());
if (edgeEndNode == null)
{
Debug.Log("Unable to find end node component.");
return;
}
edgeComponents.Add(edgeComponent);
edgeComponent.UpdateGeometry(edgeStartNode.GetVisualComponent().transform.position, edgeEndNode.GetVisualComponent().transform.position);
}
public void Update(float time)
{
graphSceneComponents.AcceptEdge(edgeComponent => {
AbstractGraphEdge edge = edgeComponent.GetGraphEdge();
AbstractGraphNode startNode = edge.GetStartGraphNode();
AbstractGraphNode endNode = edge.GetEndGraphNode();
NodeComponent startNodeComponent = graphSceneComponents.GetNodeComponent(startNode.GetId());
NodeComponent endNodeComponent = graphSceneComponents.GetNodeComponent(endNode.GetId());
// ---- distance function start
float distance = 1F / startNode.GetDegree();
distance *= 100;
distance *= 4;
SpringJoint springJoint = endNodeComponent.GetVisualComponent().GetComponent <SpringJoint>();
springJoint.maxDistance = distance;
springJoint.minDistance = distance;
SphereCollider collider = endNodeComponent.GetVisualComponent().GetComponent <SphereCollider>();
collider.radius = distance * 0.3F;
endNodeComponent.GetVisualComponent().GetComponent <Rigidbody>().mass = 5;
// ---- distance function end
// magic
Rigidbody body = endNodeComponent.GetVisualComponent().GetComponent <Rigidbody>();
body.inertiaTensor = new Vector3(0.01F, 0.01F, 0.01F);
edgeComponent.UpdateGeometry(
startNodeComponent.GetVisualComponent().transform.position,
endNodeComponent.GetVisualComponent().transform.position);
});
// graphLayout.DoLayout (time);
}
private void RelaxEdges(GraphSceneComponents sceneComponents)
{
sceneComponents.AcceptEdge(edgeComponent => {
NodeComponent v1 = sceneComponents.GetNodeComponent(edgeComponent.GetGraphEdge().GetStartGraphNode().GetId());
NodeComponent v2 = sceneComponents.GetNodeComponent(edgeComponent.GetGraphEdge().GetEndGraphNode().GetId());
float vx = v1.GetPosition().x - v2.GetPosition().x;
float vy = v1.GetPosition().y - v2.GetPosition().y;
float vz = v1.GetPosition().z - v2.GetPosition().z;
float desiredLen = 5;
float len = (float)Mathf.Sqrt(vx * vx + vy * vy + vz * vz);
if (len == 0)
{
len = 1F;
}
double f = force_multiplier * (desiredLen - len) / len;
f = f * Mathf.Pow(stretch, 2);
double dx = f * vx;
double dy = f * vy;
double dz = f * vz;
SpringLayoutData v1d = GetLayoutData(v1);
SpringLayoutData v2d = GetLayoutData(v2);
v1d.x += dx;
v1d.y += dy;
v1d.z += dz;
v2d.x -= dx;
v2d.y -= dy;
v2d.z -= dz;
});
}
public void AddNodeComponent(NodeComponent nodeComponent)
{
nodeComponents.Add(nodeComponent);
}
private SpringLayoutData GetLayoutData(NodeComponent nodeComponent)
{
return(nodeComponent.GetVisualComponent().GetComponent <SpringLayoutData>());
}
protected override void UpdateGraphLayout(GraphSceneComponents sceneComponents, float time)
{
float maxDisplace = (float)(Math.Sqrt(AREA_MULTIPLICATOR * area) / 10F);
float k = (float)Math.Sqrt(AREA_MULTIPLICATOR * area / (1F + sceneComponents.GetNodesCount()));
sceneComponents.AcceptNode(n1 => {
sceneComponents.AcceptNode(n2 => {
if (n1.GetGraphNode().GetId() != n2.GetGraphNode().GetId())
{
float xDist = n1.GetPosition().x - n2.GetPosition().x;
float yDist = n1.GetPosition().y - n2.GetPosition().y;
float zDist = n1.GetPosition().z - n2.GetPosition().z;
float dist = (float)Math.Sqrt(xDist * xDist + yDist * yDist + zDist * zDist);
if (dist > 0)
{
float repulsiveF = k * k / dist;
ForceVectorNodeLayoutData layoutData = GetLayoutData(n1);
layoutData.dx = xDist / dist * repulsiveF;
layoutData.dy = yDist / dist * repulsiveF;
layoutData.dz = zDist / dist * repulsiveF;
}
}
});
});
sceneComponents.AcceptEdge(e => {
NodeComponent nf = sceneComponents.GetNodeComponent(e.GetGraphEdge().GetStartGraphNode().GetId());
NodeComponent nt = sceneComponents.GetNodeComponent(e.GetGraphEdge().GetEndGraphNode().GetId());
float xDist = nf.GetPosition().x - nt.GetPosition().x;
float yDist = nf.GetPosition().y - nt.GetPosition().y;
float zDist = nf.GetPosition().z - nt.GetPosition().z;
float dist = (float)Math.Sqrt(xDist * xDist + yDist * yDist + zDist * zDist);
float attractiveF = dist * dist / k;
if (dist > 0)
{
ForceVectorNodeLayoutData sourceLayoutData = GetLayoutData(nf);
ForceVectorNodeLayoutData targetLayoutData = GetLayoutData(nt);
sourceLayoutData.dx -= xDist / dist * attractiveF;
sourceLayoutData.dy -= yDist / dist * attractiveF;
sourceLayoutData.dz -= zDist / dist * attractiveF;
targetLayoutData.dx += xDist / dist * attractiveF;
targetLayoutData.dy += yDist / dist * attractiveF;
targetLayoutData.dz += zDist / dist * attractiveF;
}
});
sceneComponents.AcceptNode(n => {
ForceVectorNodeLayoutData layoutData = GetLayoutData(n);
float d = (float)Math.Sqrt(n.GetPosition().x *n.GetPosition().x + n.GetPosition().y *n.GetPosition().y + n.GetPosition().z *n.GetPosition().z);
float gf = 0.01F * k * (float)gravity * d;
layoutData.dx -= gf * n.GetPosition().x / d;
layoutData.dy -= gf * n.GetPosition().y / d;
layoutData.dz -= gf * n.GetPosition().z / d;
});
// speed
sceneComponents.AcceptNode(n => {
ForceVectorNodeLayoutData layoutData = GetLayoutData(n);
layoutData.dx *= speed / SPEED_DIVISOR;
layoutData.dy *= speed / SPEED_DIVISOR;
layoutData.dz *= speed / SPEED_DIVISOR;
});
sceneComponents.AcceptNode(n => {
ForceVectorNodeLayoutData layoutData = GetLayoutData(n);
float xDist = layoutData.dx;
float yDist = layoutData.dy;
float zDist = layoutData.dz;
float dist = (float)Math.Sqrt(layoutData.dx * layoutData.dx + layoutData.dy * layoutData.dy + layoutData.dz * layoutData.dz);
if (dist > 0)
{
float limitedDist = Math.Min(maxDisplace * ((float)speed / SPEED_DIVISOR), dist);
n.SetPosition(new UnityEngine.Vector3(
n.GetPosition().x + xDist / dist * limitedDist,
n.GetPosition().y + yDist / dist * limitedDist,
n.GetPosition().z + zDist / dist * limitedDist
));
}
});
UpdateEdges();
}
private ForceVectorNodeLayoutData GetLayoutData(NodeComponent nodeComponent)
{
return(nodeComponent.GetVisualComponent().GetComponent <ForceVectorNodeLayoutData>());
}
