tests/org.eclipse.ui.tests/Eclipse JFace Tests/org/eclipse/jface/tests/viewers/interactive/TestCheckboxTree.java

package org.eclipse.jface.tests.viewers.interactive;

/*
 * Licensed Materials - Property of IBM,
 * WebSphere Studio Workbench
 * (c) Copyright IBM Corp 1999, 2000
 */
import org.eclipse.jface.*;
import org.eclipse.jface.action.*;
import org.eclipse.jface.tests.viewers.*;
import org.eclipse.jface.viewers.*;
import org.eclipse.swt.*;
import org.eclipse.swt.widgets.*;
import java.util.*;

public class TestCheckboxTree extends TestTree {
	CheckboxTreeViewer fCheckboxViewer;

	public TestCheckboxTree() {
		super();
	}
public void checkChildren(TestElement element, boolean state) {
	int numChildren = element.getChildCount();
	for (int i = 0; i < numChildren; i++) {
		TestElement child = element.getChildAt(i);
		if (fCheckboxViewer.setChecked(child, state))
			checkChildren(child, state);
	}
}
public Viewer createViewer(Composite parent) {
	CheckboxTreeViewer viewer = new CheckboxTreeViewer(parent);
	viewer.setContentProvider(new TestModelContentProvider());
	viewer.setLabelProvider(new TestLabelProvider());

	viewer.addTreeListener(new ITreeViewerListener() {
		public void treeExpanded(TreeExpansionEvent e) {
			handleTreeExpanded((TestElement) e.getElement());
		}
		public void treeCollapsed(TreeExpansionEvent e) {
		}
	});

	viewer.addCheckStateListener(new ICheckStateListener() {
		public void checkStateChanged(CheckStateChangedEvent e) {
			checkChildren((TestElement) e.getElement(), e.getChecked());
		}
	});

	fCheckboxViewer = viewer;
	fViewer = viewer;
	return viewer;
}
	public void handleTreeExpanded(TestElement element) {
		// apply the same check recursively to all children
		boolean checked = fCheckboxViewer.getChecked(element);
		int numChildren = element.getChildCount();
		for (int i= 0; i < numChildren; i++) {
			TestElement child= element.getChildAt(i);
			fCheckboxViewer.setChecked(child, checked);
		}
	}
public static void main(String[] args) {
	TestBrowser browser = new TestCheckboxTree();
	if (args.length > 0 && args[0].equals("-twopanes"))
		browser.show2Panes();
	browser.setBlockOnOpen(true);
	browser.open(TestElement.createModel(3, 10));
}
}