SWT中实现带复选框的Tree
package com.yc.treecheck;
import java.util.List;
/**
* 树节点接口
* @company 源辰
* @author navy
*/
@SuppressWarnings("rawtypes")
public interface ITreeNode {
public String getName(); // 获取节点名
public void setName(String name); // 设置节点名
public boolean getChecked(); // 获取节点的状态
public void setChecked(boolean checked); // 设置节点的状态
public List getChildren(); // 当前节点的子节点
public void addChildren(Object obj); // 添加子节点
public void setChildren(List list); // 设置当前节点的子节点
}
package com.yc.treecheck;
import java.util.List;
/**
* 树节点接口
* @company 源辰
* @author navy
*/
@SuppressWarnings("rawtypes")
public interface ITreeNode {
public String getName(); // 获取节点名
public void setName(String name); // 设置节点名
public boolean getChecked(); // 获取节点的状态
public void setChecked(boolean checked); // 设置节点的状态
public List getChildren(); // 当前节点的子节点
public void addChildren(Object obj); // 添加子节点
public void setChildren(List list); // 设置当前节点的子节点
}
package com.yc.treecheck;
import java.util.ArrayList;
import java.util.List;
@SuppressWarnings({ "rawtypes", "unchecked" })
public class Item implements ITreeNode {
private String name;
private boolean checked;
private List children = new ArrayList();
public Item() {
super();
}
public Item(String name, boolean checked, List children) {
super();
this.name = name;
this.checked = checked;
this.children = children;
}
public Item(String name) {
super();
this.name = name;
}
@Override
public String toString() {
return "Item [name=" + name + ", checked=" + checked + ", children=" + children + "]";
}
@Override
public String getName() {
return name;
}
@Override
public void setName(String name) {
this.name = name;
}
@Override
public boolean getChecked() {
return checked;
}
@Override
public void setChecked(boolean checked) {
this.checked = checked;
}
@Override
public List getChildren() {
return this.children;
}
@Override
public void addChildren(Object obj) {
this.children.add(obj);
}
@Override
public void setChildren(List list) {
this.children.clear();
this.children.addAll(list);
}
}
package com.yc.treecheck;
import java.util.List;
import org.eclipse.jface.viewers.ITreeContentProvider;
import org.eclipse.jface.viewers.Viewer;
/**
* 树形结构:提供了获取下级元素和获取元素的方法
* @company 源辰
* @author navy
*
*/
@SuppressWarnings("unchecked")
public class TreeContentProvider implements ITreeContentProvider {
public Object[] getElements(Object inputElement) {
if (inputElement instanceof List) {
List<Item> input = (List<Item>) inputElement;
return input.toArray();
}
return new Object[0];
}
public Object[] getChildren(Object parentElement) {
ITreeNode node = (ITreeNode) parentElement;
List<Item> list = node.getChildren();
if (list == null) {
return new Object[0];
}
return list.toArray();
}
public boolean hasChildren(Object element) {
ITreeNode node = (ITreeNode) element;
List<Item> list = node.getChildren();
return !(list == null || list.isEmpty());
}
public Object getParent(Object element) {
return null;
}
public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
}
public void dispose() {
}
}
package com.yc.treecheck;
import org.eclipse.jface.viewers.LabelProvider;
import org.eclipse.swt.graphics.Image;
/**
* 主要用于绘制菜单是显示的内容，如图标、标题等
* @company 源辰
* @author navy
*/
public class TreeLabelProvider extends LabelProvider {
public String getText(Object element) {
ITreeNode node = (ITreeNode) element;
return node.getName();
}
public Image getImage(Object element) {
return null;
}
}
package com.yc.treecheck;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import org.eclipse.jface.viewers.CheckStateChangedEvent;
import org.eclipse.jface.viewers.CheckboxTreeViewer;
import org.eclipse.jface.viewers.ICheckStateListener;
import org.eclipse.swt.SWT;
import org.eclipse.swt.custom.SashForm;
import org.eclipse.swt.layout.FillLayout;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.swt.widgets.Tree;
import org.eclipse.swt.widgets.TreeItem;
import com.swtdesigner.SWTResourceManager;
import com.yc.login.Login;
public class TreeDialog {
protected Shell shell;
private CheckboxTreeViewer treeViewer;
private Tree tree;
/**
* Launch the application.
*
* @param args
*/
public static void main(String[] args) {
try {
TreeDialog window = new TreeDialog();
window.open();
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* Open the window.
*/
public void open() {
Display display = Display.getDefault();
createContents();
shell.open();
shell.layout();
while (!shell.isDisposed()) {
if (!display.readAndDispatch()) {
display.sleep();
}
}
}
/**
* Create contents of the window.
*/
protected void createContents() {
shell = new Shell();
shell.setSize(450, 300);
shell.setText("带复选框的树");
shell.setLayout(new FillLayout());
shell.setImage(SWTResourceManager.getImage(Login.class, "/images/yc.ico"));
SashForm sashForm = new SashForm(shell, SWT.NONE);
Composite composite = new Composite(sashForm, SWT.NONE);
composite.setLayout(new FillLayout());
treeViewer = new CheckboxTreeViewer(composite, SWT.BORDER | SWT.CHECK | SWT.MULTI);
treeViewer.setLabelProvider(new TreeLabelProvider());
treeViewer.setContentProvider(new TreeContentProvider());
tree = treeViewer.getTree();
initTree();
Composite composite_1 = new Composite(sashForm, SWT.NONE);
composite_1.setLayout(new FillLayout(SWT.HORIZONTAL));
sashForm.setWeights(new int[] { 2, 3 });
treeViewer.addCheckStateListener(new ICheckStateListener() {
public void checkStateChanged(CheckStateChangedEvent arg0) { // 当状态改变是触发
CheckboxTreeViewer checkboxTreeViewer = (CheckboxTreeViewer)arg0.getSource();
boolean checked = arg0.getChecked(); // 当前节点的选中状态
checkboxTreeViewer.setSubtreeChecked(arg0.getElement(), checked); // 设置当前节点子节点的状态与当前节点一致
// TODO 当所有子节点都已经选中时，父节点也要选中
// 获取指定的选中的节点 -> 不能直接用arg0.getElement()这个获取到的是我们自定义的Item而不是TreeItem
TreeItem ti = (TreeItem)checkboxTreeViewer.testFindItem(arg0.getElement());
tree.setSelection(ti); // 选中这个节点
ti.setChecked(checked);
checkObject(ti);
}
});
}
private void checkObject(TreeItem ti) {
TreeItem parent = ti.getParentItem(); // 获取这个节点的节点，通过父节点获取这个父节点下的所有子节点，然后判断是否已经被选中
if(parent == null){
return;
}
TreeItem[] items = parent.getItems(); // 获取当前节点的所有兄弟节点
int checkItems = 0; // 记录被选中的节点的数量
for (TreeItem treeItem : items) {
if(treeItem.getChecked() && !treeItem.getGrayed()){ //Grayed：变灰的
checkItems = checkItems + 1;
}
}
if(checkItems == 0){
parent.setChecked(false);
parent.setGrayed(false);
} else if(checkItems == items.length){
parent.setGrayed(false);
parent.setChecked(true);
} else if(checkItems != items.length){
parent.setChecked(true);
parent.setGrayed(true);
}
checkObject(parent);
}
private void initTree() {
Item cf1 = new Item("湖南");
cf1.addChildren(new Item("长沙"));
cf1.addChildren(new Item("益阳"));
cf1.addChildren(new Item("衡阳"));
Item cf2 = new Item("湖北");
cf2.addChildren(new Item("武汉"));
cf2.addChildren(new Item("咸宁"));
List<Item> cfs = new ArrayList<Item>();
Collections.addAll(cfs, cf1, cf2);
Item cf = new Item("中国", false, cfs);
treeViewer.setInput(Arrays.asList(cf));
}
}