public override void LayoutDiagram(PSMDiagramView psmDiagramView)
{
base.LayoutDiagram(psmDiagramView);
foreach (PSMAssociationView psmAssociationView in psmDiagramView.RepresentantsCollection.Values.OfType <PSMAssociationView>())
{
DrawAssociation(psmAssociationView);
}
foreach (PSMGeneralizationView generalizationView in psmDiagramView.RepresentantsCollection.Values.OfType <PSMGeneralizationView>())
{
DrawGeneralization(generalizationView);
}
}
public void DoLayout(PSMDiagramView psmDiagramView)
{
Stopwatch stopwatch = Stopwatch.StartNew();
if (psmDiagramView.Diagram != null && !layouting)
{
layouting = true;
//verticalTree.LayoutDiagram(psmDiagramView);
verticalTreeRightAngles.LayoutDiagram(psmDiagramView);
psmDiagramView.ExolutioCanvas.InvalidateMeasure();
layouting = false;
stopwatch.Stop();
Debug.WriteLine("Layouting {0} took {1}.", psmDiagramView.Diagram, stopwatch.Elapsed);
}
}
/// <summary>
/// Draws given root element and all its children and counts width of its subtree.
/// </summary>
/// <param name="psmDiagramView">Diagram to be layouted</param>
/// <param name="root">Root element of layouted subtree</param>
/// <param name="top">Location of the upper border of the root</param>
/// <param name="left">Location of the left border of the entire subtree</param>
/// <returns>Width of the subtree (root included)</returns>
protected virtual double DrawTree(PSMDiagramView psmDiagramView, PSMComponent root, double top, double left)
{
if (!psmDiagramView.RepresentantsCollection.IsElementPresent(root))
{
return(-horizontalSpace);
}
INodeComponentViewBase element = (psmDiagramView.RepresentantsCollection[root] as INodeComponentViewBase);
if (element.MainNode == null)
{
return(-1);
}
double height = element.ActualHeight;
double width = element.ActualWidth;
double right = left + DrawSubtree(psmDiagramView, root, top + height + verticalSpace, left);
if (right == left)
{
right = left + width;
}
else
{
if (right < left + width)
{
double subtreeWidth = right - left;
DrawSubtree(psmDiagramView, root, top + height + verticalSpace, left + (width - subtreeWidth) / 2);
right = left + width;
}
}
double x = Math.Round((right + left) / 2 - width / 2);
double y = Math.Round(top);
// ReSharper disable RedundantCheckBeforeAssignment
//if (element.X != x)
{
element.X = x;
}
//if (element.Y != y)
{
element.Y = y;
}
// ReSharper restore RedundantCheckBeforeAssignment
return(right - left);
}
/// <summary>
/// Completely performs layouting of a PSM diagram.
/// </summary>
/// <param name="psmDiagramView">The diagram (resp. canvas) to be layouted.</param>
public virtual void LayoutDiagram(PSMDiagramView psmDiagramView)
{
double left = horizontalSpace;
foreach (PSMAssociationMember root in psmDiagramView.PSMDiagram.PSMSchema.Roots)
{
if (root.ParentAssociation != null)
{
continue;
}
left += DrawTree(psmDiagramView, root, verticalSpace / 2, left) + horizontalSpace;
}
//foreach (PSMDiagramReference reference in diagram.DiagramReferences)
//{
// left += TreeLayout.DrawTree(psmDiagramView, reference, TreeLayout.verticalSpace/2, left) +
// TreeLayout.horizontalSpace;
//}
}
/// <summary>
/// Draws all children of given root element and counts width of its subtree.
/// </summary>
/// <param name="psmDiagramView">Diagram to be layouted</param>
/// <param name="root">Root element of layouted subtree</param>
/// <param name="top">Location of the upper border of the root's children</param>
/// <param name="left">Location of the left border of the entire subtree</param>
/// <returns>Width of the subtree (root not included)</returns>
protected virtual double DrawSubtree(PSMDiagramView psmDiagramView, PSMComponent root, double top, double left)
{
double right = left;
if (root is PSMAssociationMember)
{
ComponentViewBase componentView = psmDiagramView.RepresentantsCollection[root];
if (componentView is IComponentViewBaseVH && ((IComponentViewBaseVH)componentView).ViewHelper is IFoldableComponentViewHelper)
{
if (((IFoldableComponentViewHelper)((IComponentViewBaseVH)componentView).ViewHelper).IsFolded)
{
return(right - left);
}
}
PSMAssociationMember rootAM = (PSMAssociationMember)root;
if (rootAM.ChildPSMAssociations.Count > 0)
{
foreach (PSMAssociation childAssociation in rootAM.ChildPSMAssociations)
{
if (childAssociation.Child.ParentAssociation != childAssociation)
{
continue;
}
right += DrawTree(psmDiagramView, (childAssociation).Child, top, right) + horizontalSpace;
}
if (rootAM is PSMClass)
{
foreach (PSMGeneralization generalization in (rootAM as PSMClass).GeneralizationsAsGeneral)
{
right += DrawTree(psmDiagramView, (generalization).Specific, top, right) + horizontalSpace;
}
}
}
if (right != left)
{
right -= horizontalSpace;
}
}
return(right - left);
}