com.windowtester.test.locator.swt.shells.TreeCellLocatorShell.java Source code

Java tutorial

Introduction

Here is the source code for com.windowtester.test.locator.swt.shells.TreeCellLocatorShell.java

Source

package com.windowtester.test.locator.swt.shells;

import java.util.ArrayList;

import org.eclipse.jface.resource.FontRegistry;
import org.eclipse.jface.viewers.CellEditor;
import org.eclipse.jface.viewers.ICellModifier;
import org.eclipse.jface.viewers.ISelectionChangedListener;
import org.eclipse.jface.viewers.ITableLabelProvider;
import org.eclipse.jface.viewers.ITreeContentProvider;
import org.eclipse.jface.viewers.LabelProvider;
import org.eclipse.jface.viewers.SelectionChangedEvent;
import org.eclipse.jface.viewers.TextCellEditor;
import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.MouseAdapter;
import org.eclipse.swt.events.MouseEvent;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.graphics.Rectangle;
import org.eclipse.swt.layout.FillLayout;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Item;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.swt.widgets.Tree;
import org.eclipse.swt.widgets.TreeColumn;
import org.eclipse.swt.widgets.TreeItem;

/*******************************************************************************
 *  Copyright (c) 2012 Google, Inc.
 *  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:
 *  Google, Inc. - initial API and implementation
 *******************************************************************************/
public class TreeCellLocatorShell {

    public TreeItem lastSelectedItem;
    public int lastSelectedItemColumn;

    private class MyContentProvider implements ITreeContentProvider {

        /*
         * (non-Javadoc)
         * 
         * @see org.eclipse.jface.viewers.IStructuredContentProvider#getElements(java.lang.Object)
         */
        public Object[] getElements(Object inputElement) {
            return ((MyModel) inputElement).child.toArray();
        }

        /*
         * (non-Javadoc)
         * 
         * @see org.eclipse.jface.viewers.IContentProvider#dispose()
         */
        public void dispose() {

        }

        /*
         * (non-Javadoc)
         * 
         * @see org.eclipse.jface.viewers.IContentProvider#inputChanged(org.eclipse.jface.viewers.Viewer,
         *      java.lang.Object, java.lang.Object)
         */
        public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {

        }

        /*
         * (non-Javadoc)
         * 
         * @see org.eclipse.jface.viewers.ITreeContentProvider#getChildren(java.lang.Object)
         */
        public Object[] getChildren(Object parentElement) {
            return getElements(parentElement);
        }

        /*
         * (non-Javadoc)
         * 
         * @see org.eclipse.jface.viewers.ITreeContentProvider#getParent(java.lang.Object)
         */
        public Object getParent(Object element) {
            if (element == null) {
                return null;
            }

            return ((MyModel) element).parent;
        }

        /*
         * (non-Javadoc)
         * 
         * @see org.eclipse.jface.viewers.ITreeContentProvider#hasChildren(java.lang.Object)
         */
        public boolean hasChildren(Object element) {
            return ((MyModel) element).child.size() > 0;
        }

    }

    public class MyModel {
        public MyModel parent;

        public ArrayList child = new ArrayList();

        public int counter;

        public MyModel(int counter, MyModel parent) {
            this.parent = parent;
            this.counter = counter;
        }

        public String toString() {
            String rv = "";
            if (parent != null) {
                rv = parent.toString() + ".";
            }

            rv += counter;

            return rv;
        }
    }

    public class MyLabelProvider extends LabelProvider implements ITableLabelProvider {
        FontRegistry registry = new FontRegistry();

        public Image getColumnImage(Object element, int columnIndex) {
            return null;
        }

        public String getColumnText(Object element, int columnIndex) {
            return columnIndex + "." + element.toString();
        }
    }

    public TreeCellLocatorShell(Shell shell) {
        final TreeViewer v = new TreeViewer(shell, SWT.FULL_SELECTION);

        TreeColumn column = new TreeColumn(v.getTree(), SWT.NONE);
        column.setWidth(200);
        column.setText("Column 1");

        column = new TreeColumn(v.getTree(), SWT.NONE);
        column.setWidth(200);
        column.setText("Column 2");

        v.setCellEditors(new CellEditor[] { new TextCellEditor(v.getTree()), new TextCellEditor(v.getTree()) });
        v.setColumnProperties(new String[] { "col1", "col2" });
        v.setCellModifier(new ICellModifier() {

            public boolean canModify(Object element, String property) {
                return true;
            }

            public Object getValue(Object element, String property) {
                return ((MyModel) element).counter + "";
            }

            public void modify(Object element, String property, Object value) {
                ((MyModel) ((TreeItem) element).getData()).counter = Integer.parseInt(value.toString());
                v.update(((TreeItem) element).getData(), null);
            }

        });
        v.setLabelProvider(new MyLabelProvider());
        v.setContentProvider(new MyContentProvider());
        v.setInput(createModel());

        v.addSelectionChangedListener(new ISelectionChangedListener() {

            public void selectionChanged(SelectionChangedEvent event) {
                System.out.println(event);
            }

        });

        v.getTree().addMouseListener(new MouseAdapter() {
            /* (non-Javadoc)
             * @see org.eclipse.swt.events.MouseAdapter#mouseDown(org.eclipse.swt.events.MouseEvent)
             */
            @Override
            public void mouseDown(MouseEvent e) {
                Tree t = v.getTree();
                TreeItem[] items = t.getSelection();
                TreeItem item = items[0];

                for (int i = 0; i < t.getColumnCount(); i++) {
                    Rectangle bounds = getBounds(item, i);
                    if (bounds.contains(e.x, e.y)) {
                        selected(item, i);
                        break;
                    }
                }
            }

        });
    }

    private static Rectangle getBounds(Item item, int columnNumber) {
        return ((TreeItem) item).getBounds(columnNumber);
    }

    private void selected(TreeItem item, int column) {
        this.lastSelectedItem = item;
        this.lastSelectedItemColumn = column;
        System.out.println("Active Column: " + column);

    }

    private MyModel createModel() {

        MyModel root = new MyModel(0, null);
        root.counter = 0;

        MyModel tmp;
        for (int i = 1; i < 10; i++) {
            tmp = new MyModel(i, root);
            root.child.add(tmp);
            for (int j = 1; j < i; j++) {
                tmp.child.add(new MyModel(j, tmp));
            }
        }

        return root;
    }

    public static void main(String[] args) {
        Display display = new Display();
        Shell shell = new Shell(display);
        shell.setLayout(new FillLayout());
        new TreeCellLocatorShell(shell);
        shell.open();

        while (!shell.isDisposed()) {
            if (!display.readAndDispatch())
                display.sleep();
        }

        display.dispose();
    }

}