load_generator/commons/plugins/org.eclipse.app4mc.slg.commons.m2t/src/org/eclipse/app4mc/slg/commons/m2t/transformers/sw/TicksTransformer.java

/**
 ********************************************************************************
 * Copyright (c) 2020-2021 Robert Bosch GmbH.
 * 
 * This program and the accompanying materials are made
 * available under the terms of the Eclipse Public License 2.0
 * which is available at https://www.eclipse.org/legal/epl-2.0/
 * 
 * SPDX-License-Identifier: EPL-2.0
 * 
 * Contributors:
 *     Robert Bosch GmbH - initial API and implementation
 ********************************************************************************
 */

package org.eclipse.app4mc.slg.commons.m2t.transformers.sw;

import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;

import org.eclipse.app4mc.amalthea.model.IDiscreteValueDeviation;
import org.eclipse.app4mc.amalthea.model.ProcessingUnitDefinition;
import org.eclipse.app4mc.amalthea.model.Ticks;
import org.eclipse.app4mc.slg.commons.m2t.generators.TicksGenerator;
import org.eclipse.app4mc.slg.commons.m2t.transformers.SLGBaseTransformer;
import org.eclipse.app4mc.slg.commons.m2t.transformers.SLGTranslationUnit;
import org.eclipse.emf.common.util.EMap;

import com.google.inject.Inject;
import com.google.inject.Singleton;

@Singleton
public class TicksTransformer extends SLGBaseTransformer {

	@Inject private TicksUtilsTransformer ticksUtilsTransformer;

	public SLGTranslationUnit transform(final Ticks ticks) {
		if (ticks.getDefault() != null) {
			final SLGTranslationUnit ticksUtilsTU = ticksUtilsTransformer.transform(ticks.getDefault());

			return createTranslationUnit(ticks, ticksUtilsTU);
		}

		return null;
	}

	private SLGTranslationUnit createTranslationUnit(Ticks ticks, SLGTranslationUnit ticksUtilsTU) {
		if ((ticksUtilsTU == null)) {
			return new SLGTranslationUnit("UNSPECIFIED TICKS");			
		}

		String basePath = ticksUtilsTU.getBasePath();
		String moduleName = ticksUtilsTU.getModuleName();
		String call = computeCall(ticks);
		return new SLGTranslationUnit(basePath, moduleName, call);
	}

	private String computeCall(Ticks ticks) {
		String parameters = TicksGenerator.getParameters(ticks.getDefault());
		String className = ticks.getDefault().eClass().getName();
		return "executeTicks_" + className +"(" + parameters + ")";
	}

	public Map<String, SLGTranslationUnit> transformAllItems(final Ticks ticks) {
		final Map<String, SLGTranslationUnit> result = new HashMap<>();
		
		final IDiscreteValueDeviation defaultTicks = ticks.getDefault();
		if ((defaultTicks != null)) {
			final SLGTranslationUnit ticksUtilsTU = ticksUtilsTransformer.transform(defaultTicks);
			result.put("default", createTranslationUnit(ticks, ticksUtilsTU));
		}

		final EMap<ProcessingUnitDefinition, IDiscreteValueDeviation> extendedTicksMap = ticks.getExtended();
		for (final Entry<ProcessingUnitDefinition, IDiscreteValueDeviation> entry : extendedTicksMap.entrySet()) {
			final ProcessingUnitDefinition puDefinition = entry.getKey();
			final IDiscreteValueDeviation ticksDeviation = entry.getValue();

			if (puDefinition != null && ticksDeviation != null) {
				final SLGTranslationUnit ticksUtilsTU = this.ticksUtilsTransformer.transform(ticksDeviation);
				result.put(puDefinition.getName(), createTranslationUnit(ticks, ticksUtilsTU));
			}

		}
		return result;
	}

}