aas/plugins/modeling/org.eclipse.papyrus.aas.table.configuration/src/org/eclipse/papyrus/aas/tables/configurations/manager/axis/RTSynchronizedOnFeatureAxisManager.java

/*****************************************************************************
 * Copyright (c) 2017 Christian W. Damus 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:
 *   Christian W. Damus - Initial API and implementation
 *   
 *****************************************************************************/

package org.eclipse.papyrus.aas.tables.configurations.manager.axis;

import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;

import org.eclipse.emf.ecore.EReference;
import org.eclipse.papyrus.infra.emf.nattable.manager.axis.AbstractSynchronizedOnEStructuralFeatureAxisManager;
import org.eclipse.uml2.uml.Element;

/**
 * A specialization of the synchronized-on-feature axis manager that accounts for
 * inheritance by accessing the extended contents-lists of {@link EReference}s in
 * the model elements.
 */
public class RTSynchronizedOnFeatureAxisManager extends AbstractSynchronizedOnEStructuralFeatureAxisManager {

	private boolean showExclusions;

	public RTSynchronizedOnFeatureAxisManager() {
		super();
	}

	public void setShowExclusions(boolean showExclusions) {
		if (showExclusions == this.showExclusions) {
			return;
		}

		this.showExclusions = showExclusions;

		List<Object> exclusions = getFeaturesValue().stream()
				.filter(Element.class::isInstance)
				.map(Element.class::cast)
				.collect(Collectors.toList());

		if (!exclusions.isEmpty()) {
			if (showExclusions) {
				updateManagedList(exclusions, Collections.emptyList());
			} else {
				updateManagedList(Collections.emptyList(), exclusions);
			}
		}
	}

	public boolean isShowExclusions() {
		return showExclusions;
	}



	@Override
	public boolean isAllowedContents(Object object) {
		// Filter out excluded elements when configured thus
		return (showExclusions || !((object instanceof Element)))
				&& super.isAllowedContents(object);
	}

	public void handleExclusion(Object possiblyExcluded) {
		if (!showExclusions && (possiblyExcluded instanceof Element)) {
			// Is it, in fact, just excluded?
			Element element = (Element) possiblyExcluded;
			if ((element.eResource() != null)) {
				updateManagedList(Collections.emptyList(), Collections.singletonList(element));
			}
		}
	}
}