package com.yc.treecheck;
import java.io.File;
import org.eclipse.jface.viewers.ILabelProvider;
import org.eclipse.jface.viewers.LabelProvider;
import org.eclipse.swt.graphics.Image;
import com.swtdesigner.SWTResourceManager;
/**
* 主要用於繪製菜單是顯示的內容，如圖標、標題等
* @company 源辰
* @author navy
*/
public class TreeLabelProvider2 extends LabelProvider implements ILabelProvider {
private static final String PACKAGE = "src";
private static final String JAVA_FILE = ".java";
private static final String XML_FILE = ".xml";
private static final String PROP_FILE = ".properties";
private Image javaFileImage = SWTResourceManager.getImage(TreeLabelProvider2.class, "/imgs/class.png");
private Image fileImage = SWTResourceManager.getImage(TreeLabelProvider2.class, "/imgs/file.png");
private Image folderFileImage = SWTResourceManager.getImage(TreeLabelProvider2.class, "/imgs/srcfolder.png");
private Image packageFolderFileImage = SWTResourceManager.getImage(TreeLabelProvider2.class, "/imgs/package.png");
@Override
public Image getImage(Object element) {
if (element instanceof File) {
File file = (File) element;
String fileName = file.getName();
if (fileName.equals(PACKAGE)) {
return packageFolderFileImage;
}
if (file.isDirectory()) {
return folderFileImage;
}
if (fileName.endsWith(JAVA_FILE)) {
return javaFileImage;
}
if (fileName.endsWith(XML_FILE)) {
return fileImage;
}
if (fileName.endsWith(PROP_FILE)) {
return fileImage;
}
}
return fileImage;
}
@Override
public String getText(Object element) {
File file = (File) element;
return file.getName();
}
@Override
public void dispose() {
javaFileImage.dispose();
packageFolderFileImage.dispose();
folderFileImage.dispose();
fileImage.dispose();
super.dispose();
}
}
package com.yc.treecheck;
import java.io.File;
import org.eclipse.jface.viewers.ITreeContentProvider;
import org.eclipse.jface.viewers.Viewer;
/**
* 樹形結構:提供了獲取下級元素和獲取元素的方法
* @company 源辰
* @author navy
*/
public class TreeContentProvider2 implements ITreeContentProvider {
public Object[] getElements(Object inputElement) {
if (inputElement instanceof File[]) {
File[] fileList = (File[]) inputElement;
if (fileList == null || fileList.length <= 0) {
return new Object[0];
}
return fileList;
}
return new Object[0];
}
public Object[] getChildren(Object parentElement) {
if (parentElement instanceof File) {
File file = (File) parentElement;
File[] list = file.listFiles();
if (list == null || list.length <= 0) {
return new Object[0];
}
return list;
}
return new Object[0];
}
public boolean hasChildren(Object inputElement) {
if (inputElement instanceof File) {
File file = (File) inputElement;
File[] list = file.listFiles();
if (list == null || list.length <= 0) {
return false;
}
return true;
}
return false;
}
public Object getParent(Object element) {
return null;
}
public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
}
public void dispose() {
}
}
package com.yc.treecheck;
import java.io.File;
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 TreeDialog2 {
protected Shell shell;
private CheckboxTreeViewer treeViewer;
private Tree tree;
/**
* Launch the application.
*
* @param args
*/
public static void main(String[] args) {
try {
TreeDialog2 window = new TreeDialog2();
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 TreeLabelProvider2());
treeViewer.setContentProvider(new TreeContentProvider2());
tree = treeViewer.getTree();
treeViewer.setInput(new File("E:\\workspace\\SWT").listFiles());
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);
}
}
運行效果：
