using System.Drawing;
using System.Windows.Forms;
using System.ComponentModel; namespace SimpleCustomControl
{
[ToolboxBitmap(typeof(TreeView))]
public partial class MyTreeView : TreeView
{
private int imageWidth = 0x12; private bool isFirstNodeDifferent = true;
[Category("行为"), DefaultValue(true), Description("确定根节点第一个节点是否作为取消所择使用。")]
public bool IsFirstNodeDifferent
{
get { return isFirstNodeDifferent; }
set { isFirstNodeDifferent = value; }
} public MyTreeView()
{
InitializeComponent();
// 如果更改 ImageList 中图片大小,此处设置可能有用.未测试.可注释掉.
imageWidth = imageList1.ImageSize.Width + ;
} //规则1：取消选定
//规则1.1：检查是否有子节点，需清除所有子节点的选定状态；
//规则1.2：检查是否有父节点，如有，则根据兄弟节点的选定状态修改父节点的选定状态
//规则2：选定
//规则2.1：检查是否有子节点，设置所有子节点为选定状态
//规则2.2：检查是否有父节点，如有，则根据兄弟节点的选定状态修改父节点的选定状态
/// <summary>
/// 鼠标点击节点触发事件
/// </summary>
private void MyTreeView_NodeMouseClick(object sender, TreeNodeMouseClickEventArgs e)
{
if (e.Button == MouseButtons.Left)
{
Rectangle rc = e.Node.Bounds;
rc = new Rectangle(rc.X - imageWidth, rc.Y, rc.Width + imageWidth, rc.Height);
if (rc.Contains(e.Location))
{
if (isFirstNodeDifferent && e.Node.Level == && e.Node.Index == )
NodeUnSelected();
else
NodeClick(e.Node);
}
}
} /// <summary>
/// 节点点击
/// </summary>
/// <param name="tn">点击的节点</param>
public void NodeClick(TreeNode tn)
{
if (tn.ImageIndex == )
{
NodeUnSelected(tn);
}
else
{
NodeSelected(tn);
}
} /// <summary>
/// 不管现在节点状态
/// 将节点设置为选中
/// </summary>
/// <param name="tn">待更改状态的节点</param>
public void NodeSelected(TreeNode tn)
{
tn.SelectedImageIndex = ;
tn.ImageIndex = ;
SetNodeImg21(tn);
SetNodeImg22(tn);
} /// <summary>
/// 不管现在节点状态
/// 去掉节点选中状态
/// </summary>
/// <param name="tn">待更改状态的节点</param>
public void NodeUnSelected(TreeNode tn)
{
tn.SelectedImageIndex = ;
tn.ImageIndex = ;
SetNodeImg11(tn);
SetNodeImg12(tn);
} /// <summary>
/// 查找节点状态 是否被选中
/// </summary>
/// <param name="tn">查看的节点</param>
/// <returns>节点是否被选中</returns>
public bool NodeIsChecked(TreeNode tn)
{
if (tn.ImageIndex == || tn.ImageIndex == )
return true;
return false;
} private bool nodeIsAllChecked;
/// <summary>
/// 返回整个TreeView 选中状态
/// </summary>
/// <returns>返回选中状态信息</returns>
[Browsable(false)]
public bool NodeIsAllChecked
{
get
{
if (isFirstNodeDifferent)
{
// 两种方式 两种效率 未测试哪个更效率
//foreach (TreeNode tn in Nodes)
//{
// if (tn.Index != 0 && NodeCheckState(tn) != CheckState.Checked)
// return false;
//}
int count = Nodes.Count;
for (int i = ; i < count; i++)
{
if (NodeCheckState(Nodes[i]) != CheckState.Checked)
return false;
}
}
else
{
foreach (TreeNode tn in Nodes)
{
if (NodeCheckState(tn) != CheckState.Checked)
return false;
}
}
return true;
}
} /// <summary>
/// 返回指定节点选中状态
/// 有三种状态 Checked 选中 Indeterminate 部分选中 Unchecked 未选中
/// </summary>
/// <param name="tn">节点是否被选中</param>
/// <returns>返回选中状态信息</returns>
public CheckState NodeCheckState(TreeNode tn)
{
switch (tn.ImageIndex)
{
case :
return CheckState.Indeterminate;
case :
return CheckState.Checked;
case :
default:
return CheckState.Unchecked;
}
} /// <summary>
/// 将所有子节点全不选
/// </summary>
public void NodeUnSelected()
{
foreach (TreeNode tn in this.Nodes)
{
NodeUnSelected(tn);
}
} /// <summary>
/// 将所有子节点全选
/// </summary>
public void NodeSelected()
{
foreach (TreeNode tn in this.Nodes)
{
NodeSelected(tn);
}
if (isFirstNodeDifferent && Nodes.Count > )
{
Nodes[].SelectedImageIndex = ;
Nodes[].ImageIndex = ;
}
} //设置节点选定状态：
//规则.1：检查是否有子节点，需清除所有子节点的选定状态；
void SetNodeImg11(TreeNode tn)
{
foreach (TreeNode t in tn.Nodes)
{
t.SelectedImageIndex = ;
t.ImageIndex = ;
if (t.Nodes.Count != )
SetNodeImg11(t);
}
} //设置节点选定状态：
//规则.2：检查是否有父节点，如有，则根据兄弟节点的选定状态修改父节点的选定状态
void SetNodeImg12(TreeNode tn)
{
if (tn.Parent == null)
return;
int Img0Num = , Img1Num = , Img2Num = ;
//统计同级节点选中情况
foreach (TreeNode t in tn.Parent.Nodes)
{
switch (t.ImageIndex)
{
case :
Img1Num++;
break;
case :
Img2Num++;
break;
case :
default:
Img0Num++;
break;
}
}
//如果同级节点选中和未选中的都有
if (Img1Num != || (Img0Num != && Img2Num != ))
{
tn.Parent.SelectedImageIndex = ;
tn.Parent.ImageIndex = ;
}
else
{
tn.Parent.StateImageIndex = ;
tn.Parent.ImageIndex = ;
}
SetNodeImg12(tn.Parent);
} //设置节点选定状态：
//规则.1：检查是否有子节点，设置所有子节点为选定状态
void SetNodeImg21(TreeNode tn)
{
foreach (TreeNode t in tn.Nodes)
{
t.SelectedImageIndex = ;
t.ImageIndex = ;
if (t.Nodes.Count != )
{
SetNodeImg21(t);
}
}
} //设置节点选定状态：
//规则.2：检查是否有父节点，如有，则根据兄弟节点的选定状态修改父节点的选定状态
void SetNodeImg22(TreeNode tn)
{
if (tn.Parent == null)
return;
int Img0Num = , Img1Num = , Img2Num = ;
foreach (TreeNode t in tn.Parent.Nodes)
{
//if (t.ImageIndex == 0)
// Img0Num++;
//if (t.ImageIndex == 1)
// Img1Num++;
//if (t.ImageIndex == 2)
// Img2Num++;
switch (t.ImageIndex)
{
case :
Img1Num++;
break;
case :
Img2Num++;
break;
case :
default:
Img0Num++;
break;
}
}
if (Img1Num != || (Img0Num != && Img2Num != ))
{
tn.Parent.SelectedImageIndex = ;
tn.Parent.ImageIndex = ;
}
//else if (Img1Num == 0 && Img2Num == 0)
//{
// tn.Parent.SelectedImageIndex = 0;
// tn.Parent.ImageIndex = 0;
//}
else
{
tn.Parent.StateImageIndex = ;
tn.Parent.ImageIndex = ;
}
SetNodeImg22(tn.Parent);
} }
}