archive/plugins/org.eclipse.app4mc.multicore.openmapping/src/org/eclipse/app4mc/multicore/openmapping/model/AmaltheaModelBuilder.java

/*******************************************************************************
 * Copyright (c) 2017, 2018 Dortmund University of Applied Sciences and Arts and others.
 *
 * 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:
 *   Dortmund University of Applied Sciences and Arts - initial API and implementation
 *******************************************************************************/

package org.eclipse.app4mc.multicore.openmapping.model;

import java.util.HashMap;
import java.util.Iterator;

import org.eclipse.app4mc.amalthea.model.Amalthea;
import org.eclipse.app4mc.amalthea.model.AmaltheaFactory;
import org.eclipse.app4mc.amalthea.model.MappingModel;
import org.eclipse.app4mc.amalthea.model.OSModel;
import org.eclipse.app4mc.amalthea.model.OperatingSystem;
import org.eclipse.app4mc.amalthea.model.SchedulerAllocation;
import org.eclipse.app4mc.amalthea.model.TaskAllocation;

public class AmaltheaModelBuilder {
	private final OMMapping mapping;
	private Amalthea centralModel;

	public AmaltheaModelBuilder(final OMMapping mapping) {
		this.mapping = mapping;
	}

	public Amalthea getAmaltheaModel() {
		assert null != this.mapping;
		// Build the model if it is not present or has been updated
		if (null == this.centralModel) {
			buildAmaltheaModel();
		}

		return this.centralModel;
	}

	private void buildAmaltheaModel() {
		// Create an empty MappingModel
		final MappingModel mappingModel = AmaltheaFactory.eINSTANCE.createMappingModel();
		// TODO Ideally, this should be handled by an OSModel Builder
		// Create an OSModel with a parent element "OS"
		final OSModel osModel = AmaltheaFactory.eINSTANCE.createOSModel();
		final OperatingSystem os = AmaltheaFactory.eINSTANCE.createOperatingSystem();
		os.setName("OS");
		osModel.getOperatingSystems().add(os);

		// Create the schedulers and the allocations (core<->scheduler && task<->scheduler)
		final HashMap<OMCore, OMScheduler> coreSchedulerMap = new HashMap<>();
		final Iterator<OMAllocation> itAllocations = this.mapping.getAllocationList().iterator();
		while (itAllocations.hasNext()) {
			final OMAllocation allocation = itAllocations.next();
			// TODO Allocation Attributes are still missing!
			final OMCore core = allocation.getCore();
			final OMTask task = allocation.getTask();
			OMScheduler scheduler;

			// Check if scheduler has been created and store / create and store it
			if (coreSchedulerMap.containsKey(core)) {
				scheduler = coreSchedulerMap.get(core);
			}
			else {
				// Create scheduler
				scheduler = new OMScheduler(core);
				// Add scheduler to os model
				os.getTaskSchedulers().add(scheduler.getSchedulerRef());
				// Create entries in mapping model
				final SchedulerAllocation coreAlloc = AmaltheaFactory.eINSTANCE.createSchedulerAllocation();
				coreAlloc.getResponsibility().add(core.getCoreRef());
				coreAlloc.setScheduler(scheduler.getSchedulerRef());
				mappingModel.getSchedulerAllocation().add(coreAlloc);
				// Remember Scheduler in map
				coreSchedulerMap.put(core, scheduler);
			}
			// Create Task<->Scheduler allocation entries in mapping model
			final TaskAllocation taskAlloc = AmaltheaFactory.eINSTANCE.createTaskAllocation();
			taskAlloc.setTask(task.getTaskRef());
			taskAlloc.setScheduler(scheduler.getSchedulerRef());
			mappingModel.getTaskAllocation().add(taskAlloc);
		}
		// Create AMALTHEA (Central) model and store results
		this.centralModel = AmaltheaFactory.eINSTANCE.createAmalthea();
		this.centralModel.setOsModel(osModel);
		this.centralModel.setMappingModel(mappingModel);
	}
}