plugins/org.eclipse.egf.producer.fprod/src/org/eclipse/egf/producer/fprod/internal/manager/ProductionPlanManager.java

/**
 * Copyright (c) 2009-2010 Thales Corporate Services S.A.S.
 * 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:
 * Thales Corporate Services S.A.S - initial API and implementation
 */
package org.eclipse.egf.producer.fprod.internal.manager;

import java.util.HashMap;
import java.util.Map;

import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.OperationCanceledException;
import org.eclipse.core.runtime.SubMonitor;
import org.eclipse.egf.common.helper.EMFHelper;
import org.eclipse.egf.core.producer.InvocationException;
import org.eclipse.egf.core.producer.context.IProductionContext;
import org.eclipse.egf.core.producer.context.ProductionContext;
import org.eclipse.egf.model.fcore.Contract;
import org.eclipse.egf.model.fcore.ContractMode;
import org.eclipse.egf.model.fcore.FactoryComponent;
import org.eclipse.egf.model.fcore.Invocation;
import org.eclipse.egf.model.fcore.InvocationContract;
import org.eclipse.egf.model.fcore.OrchestrationParameter;
import org.eclipse.egf.model.fprod.ProductionPlan;
import org.eclipse.egf.model.fprod.ProductionPlanInvocation;
import org.eclipse.egf.model.types.Type;
import org.eclipse.egf.producer.EGFProducerPlugin;
import org.eclipse.egf.producer.fprod.internal.context.ProducerFprodContextFactory;
import org.eclipse.egf.producer.fprod.l10n.FprodProducerMessages;
import org.eclipse.egf.producer.internal.manager.OrchestrationManager;
import org.eclipse.egf.producer.manager.IActivityManager;
import org.eclipse.emf.common.util.BasicDiagnostic;
import org.eclipse.emf.common.util.Diagnostic;
import org.eclipse.osgi.util.NLS;

/**
 * @author Xavier Maysonnave
 * 
 */
public class ProductionPlanManager extends OrchestrationManager<ProductionPlan> {

    private Map<ProductionPlanInvocation, ProductionPlanInvocationManager> _managers;

    public ProductionPlanManager(IActivityManager<FactoryComponent> parent, ProductionPlan productionPlan) throws InvocationException {
        super(parent, productionPlan);
    }

    @Override
    @SuppressWarnings("unchecked")
    public ProductionContext<ProductionPlan, OrchestrationParameter> getInternalProductionContext() throws InvocationException {
        if (_productionContext == null) {
            _productionContext = ProducerFprodContextFactory.createContext((IProductionContext<FactoryComponent, Contract>) getParent().getProductionContext(), getProjectBundleSession(), getElement());
        }
        return (ProductionContext<ProductionPlan, OrchestrationParameter>) _productionContext;
    }

    private Map<ProductionPlanInvocation, ProductionPlanInvocationManager> getProductionPlanManagers() throws InvocationException {
        if (_managers == null && getElement().getInvocations() != null) {
            _managers = new HashMap<ProductionPlanInvocation, ProductionPlanInvocationManager>(getElement().getInvocations().size());
            for (ProductionPlanInvocation invocation : getElement().getInvocations()) {
                _managers.put(invocation, new ProductionPlanInvocationManager(this, invocation));
            }
        }
        return _managers;
    }

    @Override
    public void dispose() throws InvocationException {
        super.dispose();
        Map<ProductionPlanInvocation, ProductionPlanInvocationManager> managers = getProductionPlanManagers();
        if (managers != null) {
            for (Invocation invocation : getElement().getInvocations()) {
                managers.get(invocation).dispose();
            }
        }
    }

    @Override
    public Diagnostic canInvoke() throws InvocationException {
        BasicDiagnostic diagnostic = checkInputElement(false);
        Map<ProductionPlanInvocation, ProductionPlanInvocationManager> managers = getProductionPlanManagers();
        if (managers != null) {
            for (Invocation invocation : getElement().getInvocations()) {
                diagnostic.add(managers.get(invocation).canInvoke());
            }
        }
        return diagnostic;
    }

    @Override
    public void initializeContext() throws InvocationException {
        super.initializeContext();
        Map<ProductionPlanInvocation, ProductionPlanInvocationManager> managers = getProductionPlanManagers();
        if (managers != null) {
            for (Invocation invocation : getElement().getInvocations()) {
                managers.get(invocation).initializeContext();
            }
        }
    }

    public int getSteps() throws InvocationException {
        int steps = 0;
        Map<ProductionPlanInvocation, ProductionPlanInvocationManager> managers = getProductionPlanManagers();
        if (managers != null) {
            for (Invocation invocation : getElement().getInvocations()) {
                steps += managers.get(invocation).getSteps();
            }
        }
        return steps;
    }

    public Diagnostic invoke(IProgressMonitor monitor) throws InvocationException {
        SubMonitor subMonitor = SubMonitor.convert(monitor, null, getSteps() * 100);
        // Check Input
        BasicDiagnostic diagnostic = checkInputElement(true);
        // Do not further process if we are on error
        if (diagnostic.getSeverity() != Diagnostic.ERROR) {
            Map<ProductionPlanInvocation, ProductionPlanInvocationManager> managers = getProductionPlanManagers();
            if (managers != null) {
                for (Invocation invocation : getElement().getInvocations()) {
                    ProductionPlanInvocationManager manager = managers.get(invocation);
                    // Invoke
                    try {
                        diagnostic.add(manager.invoke(subMonitor.newChild(manager.getSteps() * 100, SubMonitor.SUPPRESS_NONE)));
                    } catch (InvocationException ie) {
                        if (ie.getDiagnostic() != null) {
                            diagnostic.add(ie.getDiagnostic());
                        }
                        ie.setDiagnostic(diagnostic);
                        throw ie;
                    }
                    // Do not further process if we are on error
                    if (diagnostic.getSeverity() != Diagnostic.ERROR) {
                        // populate target runtime context
                        populateTargetRuntimeContext(manager);
                    }
                    // Monitor
                    if (subMonitor.isCanceled()) {
                        throw new OperationCanceledException();
                    }
                }
            }
        }
        return diagnostic;
    }

    private void populateTargetRuntimeContext(ProductionPlanInvocationManager manager) throws InvocationException {
        for (InvocationContract contract : manager.getElement().getInvocationContracts(ContractMode.OUT)) {
            // Target Contract
            for (InvocationContract targetInvocationContract : contract.getTargetInvocationContract()) {
                // Nothing to do
                if (targetInvocationContract == null || targetInvocationContract.getInvocation() == null) {
                    continue;
                }
                // Locate target manager
                ProductionPlanInvocationManager targetManager = getProductionPlanManagers().get(targetInvocationContract.getInvocation());
                if (targetManager == null) {
                    throw new InvocationException(new CoreException(EGFProducerPlugin.getDefault().newStatus(IStatus.ERROR, NLS.bind(FprodProducerMessages.ProductionPlanManager_unknown_manager, EMFHelper.getText(targetInvocationContract.getInvocation())), null)));
                }
                // Populate value accordingly
                Type type = contract.getType();
                if (type == null) {
                    type = contract.getInvokedContract().getType();
                }
                Object value = manager.getProductionContext().getOutputValue(contract.getInvokedContract(), type.getType());
                if (targetInvocationContract.getInvokedMode() == ContractMode.IN) {
                    targetManager.getInternalProductionContext().addInputData(targetInvocationContract, type.getType(), value, false);
                } else {
                    targetManager.getInternalProductionContext().addInputData(targetInvocationContract, type.getType(), value, false);
                    targetManager.getInternalProductionContext().addOutputData(targetInvocationContract, type.getType(), value, false);
                }
            }
        }
    }

}