bundles/common/org.eclipse.apogy.common.emf.ui/src-gen-custom/org/eclipse/apogy/common/emf/ui/impl/EOperationEParametersUnitsProviderCustomImpl.java

/*******************************************************************************
 * Copyright (c) 2018 Agence spatiale canadienne / Canadian Space Agency 
 * 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:
 *     Pierre Allard, 
 *     Regent L'Archeveque,
 *     Olivier L. Larouche - initial API and implementation
 * SPDX-License-Identifier: EPL-1.0
 *     
 *******************************************************************************/
package org.eclipse.apogy.common.emf.ui.impl;

import javax.measure.unit.Unit;

import org.eclipse.apogy.common.emf.ui.ApogyCommonEMFUIFactory;
import org.eclipse.apogy.common.emf.ui.EOperationEParametersUnitsProviderParameters;
import org.eclipse.apogy.common.emf.ui.ETypedElementToUnitsMap;
import org.eclipse.apogy.common.emf.ui.SimpleUnitsProvider;
import org.eclipse.apogy.common.emf.ui.UnitsProvider;
import org.eclipse.apogy.common.emf.ui.UnitsProviderParameters;

public class EOperationEParametersUnitsProviderCustomImpl extends EOperationEParametersUnitsProviderImpl {

	@Override
	public ETypedElementToUnitsMap getMap() {
		ETypedElementToUnitsMap map = super.getMap();

		if (map == null) {
			map = ApogyCommonEMFUIFactory.eINSTANCE.createETypedElementToUnitsMap();
			setMap(map);
		}

		return map;
	}

	public Unit<?> getProvidedUnit(UnitsProviderParameters parameters) {
		if (parameters instanceof EOperationEParametersUnitsProviderParameters) {
			UnitsProvider provider = getMap().getEntries()
					.get(((EOperationEParametersUnitsProviderParameters) parameters).getParam());

			if (provider instanceof SimpleUnitsProvider) {
				return ((SimpleUnitsProvider) provider).getProvidedUnit(null);
			}
		}
		return null;
	}

} // EOperationEParametersUnitsProviderImpl