/// <summary>
/// Returns all the figures to be drawn
/// </summary>
/// <returns>The figures.</returns>
/// <param name="cls">Project details.</param>
public IEnumerable<IFigure> GetFigures(UMLClass cls)
{
// We build trees with the input
Roots = GetTreeRoots(cls);
#region Class Blocks
double x = X_START;
double y = Y_START;
// For each tree, we perform a level order traversal
// and draw the tree in that order. This, will make sure
// that links don't overlap each other.
foreach(var root in Roots)
{
if(root.children.Count == 0){
//figures.Add(new ClassFigure((ClassNode)root.Node));
Singles.Add(new ClassFigure((ClassNode)root.Node));
continue;
}
Queue<TreeNode> queue1 = new Queue<TreeNode>();
Queue<TreeNode> queue2 = new Queue<TreeNode>();
queue1.Enqueue(root);
int level = 0;
x = X_START;
var root_figure = new ClassFigure((ClassNode)root.Node);
root_figure.MoveTo(x,y);
figures.Add(root_figure);
while(queue1.Count != 0 || queue2.Count != 0){
while(queue1.Count !=0){
var parent = queue1.Dequeue();
foreach(var child in parent.children){
queue2.Enqueue(child);
}
}
level = level + 1;
x = X_START;
y+= Y_INCREMENT;
foreach(var node in queue2){
var figure = new ClassFigure((ClassNode)node.Node);
figure.MoveTo(x,y);
x += figure.DisplayBox.Width + 50.0;
figures.Add(figure);
//connections.Add(new InheritanceConnectionFigure(figure, node.P
}
if(queue2.Count == 0) continue;
while(queue2.Count !=0){
var parent = queue2.Dequeue();
foreach(var child in parent.children){
queue1.Enqueue(child);
}
}
level = level + 1;
x = X_START;
y+= Y_INCREMENT;
foreach(var node in queue1){
var figure = new ClassFigure((ClassNode)node.Node);
figure.MoveTo(x,y);
x += figure.DisplayBox.Width + 50.0;
figures.Add(figure);
}
}
}
#endregion Class Blocks
#region Links
foreach(var node in cls.ClassNodes){
// Add all inheritance links
TypeFigure subclass = GetFigure(node.Namespace);
foreach(var link in node.Links){
TypeFigure superclass = GetFigure(link);
if (subclass != null && superclass != null) {
InheritanceConnectionFigure connection = new InheritanceConnectionFigure(subclass, superclass);
yield return connection;
}
}
if(node.Implementations.Count ==0) continue;
// Handle implementations
string imps = "";
//Position the text to the center of the box
string space_string = "";
int len = subclass.Namespace.Length;
foreach(var implementation in node.Implementations)
{
for(int i =0;i<(len - implementation.Length) /2;i++) space_string +=" ";
imps= imps + space_string + implementation + space_string;
imps+="\n";
}
SimpleTextFigure interf = new SimpleTextFigure(imps);
figureToInterfaces.Add(subclass.Namespace,interf);
InheritanceConnectionFigure connex = new InheritanceConnectionFigure(subclass, interf);
yield return connex;
}
//Positioning the interface names on top of the class figures
foreach(var item in figures){
if(figureToInterfaces.ContainsKey(item.Namespace)){
var simple = figureToInterfaces[item.Namespace];
simple.MoveTo(item.DisplayBox.X,item.DisplayBox.Y-50);
yield return simple;
}
yield return item;
}
#endregion Links
#region Single Nodes
// Draw single nodes in a row
x = X_START;
foreach(var item in Singles){
item.MoveTo(x,y);
x += item.DisplayBox.Width + SINGLES_GAP;
if(x>PAGE_WIDTH_LIMIT){
x = X_START;
y+= Y_INCREMENT;
}
if(figureToInterfaces.ContainsKey(item.Namespace)){
var simple = figureToInterfaces[item.Namespace];
simple.MoveTo(item.DisplayBox.X,item.DisplayBox.Y-50);
yield return simple;
}
yield return item;
}
// Next, all structures, enum and interfaces are drawn.
foreach(var node in cls.StructNodes){
var item = new StructFigure(node);
item.MoveTo(x,y);
x += item.DisplayBox.Width + SINGLES_GAP;
if(x>PAGE_WIDTH_LIMIT){
x = X_START;
y+= Y_INCREMENT;
}
yield return item;
}
foreach(var node in cls.EnumNodes){
var item = new EnumFigure(node);
item.MoveTo(x,y);
x += item.DisplayBox.Width + SINGLES_GAP;
if(x> PAGE_WIDTH_LIMIT){
x = X_START;
y+= Y_INCREMENT;
}
yield return item;
}
foreach(var node in cls.InterfaceNodes){
var item = new InterfaceFigure(node);
item.MoveTo(x,y);
x += item.DisplayBox.Width + SINGLES_GAP;
if(x>PAGE_WIDTH_LIMIT){
x = X_START;
y+= Y_INCREMENT;
}
yield return item;
}
#endregion Single Nodes
}
public IEnumerable<IFigure> GetFigures(UMLClass cls)
{
// Add all entities
foreach(var classnode in cls.ClassNodes){
figures.Add(new ClassFigure(classnode));
AllNodes.Add(classnode);
//NodeMapping.Add(classnode.Namespace,classnode);
}
foreach(var interfacenode in cls.InterfaceNodes){
figures.Add(new InterfaceFigure(interfacenode));
AllNodes.Add(interfacenode);
//NodeMapping.Add(interfacenode.Namespace,interfacenode);
}
foreach(var structnode in cls.StructNodes){
figures.Add(new StructFigure(structnode));
AllNodes.Add(structnode);
//NodeMapping.Add(structnode.Namespace,structnode);
}
foreach(var enumnode in cls.EnumNodes){
figures.Add(new EnumFigure(enumnode));
AllNodes.Add(enumnode);
//NodeMapping.Add(enumnode.Namespace,enumnode);
}
// Iterate over links of all entities and draw links.
foreach (var node in AllNodes) {
TypeFigure subclass = GetFigure(node.Namespace);
foreach(var link in node.Links){
TypeFigure superclass = GetFigure(link);
if (subclass != null && superclass != null) {
InheritanceConnectionFigure connection = new InheritanceConnectionFigure(subclass, superclass);
//connection.EndHandle.Owner.
yield return connection;
}
}
if(node.Implementations.Count ==0) continue;
string imps = "";
foreach(var implementation in node.Implementations)
{
imps+=implementation;
imps+="\n";
}
SimpleTextFigure interf = new SimpleTextFigure(imps);
figureToInterfaces.Add(subclass.Namespace,interf);
implementations.Add(interf);
InheritanceConnectionFigure connex = new InheritanceConnectionFigure(subclass, interf);
yield return connex;
}
double x = 50.0;
double y = 50.0;
foreach (TypeFigure figure in figures) {
figure.MoveTo(x, y);
if(figureToInterfaces.ContainsKey(figure.Namespace)){
var simple = figureToInterfaces[figure.Namespace];
simple.MoveTo(x,y-50);
yield return simple;
}
yield return figure;
x += figure.DisplayBox.Width + 50.0;
if (x > 1000.0) {
x = 50.0;
y += figure.DisplayBox.Height + 100.0;
}
}
//foreach(var figure in implementations)
//{
// figure.MoveTo(x,y);
// yield return figure;
// x += figure.DisplayBox.Width + 50.0;
// if (x > 1000.0) {
// x = 50.0;
// y += figure.DisplayBox.Height + 100.0;
// }
//}
}