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); }