/*******************************************************************************
* Copyright (c) 2010, 2015 Willink Transformations 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:
* E.D.Willink - initial API and implementation
*******************************************************************************/
package org.eclipse.ocl.examples.build.utilities;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.log4j.Logger;
import org.eclipse.core.runtime.Path;
import org.eclipse.emf.codegen.ecore.genmodel.GenModelPackage;
import org.eclipse.emf.codegen.ecore.genmodel.GenPackage;
import org.eclipse.emf.common.util.BasicMonitor;
import org.eclipse.emf.common.util.Diagnostic;
import org.eclipse.emf.common.util.Monitor;
import org.eclipse.emf.common.util.TreeIterator;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.common.util.UniqueEList;
import org.eclipse.emf.ecore.EAnnotation;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.xmi.XMLResource;
import org.eclipse.emf.importer.ModelImporter;
import org.eclipse.emf.importer.ecore.EcoreImporter;
import org.eclipse.emf.mwe.core.WorkflowContext;
import org.eclipse.emf.mwe.core.issues.Issues;
import org.eclipse.emf.mwe.core.monitor.ProgressMonitor;
import org.eclipse.jdt.annotation.NonNull;
import org.eclipse.ocl.pivot.internal.resource.ProjectMap;
import org.eclipse.ocl.pivot.internal.resource.StandaloneProjectMap;
import org.eclipse.ocl.pivot.utilities.ClassUtil;
/**
* Reloads the designated <tt>genModel</tt>.
*/
public class GenmodelReloader extends AbstractProjectComponent
{
protected Logger log = Logger.getLogger(getClass());
// protected String modelImporter = UMLImporter.class.getName();
protected String genModel; // URI of the genmodel
protected String ecoreFile = null; // Explicit file URI of the Ecore
protected boolean showProgress = false; // Set true to show genmodel new tasks
@Override
public void checkConfiguration(Issues issues) {
super.checkConfiguration(issues);
if (genModel == null) {
issues.addError(this, "genModel name not specified.");
}
// if (modelImporter == null) {
// issues.addError(this, "modelImporter class not specified.");
// }
}
public String getEcoreFile() {
return ecoreFile;
}
public String getGenModel() {
return genModel;
}
// public String getModelImporter() {
// return modelImporter;
// }
@Override
public void invokeInternal(WorkflowContext ctx, ProgressMonitor arg1, Issues arg2) {
URI genModelURI = URI.createPlatformResourceURI(genModel, true);
log.info("Reloading '" + genModelURI + "'");
Monitor monitor = showProgress ? new LoggerMonitor(log) : new BasicMonitor();
StandaloneProjectMap.IProjectDescriptor projectDescriptor = ClassUtil.nonNullState(getProjectDescriptor());
// @SuppressWarnings("null")@NonNull URI nsURI = URI.createURI(PivotPackage.eNS_URI);
// IPackageDescriptor packageDescriptor = projectDescriptor.getPackageDescriptor(nsURI);
// packageDescriptor.configure(null, StandaloneProjectMap.LoadModelStrategy.INSTANCE, null);
EcoreImporter modelImporterInstance = new EcoreImporter()
{
@Override
public ResourceSet createResourceSet() {
ResourceSet resourceSet = super.createResourceSet();
ProjectMap.initializeURIResourceMap(resourceSet);
return resourceSet;
}
};
// Class<?> clazz = ResourceLoaderFactory.createResourceLoader().loadClass(modelImporter);
// if (clazz == null)
// throw new ConfigurationException("Couldn't find class " + modelImporter);
// try {
// modelImporterInstance = (ModelImporter) clazz.newInstance();
// } catch (Exception e) {
// throw new ConfigurationException("Couldn't create instance of class " + modelImporter, e);
// }
try {
Path path = new Path(genModel);
modelImporterInstance.defineOriginalGenModelPath(path);
Diagnostic diagnostic = modelImporterInstance.computeEPackages(monitor);
if (diagnostic.getSeverity() != Diagnostic.OK) {
log.info(diagnostic);
}
modelImporterInstance.adjustEPackages(monitor);
if (ecoreFile != null) {
ResourceSet genModelResourceSet = modelImporterInstance.getGenModelResourceSet();
URI ecoreURI = URI.createPlatformResourceURI(ecoreFile, true);
Resource ecoreResource = genModelResourceSet.getResource(ecoreURI, true);
ecoreResource.setURI(ecoreURI); // change file:... to platform:...
List<@NonNull EPackage> ePackages = ClassUtil.nullFree(modelImporterInstance.getEPackages());
ecoreResource.getContents().clear();
ecoreResource.getContents().addAll(ePackages);
for (TreeIterator<EObject> tit = ecoreResource.getAllContents(); tit.hasNext(); ) {
EObject eObject = tit.next();
if (eObject instanceof EAnnotation) {
EAnnotation eAnnotation = (EAnnotation)eObject;
if (GenModelPackage.eNS_URI.equals(eAnnotation.getSource())) {
String string = eAnnotation.getDetails().get("documentation");
if (string != null) {
String normalizedString = string.replaceAll("\\r\\n", "\n");
if (!string.equals(normalizedString)) {
eAnnotation.getDetails().put("documentation", normalizedString);
}
}
}
}
}
projectDescriptor.configure(genModelResourceSet, StandaloneProjectMap.LoadBothStrategy.INSTANCE, null);
}
modelImporterInstance.prepareGenModelAndEPackages(monitor);
// modelImporterInstance.saveGenModelAndEPackages(monitor); -- assumes Eclipse running
List<Resource> resources = computeResourcesToBeSaved(modelImporterInstance);
// String readOnlyFiles = ConverterUtil.WorkspaceResourceValidator.validate(resources);
// if (readOnlyFiles != null)
// {
// throw new Exception(ImporterPlugin.INSTANCE.getString("_UI_ReadOnlyFiles_error", new String[]{readOnlyFiles}));
// }
ResourceUtils.checkResourceSet(resources.get(0).getResourceSet());
// if (ecoreFile != null) {
// Resource ecoreResource = resources.get(1);
// URI ecoreURI = URI.createPlatformResourceURI(ecoreFile, true);
// ecoreResource.setURI(ecoreURI); // change file:... to platform:...
// }
for (Resource resource : resources)
{
resource.save(getGenModelSaveOptions());
}
} catch (RuntimeException e) {
throw e;
} catch (Exception e) {
throw new RuntimeException("Problems running " + getClass().getSimpleName(), e);
}
}
public void setEcoreFile(String ecoreFile) {
this.ecoreFile = ecoreFile;
}
public void setGenModel(String genModel) {
this.genModel = genModel;
}
// public void setModelImporter(String modelImporter) {
// this.modelImporter = modelImporter;
// }
public static List<Resource> computeResourcesToBeSaved(
ModelImporter modelImporter) { // This is a clone of the protected
// ModelImporter method
List<Resource> resources = new UniqueEList.FastCompare<Resource>();
Resource genModelResource = modelImporter.getGenModel().eResource();
resources.add(genModelResource);
for (GenPackage genPackage : modelImporter.getGenModel()
.getGenPackages()) {
resources.add(genPackage.getEcorePackage().eResource());
}
// Handle application genmodel stub
//
for (GenPackage genPackage : modelImporter.getGenModel()
.getUsedGenPackages()) {
if (genPackage.eResource() == genModelResource) {
resources.add(genPackage.getEcorePackage().eResource());
}
}
return resources;
}
protected Map<?, ?> getGenModelSaveOptions() {
Map<Object, Object> result = new HashMap<Object, Object>();
result.put(Resource.OPTION_SAVE_ONLY_IF_CHANGED, Resource.OPTION_SAVE_ONLY_IF_CHANGED_MEMORY_BUFFER);
// result.put(XMLResource.OPTION_LINE_WIDTH, Integer.valueOf(132));
result.put(XMLResource.OPTION_LINE_DELIMITER, "\n");
return result;
}
}