org.eclipse.stem/modelgen/org.eclipse.stem.model.codegen/src/org/eclipse/stem/model/xtext/ExpressionMethodGenerator.java

/*******************************************************************************
 * Copyright (c) 2013 IBM Corporation and others.
 * 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:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.stem.model.xtext;

import java.util.List;

import org.eclipse.emf.codegen.ecore.genmodel.GenClass;
import org.eclipse.stem.model.codegen.STEMGenFeature;
import org.eclipse.stem.model.ctdl.generator.CTDLGenerator;
import org.eclipse.stem.model.metamodel.Transition;
import org.eclipse.xtext.generator.IGenerator;
import org.eclipse.xtext.generator.InMemoryFileSystemAccess;
import org.eclipse.xtext.resource.XtextResource;

import com.google.inject.Guice;
import com.google.inject.Injector;

public class ExpressionMethodGenerator 
{
	private Injector injector;
	
	private IGenerator generator;
	
	private GenClass modelGenClass;
	
	private GenClass labelValueGenClass;
	
	public ExpressionMethodGenerator(GenClass modelGenClass, GenClass labelValueGenClass)
	{
		this.modelGenClass = modelGenClass;
		this.labelValueGenClass = labelValueGenClass;
		init();
	}
	
	private void init()
	{
		injector = Guice.createInjector(new org.eclipse.stem.model.ctdl.CTDLRuntimeModule());
		generator = injector.getInstance(IGenerator.class);
	}
	
	private CTDLGenerator getExpressionMethodGenerator()
	{
		return (CTDLGenerator)generator;
	}

	public ExpressionMethodDefinition generateExpressionMethod(Transition transition, STEMGenFeature sourceFeature, STEMGenFeature targetFeature, List<STEMGenFeature> forIncidenceFeatures)
	{
		XtextResource r = XtextUtils.loadAndResolveXtextResourceForTransition(transition);
		if (r == null) {
			return null;
		}
		InMemoryFileSystemAccess fsa = new InMemoryFileSystemAccess();
		
		CTDLGenerator gen = getExpressionMethodGenerator();
		gen.setModelGenClass(modelGenClass);
		gen.setLabelValueGenClass(labelValueGenClass);
		
		try {
			gen.doGenerate(r, fsa);
		} catch (Throwable t0) {
			t0.printStackTrace();
		}
		
		ExpressionMethodDefinition definition = null;
		if (fsa.getTextFiles().size() > 0) {
			String methodBody = fsa.getTextFiles().values().iterator().next().toString();
			definition = new ExpressionMethodDefinition(transition, methodBody);
		}
		
		definition.setSourceFeature(sourceFeature);
		definition.setTargetFeature(targetFeature);
		definition.setForIncidenceFeatures(forIncidenceFeatures);
		
		return definition;
	}
}