bundles/core/org.eclipse.apogy.core.invocator.ui/src/org/eclipse/apogy/core/invocator/ui/models/AbstractResultViewModel.java

/*******************************************************************************
 * Copyright (c) 2018 Agence spatiale canadienne / Canadian Space Agency 
 * 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:
 *     Pierre Allard,
 *     Regent L'Archeveque,
 *     Olivier L. Larouche - initial API and implementation
 *
 * SPDX-License-Identifier: EPL-1.0
 *     
 *******************************************************************************/
package org.eclipse.apogy.core.invocator.ui.models;

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

import org.eclipse.apogy.common.emf.ui.emfforms.ApogyCommonEMFUiEMFFormsFacade;
import org.eclipse.apogy.common.emf.ui.emfforms.PropertyType;
import org.eclipse.apogy.core.invocator.AbstractResult;
import org.eclipse.apogy.core.invocator.ApogyCoreInvocatorPackage;
import org.eclipse.emf.ecore.EAttribute;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EReference;
import org.eclipse.emf.ecp.view.spi.model.VControl;
import org.eclipse.emf.ecp.view.spi.model.VView;
import org.eclipse.emf.ecp.view.spi.model.VViewFactory;
import org.eclipse.emf.ecp.view.spi.model.VViewModelProperties;
import org.eclipse.emf.ecp.view.spi.provider.IViewProvider;

public class AbstractResultViewModel implements IViewProvider {

	@Override
	public double canProvideViewModel(EObject eObject, VViewModelProperties properties) {
		if (eObject instanceof AbstractResult) {
			return 10;
		}
		return NOT_APPLICABLE;
	}

	@Override
	public VView provideViewModel(EObject eObject, VViewModelProperties properties) {
		EClass eClass = eObject.eClass();

		VView vView = VViewFactory.eINSTANCE.createView();
		vView.setRootEClass(eClass);
		vView.setVisible(true);

		List<VControl> vControls = new ArrayList<>();

		for (EAttribute attribute : eClass.getEAllAttributes()) {
			PropertyType type = ApogyCommonEMFUiEMFFormsFacade.INSTANCE.getPropertyType(attribute);
			if (type != PropertyType.NONE) {
				VControl vControl = ApogyCommonEMFUiEMFFormsFacade.INSTANCE.createVControl(attribute);
				if (vControl != null)
					vControls.add(vControl);
			}
		}

		// Adds all References
		for (EReference eReference : eClass.getEAllReferences()) {
			// Add the Exception Container only if it is not null.
			if (eReference == ApogyCoreInvocatorPackage.Literals.OPERATION_CALL_RESULT__EXCEPTION_CONTAINER) {
				if (eObject.eGet(eReference) != null) {
					VControl vControl = ApogyCommonEMFUiEMFFormsFacade.INSTANCE.createVControl(eReference);
					if (vControl != null)
						vControls.add(vControl);
				}
			} else {
				PropertyType type = ApogyCommonEMFUiEMFFormsFacade.INSTANCE.getPropertyType(eReference);
				if (type != PropertyType.NONE) {
					VControl vControl = ApogyCommonEMFUiEMFFormsFacade.INSTANCE.createVControl(eReference);
					if (vControl != null)
						vControls.add(vControl);
				}
			}
		}

		SortedSet<VControl> sortedVControls = ApogyCommonEMFUiEMFFormsFacade.INSTANCE
				.sortVControlAlphabetically(vControls);
		vView.getChildren().addAll(sortedVControls);

		return vView;
	}
}