org.bimserver.tools.generators.DataObjectGenerator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of BimServer Show documentation
There is a newer version: 1.5.9
package org.bimserver.tools.generators;

/******************************************************************************
 * Copyright (C) 2009-2016  BIMserver.org
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see {@literal}.
 *****************************************************************************/

import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.bimserver.database.migrations.Schema;
import org.bimserver.plugins.VirtualFile;
import org.bimserver.utils.StringUtils;
import org.eclipse.emf.codegen.ecore.generator.Generator;
import org.eclipse.emf.codegen.ecore.generator.GeneratorAdapterFactory;
import org.eclipse.emf.codegen.ecore.genmodel.GenClass;
import org.eclipse.emf.codegen.ecore.genmodel.GenDelegationKind;
import org.eclipse.emf.codegen.ecore.genmodel.GenEnum;
import org.eclipse.emf.codegen.ecore.genmodel.GenJDKLevel;
import org.eclipse.emf.codegen.ecore.genmodel.GenModel;
import org.eclipse.emf.codegen.ecore.genmodel.GenModelFactory;
import org.eclipse.emf.codegen.ecore.genmodel.GenModelPackage;
import org.eclipse.emf.codegen.ecore.genmodel.GenPackage;
import org.eclipse.emf.codegen.ecore.genmodel.GenRuntimeVersion;
import org.eclipse.emf.codegen.ecore.genmodel.generator.GenBaseGeneratorAdapter;
import org.eclipse.emf.codegen.ecore.genmodel.generator.GenModelGeneratorAdapterFactory;
import org.eclipse.emf.codegen.ecore.templates.model.AdapterFactoryClass;
import org.eclipse.emf.codegen.ecore.templates.model.EnumClass;
import org.eclipse.emf.codegen.ecore.templates.model.FactoryClass;
import org.eclipse.emf.codegen.ecore.templates.model.PackageClass;
import org.eclipse.emf.codegen.ecore.templates.model.ResourceClass;
import org.eclipse.emf.codegen.ecore.templates.model.ResourceFactoryClass;
import org.eclipse.emf.codegen.ecore.templates.model.SwitchClass;
import org.eclipse.emf.codegen.ecore.templates.model.XMLProcessorClass;
import org.eclipse.emf.codegen.util.ImportManager;
import org.eclipse.emf.common.util.BasicMonitor;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import org.eclipse.emf.ecore.xmi.impl.EcoreResourceFactoryImpl;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.sleepycat.je.DatabaseException;

public class DataObjectGenerator {

	private static final Logger LOGGER = LoggerFactory.getLogger(DataObjectGenerator.class);
	private static final String MODEL_PACKAGE = "org.bimserver.models.";
	private final Schema schema;

	public DataObjectGenerator(Schema schema) {
		this.schema = schema;
	}

	public VirtualFile generate(VirtualFile basedir) throws DatabaseException {
		GenModel genModel = GenModelFactory.eINSTANCE.createGenModel();
		genModel.setRuntimeVersion(GenRuntimeVersion.EMF25);
		genModel.setComplianceLevel(GenJDKLevel.JDK60_LITERAL);
		genModel.setFeatureDelegation(GenDelegationKind.REFLECTIVE_LITERAL);
		genModel.setRootExtendsClass("org.bimserver.emf.IdEObjectImpl");
		genModel.setRootExtendsInterface("org.bimserver.emf.IdEObject");
		genModel.setSuppressContainment(true);
		genModel.setCodeFormatting(true);
		genModel.setCopyrightText(StringUtils.readFromFile(new File("copyright.txt")));
		genModel.setCanGenerate(true);
		genModel.setModelDirectory("test");
		genModel.setModelName("model name");
		genModel.setForceOverwrite(true);
		genModel.setSuppressNotification(false);
		genModel.setContainmentProxies(false);
		genModel.setBinaryCompatibleReflectiveMethods(false);
		genModel.setPublicConstructors(false);
		genModel.setMinimalReflectiveMethods(true); // More code, but faster
		genModel.setUpdateClasspath(false);

		genModel.initialize(schema.getEPackages());
		
		GeneratorAdapterFactory.Descriptor.Registry.INSTANCE.addDescriptor(GenModelPackage.eNS_URI, GenModelGeneratorAdapterFactory.DESCRIPTOR);
		Generator generator = new Generator();
		generator.setInput(genModel);
		generator.generate(genModel, GenBaseGeneratorAdapter.MODEL_PROJECT_TYPE, new BasicMonitor.Printing(System.out));

		
//		List genPackages = createGenPackages(genModel, basedir);
//		generatePackages(genModel, genPackages, basedir);

		return basedir;
	}

	public void generatePackages(GenModel genModel, List genPackages, VirtualFile basedir) {
		for (GenPackage genPackage : genPackages) {
			genPackage.prepareCache();

			genModel.setImportManager(new ImportManager("org.bimserver.models"));

			String packageClassPathImpl = MODEL_PACKAGE + genPackage.getPackageName() + ".impl.";
			packageClassPathImpl = packageClassPathImpl.replace(".", File.separator) + genPackage.getPackageClassName() + ".java";
			VirtualFile packageVirtualFileImpl = basedir.createFile(packageClassPathImpl);
			packageVirtualFileImpl.setStringContent(new PackageClass().generate(new Object[] { genPackage, false, true }));

			genModel.setImportManager(new ImportManager("org.bimserver.models"));

			String packageClassPathInterface = MODEL_PACKAGE + genPackage.getPackageName() + ".";
			packageClassPathInterface = packageClassPathInterface.replace(".", File.separator) + genPackage.getPackageInterfaceName() + ".java";
			VirtualFile packageVirtualFileInterface = basedir.createFile(packageClassPathInterface);
			packageVirtualFileInterface.setStringContent(new PackageClass().generate(new Object[] { genPackage, true, false }));

			genModel.setImportManager(new ImportManager("org.bimserver.models"));

			String factoryClassPathImpl = MODEL_PACKAGE + genPackage.getPackageName() + ".impl.";
			factoryClassPathImpl = factoryClassPathImpl.replace(".", File.separator) + genPackage.getFactoryClassName() + ".java";
			VirtualFile factoryVirtualFileImpl = basedir.createFile(factoryClassPathImpl);
			factoryVirtualFileImpl.setStringContent(new FactoryClass().generate(new Object[] { genPackage, false, true }));

			genModel.setImportManager(new ImportManager("org.bimserver.models"));

			String factoryClassPathInterface = MODEL_PACKAGE + genPackage.getPackageName() + ".";
			factoryClassPathInterface = factoryClassPathInterface.replace(".", File.separator) + genPackage.getFactoryInterfaceName() + ".java";
			VirtualFile factoryVirtualFileInterface = basedir.createFile(factoryClassPathInterface);
			factoryVirtualFileInterface.setStringContent(new FactoryClass().generate(new Object[] { genPackage, true, false }));

			genModel.setImportManager(new ImportManager("org.bimserver.models"));

			String adapterFactoryClassPath = MODEL_PACKAGE + genPackage.getPackageName() + ".util.";
			adapterFactoryClassPath = adapterFactoryClassPath.replace(".", File.separator) + genPackage.getAdapterFactoryClassName() + ".java";
			VirtualFile adapterFactoryVirtualFile = basedir.createFile(adapterFactoryClassPath);
			adapterFactoryVirtualFile.setStringContent(new AdapterFactoryClass().generate(genPackage));

			genModel.setImportManager(new ImportManager("org.bimserver.models"));

			String switchClassPath = MODEL_PACKAGE + genPackage.getPackageName() + ".util.";
			switchClassPath = switchClassPath.replace(".", File.separator) + genPackage.getSwitchClassName() + ".java";
			VirtualFile switchClassVirtualFile = basedir.createFile(switchClassPath);
			switchClassVirtualFile.setStringContent(new SwitchClass().generate(genPackage));

			genModel.setImportManager(new ImportManager("org.bimserver.models"));

			String xmlClassPath = MODEL_PACKAGE + genPackage.getPackageName() + ".util.";
			xmlClassPath = xmlClassPath.replace(".", File.separator) + genPackage.getXMLProcessorClassName() + ".java";
			VirtualFile xmlClassVirtualFile = basedir.createFile(xmlClassPath);
			xmlClassVirtualFile.setStringContent(new XMLProcessorClass().generate(genPackage));

			genModel.setImportManager(new ImportManager("org.bimserver.models"));

			String resourceFactoryClassPath = MODEL_PACKAGE + genPackage.getPackageName() + ".util.";
			resourceFactoryClassPath = resourceFactoryClassPath.replace(".", File.separator) + genPackage.getResourceFactoryClassName() + ".java";
			VirtualFile resourceFactoryClassVirtualFile = basedir.createFile(resourceFactoryClassPath);
			resourceFactoryClassVirtualFile.setStringContent(new ResourceFactoryClass().generate(genPackage));

			genModel.setImportManager(new ImportManager("org.bimserver.models"));

			String resourceClassPath = MODEL_PACKAGE + genPackage.getPackageName() + ".util.";
			resourceClassPath = resourceClassPath.replace(".", File.separator) + genPackage.getResourceClassName() + ".java";
			VirtualFile resourceClassVirtualFile = basedir.createFile(resourceClassPath);
			resourceClassVirtualFile.setStringContent(new ResourceClass().generate(genPackage));

			// String validatorClassPath = "com.logiclabs.streamdb.models." +
			// genPackage.getPackageName() + ".util.";
			// validatorClassPath = validatorClassPath.replace(".",
			// File.separator) + genPackage.getValidatorClassName() + ".java";
			// VirtualFile validatorClassVirtualFile =
			// basedir.createFile(validatorClassPath);
			// validatorClassVirtualFile.setStringContent(new
			// ValidatorClass().generate(genPackage));

			for (GenClass genClass : genPackage.getGenClasses()) {
				genModel.setImportManager(new ImportManager("org.bimserver.models"));
				String implFileName = MODEL_PACKAGE + genPackage.getPackageName() + ".impl.";
				implFileName = implFileName.replace(".", File.separator) + genClass.getName() + "Impl.java";
				VirtualFile implVirtualFile = basedir.createFile(implFileName);
				implVirtualFile.setStringContent(new org.eclipse.emf.codegen.ecore.templates.model.Class().generate(new Object[] { genClass, false, true }));

				genModel.setImportManager(new ImportManager("org.bimserver.models"));
				String interfaceFileName = MODEL_PACKAGE + genPackage.getPackageName() + ".";
				interfaceFileName = interfaceFileName.replace(".", File.separator) + genClass.getName() + ".java";
				VirtualFile interfaceVirtualFile = basedir.createFile(interfaceFileName);
				interfaceVirtualFile.setStringContent(new org.eclipse.emf.codegen.ecore.templates.model.Class().generate(new Object[] { genClass, true, false }));
			}
			for (GenEnum genEnum : genPackage.getGenEnums()) {
				genModel.setImportManager(new ImportManager("org.bimserver.models"));

				String enumFileName = MODEL_PACKAGE + genPackage.getPackageName() + ".";
				enumFileName = enumFileName.replace(".", File.separator) + genEnum.getName() + ".java";
				VirtualFile enumVirtualFile = basedir.createFile(enumFileName);
				enumVirtualFile.setStringContent(new EnumClass().generate(genEnum));
			}
		}
	}

	public List createGenPackages(GenModel genModel, VirtualFile basedir) throws DatabaseException {
		ResourceSetImpl resourceSet = new ResourceSetImpl();
		resourceSet.getResourceFactoryRegistry().getExtensionToFactoryMap().put("ecore", new EcoreResourceFactoryImpl());
		Map resources = new HashMap();
		genModel.initialize(schema.getEPackages());
		for (GenPackage genPackage : genModel.getGenPackages()) {
			genPackage.setBasePackage("org.bimserver.models");
		}
		for (EPackage ePackage : schema.getEPackages()) {
			// GenPackage genPackage = genModel.get
			// result.add(genPackage);
			// genModel.getGenPackages().add(genPackage);
			// genPackage.initialize(ePackage);
			// for (EClassifier eClassifier : ePackage.getEClassifiers()) {
			// if (eClassifier instanceof EClass) {
			// EClass eClass = (EClass) eClassifier;
			// GenClass genClass = GenModelFactory.eINSTANCE.createGenClass();
			// genClass.initialize(eClass);
			// genPackage.getGenClasses().add(genClass);
			// for (EStructuralFeature eStructuralFeature :
			// eClass.getEStructuralFeatures()) {
			// GenFeature genFeature =
			// GenModelFactory.eINSTANCE.createGenFeature();
			// genFeature.initialize(eStructuralFeature);
			// genFeature.setEcoreFeature(eStructuralFeature);
			// genClass.getGenFeatures().add(genFeature);
			// }
			// } else if (eClassifier instanceof EEnum) {
			// EEnum eEnum = (EEnum) eClassifier;
			// GenEnum genEnum = GenModelFactory.eINSTANCE.createGenEnum();
			// genEnum.setEcoreEnum(eEnum);
			// genPackage.getGenEnums().add(genEnum);
			// for (EEnumLiteral eEnumLiteral : eEnum.getELiterals()) {
			// GenEnumLiteral genEnumLiteral =
			// GenModelFactory.eINSTANCE.createGenEnumLiteral();
			// genEnumLiteral.setEcoreEnumLiteral(eEnumLiteral);
			// genEnum.getGenEnumLiterals().add(genEnumLiteral);
			// }
			// } else if (eClassifier instanceof EDataType) {
			// GenDataType genDataType =
			// GenModelFactory.eINSTANCE.createGenDataType();
			// genDataType.setEcoreDataType((EDataType) eClassifier);
			// genPackage.getGenDataTypes().add(genDataType);
			// }
			// }
			Resource resource = resourceSet.createResource(URI.createURI(ePackage.getName() + ".ecore"));
			resource.getContents().add(ePackage);
			VirtualFile ecoreFile = basedir.createFile(MODEL_PACKAGE.replace(".", "/") + ePackage.getName() + "/impl/" + ePackage.getName() + ".ecore");
			resources.put(ecoreFile, resource);
		}
		for (VirtualFile virtualFile : resources.keySet()) {
			try {
				Resource resource = resources.get(virtualFile);
				OutputStream openOutputStream = virtualFile.openOutputStream();
				resource.save(openOutputStream, null);
				openOutputStream.close();
			} catch (IOException e) {
				LOGGER.error("", e);
			}
		}
		return genModel.getGenPackages();
	}
}