tests/org.eclipse.sphinx.tests.emf.serialization.generators/src/org/eclipse/sphinx/tests/emf/serialization/generators/model/EReferenceReferencedWithoutSubtypesModelBuilder.java

/**
 * <copyright>
 *
 * Copyright (c) 2014 itemis and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v2.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/org/documents/epl-2.0/EPL-2.0.html
 *
 * Contributors:
 *     itemis - Initial API and implementation
 *
 * </copyright>
 */
package org.eclipse.sphinx.tests.emf.serialization.generators.model;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.emf.common.util.EList;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.ecore.EReference;
import org.eclipse.sphinx.emf.serialization.XMLPersistenceMappingExtendedMetaData;

@SuppressWarnings("nls")
public class EReferenceReferencedWithoutSubtypesModelBuilder extends AbstractNodeModelBuilder {

	public EReferenceReferencedWithoutSubtypesModelBuilder(String name, int persistenceMappingStrategy) {
		super(name, persistenceMappingStrategy);
	}

	@Override
	protected List<EPackage> createMetaModel(String name, boolean isMany, int persistenceMappingStrategy) {
		List<EPackage> ePackages = new ArrayList<EPackage>();
		EPackage ePackage = createEPackage(name, "http://www.eclipse.org/sphinx/tests/emf/serialization/generators/" + name, "nodes");
		ePackages.add(ePackage);

		EClass nodeEClass = createEClass(ePackage, "Node", "NODE", "NODES", true);
		createEReference(nodeEClass, "child", true, isMany, nodeEClass, "CHILD", "CHILDREN",
				XMLPersistenceMappingExtendedMetaData.XML_PERSISTENCE_MAPPING_STRATEGY__1001__FEATURE_WRAPPER_ELEMENT__CLASSIFIER_ELEMENT);
		createEReference(nodeEClass, "related", false, isMany, nodeEClass, "RELATED-REF", "RELATED-REFS", "-REFTYPE", persistenceMappingStrategy);

		return ePackages;
	}

	@Override
	@SuppressWarnings("unchecked")
	protected EObject createModel(EPackage rootEPackage, EPackage childEPackage, int numberOfChildren) {
		EObject eObject = createNodeEObject(rootEPackage, "Node");

		EReference relatedEReference = (EReference) ((EClass) rootEPackage.getEClassifier("Node")).getEStructuralFeature("related");
		EReference childEReference = (EReference) ((EClass) rootEPackage.getEClassifier("Node")).getEStructuralFeature("child");

		if (null != relatedEReference) {
			if (relatedEReference.isMany()) {
				for (int i = 0; i < numberOfChildren; i++) {
					EObject childNode = createNodeEObject(childEPackage, "Node");
					((EList<EObject>) eObject.eGet(childEReference)).add(childNode);
					((EList<EObject>) eObject.eGet(relatedEReference)).add(childNode);
				}
			} else if (0 < numberOfChildren) {
				EObject childNode = createNodeEObject(childEPackage, "Node");
				eObject.eSet(childEReference, childNode);
				eObject.eSet(relatedEReference, childNode);
			} else {
				// intentionally no child references
			}
		} else {
			// no children reference in meta model
		}

		return eObject;
	}

}