package cn.com.chengang.jface.treeviewer;

import java.util.List;

import org.eclipse.jface.action.MenuManager;
import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.swt.SWT;
import org.eclipse.swt.layout.FillLayout;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.swt.widgets.Tree;
import org.eclipse.swt.widgets.TreeColumn;

public class TreeViewer3 {
	public static void main(String[] args) {
		new TreeViewer3().open();
	}

	public void open() {
		final Display display = new Display();
		final Shell shell = new Shell();
		shell.setSize(300, 300);
		// -----------界面核心代碼-------------------
		shell.setLayout(new FillLayout());
		TreeViewer tv = new TreeViewer(shell, SWT.BORDER);

		Tree tree = tv.getTree();
		tree.setHeaderVisible(true);
		TreeColumn column = new TreeColumn(tree, SWT.LEFT);
		column.setText("名稱");
		column.setWidth(150);
		column = new TreeColumn(tree, SWT.LEFT);
		column.setText("子結點數");
		column.setWidth(80);
		column = new TreeColumn(tree, SWT.LEFT);
		column.setText("性別");
		column.setWidth(40);

		tv.setContentProvider(new TreeViewerContentProvider());
		tv.setLabelProvider(new MyTableLableProvider());
		// 和TableViewer一樣，數據的入口也是setInput方法
		List<CountryEntity> input = DataFactory.createTreeData();
		tv.setInput(input);
		// --------------插入代碼：START------------------
		MyActionGroup actionGroup = new MyActionGroup(tv);// 生成一個ActionGroup對象
		actionGroup.fillContextMenu(new MenuManager());// 將按鈕注入到菜單對象中
		// -----------END------------------
		shell.open();
		while (!shell.isDisposed()) {
			if (!display.readAndDispatch())
				display.sleep();
		}
		display.dispose();
	}
}