plugins/org.eclipse.qvtd.cs2as.compiler/src/org/eclipse/qvtd/cs2as/compiler/internal/OCL2QVTiCompilerChain.java

package org.eclipse.qvtd.cs2as.compiler.internal;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.jdt.annotation.NonNull;
import org.eclipse.jdt.annotation.Nullable;
import org.eclipse.ocl.pivot.Element;
import org.eclipse.qvtd.compiler.AbstractCompilerChain;
import org.eclipse.qvtd.compiler.CompilerChain;
import org.eclipse.qvtd.pivot.qvtbase.Transformation;
import org.eclipse.qvtd.pivot.qvtbase.utilities.QVTbaseUtil;
import org.eclipse.qvtd.pivot.qvtimperative.utilities.QVTimperative;

public class OCL2QVTiCompilerChain extends AbstractCompilerChain {

	public @NonNull static final String DEFAULT_TRACE_PROPERTY_NAME = "ast"; 
	public @NonNull static final Key<String> TRACE_PROPERTY_NAME_KEY = new Key<String>("ocl2qvtp.tracePropName");

	private @NonNull URI oclASUri;
	private @NonNull List<URI> extendedASUris = new ArrayList<URI>();	
	private @Nullable String traceabilityPropName;
	
	/**
	 * To provide a different traceabilityPropName different to the default {@link OCL2QVTiCompilerChain#DEFAULT_TRACE_PROPERTY_NAME "ast"} one,
	 * it must be passed as an option using the {@link #TRACE_PROPERTY_NAME_KEY TRACE_PROPERTY_NAME_KEY} and the {@link CompilerChain#QVTP_STEP QVTP_STEP}
	 * 
	 * @param qvti mandatory {@link QVTimperative} instance
	 * @param options optional options
	 * @param oclDocURI the mandatory main OCL document URI to compile
	 * @param extendedDocURIs optional OCL document URIs that the main one extends
	 */
	public OCL2QVTiCompilerChain(@NonNull QVTimperative qvti, @Nullable Map<@NonNull String, @NonNull Map<@NonNull Key<?>, @Nullable Object>> options,			
			@NonNull URI oclDocURI, URI... extendedDocURIs) { 
		super(qvti.getEnvironmentFactory(), oclDocURI, options);
		this.traceabilityPropName = getTraceabilityPropertyName();
		this.oclASUri = qvti.parse(oclDocURI).getURI();
		for (URI oclDocUri : extendedDocURIs) {
			this.extendedASUris.add(qvti.parse(oclDocUri).getURI()); // We add the AS URI
		}
	}

	@Override
	public @NonNull Transformation compile(@NonNull String enforcedOutputName) throws IOException {
		return qvtp2qvti(qvtm2qvtp(ocl2qvtm(oclASUri)));
		// return qvtp2qvti(ocl2qvtp());
	}
	
	public @NonNull Transformation compile() throws IOException {
		return compile("");
	}
	
	protected Resource ocl2qvtp() throws IOException { 
		Resource pModel = ocl2qvtp(oclASUri);
		if (!extendedASUris.isEmpty()) {
			List<Resource> qvtpModels = new ArrayList<Resource>();
			for (URI extendedQVTpModel : extendedASUris) {
				qvtpModels.add(ocl2qvtp(extendedQVTpModel));
			}
			QVTpModelsMerger.merge(environmentFactory , pModel, qvtpModels);
			saveResource(pModel, QVTP_STEP);
			for(Resource qvtpModel : qvtpModels) {	// unload unnecessary qvtpModels
				qvtpModel.unload();
			}
		}
		return pModel;
	}
	
	protected Resource ocl2qvtp(URI oclURI) throws IOException {
		OCL2QVTp ocl2qvtp = new OCL2QVTp(environmentFactory, traceabilityPropName);
		Resource pResource = ocl2qvtp.run(environmentFactory.getMetamodelManager().getASResourceSet(), oclURI);
		saveResource(pResource, QVTP_STEP);
		return pResource;
	}
	
	protected Resource ocl2qvtm(URI oclURI) throws IOException {
		OCL2QVTp ocl2qvtm = new OCL2QVTp(environmentFactory, traceabilityPropName);
		Resource mResource = ocl2qvtm.run(environmentFactory.getMetamodelManager().getASResourceSet(), oclURI);
		saveResource(mResource, QVTM_STEP);
		return mResource;
	}
	
	private @NonNull String getTraceabilityPropertyName() {
		String tracePropName = getOption(QVTP_STEP, TRACE_PROPERTY_NAME_KEY);
		return tracePropName == null ? DEFAULT_TRACE_PROPERTY_NAME : tracePropName;
	}
	@Override
	protected @NonNull Transformation qvtp2qvti(@NonNull Resource pResource) throws IOException {
		rewriteSafeNavigations(pResource);
		return super.qvtp2qvti(pResource);
	}

	// FIXME this workaround produces a new traversal of the resource
	private void rewriteSafeNavigations(@NonNull Resource resource) {
		for (EObject rootObject : resource.getContents()) {
			if (rootObject instanceof Element) {
				QVTbaseUtil.rewriteSafeNavigations(environmentFactory, (Element) rootObject);
			}
		}
	}
}