cda/plugins/org.eclipse.mdht.uml.cda/src/org/eclipse/mdht/uml/cda/tests/TestSTDC.java

/*******************************************************************************
 * Copyright (c) 2017 seanmuir.
 * 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:
 *     seanmuir - initial API and implementation
 *
 *******************************************************************************/
package org.eclipse.mdht.uml.cda.tests;

import java.util.List;

import org.eclipse.emf.ecore.EAttribute;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EClassifier;
import org.eclipse.emf.ecore.EDataType;
import org.eclipse.emf.ecore.EEnum;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EReference;
import org.eclipse.emf.ecore.EStructuralFeature;
import org.eclipse.emf.ecore.util.EcoreUtil;
import org.eclipse.emf.ecore.util.FeatureMapUtil;
import org.eclipse.mdht.uml.cda.CDAFactory;
import org.eclipse.mdht.uml.cda.EncompassingEncounter;
import org.eclipse.mdht.uml.cda.Patient;
import org.eclipse.mdht.uml.cda.Person;
import org.eclipse.mdht.uml.cda.PlayingEntity;
import org.eclipse.mdht.uml.cda.Subject;
import org.eclipse.mdht.uml.cda.SubjectPerson;
import org.eclipse.mdht.uml.cda.util.CDAUtil;
import org.eclipse.mdht.uml.hl7.datatypes.DatatypesFactory;
import org.eclipse.mdht.uml.hl7.datatypes.INT;
import org.eclipse.mdht.uml.hl7.datatypes.ST;

/**
 * @author seanmuir
 *
 */
public class TestSTDC {

	static void populate(EObject cd, int depth) {
		if (depth++ > 8) {
			return;
		}

		for (EAttribute ea : cd.eClass().getEAllAttributes()) {

			Object value = null;
			EDataType type = (EDataType) ea.getEType();
			if (type instanceof EEnum) {
				value = EcoreUtil.createFromString(type, ((EEnum) type).getEEnumLiteral(0).getLiteral());
			} else {
				EStructuralFeature feature = cd.eClass().getEStructuralFeature(ea.getName());

				if (FeatureMapUtil.isFeatureMap(feature)) {
					// System.out.println(ea);
				} else {

					try {
						value = EcoreUtil.createFromString(type, "Test");
					} catch (Exception e) {
						// System.out.println(ea);
						// System.out.println(type);
					}
				}
			}
			// System.out.println(value);
			if (value != null) {
				// System.out.println(ea);
				// System.out.println(value);
				// cd.eSet(ea, value);

				if (ea.isMany()) {
					List list = (List) cd.eGet(ea);
					list.add(value);
				} else {
					cd.eSet(ea, value);
				}

			}

			// if (value instanceof String && !type.isInstance(value)) {

			//
			// }

		}

		// System.out.println(EcoreUtil.getURI(cd));
		//
		// // if (cd.eClass().) {
		// //
		// // }

		for (EReference ea : cd.eClass().getEAllReferences()) {

			EClassifier ec = ea.getEType();

			// System.out.println(ec);

			if (ec instanceof EClass) {
				EClass ec1 = (EClass) ec;
				if (!ec1.isAbstract()) {
					EObject eo = EcoreUtil.create((EClass) ec);

					if (!"http://www.openhealthtools.org/mdht/uml/hl7/datatypes".equals(ec1.getEPackage().getNsURI())) {
						// System.out.println(ec1.getEPackage().getNsURI());
						populate(eo, depth);
					}

					if (ea.isMany()) {
						List<EObject> list = (List<EObject>) cd.eGet(ea);
						list.add(eo);
					} else {
						cd.eSet(ea, eo);
					}
				} else {

					ST st = DatatypesFactory.eINSTANCE.createST("asdasdfasdf");
					if (ea.isMany()) {
						List<EObject> list = (List<EObject>) cd.eGet(ea);
						list.add(st);
					} else {
						cd.eSet(ea, st);
					}

				}

			}
		}

	}

	/**
	 * @param args
	 * @throws Exception
	 */
	public static void main(String[] args) throws Exception {
		Patient p = CDAFactory.eINSTANCE.createPatient();

		// p.setEthnicGroupCode(DatatypesFactory.eINSTANCE.createCE("11a", "22"));
		p.getSDTCEthnicGroupCodes().add(
			DatatypesFactory.eINSTANCE.createCE("EthnicGroupCode", "EthnicGroupCodeSystem"));
		p.getSDTCRaceCodes().add(DatatypesFactory.eINSTANCE.createCE("RaceCode", "RaceCodeSystem"));
		p.setSDTCDeceasedInd(DatatypesFactory.eINSTANCE.createBL(false));
		p.setSDTCDeceasedTime(DatatypesFactory.eINSTANCE.createTS("1900"));
		p.setSDTCDesc(DatatypesFactory.eINSTANCE.createED("Description"));
		p.setSDTCMultipleBirthInd(DatatypesFactory.eINSTANCE.createBL(false));
		INT i = DatatypesFactory.eINSTANCE.createINT();
		i.setValue(Integer.valueOf(3));
		p.setSDTCMultipleBirthOrderNumber(i);
		// p.getSDTCEthnicGroupCodes().add(DatatypesFactory.eINSTANCE.createCE("11c", "22"));

		// p.getSDTCEthnicGroupCodes().add(DatatypesFactory.eINSTANCE.createCE("11d", "22"));

		// p.getGuardians().add(CDAFactory.eINSTANCE.createGuardian());
		// p.getGuardians().add(CDAFactory.eINSTANCE.createGuardian());
		// p.getGuardians().add(CDAFactory.eINSTANCE.createGuardian());
		CDAUtil.saveSnippet(p, System.out);

		SubjectPerson sp = CDAFactory.eINSTANCE.createSubjectPerson();

		populate(sp, 4);
		CDAUtil.saveSnippet(sp, System.out);

		Person person = CDAFactory.eINSTANCE.createPerson();

		populate(person, 4);
		CDAUtil.saveSnippet(person, System.out);

		Subject subject = CDAFactory.eINSTANCE.createSubject();
		populate(subject, 4);
		CDAUtil.saveSnippet(subject, System.out);

		PlayingEntity playingEntity = CDAFactory.eINSTANCE.createPlayingEntity();
		populate(playingEntity, 4);
		CDAUtil.saveSnippet(playingEntity, System.out);

		EncompassingEncounter ee = CDAFactory.eINSTANCE.createEncompassingEncounter();
		populate(ee, 1);
		System.out.println("");
		System.out.println("EncompassingEncounter");
		CDAUtil.saveSnippet(ee, System.out);

		showSDTC(CDAFactory.eINSTANCE.createEncompassingEncounter());

		showSDTC(CDAFactory.eINSTANCE.createParticipant2());

		showSDTC(CDAFactory.eINSTANCE.createPerformer2());

		showSDTC(CDAFactory.eINSTANCE.createOrganizer());

		//
		// System.out.println("asdfasdf");
		// Performer2 performer2 = CDAFactory.eINSTANCE.createPerformer2();
		// populate(performer2, 4);
		// CDAUtil.saveSnippet(performer2, System.out);

		// Observation observation = CDAFactory.eINSTANCE.createObservation();
		// IVL_REAL real = DatatypesFactory.eINSTANCE.createIVL_REAL();
		// real.setLow(DatatypesFactory.eINSTANCE.createIVXB_REAL());
		// real.getLow().setValue(9.0);
		// real.setHigh(DatatypesFactory.eINSTANCE.createIVXB_REAL());
		// real.getHigh().setValue(9.0);
		// real.setCenter(DatatypesFactory.eINSTANCE.createIVXB_REAL());
		//
		// observation.getValues().add(real);
		// CDAUtil.saveSnippet(observation, System.out);

		// sp.setAdministrativeGenderCode(DatatypesFactory.eINSTANCE);

	}

	public static void showSDTC(EObject eObject) throws Exception {
		System.out.println("");
		System.out.println("=================================================");
		System.out.println(eObject.eClass().getName());
		populate(eObject, 6);
		CDAUtil.saveSnippet(eObject, System.out);
		System.out.println("");
		System.out.println("=================================================");
	}

}