BaseLabelProvider.java

package org.openlca.app.viewers;

import org.eclipse.jface.viewers.ColumnLabelProvider;
import org.eclipse.swt.graphics.Image;
import org.openlca.app.navigation.Group;
import org.openlca.app.navigation.GroupType;
import org.openlca.app.rcp.images.Images;
import org.openlca.app.util.Labels;
import org.openlca.core.model.AllocationMethod;
import org.openlca.core.model.Exchange;
import org.openlca.core.model.Flow;
import org.openlca.core.model.FlowPropertyFactor;
import org.openlca.core.model.FlowPropertyType;
import org.openlca.core.model.FlowType;
import org.openlca.core.model.ImpactCategory;
import org.openlca.core.model.Location;
import org.openlca.core.model.Process;
import org.openlca.core.model.ProcessType;
import org.openlca.core.model.RootEntity;
import org.openlca.core.model.UncertaintyType;
import org.openlca.core.model.Unit;
import org.openlca.core.model.descriptors.BaseDescriptor;
import org.openlca.util.Strings;

public class BaseLabelProvider extends ColumnLabelProvider {

	@Override
	public Image getImage(Object element) {
		if (element instanceof RootEntity)
			return Images.get((RootEntity) element);
		if (element instanceof BaseDescriptor)
			return Images.get((BaseDescriptor) element);
		if (element instanceof Exchange)
			return Images.get(((Exchange) element).getFlow());
		if (element instanceof FlowType)
			return Images.get((FlowType) element);
		if (element instanceof ProcessType)
			return Images.get((ProcessType) element);
		if (element instanceof Group)
			return Images.get((Group) element);
		if (element instanceof GroupType)
			return Images.get((GroupType) element);
		return null;
	}

	@Override
	public String getText(Object element) {
		if (element instanceof BaseDescriptor)
			return getModelLabel((BaseDescriptor) element);
		if (element instanceof RootEntity)
			return getModelLabel((RootEntity) element);
		if (element instanceof ImpactCategory)
			return ((ImpactCategory) element).getName();
		if (element instanceof Exchange)
			return getModelLabel(((Exchange) element).getFlow());
		if (element instanceof FlowPropertyFactor)
			return getModelLabel(((FlowPropertyFactor) element).getFlowProperty());
		if (element instanceof Unit)
			return ((Unit) element).getName();
		if (element instanceof Location)
			return ((Location) element).getName();
		if (element instanceof BaseDescriptor)
			return Labels.getDisplayName((BaseDescriptor) element);
		if (element instanceof Enum<?>)
			return getEnumText(element);
		if (element != null)
			return element.toString();
		return null;
	}

	private String getEnumText(Object enumValue) {
		if (enumValue instanceof AllocationMethod)
			return Labels.allocationMethod((AllocationMethod) enumValue);
		if (enumValue instanceof FlowPropertyType)
			return Labels.flowPropertyType((FlowPropertyType) enumValue);
		if (enumValue instanceof FlowType)
			return Labels.flowType((FlowType) enumValue);
		if (enumValue instanceof ProcessType)
			return Labels.processType((ProcessType) enumValue);
		if (enumValue instanceof UncertaintyType)
			return Labels.uncertaintyType((UncertaintyType) enumValue);
		if (enumValue != null)
			return enumValue.toString();
		return null;
	}

	protected String getModelLabel(RootEntity o) {
		if (o == null)
			return "";
		String label = Strings.cut(o.getName(), 75);
		Location location = null;
		if (o instanceof Flow)
			location = ((Flow) o).getLocation();
		else if (o instanceof Process)
			location = ((Process) o).getLocation();
		if (location != null && location.getCode() != null)
			label += " (" + location.getCode() + ")";
		return label;
	}

	protected String getModelLabel(BaseDescriptor d) {
		if (d == null)
			return null;
		return Strings.cut(Labels.getDisplayName(d), 75);
	}

	@Override
	public String getToolTipText(Object element) {
		if (element instanceof BaseDescriptor)
			return Labels.getDisplayInfoText((BaseDescriptor) element);
		return null;
	}

}