source: trunk/CrossPare/src/de/ugoe/cs/cpdp/decentApp/DECENTEpsilonModelHandler.java @ 34

package de.ugoe.cs.cpdp.decentApp;

import java.io.File;
import java.net.URISyntaxException;
import java.util.HashMap;

import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.epsilon.common.util.StringProperties;
import org.eclipse.epsilon.emc.emf.EmfModel;
import org.eclipse.epsilon.emc.emf.InMemoryEmfModel;
import org.eclipse.epsilon.eol.exceptions.models.EolModelLoadingException;
import org.eclipse.epsilon.eol.models.IModel;

import ARFFx.ARFFxPackage;
import DECENT.DECENTPackage;

/**
 * Class for handling decent and arffx model files.
 * 
 * @author Philip Makedonski, Fabian Trautsch
 *
 */

public class DECENTEpsilonModelHandler {
        private HashMap<String, Object> metaModelCache = new HashMap<>();
        private boolean useDECENTBinary = false;
        private boolean useARFFxBinary = false;

        public static String metaPath = "./decent/models/";

        /**
         * Returns the decent model as IModel instance
         * 
         * @param decentModelLocation location of the decent model file
         * @param read indicates if the model should be read from
         * @param write indicates if data should be written in the model
         * @return EmFModel (IModel) instance from the decent model, which was loaded
         * @throws Exception
         */
        public IModel getDECENTModel(String decentModelLocation, boolean read, boolean write) throws Exception { 

                EmfModel model;
                
                if (isUseDECENTBinary()) {
                        unregisterMetaModels("");
                        if (!read) {
                                new File(decentModelLocation).delete();
                                new File(decentModelLocation+"bin").delete();
                        }
                        DECENTResourceTool tool = new DECENTResourceTool();
                        if (new File(decentModelLocation).exists() && !new File(decentModelLocation+"bin").exists()) {
                                Resource resource = tool.loadResourceFromXMI(decentModelLocation,"decent", DECENTPackage.eINSTANCE);
                                tool.storeBinaryResourceContents(resource.getContents(), decentModelLocation+"bin", "decentbin");
                        }
                        
                        Resource resourceBin = tool.loadResourceFromBinary(decentModelLocation+"bin","decentbin", DECENTPackage.eINSTANCE);
                        //alternative pattern
//                      model = createInMemoryEmfModel("DECENT", resourceLocation, "../DECENT.Meta/model/DECENTv3.ecore", read, write, resourceBin, DECENTPackage.eINSTANCE);
//                      restoreMetaModels();

                        //NOTE: Adding the package is essential as otherwise epsilon breaks
                        model = new InMemoryEmfModel("DECENT", resourceBin, DECENTPackage.eINSTANCE);
                        model.setStoredOnDisposal(write);
                        model.setReadOnLoad(read);
                        model.setCachingEnabled(true);
                        restoreMetaModels();            
                } else {
                        model = createEmfModel("DECENT", decentModelLocation, metaPath+"DECENTv3.ecore", read, write);
                }

                return model;
        }

        /**
         * Converts the decent model to a binary form
         * 
         * @param location of the decent model file
         */
        public void convertDECENTModelToBinary(String location) {
                unregisterMetaModels("");
                DECENTResourceTool tool = new DECENTResourceTool();
                Resource resource = tool.loadResourceFromXMI(location+"/model.decent","decent", DECENTPackage.eINSTANCE);
                tool.storeBinaryResourceContents(resource.getContents(), location+"/model.decent"+"bin", "decentbin");
                restoreMetaModels();            
        }

        /**
         * Converts the decent model to a xmi form
         * 
         * @param location of the decent model file
         */
        
        public void convertDECENTModelToXMI(String location) {
                unregisterMetaModels("");
                DECENTResourceTool tool = new DECENTResourceTool(); 
                Resource resource = tool.loadResourceFromBinary(location+"/model.decentbin","decentbin", DECENTPackage.eINSTANCE);
                restoreMetaModels();            
                tool.storeResourceContents(resource.getContents(), location+"/model.decent", "decent");
        }

        /**
         * Returns the arffx model as IModel instance
         * 
         * @param arffxModelLocation location of the arffx model file
         * @param read indicates if the model should be read from
         * @param write indicates if data should be written in the model
         * @return EmFModel (IModel) instance from the arffx model, which was loaded
         * @throws Exception
         */
        
        public IModel getARFFxModel(String arffxModelLocation, boolean read, boolean write) throws Exception {
                
                EmfModel model;
                
                if (isUseARFFxBinary()) {
                        unregisterMetaModels("");
                        if (!read) {
                                new File(arffxModelLocation).delete();
                                new File(arffxModelLocation+"bin").delete();
                        }
                        ARFFxResourceTool tool = new ARFFxResourceTool();
                        if (new File(arffxModelLocation).exists() && !new File(arffxModelLocation+"bin").exists()) {
                                Resource resource = tool.loadResourceFromXMI(arffxModelLocation,"arffx", ARFFxPackage.eINSTANCE);
                                tool.storeBinaryResourceContents(resource.getContents(), arffxModelLocation+"bin", "arffxbin");
                        }
                        
                        Resource resourceBin = tool.loadResourceFromBinary(arffxModelLocation+"bin","arffxbin", ARFFxPackage.eINSTANCE);
                        //alternative pattern
//                      model = createInMemoryEmfModel("DECENT", resourceLocation, "../DECENT.Meta/model/DECENTv3.ecore", read, write, resourceBin, DECENTPackage.eINSTANCE);
//                      restoreMetaModels();

                        //NOTE: Adding the package is essential as otherwise epsilon breaks
                        model = new InMemoryEmfModel("ARFFx", resourceBin, ARFFxPackage.eINSTANCE);
//                      model.getModelImpl().getURI().toFileString()
                        model.setStoredOnDisposal(write);
                        model.setReadOnLoad(read);
                        model.setCachingEnabled(true);
                        restoreMetaModels();            
                } else {
                        model = createEmfModel("ARFFx", arffxModelLocation, metaPath+"ARFFx.ecore", read, write);
                }
                
                return model;
        }


        /**
         * Converts an arffx model to a binary version
         * 
         * @param location of the arffx model
         */
        public void convertARFFxModelToBinary(String location) {
                unregisterMetaModels("");
                ARFFxResourceTool tool = new ARFFxResourceTool();
                Resource resource = tool.loadResourceFromXMI(location+"/model.arffx","arffx", ARFFxPackage.eINSTANCE);
                tool.storeBinaryResourceContents(resource.getContents(), location+"/model.arffx"+"bin", "arffxbin");
                restoreMetaModels();            
        }
        
        /**
         * Converts an arffx model to xmi
         * 
         * @param location of the arffx model
         */

        public void convertARFFxModelToXMI(String location) {
                unregisterMetaModels("");
                ARFFxResourceTool tool = new ARFFxResourceTool(); 
                Resource resource = tool.loadResourceFromBinary(location+"/model.arffxbin","arffxbin", DECENTPackage.eINSTANCE);
                restoreMetaModels();            
                tool.storeResourceContents(resource.getContents(), location+"/model.arffx", "arffx");
        }


        /**
         * Returns the log model as IModel instance
         * 
         * @param logModelLocation location of the log model file
         * @param read indicates if the model should be read from
         * @param write indicates if data should be written in the model
         * @return EmFModel (IModel) instance from the log model, which was loaded
         * @throws Exception
         */
        
        public IModel getLOGModel(String logModelLocation, boolean read, boolean write) throws Exception {
                if (!new File(logModelLocation).exists()) {
                        read = false;
                }
                IModel model = createEmfModel("LOG", logModelLocation, metaPath +"LOG.ecore", read, write);
                System.setProperty("epsilon.logFileAvailable", "true");
                return model;
        }

        /**
         * Creates an EMF Model
         * 
         * @param name of the  emf model
         * @param model  name of the model
         * @param metamodel name of the metamodel
         * @param readOnLoad indicates if the model should be read on load
         * @param storeOnDisposal indicates if the model should be stored on disposal
         * @return
         * @throws EolModelLoadingException
         * @throws URISyntaxException
         */
        
        @SuppressWarnings("deprecation")
        protected EmfModel createEmfModel(String name, String model, 
                        String metamodel, boolean readOnLoad, boolean storeOnDisposal) 
                                        throws EolModelLoadingException, URISyntaxException {
                EmfModel emfModel = new EmfModel();
                StringProperties properties = new StringProperties();
                properties.put(EmfModel.PROPERTY_NAME, name);
                properties.put(EmfModel.PROPERTY_ALIASES, name);
                properties.put(EmfModel.PROPERTY_FILE_BASED_METAMODEL_URI, 
                                "file:/" + getFile(metamodel).getAbsolutePath());
                properties.put(EmfModel.PROPERTY_MODEL_URI, 
                                "file:/" + getFile(model).getAbsolutePath());
                properties.put(EmfModel.PROPERTY_IS_METAMODEL_FILE_BASED, "true");
                properties.put(EmfModel.PROPERTY_READONLOAD, readOnLoad + "");
                properties.put(EmfModel.PROPERTY_CACHED, "true");
                properties.put(EmfModel.PROPERTY_STOREONDISPOSAL, 
                                storeOnDisposal + "");
                emfModel.load(properties, "");
                //System.out.println(emfModel.allContents());
                return emfModel;
        }

        /**
         * Returns a new File instance from the given filename
         * 
         * @param fileName of the file
         * @return
         * @throws URISyntaxException
         */
        public File getFile(String fileName) throws URISyntaxException {;
                return new File(fileName);
        }

        /**
         * Restores the metamodels, so that they are registered in the
         * EPackage registry
         */
        private void restoreMetaModels() {
                for (String key : metaModelCache .keySet()) {
                        EPackage.Registry.INSTANCE.put(key, metaModelCache.get(key));
                };
        }

        /**
         * Unregister the metamodels from the EPackage registry
         * 
         * @param filter for filtering out certain instances
         */
        private void unregisterMetaModels(String filter) {
                for (String key : EPackage.Registry.INSTANCE.keySet()) {
                        if (key.contains(filter)) {
                                metaModelCache.put(key, EPackage.Registry.INSTANCE.get(key));
                        }
                };
                for (String key : metaModelCache .keySet()) {
                        EPackage.Registry.INSTANCE.remove(key);
                };
        }
        
        /**
         * Returns true if decent binary model is used
         * @return
         */

        public boolean isUseDECENTBinary() {
                return useDECENTBinary;
        }

        /**
         * Sets the boolean which indicates, if the decent binary
         * model is used
         * @param useDECENTBinary
         */
        public void setUseDECENTBinary(boolean useDECENTBinary) {
                this.useDECENTBinary = useDECENTBinary;
        }

        /**
         * Returns true if arffx binary model is used
         * @return
         */
        public boolean isUseARFFxBinary() {
                return useARFFxBinary;
        }
        
        /**
         * Sets the boolean which indicates, if the arffx binary
         * model is used
         * @param useARFFxBinary
         */

        public void setUseARFFxBinary(boolean useARFFxBinary) {
                this.useARFFxBinary = useARFFxBinary;
        }

        
}