| /** |
| * Copyright (c) 2005-2007 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 - Initial API and implementation |
| */ |
| package org.eclipse.emf.importer.ecore; |
| |
| import java.util.List; |
| |
| import org.eclipse.emf.codegen.ecore.genmodel.GenPackage; |
| import org.eclipse.emf.common.util.BasicDiagnostic; |
| import org.eclipse.emf.common.util.Diagnostic; |
| import org.eclipse.emf.common.util.DiagnosticException; |
| import org.eclipse.emf.common.util.Monitor; |
| import org.eclipse.emf.common.util.URI; |
| import org.eclipse.emf.common.util.UniqueEList; |
| import org.eclipse.emf.converter.ConverterPlugin; |
| import org.eclipse.emf.converter.util.ConverterUtil; |
| import org.eclipse.emf.ecore.EPackage; |
| import org.eclipse.emf.ecore.EcorePackage; |
| import org.eclipse.emf.ecore.resource.Resource; |
| import org.eclipse.emf.ecore.resource.ResourceSet; |
| import org.eclipse.emf.ecore.util.Diagnostician; |
| import org.eclipse.emf.ecore.util.EcoreUtil; |
| import org.eclipse.emf.importer.ModelImporter; |
| |
| |
| public class EcoreImporter extends ModelImporter |
| { |
| @Override |
| public String getID() |
| { |
| return "org.eclipse.emf.importer.ecore"; |
| } |
| |
| @Override |
| protected Diagnostic doComputeEPackages(Monitor monitor) throws Exception |
| { |
| Diagnostic diagnostic = Diagnostic.OK_INSTANCE; |
| |
| List<URI> locationURIs = getModelLocationURIs(); |
| if (locationURIs.isEmpty()) |
| { |
| diagnostic = new BasicDiagnostic( |
| Diagnostic.ERROR, |
| EcoreImporterPlugin.getPlugin().getBundle().getSymbolicName(), |
| 0, |
| EcoreImporterPlugin.INSTANCE.getString("_UI_SpecifyAValidCoreModel_message"), |
| null); |
| } |
| else |
| { |
| monitor.beginTask("", 2); |
| monitor.subTask(EcoreImporterPlugin.INSTANCE.getString("_UI_Loading_message", new Object []{ locationURIs })); |
| |
| ResourceSet ecoreResourceSet = createResourceSet(); |
| for (URI ecoreModelLocation : locationURIs) |
| { |
| ecoreResourceSet.getResource(ecoreModelLocation, true); |
| } |
| EcoreUtil.resolveAll(ecoreResourceSet); |
| |
| for (Resource resource : ecoreResourceSet.getResources()) |
| { |
| getEPackages().addAll(EcoreUtil.<EPackage>getObjectsByType(resource.getContents(), EcorePackage.Literals.EPACKAGE)); |
| } |
| |
| BasicDiagnostic diagnosticChain = |
| new BasicDiagnostic |
| (ConverterPlugin.ID, |
| ConverterUtil.ACTION_MESSAGE_NONE, |
| EcoreImporterPlugin.INSTANCE.getString("_UI_ErrorsWereDetectedEcore_message"), |
| null); |
| for (EPackage ePackage : getEPackages()) |
| { |
| Diagnostician.INSTANCE.validate(ePackage, diagnosticChain); |
| } |
| if (diagnosticChain.getSeverity() != Diagnostic.OK) |
| { |
| diagnostic = diagnosticChain; |
| } |
| } |
| return diagnostic; |
| } |
| |
| @Override |
| public void addToResource(EPackage ePackage, ResourceSet resourceSet) |
| { |
| if (ePackage.eResource() != null && getGenModel().eResource() != null) |
| { |
| URI ePackageURI = ePackage.eResource().getURI(); |
| URI genModelURI = getGenModel().eResource().getURI(); |
| |
| if (!ePackageURI.trimSegments(1).equals(genModelURI.trimSegments(1))) |
| { |
| ePackage.eResource().getContents().remove(ePackage); |
| } |
| } |
| super.addToResource(ePackage, resourceSet); |
| } |
| |
| @Override |
| protected void adjustGenModel(Monitor monitor) |
| { |
| super.adjustGenModel(monitor); |
| |
| URI genModelURI = createFileURI(getGenModelPath().toString()); |
| for (URI uri : getModelLocationURIs()) |
| { |
| getGenModel().getForeignModel().add(makeRelative(uri, genModelURI).toString()); |
| } |
| } |
| |
| @Override |
| protected void handleOriginalGenModel() throws DiagnosticException |
| { |
| URI genModelURI = getOriginalGenModel().eResource().getURI(); |
| StringBuffer text = new StringBuffer(); |
| for (String value : getOriginalGenModel().getForeignModel()) |
| { |
| if (value.endsWith(".ecore") || value.endsWith(".emof")) |
| { |
| text.append(makeAbsolute(URI.createURI(value), genModelURI).toString()); |
| text.append(" "); |
| } |
| } |
| |
| if (text.length() == 0) |
| { |
| List<URI> locations = new UniqueEList<URI>(); |
| for (GenPackage genPackage : getOriginalGenModel().getGenPackages()) |
| { |
| URI ecoreURI = genPackage.getEcorePackage().eResource().getURI(); |
| if (locations.add(ecoreURI)) |
| { |
| text.append(makeAbsolute(URI.createURI(ecoreURI.toString()), genModelURI).toString()); |
| text.append(" "); |
| } |
| } |
| } |
| |
| setModelLocation(text.toString().trim()); |
| } |
| } |