模仿Word中组织结构图的特点生成流程图

//基本原理： // 1. 从下往上计算位置 // 2. 模仿Word中组织结构图的特点 //调用代码： Tree<string> tree = new Tree<string>(null, "董事会");
tree.Add("北京公司");
tree.Add("董事秘书室特殊机构");
tree.Add("上海公司"); tree.Childs[].Add("总经理办公室");
tree.Childs[].Add("财务部");
tree.Childs[].Add("销售部"); tree.Childs[].Add("上海销售部"); Bitmap bmp = tree.DrawAsImage(); //实现代码： using System;
using System.Collections.Generic;
using System.Text;
using System.Drawing; namespace Test {
/// <summary>
/// 用来输出组织结构图的类
/// </summary>
/// <typeparam name="T"></typeparam>
public class Tree<T>
{
Tree<T> _Parent = null;
T _Content;
List<Tree<T>> _Childs = new List<Tree<T>>();
SizeF _Size;
Rectangle _Rec; public Tree(Tree<T> parent, T content)
{
_Parent = parent;
_Content = content;
} public Tree<T> Add(T content)
{
Tree<T> tree = new Tree<T>(this, content);
_Childs.Add(tree);
return tree;
} public Tree<T> Parent { get { return _Parent; } } public T Content { get { return _Content; } } public List<Tree<T>> Childs { get { return _Childs; } } public SizeF Size { get { return _Size; } set { _Size = value; } } public Rectangle Rec { get { return _Rec; } set { _Rec = value; } } void MeatureAllSize(Graphics g, Font font, int addWidth)
{
_Size = g.MeasureString(_Content.ToString(), font);
_Size.Width += addWidth;
foreach (Tree<T> tree in Childs)
tree.MeatureAllSize(g, font, addWidth);
} List<List<Tree<T>>> GetTreeLayers()
{
List<List<Tree<T>>> layers = new List<List<Tree<T>>>();
GetTreeLayers(layers, new List<Tree<T>>(new Tree<T>[] { this }), ); return layers;
} void GetTreeLayers(List<List<Tree<T>>> layers, List<Tree<T>> childs, int level)
{
if (childs.Count == ) return;
if (layers.Count <= level) layers.Add(new List<Tree<T>>()); for (int i = ; i < childs.Count; i++)
{
layers[level].Add(childs[i]);
GetTreeLayers(layers, childs[i].Childs, level + );
}
} /// <summary>
/// 设置显示区域（从最后一层最左开始）
/// </summary>
/// <param name="level"></param>
/// <param name="height"></param>
/// <param name="interval"></param>
/// <param name="left"></param>
void SetRectangle(int level, int height, int hInterval, int vInterval, int left)
{
int index = ;
if (Parent != null) index = Parent.Childs.IndexOf(this); if (Childs.Count == )
{
// 没有儿子，就向前靠
if (left > ) left += hInterval;
}
else
{
// 有儿子，就在儿子中间
int centerX = (Childs[].Rec.Left + Childs[Childs.Count - ].Rec.Right) / ;
left = centerX - (int)_Size.Width / ; // 并且不能和前面的重复，如果重复，联同子孙和子孙的右边节点右移
if (Parent != null && index > )
{
int ex = (Parent.Childs[index - ].Rec.Right + hInterval) - left;
if (index > && ex > )
{
for (int i = index; i < Parent.Childs.Count; i++)
Parent.Childs[i].RightChilds(ex);
left += ex;
}
}
}
_Rec = new Rectangle(left, (height + vInterval) * level, (int)_Size.Width, height);
} /// <summary>
/// 所有子孙向右平移
/// </summary>
/// <param name="ex"></param>
void RightChilds(int ex)
{
Rectangle rec;
for (int i = ; i < _Childs.Count; i++)
{
rec = _Childs[i].Rec;
rec.Offset(ex, );
_Childs[i].Rec = rec;
_Childs[i].RightChilds(ex);
}
} void Offset(int x, int y)
{
_Rec.Offset(x, y);
for (int i = ; i < _Childs.Count; i++)
_Childs[i].Offset(x, y);
} public Bitmap DrawAsImage()
{
return DrawAsImage(Pens.Black, new Font("宋体", 10.5f), , , , , );
} public Bitmap DrawAsImage(Pen pen, Font font, int h, int horPadding,
int horInterval, int verInterval, int borderWidth)
{
Bitmap bmp = new Bitmap(, );
Graphics g = Graphics.FromImage(bmp);
// 把树扁平化
List<List<Tree<T>>> layers = GetTreeLayers(); // 算出每个单元的大小
MeatureAllSize(g, font, horPadding);
g.Dispose();
bmp.Dispose(); // 从最后一层开始排列
int left = ;
for (int i = layers.Count - ; i >= ; i--)
{
for (int j = ; j < layers[i].Count; j++)
{
layers[i][j].SetRectangle(i, h, horInterval, verInterval, left);
left = layers[i][j].Rec.Right;
}
} Offset(borderWidth, borderWidth); // 获取画布需要的大小
int maxHeight = (h + verInterval) * layers.Count - verInterval + borderWidth * ;
int maxWidth = ;
for (int i = layers.Count - ; i >= ; i--)
{
for (int j = ; j < layers[i].Count; j++)
{
if (layers[i][j].Rec.Right > maxWidth)
maxWidth = layers[i][j].Rec.Right;
}
}
maxWidth += borderWidth; // 边宽 // 画
bmp = new Bitmap(maxWidth, maxHeight);
g = Graphics.FromImage(bmp);
g.Clear(Color.White);
StringFormat format = (StringFormat)StringFormat.GenericDefault.Clone();
format.Alignment = StringAlignment.Center;
format.LineAlignment = StringAlignment.Center; Rectangle rec, recParent;
for (int i = ; i < layers.Count; i++)
{
for (int j = ; j < layers[i].Count; j++)
{
// 画字
rec = (Rectangle)layers[i][j].Rec;
g.DrawRectangle(pen, rec);
g.DrawString(layers[i][j].Content.ToString(), font, new SolidBrush(pen.Color),
rec, format);
// 画到父亲的线
if (layers[i][j].Parent != null)
{
recParent = layers[i][j].Parent.Rec;
g.DrawLine(pen, rec.Left + rec.Width / , rec.Top,
rec.Left + rec.Width / , rec.Top - verInterval / );
g.DrawLine(pen, recParent.Left + recParent.Width / , recParent.Bottom,
recParent.Left + recParent.Width / , rec.Top - verInterval / );
g.DrawLine(pen, rec.Left + rec.Width / , rec.Top - verInterval / ,
recParent.Left + recParent.Width / , rec.Top - verInterval / );
}
}
} g.Flush();
g.Dispose(); return bmp;
}
}
}