plugins/components/org.eclipse.papyrus.designer.components.transformation.ui/src/org/eclipse/papyrus/designer/components/transformation/ui/dialogs/PortLabelProvider.java

/*****************************************************************************
 * Copyright (c) 2013 CEA LIST.
 *
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License 2.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *  Ansgar Radermacher  ansgar.radermacher@cea.fr
 *
 *****************************************************************************/

package org.eclipse.papyrus.designer.components.transformation.ui.dialogs;

import org.eclipse.jface.viewers.ITableLabelProvider;
import org.eclipse.jface.viewers.LabelProvider;
import org.eclipse.papyrus.designer.languages.common.base.StringConstants;
import org.eclipse.swt.graphics.Image;
import org.eclipse.uml2.uml.Port;

/**
 * A label provider for either UML or FCM ports
 *
 * @author ansgar
 *
 */
public class PortLabelProvider extends LabelProvider implements ITableLabelProvider {

	/**
	 * small helper function: simply retrieval of port (element might be either a port or an FCM port)
	 *
	 * @param element
	 * @return
	 */
	private Port getUMLport(Object element) {
		if (element instanceof org.eclipse.papyrus.designer.components.FCM.Port) {
			org.eclipse.papyrus.designer.components.FCM.Port port = (org.eclipse.papyrus.designer.components.FCM.Port) element;
			if (port != null) {
				return port.getBase_Port();
			}
		} else if (element instanceof Port) {
			return (Port) element;
		}
		return null;
	}

	@Override
	public String getText(Object element) {
		Port umlPort = getUMLport(element);
		if (umlPort != null) {
			return umlPort.getName();
		}
		return StringConstants.EMPTY;
	}

	/*
	 * (non-Javadoc)
	 *
	 * @see org.eclipse.jface.viewers.ITableLabelProvider#getColumnText(java.lang.Object, int)
	 */
	@Override
	public String getColumnText(Object element, int columnIndex) {
		if (columnIndex == 0) {
			return getText(element);
		} else if (columnIndex == 1) {
			Port umlPort = getUMLport(element);
			if (umlPort != null) {
				return umlPort.getType().getName();
			}
		}
		return StringConstants.EMPTY;
	}

	@Override
	public Image getColumnImage(Object element, int columnIndex) {
		// TODO Auto-generated method stub
		return null;
	}
};