plugins/org.eclipse.m2m.atl.engine.vm/src/org/eclipse/m2m/atl/engine/vm/ModelLoader.java - mmt/org.eclipse.atl - Git at Google

 /*******************************************************************************
  * Copyright (c) 2004 INRIA.
  * 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:
  * 	   Frederic Jouault (INRIA) - initial API and implementation
  *     Dennis Wagelaar (Vrije Universiteit Brussel)
  *******************************************************************************/
 package org.eclipse.m2m.atl.engine.vm;

 import java.io.FileInputStream;
 import java.io.FileNotFoundException;
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
 import java.io.RandomAccessFile;
 import java.net.URL;
 import java.util.HashMap;
 import java.util.Iterator;
 import java.util.Map;
 import java.util.logging.Level;

 import org.eclipse.m2m.atl.common.ATLLogger;
 import org.eclipse.m2m.atl.engine.extractors.Extractor;
 import org.eclipse.m2m.atl.engine.extractors.xml.XMLExtractor;
 import org.eclipse.m2m.atl.engine.injectors.Injector;
 import org.eclipse.m2m.atl.engine.injectors.xml.XMLInjector;
 import org.eclipse.m2m.atl.engine.vm.nativelib.ASMModel;
 import org.eclipse.m2m.atl.engine.vm.nativelib.ASMModelElement;

 /**
  * Model loading and saving facility. Must be extended by concrete implementations
  * such as for EMF or MDR. This is only used in command-line mode at the present time.
  *
  * @author <a href="mailto:frederic.jouault@univ-nantes.fr">Frederic Jouault</a>
  * @author <a href="mailto:dennis.wagelaar@vub.ac.be">Dennis Wagelaar</a>
  */
 public abstract class ModelLoader {

 	protected Map loadedModels = new HashMap();

 	private Map injectors = new HashMap();
 	{
 		injectors.put("xml", XMLInjector.class);
 	}

 	public void addInjector(String prefix, Class inj) {
 		injectors.put(prefix, inj);
 	}

 	private Map extractors = new HashMap();
 	{
 		extractors.put("xml", XMLExtractor.class);
 	}

 	public void addExtractor(String prefix, Class ext) {
 		extractors.put(prefix, ext);
 	}

 	/**
 	 * Loads a model with given name and metamodel from in.
 	 * Use this method only if there is no real URI available!
 	 * @param name The model name.
 	 * @param metamodel The metamodel of the model to be loaded.
 	 * @param in The input stream from which to load.
 	 * @return The loaded ASMModel.
 	 * @throws IOException
 	 */
 	public abstract ASMModel loadModel(String name, ASMModel metamodel, InputStream in) throws IOException;

 	protected abstract ASMModel realLoadModel(String name, ASMModel metamodel, String href) throws IOException;

 	/**
 	 * Loads a model from the URI represented by href.
 	 * @param name The model name.
 	 * @param metamodel The metamodel of the model to be loaded.
 	 * @param href The model URI.
 	 * @return The loaded ASMModel.
 	 * @throws IOException
 	 */
 	public ASMModel loadModel(String name, ASMModel metamodel, String href) throws IOException {
 		ASMModel ret = null;

 		href = href.replaceAll("\\\\:", "<colon>");
 		String[] ss = href.split(":");
 		for (int i = 0; i < ss.length; i++)
 			ss[i] = ss[i].replaceAll("<colon>", ":");
 		if (ss.length == 1) {
 			ret = realLoadModel(name, metamodel, ss[0]);
 		} else if(ss[0].equals("uri") || ss[0].equals("platform")) {
 			ret = realLoadModel(name, metamodel, href);
 		} else if(ss[0].equals("xmi")) {
 			final String url = ss[ss.length - 1];
 			ret = realLoadModel(name, metamodel, url);
 		} else {
 			ret = newModel(name, ss[ss.length - 1], metamodel);
 			inject(ret, ss[0], (ss.length == 3) ? ss[1] : null, ss[ss.length - 1], null);
 		}

 		loadedModels.put(name, ret);
 		return ret;
 	}

 	public ASMModelElement inject(ASMModel ret, String kind, String params, String uri, InputStream in) {
 		ASMModelElement root = null;
 		try {
 			Map paramsMap = new HashMap();
 			if (uri != null) {
 				try {
 					in = new FileInputStream(uri);
 				} catch (FileNotFoundException fnfe) {
 					in = new URL(uri).openStream();
 				}
 			}

 			Class injectorClass = (Class)injectors.get(kind);
 			if (injectorClass != null) {
 				Injector inj = (Injector)injectorClass.newInstance();
 				if (params != null) {
 					String[] sparams = params.split(",");
 					Map args = new HashMap();
 					for (int i = 0; i < sparams.length; i++) {
 						String p = sparams[i];
 						String[] pair = p.split("=");
 						if (pair.length == 1) {
 							if (p.indexOf("=") == -1) {
 								args.put("name", pair[0]);
 							} else {
 								args.put(pair[0], "");
 							}
 						} else {
 							args.put(pair[0], pair[1]);
 						}

 					}
 					Map types = inj.getParameterTypes();
 					for (Iterator i = types.keySet().iterator(); i.hasNext();) {
 						String pname = (String)i.next();
 						String type = (String)types.get(pname);
 						if (type.equals("String")) {
 							String val = (String)args.get(pname);
 							if (val != null) {
 								paramsMap.put(pname, val);
 							} else {
 								ATLLogger.warning("could not find value for parameter \"" + pname + "\" : "
 										+ type + ".");
 							}
 						} else if (type.startsWith("Model:")) {
 							paramsMap.put(pname, loadedModels.get(args.get(pname)));
 						} else if (type.equals("RandomAccessFile") && (uri != null)) {
 							paramsMap.put(pname, new RandomAccessFile(uri, "r"));
 						} else {
 							ATLLogger.warning("unknown parameter type \"" + type + "\" of \"" + pname + "\".");
 						}
 					}
 				}
 				// inj.performImportation(metamodel, ret, in, other);
 				root = inj.inject(ret, in, paramsMap);
 				ret.setIsTarget(false);
 				// getAllAcquaintances ?
 			} else {
 				ATLLogger.severe("ERROR: could not find injector for \"" + kind + "\"");
 			}
 		} catch (Exception e) {
 			ATLLogger.log(Level.SEVERE, e.getLocalizedMessage(), e);

 		}
 		return root;
 	}

 	/**
 	 * @param name
 	 * @param metamodel
 	 * @return A new ASMModel
 	 * @deprecated Use {@link #newModel(String, String, ASMModel)} instead
 	 */
 	public abstract ASMModel newModel(String name, ASMModel metamodel);

 	/**
 	 * @param name
 	 * @param uri
 	 * @param metamodel
 	 * @return A new ASMModel
 	 */
 	public ASMModel newModel(String name, String uri, ASMModel metamodel) {
 		return newModel(name, metamodel);
 	}

 	protected abstract void setParameter(String name, Object value);

 	protected abstract void realSave(ASMModel model, String href) throws IOException;

 	/**
 	 * Saves the model to a writable URL.
 	 * @param model The model to save
 	 * @param href The writable URL
 	 * @throws IOException
 	 */
 	public void save(ASMModel model, String href) throws IOException {
 		String[] ss = href.split(":");
 		if (ss.length == 1) {
 			realSave(model, href);
 		} else if (ss[0].equals("xmi")) {
 			String url = ss[ss.length - 1];

 			// setting default values
 			// EMF-specific
 			setParameter("useIDs", "false");
 			setParameter("removeIDs", "false");
 			setParameter("encoding", "ISO-8859-1");

 			// for MDR-specific
 			setParameter("xmiVersion", null);

 			// parsing user-provided values
 			if (ss.length == 3) {
 				String[] sparams = ss[1].split(",");
 				for (int i = 0; i < sparams.length; i++) {
 					String p = sparams[i];
 					String[] pair = p.split("=");
 					if (pair.length == 2) {
 						setParameter(pair[0], pair[1]);
 					}
 				}
 			}
 			realSave(model, url);
 		} else if (ss[0].equals("file")) {
 			realSave(model, href);
 		} else if (ss[0].equals("platform")) {
 			realSave(model, href);
 		} else {
 			extract(model, ss[0], (ss.length == 3) ? ss[1] : null, ss[ss.length - 1], null);
 		}
 	}

 	public void extract(ASMModel model, String kind, String params, String uri, OutputStream out) {
 		try {
 			Map paramsMap = new HashMap();
 			if (uri != null) {
 				out = new FileOutputStream(uri);
 			}

 			Class extractorClass = (Class)extractors.get(kind);
 			if (extractorClass != null) {
 				Extractor ext = (Extractor)extractorClass.newInstance();
 				if (params != null) {
 					String[] sparams = params.split(",");
 					Map args = new HashMap();
 					for (int i = 0; i < sparams.length; i++) {
 						String p = sparams[i];
 						String[] pair = p.split("=");
 						if (pair.length == 1) {
 							if (p.indexOf("=") == -1) {
 								args.put("name", pair[0]);
 							} else {
 								args.put(pair[0], "");
 							}
 						} else {
 							args.put(pair[0], pair[1]);
 						}
 					}
 					Map types = ext.getParameterTypes();
 					for (Iterator i = types.keySet().iterator(); i.hasNext();) {
 						String pname = (String)i.next();
 						String type = (String)types.get(pname);
 						if (type.equals("String")) {
 							String val = (String)args.get(pname);
 							if (val != null) {
 								paramsMap.put(pname, val);
 							} else {
 								ATLLogger.warning("could not find value for parameter \"" + pname + "\" : "
 										+ type + ".");
 							}
 						} else if (type.startsWith("Model:")) {
 							paramsMap.put(pname, loadedModels.get(args.get(pname)));
 						} else {
 							ATLLogger
 									.warning("unknown parameter type \"" + type + "\" of \"" + pname + "\".");
 						}
 					}
 				}
 				ext.extract(model, out, paramsMap);
 			} else {
 				ATLLogger.severe("ERROR: could not find extractor for \"" + kind + "\"");
 			}
 		} catch (Throwable e) {
 			ATLLogger.log(Level.SEVERE, e.getLocalizedMessage(), e);
 		}
 	}

 	public abstract ASMModel getMOF();

 	public abstract ASMModel getATL();

 	public abstract ASMModel getBuiltInMetaModel(String name);

 	public abstract void unload(ASMModel model);

 }
	/*******************************************************************************
	* Copyright (c) 2004 INRIA.
	* 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:
	* Frederic Jouault (INRIA) - initial API and implementation
	* Dennis Wagelaar (Vrije Universiteit Brussel)
	*******************************************************************************/
	package org.eclipse.m2m.atl.engine.vm;

	import java.io.FileInputStream;
	import java.io.FileNotFoundException;
	import java.io.FileOutputStream;
	import java.io.IOException;
	import java.io.InputStream;
	import java.io.OutputStream;
	import java.io.RandomAccessFile;
	import java.net.URL;
	import java.util.HashMap;
	import java.util.Iterator;
	import java.util.Map;
	import java.util.logging.Level;

	import org.eclipse.m2m.atl.common.ATLLogger;
	import org.eclipse.m2m.atl.engine.extractors.Extractor;
	import org.eclipse.m2m.atl.engine.extractors.xml.XMLExtractor;
	import org.eclipse.m2m.atl.engine.injectors.Injector;
	import org.eclipse.m2m.atl.engine.injectors.xml.XMLInjector;
	import org.eclipse.m2m.atl.engine.vm.nativelib.ASMModel;
	import org.eclipse.m2m.atl.engine.vm.nativelib.ASMModelElement;

	/**
	* Model loading and saving facility. Must be extended by concrete implementations
	* such as for EMF or MDR. This is only used in command-line mode at the present time.
	*
	* @author <a href="mailto:frederic.jouault@univ-nantes.fr">Frederic Jouault</a>
	* @author <a href="mailto:dennis.wagelaar@vub.ac.be">Dennis Wagelaar</a>
	*/
	public abstract class ModelLoader {

	protected Map loadedModels = new HashMap();

	private Map injectors = new HashMap();
	{
	injectors.put("xml", XMLInjector.class);
	}

	public void addInjector(String prefix, Class inj) {
	injectors.put(prefix, inj);
	}

	private Map extractors = new HashMap();
	{
	extractors.put("xml", XMLExtractor.class);
	}

	public void addExtractor(String prefix, Class ext) {
	extractors.put(prefix, ext);
	}

	/**
	* Loads a model with given name and metamodel from in.
	* Use this method only if there is no real URI available!
	* @param name The model name.
	* @param metamodel The metamodel of the model to be loaded.
	* @param in The input stream from which to load.
	* @return The loaded ASMModel.
	* @throws IOException
	*/
	public abstract ASMModel loadModel(String name, ASMModel metamodel, InputStream in) throws IOException;

	protected abstract ASMModel realLoadModel(String name, ASMModel metamodel, String href) throws IOException;

	/**
	* Loads a model from the URI represented by href.
	* @param name The model name.
	* @param metamodel The metamodel of the model to be loaded.
	* @param href The model URI.
	* @return The loaded ASMModel.
	* @throws IOException
	*/
	public ASMModel loadModel(String name, ASMModel metamodel, String href) throws IOException {
	ASMModel ret = null;

	href = href.replaceAll("\\\\:", "<colon>");
	String[] ss = href.split(":");
	for (int i = 0; i < ss.length; i++)
	ss[i] = ss[i].replaceAll("<colon>", ":");
	if (ss.length == 1) {
	ret = realLoadModel(name, metamodel, ss[0]);
	} else if(ss[0].equals("uri") \|\| ss[0].equals("platform")) {
	ret = realLoadModel(name, metamodel, href);
	} else if(ss[0].equals("xmi")) {
	final String url = ss[ss.length - 1];
	ret = realLoadModel(name, metamodel, url);
	} else {
	ret = newModel(name, ss[ss.length - 1], metamodel);
	inject(ret, ss[0], (ss.length == 3) ? ss[1] : null, ss[ss.length - 1], null);
	}

	loadedModels.put(name, ret);
	return ret;
	}

	public ASMModelElement inject(ASMModel ret, String kind, String params, String uri, InputStream in) {
	ASMModelElement root = null;
	try {
	Map paramsMap = new HashMap();
	if (uri != null) {
	try {
	in = new FileInputStream(uri);
	} catch (FileNotFoundException fnfe) {
	in = new URL(uri).openStream();
	}
	}

	Class injectorClass = (Class)injectors.get(kind);
	if (injectorClass != null) {
	Injector inj = (Injector)injectorClass.newInstance();
	if (params != null) {
	String[] sparams = params.split(",");
	Map args = new HashMap();
	for (int i = 0; i < sparams.length; i++) {
	String p = sparams[i];
	String[] pair = p.split("=");
	if (pair.length == 1) {
	if (p.indexOf("=") == -1) {
	args.put("name", pair[0]);
	} else {
	args.put(pair[0], "");
	}
	} else {
	args.put(pair[0], pair[1]);
	}

	}
	Map types = inj.getParameterTypes();
	for (Iterator i = types.keySet().iterator(); i.hasNext();) {
	String pname = (String)i.next();
	String type = (String)types.get(pname);
	if (type.equals("String")) {
	String val = (String)args.get(pname);
	if (val != null) {
	paramsMap.put(pname, val);
	} else {
	ATLLogger.warning("could not find value for parameter \"" + pname + "\" : "
	+ type + ".");
	}
	} else if (type.startsWith("Model:")) {
	paramsMap.put(pname, loadedModels.get(args.get(pname)));
	} else if (type.equals("RandomAccessFile") && (uri != null)) {
	paramsMap.put(pname, new RandomAccessFile(uri, "r"));
	} else {
	ATLLogger.warning("unknown parameter type \"" + type + "\" of \"" + pname + "\".");
	}
	}
	}
	// inj.performImportation(metamodel, ret, in, other);
	root = inj.inject(ret, in, paramsMap);
	ret.setIsTarget(false);
	// getAllAcquaintances ?
	} else {
	ATLLogger.severe("ERROR: could not find injector for \"" + kind + "\"");
	}
	} catch (Exception e) {
	ATLLogger.log(Level.SEVERE, e.getLocalizedMessage(), e);

	}
	return root;
	}

	/**
	* @param name
	* @param metamodel
	* @return A new ASMModel
	* @deprecated Use {@link #newModel(String, String, ASMModel)} instead
	*/
	public abstract ASMModel newModel(String name, ASMModel metamodel);

	/**
	* @param name
	* @param uri
	* @param metamodel
	* @return A new ASMModel
	*/
	public ASMModel newModel(String name, String uri, ASMModel metamodel) {
	return newModel(name, metamodel);
	}

	protected abstract void setParameter(String name, Object value);

	protected abstract void realSave(ASMModel model, String href) throws IOException;

	/**
	* Saves the model to a writable URL.
	* @param model The model to save
	* @param href The writable URL
	* @throws IOException
	*/
	public void save(ASMModel model, String href) throws IOException {
	String[] ss = href.split(":");
	if (ss.length == 1) {
	realSave(model, href);
	} else if (ss[0].equals("xmi")) {
	String url = ss[ss.length - 1];

	// setting default values
	// EMF-specific
	setParameter("useIDs", "false");
	setParameter("removeIDs", "false");
	setParameter("encoding", "ISO-8859-1");

	// for MDR-specific
	setParameter("xmiVersion", null);

	// parsing user-provided values
	if (ss.length == 3) {
	String[] sparams = ss[1].split(",");
	for (int i = 0; i < sparams.length; i++) {
	String p = sparams[i];
	String[] pair = p.split("=");
	if (pair.length == 2) {
	setParameter(pair[0], pair[1]);
	}
	}
	}
	realSave(model, url);
	} else if (ss[0].equals("file")) {
	realSave(model, href);
	} else if (ss[0].equals("platform")) {
	realSave(model, href);
	} else {
	extract(model, ss[0], (ss.length == 3) ? ss[1] : null, ss[ss.length - 1], null);
	}
	}

	public void extract(ASMModel model, String kind, String params, String uri, OutputStream out) {
	try {
	Map paramsMap = new HashMap();
	if (uri != null) {
	out = new FileOutputStream(uri);
	}

	Class extractorClass = (Class)extractors.get(kind);
	if (extractorClass != null) {
	Extractor ext = (Extractor)extractorClass.newInstance();
	if (params != null) {
	String[] sparams = params.split(",");
	Map args = new HashMap();
	for (int i = 0; i < sparams.length; i++) {
	String p = sparams[i];
	String[] pair = p.split("=");
	if (pair.length == 1) {
	if (p.indexOf("=") == -1) {
	args.put("name", pair[0]);
	} else {
	args.put(pair[0], "");
	}
	} else {
	args.put(pair[0], pair[1]);
	}
	}
	Map types = ext.getParameterTypes();
	for (Iterator i = types.keySet().iterator(); i.hasNext();) {
	String pname = (String)i.next();
	String type = (String)types.get(pname);
	if (type.equals("String")) {
	String val = (String)args.get(pname);
	if (val != null) {
	paramsMap.put(pname, val);
	} else {
	ATLLogger.warning("could not find value for parameter \"" + pname + "\" : "
	+ type + ".");
	}
	} else if (type.startsWith("Model:")) {
	paramsMap.put(pname, loadedModels.get(args.get(pname)));
	} else {
	ATLLogger
	.warning("unknown parameter type \"" + type + "\" of \"" + pname + "\".");
	}
	}
	}
	ext.extract(model, out, paramsMap);
	} else {
	ATLLogger.severe("ERROR: could not find extractor for \"" + kind + "\"");
	}
	} catch (Throwable e) {
	ATLLogger.log(Level.SEVERE, e.getLocalizedMessage(), e);
	}
	}

	public abstract ASMModel getMOF();

	public abstract ASMModel getATL();

	public abstract ASMModel getBuiltInMetaModel(String name);

	public abstract void unload(ASMModel model);

	}