org.eclipse.jst.j2ee.internal.ui.DoubleCheckboxTableViewer.java Source code

Introduction

Here is the source code for org.eclipse.jst.j2ee.internal.ui.DoubleCheckboxTableViewer.java

Source

/*******************************************************************************
 * Copyright (c) 2008 SAP AG 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:
 * Stefan Dimov, stefan.dimov@sap.com - initial API and implementation
 *******************************************************************************/
package org.eclipse.jst.j2ee.internal.ui;

import java.util.ArrayList;
import java.util.Hashtable;

import org.eclipse.jface.viewers.CheckboxTableViewer;
import org.eclipse.jface.viewers.ICheckStateListener;
import org.eclipse.jface.viewers.ViewerRow;
import org.eclipse.swt.SWT;
import org.eclipse.swt.custom.TableEditor;
import org.eclipse.swt.widgets.Table;
import org.eclipse.swt.widgets.TableItem;

public class DoubleCheckboxTableViewer extends CheckboxTableViewer {

    protected int mSecondCheckboxColumnIndex;
    protected ICheckStateListener lstnr = null;

    public DoubleCheckboxTableViewer(Table table, int secondCheckBoxColumnIndex) {
        super(table);
        if (secondCheckBoxColumnIndex < 1)
            throw new IllegalArgumentException(
                    "The index of the second column with check box must be bigger than zero"); //$NON-NLS-1$

        mSecondCheckboxColumnIndex = secondCheckBoxColumnIndex;
    }

    @Override
    public void addCheckStateListener(ICheckStateListener listener) {
        super.addCheckStateListener(listener);
        lstnr = listener;
    }

    @Override
    public void removeCheckStateListener(ICheckStateListener listener) {
        super.removeCheckStateListener(listener);
        lstnr = null;
        getTable().getItems();
        for (int i = 0; i < getTable().getItems().length; i++) {
            DoubleCheckboxTableItem it = (DoubleCheckboxTableItem) getTable().getItem(i);
            it.setTableListener(null);
        }
    }

    @Override
    protected ViewerRow internalCreateNewRowPart(int style, int rowIndex) {
        DoubleCheckboxTableItem item;
        if (rowIndex >= 0) {
            item = new DoubleCheckboxTableItem(getTable(), style, rowIndex, mSecondCheckboxColumnIndex, this);
        } else {
            item = new DoubleCheckboxTableItem(getTable(), style, mSecondCheckboxColumnIndex, this);
        }
        item.setTableListener(lstnr);
        TableEditor editor = new TableEditor(getTable());
        editor.minimumWidth = item.getSecondCheckBox().getSize().x;
        editor.horizontalAlignment = SWT.CENTER;
        editor.setEditor(item.getSecondCheckBox(), item, mSecondCheckboxColumnIndex);
        return getViewerRowFromItem(item);
    }

    public Object[] getSecondCheckedItems() {
        TableItem[] children = getTable().getItems();
        ArrayList v = new ArrayList(children.length);
        for (int i = 0; i < children.length; i++) {
            DoubleCheckboxTableItem item = (DoubleCheckboxTableItem) children[i];
            if (item.getSecondChecked()) {
                v.add(item);
            }
        }
        return v.toArray();
    }

    public void setAllSecondChecked(boolean state) {
        TableItem[] children = getTable().getItems();
        for (int i = 0; i < children.length; i++) {
            DoubleCheckboxTableItem item = (DoubleCheckboxTableItem) children[i];
            item.setSecondChecked(state);
        }
    }

    public Object[] getSingleCheckedElements() {
        TableItem[] children = getTable().getItems();
        ArrayList v = new ArrayList(children.length);
        for (int i = 0; i < children.length; i++) {
            DoubleCheckboxTableItem item = (DoubleCheckboxTableItem) children[i];
            if (item.getChecked() && (!item.getSecondChecked())) {
                v.add(item.getData());
            }
        }
        return v.toArray();
    }

    public Object[] getDoubleCheckedElements() {
        TableItem[] children = getTable().getItems();
        ArrayList v = new ArrayList(children.length);
        for (int i = 0; i < children.length; i++) {
            DoubleCheckboxTableItem item = (DoubleCheckboxTableItem) children[i];
            if (item.getChecked() && item.getSecondChecked()) {
                v.add(item.getData());
            }
        }
        return v.toArray();
    }

    public Object[] getUncheckedItems() {
        TableItem[] children = getTable().getItems();
        ArrayList v = new ArrayList(children.length);
        for (int i = 0; i < children.length; i++) {
            TableItem item = children[i];
            if (!item.getChecked()) {
                v.add(item);
            }
        }
        return v.toArray();
    }

    public Object[] getSecondUncheckedElements() {
        TableItem[] children = getTable().getItems();
        ArrayList v = new ArrayList(children.length);
        for (int i = 0; i < children.length; i++) {
            DoubleCheckboxTableItem item = (DoubleCheckboxTableItem) children[i];
            if (!item.getSecondChecked()) {
                v.add(item.getData());
            }
        }
        return v.toArray();
    }

    public void setSecondCheckedItems(Object[] inputItems) {
        assertElementsNotNull(inputItems);
        Hashtable set = new Hashtable();
        for (int i = 0; i < inputItems.length; ++i) {
            set.put(inputItems[i], inputItems[i]);
        }
        TableItem[] items = getTable().getItems();
        for (int i = 0; i < items.length; ++i) {
            DoubleCheckboxTableItem item = (DoubleCheckboxTableItem) items[i];
            boolean check = set.containsKey(item);
            if (item.getSecondChecked() != check) {
                item.setSecondChecked(check);
            }

        }
    }

    @Override
    protected void doRemove(int start, int end) {
        // ensure that the second check box is disposed
        for (int i = start; i <= end; i++) {
            disposeSecondCheckboxOfItem(i);
        }

        super.doRemove(start, end);
    }

    @Override
    protected void doRemove(int[] indices) {
        // ensure that the second check box is disposed
        for (int i : indices) {
            disposeSecondCheckboxOfItem(i);
        }

        super.doRemove(indices);
    }

    private void disposeSecondCheckboxOfItem(int index) {
        TableItem item = getTable().getItem(index);
        if (item instanceof DoubleCheckboxTableItem) {
            ((DoubleCheckboxTableItem) item).disposeSecondCheckbox();
        }
    }

}