cps/tests/org.eclipse.viatra.examples.cps.m2t.proto.distributed.tests/src/org/eclipse/viatra/examples/cps/m2t/proto/distributed/tests/GeneratorTest.java

/*******************************************************************************
 * Copyright (c) 2014-2016 Akos Horvath, Abel Hegedus, Marton Bur, Zoltan Ujhelyi, Zsolt Kovari, IncQuery Labs Ltd.
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-v20.html.
 * 
 * SPDX-License-Identifier: EPL-2.0
 *******************************************************************************/
package org.eclipse.viatra.examples.cps.m2t.proto.distributed.tests;

import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;

import java.util.Map;
import java.util.concurrent.TimeUnit;

import org.apache.log4j.Level;
import org.apache.log4j.Logger;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import org.eclipse.emf.ecore.xmi.impl.XMIResourceFactoryImpl;
import org.eclipse.viatra.examples.cps.deployment.Deployment;
import org.eclipse.viatra.examples.cps.deployment.DeploymentApplication;
import org.eclipse.viatra.examples.cps.deployment.DeploymentHost;
import org.eclipse.viatra.examples.cps.deployment.DeploymentPackage;
import org.eclipse.viatra.examples.cps.deployment.common.DeploymentQueries;
import org.eclipse.viatra.examples.cps.tests.util.CPSTestBase;
import org.eclipse.viatra.examples.cps.xform.m2t.api.ICPSGenerator;
import org.eclipse.viatra.examples.cps.xform.m2t.distributed.CodeGenerator;
import org.eclipse.viatra.examples.cps.xform.m2t.exceptions.CPSGeneratorException;
import org.eclipse.viatra.query.runtime.api.AdvancedViatraQueryEngine;
import org.eclipse.viatra.query.runtime.api.ViatraQueryEngine;
import org.eclipse.viatra.query.runtime.emf.EMFScope;
import org.junit.Ignore;
import org.junit.Test;

import com.google.common.base.Stopwatch;

public class GeneratorTest extends CPSTestBase {

	private static Logger logger = Logger.getLogger("cps.proto.generator");
	
//	@Ignore
	@Test
	public void testSmall() {
		logger.info("Start Generating...");
		
		Deployment model = loadModel("C:\\Eclipses\\CPSDemonstrator\\git\\incquery-examples-cps\\models\\org.eclipse.viatra.examples.cps.instances\\example.deployment");
		
		generateCode(model);
		
		assertTrue(true);
	}
	
	@Ignore
	@Test
	public void test64ClientServer() {
		logger.info("Start Generating...");
		
		Logger.getLogger("cps.proto.generator").setLevel(Level.OFF);
		
		Deployment model = loadModel("C:\\Eclipses\\CPSDemonstrator\\git\\incquery-examples-cps\\models\\org.eclipse.viatra.examples.cps.instances\\ClientServerScenario\\BatchIncQuery64_1234495999981128.deployment");
		
		Stopwatch fullTime = Stopwatch.createStarted();
		generateCode(model);
		fullTime.stop();
		Logger.getLogger("cps.proto.generator").setLevel(Level.INFO);
		logger.info("Full Time: " + fullTime.elapsed(TimeUnit.MILLISECONDS) + " ms");
		
		assertTrue(true);
	}

	private void generateCode(Deployment model) {
		ViatraQueryEngine engine = AdvancedViatraQueryEngine.on(new EMFScope(model));
		DeploymentQueries.instance().prepare(engine);
		
		ICPSGenerator generator = new CodeGenerator("org.alma", engine, false);
		try{
			for(DeploymentHost host : model.getHosts()){
				logger.info(generator.generateHostCode(host));
				logger.info("********* A P P L I C A T I O N S *********");
				for(DeploymentApplication app : host.getApplications()){
					logger.info(generator.generateApplicationCode(app));
						logger.info("********* B E H A V I O R *********");
						logger.info(generator.generateBehaviorCode(app.getBehavior()));
						logger.info("");
					logger.info("");
					logger.info("*************************************");
					logger.info("*******************************************");
					
				}
				logger.info("");
			}
		}catch(CPSGeneratorException e){
			e.printStackTrace();
			fail();
		}
	}

	public Deployment loadModel(String filePath) {
	    // Initialize the model
		DeploymentPackage.eINSTANCE.eClass();
	    
	    // Register the XMI resource factory for the .website extension

	    Resource.Factory.Registry reg = Resource.Factory.Registry.INSTANCE;
	    Map<String, Object> m = reg.getExtensionToFactoryMap();
	    m.put("website", new XMIResourceFactoryImpl());

	    // Obtain a new resource set
	    ResourceSet resSet = new ResourceSetImpl();

	    // Get the resource
	    Resource resource = resSet.getResource(URI.createFileURI(filePath), true);
	    // Get the first model element and cast it to the right type, in my
	    // example everything is hierarchical included in this first node
	    if(!resource.getContents().isEmpty() && resource.getContents().get(0) instanceof Deployment){
	    	return (Deployment) resource.getContents().get(0);
	    }
	    return null;
	  }
}