public Shape GetShape(DiagramMemberElement member)
{
Shape shape = null;
if (shapeLookup.TryGetValue(member, out shape))
{
return(shape);
}
if (member is DiagramTypeElement)
{
shape = GetUmlType(member as DiagramTypeElement);
}
if (member is DiagramCommentElement)
{
shape = GetUmlComment(member as DiagramCommentElement);
}
if (member is DiagramRelationElement)
{
shape = GetUmlAssociation(member as DiagramRelationElement);
}
shapeLookup.Add(member, shape);
return(shape);
}
private void EndDrawAssociation(Shape start, Shape end)
{
if (start == null || end == null)
{
return;
}
DiagramMemberElement startElement = GetTypeElement(start);
DiagramMemberElement endElement = GetTypeElement(end);
var association = new DiagramRelationElement();
association.Start = startElement;
association.End = endElement;
DiagramNode.AddChild(association);
}
