org.obeonetwork.dsl.uml2.design.services.internal.SemanticElementsSwitch.java Source code

Introduction

Here is the source code for org.obeonetwork.dsl.uml2.design.services.internal.SemanticElementsSwitch.java

Source

/*******************************************************************************
 * Copyright (c) 2009, 2011 Obeo.
 * 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:
 *     Obeo - initial API and implementation
 *******************************************************************************/
package org.obeonetwork.dsl.uml2.design.services.internal;

import java.util.Collection;
import java.util.List;

import org.eclipse.emf.ecore.EModelElement;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.uml2.uml.ActivityEdge;
import org.eclipse.uml2.uml.Association;
import org.eclipse.uml2.uml.Connector;
import org.eclipse.uml2.uml.Element;
import org.eclipse.uml2.uml.Extend;
import org.eclipse.uml2.uml.Include;
import org.eclipse.uml2.uml.Message;
import org.eclipse.uml2.uml.Operation;
import org.eclipse.uml2.uml.Property;
import org.eclipse.uml2.uml.Slot;
import org.eclipse.uml2.uml.Transition;
import org.eclipse.uml2.uml.ValueSpecification;
import org.eclipse.uml2.uml.util.UMLSwitch;

import com.google.common.collect.Lists;

/**
 * An EMF Switch used to compute the elements related to another one, which should be displayed on the
 * property view.
 * 
 * @author Cedric Brun <a href="mailto:cedric.brun@obeo.fr">cedric.brun@obeo.fr</a>
 */
public class SemanticElementsSwitch extends UMLSwitch<Object> {

    private List<EObject> semantics = Lists.newArrayList();

    public Collection<EObject> getSemanticElements(EObject cur) {
        semantics = Lists.newArrayList();
        doSwitch(cur);
        /*
         * We are reversing as we want the first specific before.
         */
        return Lists.reverse(semantics);
    }

    @Override
    public Object caseProperty(Property object) {
        ValueSpecification defaultValue = object.getDefaultValue();
        addIfNotNull(defaultValue);
        addIfNotNull(object.getNameExpression());
        return super.caseProperty(object);
    }

    private void addIfNotNull(EObject defaultValue) {
        if (defaultValue != null)
            semantics.add(defaultValue);
    }

    @Override
    public Object caseOperation(Operation object) {
        semantics.addAll(object.getOwnedParameters());
        return super.caseOperation(object);
    }

    @Override
    public Object caseAssociation(Association object) {
        semantics.addAll(object.getMemberEnds());
        return super.caseAssociation(object);
    }

    @Override
    public Object caseInclude(Include object) {
        addIfNotNull(object.getIncludingCase());
        addIfNotNull(object.getAddition());
        return super.caseInclude(object);
    }

    @Override
    public Object caseExtend(Extend object) {
        addIfNotNull(object.getExtension());
        addIfNotNull(object.getExtendedCase());
        return super.caseExtend(object);
    }

    @Override
    public Object caseEModelElement(EModelElement object) {
        semantics.addAll(object.getEAnnotations());
        return super.caseEModelElement(object);
    }

    @Override
    public Object caseMessage(Message object) {
        addIfNotNull(object.getSendEvent());
        addIfNotNull(object.getReceiveEvent());
        return super.caseMessage(object);
    }

    @Override
    public Object caseActivityEdge(ActivityEdge object) {
        addIfNotNull(object.getGuard());
        return super.caseActivityEdge(object);
    }

    @Override
    public Object caseTransition(Transition object) {
        if (object.getGuard() != null) {
            addIfNotNull(object.getGuard().getSpecification());
        }
        return super.caseTransition(object);
    }

    @Override
    public Object caseSlot(Slot object) {
        addIfNotNull(object.getDefiningFeature());
        return super.caseSlot(object);
    }

    @Override
    public Object caseConnector(Connector object) {
        semantics.addAll(object.getEnds());
        return super.caseConnector(object);
    }

    @Override
    public Object caseElement(Element object) {
        semantics.addAll(object.getStereotypeApplications());
        return super.caseElement(object);
    }

    public Object defaultCase(EObject object) {
        addIfNotNull(object);
        return super.defaultCase(object);
    }
}