plugins/org.apache.geronimo.st.ui/src/org/apache/geronimo/st/ui/sections/AbstractTableSection.java - geronimo-devtools - Git at Google

 /*
  * Licensed to the Apache Software Foundation (ASF) under one or more
  * contributor license agreements.  See the NOTICE file distributed with
  * this work for additional information regarding copyright ownership.
  * The ASF licenses this file to You under the Apache License, Version 2.0
  * (the "License"); you may not use this file except in compliance with
  * the License.  You may obtain a copy of the License at
  *
  *    http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
 package org.apache.geronimo.st.ui.sections;

 import org.apache.geronimo.st.ui.CommonMessages;
 import org.apache.geronimo.st.ui.wizards.AbstractTableWizard;
 import org.eclipse.emf.common.notify.AdapterFactory;
 import org.eclipse.emf.ecore.EClass;
 import org.eclipse.emf.ecore.EObject;
 import org.eclipse.emf.ecore.EReference;
 import org.eclipse.emf.ecore.util.EcoreUtil;
 import org.eclipse.emf.edit.ui.provider.AdapterFactoryContentProvider;
 import org.eclipse.emf.edit.ui.provider.AdapterFactoryLabelProvider;
 import org.eclipse.jface.dialogs.Dialog;
 import org.eclipse.jface.viewers.ColumnWeightData;
 import org.eclipse.jface.viewers.StructuredSelection;
 import org.eclipse.jface.viewers.TableLayout;
 import org.eclipse.jface.viewers.TableViewer;
 import org.eclipse.jface.viewers.Viewer;
 import org.eclipse.jface.viewers.ViewerFilter;
 import org.eclipse.jface.wizard.Wizard;
 import org.eclipse.jface.wizard.WizardDialog;
 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.layout.GridLayout;
 import org.eclipse.swt.widgets.Button;
 import org.eclipse.swt.widgets.Composite;
 import org.eclipse.swt.widgets.Display;
 import org.eclipse.swt.widgets.Table;
 import org.eclipse.swt.widgets.TableColumn;
 import org.eclipse.swt.widgets.TableItem;
 import org.eclipse.ui.forms.widgets.FormToolkit;
 import org.eclipse.ui.forms.widgets.Section;

 /**
  * @version $Rev$ $Date$
  */
 public abstract class AbstractTableSection extends AbstractSectionPart {

 	private static final String[] COLUMN_NAMES = new String[] {};

 	private Table table;

 	private TableViewer tableViewer;

 	protected Button addButton;

 	protected Button editButton;

 	protected Button removeButton;

 	public AbstractTableSection(Section section) {
 		super(section);
 	}

 	/**
 	 * @param plan
 	 * @param parent
 	 * @param toolkit
 	 * @param style
 	 *
 	 * Subclasses should call createClient() in constructor
 	 */
 	public AbstractTableSection(EObject plan, Composite parent,
 			FormToolkit toolkit, int style) {
 		super(parent, toolkit, style, plan);
 	}

 	public void createClient() {

 		if (getTableEntryObjectType() == null)
 			throw new NullPointerException();

 		getSection().setText(getTitle());
 		getSection().setDescription(getDescription());
 		getSection().setLayoutData(getSectionLayoutData());
 		Composite composite = createTableComposite(getSection());
 		getSection().setClient(composite);
 		table = createTable(composite);

 		tableViewer = new TableViewer(getTable());
 		tableViewer.setContentProvider(new AdapterFactoryContentProvider(getAdapterFactory()));
 		tableViewer.setLabelProvider(new AdapterFactoryLabelProvider(getAdapterFactory()));
 		tableViewer.setInput(getInput());

 		tableViewer.addFilter(new ViewerFilter() {
 			public boolean select(Viewer viewer, Object parentElement,
 					Object element) {
 				return AbstractTableSection.this.filter(viewer, parentElement, element);
 			}
 		});

 		if (getTableColumnNames().length > 0) {
 			tableViewer.setColumnProperties(getTableColumnNames());
 		}

 		Composite buttonComp = createButtonComposite(composite);
 		createAddButton(toolkit, buttonComp);
 		createRemoveButton(toolkit, buttonComp);
 		createEditButton(toolkit, buttonComp);

 	}

 	public Object getInput() {
 		return getPlan();
 	}

 	protected boolean filter(Viewer viewer, Object parentElement, Object element) {
 		return getTableEntryObjectType().isInstance(element);
 	}

 	protected Composite createTableComposite(Composite parent) {
 		Composite composite = toolkit.createComposite(parent);
 		composite.setLayout(getSectionCompositeLayout());
 		composite.setLayoutData(getTableCompositeLayoutData());
 		return composite;
 	}

 	protected GridData getSectionLayoutData() {
 		return new GridData(SWT.FILL, SWT.FILL, false, false);
 	}

 	protected GridData getTableCompositeLayoutData() {
 		return new GridData(SWT.FILL, SWT.FILL, false, false);
 	}

 	protected GridLayout getSectionCompositeLayout() {
 		GridLayout layout = new GridLayout();
 		layout.numColumns = 2;
 		layout.marginHeight = 5;
 		layout.marginWidth = 10;
 		layout.verticalSpacing = 5;
 		layout.horizontalSpacing = 15;
 		return layout;
 	}

 	protected Table createTable(Composite composite) {
 		Table table = new Table(composite, SWT.BORDER | SWT.FULL_SELECTION
 				| SWT.V_SCROLL | SWT.SINGLE);
 		if (isHeaderVisible()) {
 			table.setHeaderVisible(true);
 		}

 		GridData data = new GridData(SWT.FILL, SWT.FILL, false, false);
 		data.heightHint = 60;
 		data.widthHint = 400;
 		table.setLayoutData(data);

 		TableLayout tableLayout = new TableLayout();
 		table.setLayout(tableLayout);

 		for (int i = 0; i < getTableColumnNames().length; i++) {
 			tableLayout.addColumnData(new ColumnWeightData(35));
 			TableColumn tableColumn = new TableColumn(table, SWT.NONE);
 			tableColumn.setText(getTableColumnNames()[i]);
 		}

 		return table;
 	}

 	protected Composite createButtonComposite(Composite parent) {
 		Composite buttonComp = new Composite(parent, SWT.NONE);
 		GridLayout layout = new GridLayout();
 		layout.horizontalSpacing = 2;
 		layout.verticalSpacing = 2;
 		layout.marginWidth = 0;
 		layout.marginHeight = 0;
 		layout.numColumns = 1;
 		buttonComp.setLayout(layout);
 		buttonComp.setBackground(toolkit.getColors().getBackground());
 		buttonComp.setLayoutData(new GridData(SWT.FILL, SWT.FILL, false, false));
 		return buttonComp;
 	}

 	protected void createRemoveButton(FormToolkit toolkit, Composite buttonComp) {
 		removeButton = toolkit.createButton(buttonComp, CommonMessages.remove, SWT.NONE);
 		removeButton.addSelectionListener(new SelectionAdapter() {
 			public void widgetSelected(SelectionEvent e) {
 				int[] selectedIndices = table.getSelectionIndices();
 				for (int i = 0; i < selectedIndices.length; i++) {
 					TableItem tableItem = table.getItem(selectedIndices[i]);
 					EObject type = (EObject) (tableItem.getData());
 					table.remove(selectedIndices[i]);
 					EcoreUtil.remove(type);
 					markDirty();
 				}
 			}
 		});
 		removeButton.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false));
 	}

 	protected void createAddButton(FormToolkit toolkit, Composite buttonComp) {
 		addButton = toolkit.createButton(buttonComp, CommonMessages.add, SWT.NONE);

 		addButton.addSelectionListener(new SelectionAdapter() {
 			public void widgetSelected(SelectionEvent e) {
 				Wizard wizard = getWizard();
 				if (wizard != null) {
 					WizardDialog dialog = new WizardDialog(Display.getCurrent().getActiveShell(), wizard);

 					dialog.open();

 					if (dialog.getReturnCode() == Dialog.OK) {
 						markDirty();
 					}
 				}
 			}
 		});

 		addButton.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false));
 	}

 	protected void createEditButton(FormToolkit toolkit, Composite buttonComp) {
 		editButton = toolkit.createButton(buttonComp, CommonMessages.edit, SWT.NONE);

 		editButton.addSelectionListener(new SelectionAdapter() {
 			public void widgetSelected(SelectionEvent e) {
 				Object o = ((StructuredSelection) getTableViewer().getSelection()).getFirstElement();
 				if (o != null) {
 					Wizard wizard = getWizard();
 					if (wizard != null) {
 						if (wizard instanceof AbstractTableWizard) {
 							((AbstractTableWizard) wizard).setEObject((EObject) o);
 						}
 						WizardDialog dialog = new WizardDialog(Display.getCurrent().getActiveShell(), wizard);
 						dialog.open();
 						if (dialog.getReturnCode() == Dialog.OK) {
 							markDirty();
 							// TODO notify listeners
 						}
 					}
 				}
 			}
 		});

 		editButton.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false));
 	}

 	public TableViewer getTableViewer() {
 		return tableViewer;
 	}

 	protected boolean isHeaderVisible() {
 		return true;
 	}

 	protected Table getTable() {
 		return table;
 	}

 	public String[] getTableColumnNames() {
 		return COLUMN_NAMES;
 	}

 	abstract public String getTitle();

 	abstract public String getDescription();

 	abstract public Wizard getWizard();

 	abstract public EReference getEReference();

 	abstract public EClass getTableEntryObjectType();

 	abstract public AdapterFactory getAdapterFactory();

 }
	/*
	* Licensed to the Apache Software Foundation (ASF) under one or more
	* contributor license agreements. See the NOTICE file distributed with
	* this work for additional information regarding copyright ownership.
	* The ASF licenses this file to You under the Apache License, Version 2.0
	* (the "License"); you may not use this file except in compliance with
	* the License. You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/
	package org.apache.geronimo.st.ui.sections;

	import org.apache.geronimo.st.ui.CommonMessages;
	import org.apache.geronimo.st.ui.wizards.AbstractTableWizard;
	import org.eclipse.emf.common.notify.AdapterFactory;
	import org.eclipse.emf.ecore.EClass;
	import org.eclipse.emf.ecore.EObject;
	import org.eclipse.emf.ecore.EReference;
	import org.eclipse.emf.ecore.util.EcoreUtil;
	import org.eclipse.emf.edit.ui.provider.AdapterFactoryContentProvider;
	import org.eclipse.emf.edit.ui.provider.AdapterFactoryLabelProvider;
	import org.eclipse.jface.dialogs.Dialog;
	import org.eclipse.jface.viewers.ColumnWeightData;
	import org.eclipse.jface.viewers.StructuredSelection;
	import org.eclipse.jface.viewers.TableLayout;
	import org.eclipse.jface.viewers.TableViewer;
	import org.eclipse.jface.viewers.Viewer;
	import org.eclipse.jface.viewers.ViewerFilter;
	import org.eclipse.jface.wizard.Wizard;
	import org.eclipse.jface.wizard.WizardDialog;
	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.layout.GridLayout;
	import org.eclipse.swt.widgets.Button;
	import org.eclipse.swt.widgets.Composite;
	import org.eclipse.swt.widgets.Display;
	import org.eclipse.swt.widgets.Table;
	import org.eclipse.swt.widgets.TableColumn;
	import org.eclipse.swt.widgets.TableItem;
	import org.eclipse.ui.forms.widgets.FormToolkit;
	import org.eclipse.ui.forms.widgets.Section;

	/**
	* @version $Rev$ $Date$
	*/
	public abstract class AbstractTableSection extends AbstractSectionPart {

	private static final String[] COLUMN_NAMES = new String[] {};

	private Table table;

	private TableViewer tableViewer;

	protected Button addButton;

	protected Button editButton;

	protected Button removeButton;

	public AbstractTableSection(Section section) {
	super(section);
	}

	/**
	* @param plan
	* @param parent
	* @param toolkit
	* @param style
	*
	* Subclasses should call createClient() in constructor
	*/
	public AbstractTableSection(EObject plan, Composite parent,
	FormToolkit toolkit, int style) {
	super(parent, toolkit, style, plan);
	}

	public void createClient() {

	if (getTableEntryObjectType() == null)
	throw new NullPointerException();

	getSection().setText(getTitle());
	getSection().setDescription(getDescription());
	getSection().setLayoutData(getSectionLayoutData());
	Composite composite = createTableComposite(getSection());
	getSection().setClient(composite);
	table = createTable(composite);

	tableViewer = new TableViewer(getTable());
	tableViewer.setContentProvider(new AdapterFactoryContentProvider(getAdapterFactory()));
	tableViewer.setLabelProvider(new AdapterFactoryLabelProvider(getAdapterFactory()));
	tableViewer.setInput(getInput());

	tableViewer.addFilter(new ViewerFilter() {
	public boolean select(Viewer viewer, Object parentElement,
	Object element) {
	return AbstractTableSection.this.filter(viewer, parentElement, element);
	}
	});

	if (getTableColumnNames().length > 0) {
	tableViewer.setColumnProperties(getTableColumnNames());
	}

	Composite buttonComp = createButtonComposite(composite);
	createAddButton(toolkit, buttonComp);
	createRemoveButton(toolkit, buttonComp);
	createEditButton(toolkit, buttonComp);

	}

	public Object getInput() {
	return getPlan();
	}

	protected boolean filter(Viewer viewer, Object parentElement, Object element) {
	return getTableEntryObjectType().isInstance(element);
	}

	protected Composite createTableComposite(Composite parent) {
	Composite composite = toolkit.createComposite(parent);
	composite.setLayout(getSectionCompositeLayout());
	composite.setLayoutData(getTableCompositeLayoutData());
	return composite;
	}

	protected GridData getSectionLayoutData() {
	return new GridData(SWT.FILL, SWT.FILL, false, false);
	}

	protected GridData getTableCompositeLayoutData() {
	return new GridData(SWT.FILL, SWT.FILL, false, false);
	}

	protected GridLayout getSectionCompositeLayout() {
	GridLayout layout = new GridLayout();
	layout.numColumns = 2;
	layout.marginHeight = 5;
	layout.marginWidth = 10;
	layout.verticalSpacing = 5;
	layout.horizontalSpacing = 15;
	return layout;
	}

	protected Table createTable(Composite composite) {
	Table table = new Table(composite, SWT.BORDER \| SWT.FULL_SELECTION
	\| SWT.V_SCROLL \| SWT.SINGLE);
	if (isHeaderVisible()) {
	table.setHeaderVisible(true);
	}

	GridData data = new GridData(SWT.FILL, SWT.FILL, false, false);
	data.heightHint = 60;
	data.widthHint = 400;
	table.setLayoutData(data);

	TableLayout tableLayout = new TableLayout();
	table.setLayout(tableLayout);

	for (int i = 0; i < getTableColumnNames().length; i++) {
	tableLayout.addColumnData(new ColumnWeightData(35));
	TableColumn tableColumn = new TableColumn(table, SWT.NONE);
	tableColumn.setText(getTableColumnNames()[i]);
	}

	return table;
	}

	protected Composite createButtonComposite(Composite parent) {
	Composite buttonComp = new Composite(parent, SWT.NONE);
	GridLayout layout = new GridLayout();
	layout.horizontalSpacing = 2;
	layout.verticalSpacing = 2;
	layout.marginWidth = 0;
	layout.marginHeight = 0;
	layout.numColumns = 1;
	buttonComp.setLayout(layout);
	buttonComp.setBackground(toolkit.getColors().getBackground());
	buttonComp.setLayoutData(new GridData(SWT.FILL, SWT.FILL, false, false));
	return buttonComp;
	}

	protected void createRemoveButton(FormToolkit toolkit, Composite buttonComp) {
	removeButton = toolkit.createButton(buttonComp, CommonMessages.remove, SWT.NONE);
	removeButton.addSelectionListener(new SelectionAdapter() {
	public void widgetSelected(SelectionEvent e) {
	int[] selectedIndices = table.getSelectionIndices();
	for (int i = 0; i < selectedIndices.length; i++) {
	TableItem tableItem = table.getItem(selectedIndices[i]);
	EObject type = (EObject) (tableItem.getData());
	table.remove(selectedIndices[i]);
	EcoreUtil.remove(type);
	markDirty();
	}
	}
	});
	removeButton.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false));
	}

	protected void createAddButton(FormToolkit toolkit, Composite buttonComp) {
	addButton = toolkit.createButton(buttonComp, CommonMessages.add, SWT.NONE);

	addButton.addSelectionListener(new SelectionAdapter() {
	public void widgetSelected(SelectionEvent e) {
	Wizard wizard = getWizard();
	if (wizard != null) {
	WizardDialog dialog = new WizardDialog(Display.getCurrent().getActiveShell(), wizard);

	dialog.open();

	if (dialog.getReturnCode() == Dialog.OK) {
	markDirty();
	}
	}
	}
	});

	addButton.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false));
	}

	protected void createEditButton(FormToolkit toolkit, Composite buttonComp) {
	editButton = toolkit.createButton(buttonComp, CommonMessages.edit, SWT.NONE);

	editButton.addSelectionListener(new SelectionAdapter() {
	public void widgetSelected(SelectionEvent e) {
	Object o = ((StructuredSelection) getTableViewer().getSelection()).getFirstElement();
	if (o != null) {
	Wizard wizard = getWizard();
	if (wizard != null) {
	if (wizard instanceof AbstractTableWizard) {
	((AbstractTableWizard) wizard).setEObject((EObject) o);
	}
	WizardDialog dialog = new WizardDialog(Display.getCurrent().getActiveShell(), wizard);
	dialog.open();
	if (dialog.getReturnCode() == Dialog.OK) {
	markDirty();
	// TODO notify listeners
	}
	}
	}
	}
	});

	editButton.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false));
	}

	public TableViewer getTableViewer() {
	return tableViewer;
	}

	protected boolean isHeaderVisible() {
	return true;
	}

	protected Table getTable() {
	return table;
	}

	public String[] getTableColumnNames() {
	return COLUMN_NAMES;
	}

	abstract public String getTitle();

	abstract public String getDescription();

	abstract public Wizard getWizard();

	abstract public EReference getEReference();

	abstract public EClass getTableEntryObjectType();

	abstract public AdapterFactory getAdapterFactory();

	}