plugins/org.eclipse.actf.visualization.ui.report/src/org/eclipse/actf/visualization/ui/report/ProblemTree.java

/*******************************************************************************
 * Copyright (c) 2003, 2008 IBM Corporation and Others
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Kentarou FUKUDA - initial API and implementation
 *******************************************************************************/
package org.eclipse.actf.visualization.ui.report;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.eclipse.actf.util.ui.PlatformUIUtil;
import org.eclipse.actf.visualization.eval.EvaluationPlugin;
import org.eclipse.actf.visualization.eval.IEvaluationItem;
import org.eclipse.actf.visualization.eval.problem.IProblemItem;
import org.eclipse.actf.visualization.eval.problem.ProblemConst;
import org.eclipse.actf.visualization.events.IVisualizationEventListener;
import org.eclipse.actf.visualization.events.VisualizationEvent;
import org.eclipse.actf.visualization.ui.report.table.ResultTableFilter;
import org.eclipse.actf.visualization.ui.report.table.ResultTableViewer;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Tree;
import org.eclipse.swt.widgets.TreeItem;
import org.eclipse.ui.ISharedImages;



public class ProblemTree implements IVisualizationEventListener {
	
	private Tree tree;

	private TreeItem nodeRoot;

	private TreeItem node1_1;

	private TreeItem node1_2;

	private TreeItem node1_3;

	private ResultTableViewer _resultTableViewer = null;
        
	public ProblemTree(Composite parent) {

		tree = new Tree(parent, SWT.MULTI | SWT.BORDER);

		GridData gridData = new GridData();
		gridData.horizontalAlignment = GridData.FILL;
		gridData.grabExcessHorizontalSpace = true;
		gridData.verticalAlignment = GridData.FILL;
		gridData.grabExcessVerticalSpace = true;
		tree.setLayoutData(gridData);

		createTreeNodes();

		tree.addSelectionListener(new SelectionAdapter() {

			public void widgetSelected(SelectionEvent arg0) {
				if (null != _resultTableViewer) {
					TreeItem[] targets = tree.getSelection();
					if (targets != null) {
						int result = 0;
						for (int i = 0; i < targets.length; i++) {
							TreeItem item = targets[i];
							if (item == nodeRoot) {
								_resultTableViewer.changeFilterType(ResultTableFilter.ALL);
								return;
							} else if (item == node1_1) {
								result = result | IEvaluationItem.SEV_ERROR;
							} else if (item == node1_2) {
								result = result | IEvaluationItem.SEV_WARNING;
							} else if (item == node1_3) {
								result = result | IEvaluationItem.SEV_INFO;
							}
						}
						_resultTableViewer.changeFilterType(result);
						return;
					}
					_resultTableViewer.changeFilterType(ResultTableFilter.ALL);
				}
			}
		});
	}

	public void createTreeNodes() {
		tree.removeAll();

		nodeRoot = new TreeItem(tree, SWT.NULL);
		nodeRoot.setImage(PlatformUIUtil
				.getSharedImageDescriptor(ISharedImages.IMG_OBJ_FOLDER).createImage());
        
		node1_1 = new TreeItem(nodeRoot, SWT.NULL);
		node1_1.setImage(EvaluationPlugin.getImageDescriptor("icons/Err.png").createImage());

		node1_2 = new TreeItem(nodeRoot, SWT.NULL);
		node1_2.setImage(EvaluationPlugin.getImageDescriptor("icons/Warn.png").createImage());

		node1_3 = new TreeItem(nodeRoot, SWT.NULL);
		node1_3.setImage(EvaluationPlugin.getImageDescriptor("icons/Info.png").createImage());

		nodeRoot.setExpanded(true);

		clearTreeNodeAmount();
	}

	public Tree getTree() {
		return tree;
	}

	public void setRootSel() {
		tree.setSelection(new TreeItem[] { nodeRoot });
	}

	public void clearTreeNodeAmount() {
		changeTreeNodeAmount(new ArrayList());
		tree.getDisplay().update();
	}

	public void changeTreeNodeAmount(List targetList) {

		int error = 0;
		int user = 0;
		int info = 0;

		// TODO lv
		for (Iterator i = targetList.iterator(); i.hasNext();) {
			try {
				switch (((IProblemItem) i.next()).getSeverity()) {
				case IProblemItem.SEV_ERROR:
					error++;
					break;
				case IProblemItem.SEV_WARNING:
					user++;
					break;
				case IProblemItem.SEV_INFO:
					info++;
					break;
				}
			} catch (Exception e) {

			}
		}

		int all = error + user + info;

		nodeRoot.setText(ProblemConst.ALL_ERRORS + "  (" + all + ")");
		node1_1.setText(ProblemConst.ESSENTIAL + "  (" + error + ")");
		node1_2.setText(ProblemConst.USER_CHECK + "  (" + user + ")");
		node1_3.setText(ProblemConst.INFO + " (" + info + ")");

	}

	public void setResultTableViewer(ResultTableViewer resultTableViewer) {
		this._resultTableViewer = resultTableViewer;
	}

	public ResultTableViewer getResultTableViewer() {
		return this._resultTableViewer;
	}

    public void visualizerChanged(VisualizationEvent checkerEvent) {

        _resultTableViewer.visualizerChanged(checkerEvent);        
        changeTreeNodeAmount(checkerEvent.getEvaluationResult().getProblemList());        
    }
}