plugins/diagram/org.eclipse.papyrus.sysml.diagram.internalblock/src/org/eclipse/papyrus/sysml/diagram/internalblock/handler/ShowHideRelatedContentsAction.java

/*****************************************************************************
 * Copyright (c) 2011 CEA LIST.
 *
 * 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:
 *
 *		CEA LIST - Initial API and implementation
 *
 *****************************************************************************/
package org.eclipse.papyrus.sysml.diagram.internalblock.handler;

import org.eclipse.gmf.runtime.diagram.ui.editparts.IGraphicalEditPart;
import org.eclipse.gmf.runtime.notation.View;
import org.eclipse.papyrus.sysml.diagram.internalblock.messages.Messages;
import org.eclipse.papyrus.uml.diagram.common.actions.ShowHideContentsAction;
import org.eclipse.papyrus.uml.diagram.common.editpolicies.ShowHideRelatedContentsEditPolicy;
import org.eclipse.uml2.uml.Classifier;
import org.eclipse.uml2.uml.Property;

public class ShowHideRelatedContentsAction extends ShowHideContentsAction {

	/** Constructor. */
	public ShowHideRelatedContentsAction() {
		super(Messages.ShowHideRelatedContentsAction_Title, Messages.ShowHideRelatedContentsAction_Message, ShowHideRelatedContentsEditPolicy.SHOW_HIDE_RELATED_CONTENTS_POLICY);
	}

	/**
	 * @see org.eclipse.papyrus.diagram.common.actions.AbstractShowHideAction#initAction()
	 */
	@Override
	protected void initAction() {
		super.initAction();

		for (IGraphicalEditPart current : this.selectedElements) {
			// the selected elements which aren't Classifier are ignored
			if (((View) current.getModel()).getElement() instanceof Property) {
				Property property = (Property) ((View) current.getModel()).getElement();
				if (property.getType() instanceof Classifier) {
					this.representations.add(new RootEditPartRepresentation(current, (Classifier) property.getType()));
				}
			}
		}
		// setEditorLabelProvider(new EditorLabelProvider());
		// this.representations = new ArrayList<AbstractShowHideAction.EditPartRepresentation>();
		//
		// for(EditPart current : this.selectedElements) {
		// //the selected elements which aren't Classifier are ignored
		// EObject element = ((View)current.getModel()).getElement();
		// if(element instanceof Property) {
		// Type type = ((Property)element).getType();
		// if(type instanceof Classifier) {
		// this.representations.add(new CustomEditPartRepresentation(current, (Classifier)type));
		// }
		// }
		// }
		// this.setEditorLabelProvider(new CustomEditorLabelProvider());
		// this.setContentProvider(new CustomContentProvider());

	}

	// protected class CustomContentProvider extends ShowHideContentsAction.ContentProvider {
	//
	// /**
	// * @see org.eclipse.jface.viewers.ITreeContentProvider#getParent(java.lang.Object)
	// *
	// * @param element
	// * @return
	// */
	// @Override
	// public Object getParent(Object element) {
	// if(!(element instanceof EditPartRepresentation)) {
	// EditPartRepresentation rep = findEditPartRepresentation(element);
	// if(rep != null) {
	// //element can be owned by the class (rep), or by a superclass (ClassifierRepresentation)
	// List<ClassifierRepresentation> classes = ((CustomEditPartRepresentation)rep).getSuperClasses();
	// for(ClassifierRepresentation classifierRepresentation : classes) {
	// if(classifierRepresentation.getRepresentedClassifier().getOwnedMembers().contains(element)) {
	// return classifierRepresentation;
	// }
	// }
	// }
	// return rep;
	// }
	// return null;
	// }
	// }
}