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));
    }
}