ElementFilteredListChooser.java

/*
 * Copyright (c) 2008 Borland Software Corporation
 * 
 * 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:
 *    Tatiana Fesenko (Borland) - initial API and implementation
 */
package org.eclipse.uml2.diagram.common.sheet.chooser;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.LinkedList;
import java.util.List;

import org.eclipse.emf.common.notify.AdapterFactory;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EStructuralFeature;
import org.eclipse.emf.edit.provider.ItemPropertyDescriptor;
import org.eclipse.emf.edit.ui.provider.AdapterFactoryContentProvider;
import org.eclipse.emf.edit.ui.provider.AdapterFactoryLabelProvider;
import org.eclipse.emf.transaction.TransactionalEditingDomain;
import org.eclipse.jface.viewers.IDoubleClickListener;
import org.eclipse.jface.viewers.ILabelProvider;
import org.eclipse.jface.viewers.ISelectionChangedListener;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Control;
import org.eclipse.uml2.diagram.common.part.ModelElementsContentHelper;

public class ElementFilteredListChooser implements ElementChooserPage {

	private FilteredListControl myFilteredList;

	private final Validator myValidator;

	private Object[] myListElements;

	private final AdapterFactoryContentProvider myAdapterFctoryContentProvier;

	private final AdapterFactory myItemProvidersAdapterFactory;

	private final TransactionalEditingDomain myEditingDomain;

	private final EObject mySourceObject;

	private final EStructuralFeature myFeature;

	public ElementFilteredListChooser(AdapterFactory itemProvidersAdapterFactory, EObject sourceObject, EStructuralFeature feature, Validator validator, TransactionalEditingDomain editingDomain) {
		myItemProvidersAdapterFactory = itemProvidersAdapterFactory;
		mySourceObject = sourceObject;
		myFeature = feature;
		myValidator = validator;
		myAdapterFctoryContentProvier = new AdapterFactoryContentProvider(itemProvidersAdapterFactory);
		myEditingDomain = editingDomain;
	}

	public Control createControl(Composite parent) {
		myFilteredList = new FilteredListControl(parent, new AdapterFactoryLabelProvider(myItemProvidersAdapterFactory));
		Object featureValue = mySourceObject.eGet(myFeature);
		if (featureValue != null) {
			myFilteredList.setInitialSelections(new Object[] { featureValue });
		}
		ILabelProvider labelProvider = new SimpleNamedElementLabelProvider(new AdapterFactoryLabelProvider(myItemProvidersAdapterFactory));
		myFilteredList.setFilterMatcher(new ConfigurableFilterMatcher(labelProvider));
		myListElements = collectElements(mySourceObject.eResource().getResourceSet());
		myFilteredList.setListElements(myListElements);
		return myFilteredList;
	}

	public final List<?> getSelection() {
		Object[] result = myFilteredList.getSelectedElements();
		return Arrays.asList(result);
	}

	public final void setSelection(List<?> selection) {
		if (selection == null || selection.isEmpty()) {
			myFilteredList.setSelection(null);
		} else {
			addMissingElements(selection);
			myFilteredList.setSelection(selection.toArray());
		}
	}

	private void addMissingElements(List<?> selection) {
		List<Object> els = Arrays.asList(myListElements);
		if (els.containsAll(selection)) {
			return;
		}
		List<Object> elementsToAdd = new ArrayList<Object>(selection.size());
		for (Object next : selection) {
			next = myValidator.validate(next);
			if (next != null && !els.contains(next)) {
				elementsToAdd.add(next);
			}
		}
		addElements(elementsToAdd.toArray(new Object[elementsToAdd.size()]));
	}

	private void addElements(Object[] elements) {
		myListElements = concatArrays(myListElements, elements);
		myFilteredList.setListElements(myListElements);
	}

	private Object[] concatArrays(Object[] first, Object[] second) {
		Object[] result = new Object[first.length + second.length];
		System.arraycopy(second, 0, result, 0, second.length);
		System.arraycopy(first, 0, result, second.length, first.length);
		return result;
	}

	protected EObject[] collectElements(Object inputElement) {
		List<EObject> result = new LinkedList<EObject>();
		for (Object next : getAllChildren(inputElement)) {
			if (next instanceof EObject) {
				EObject transformed = myValidator.validate((EObject) next);
				if (transformed != null) {
					result.add(transformed);
				}
			}
		}
		return result.toArray(new EObject[result.size()]);
	}

	protected Collection<EObject> getAllChildren(Object parentElement) {
		return ItemPropertyDescriptor.getReachableObjectsOfType(mySourceObject, myFeature.getEType());
		//		List<Object> result = new LinkedList<Object>();
		//		for (Object next : ModelElementsContentHelper.getChildren(parentElement, myAdapterFctoryContentProvier, myEditingDomain.getResourceSet())) {
		//			result.add(next);
		//			result.addAll(getAllChildren(next));
		//		}
		//		return result;
	}

	public void addDoubleClickListener(IDoubleClickListener l) {
		myFilteredList.addDoubleClickListener(l);
	}

	public void addSelectionListener(ISelectionChangedListener l) {
		myFilteredList.addSelectionListener(l);
	}

}