Commit 321b31b3 authored by Martti Käärik's avatar Martti Käärik
Browse files

Updated according to Sirius API changes.

Change-Id: I0584fc45a09bb3c930aa83808887f421655c127c
parent f5123ed0
Loading
Loading
Loading
Loading
+8 −4
Original line number Diff line number Diff line
@@ -28,6 +28,7 @@ import org.eclipse.gmf.runtime.diagram.ui.requests.ZOrderRequest;
import org.eclipse.gmf.runtime.diagram.ui.services.layout.AbstractLayoutEditPartProvider;
import org.eclipse.sirius.diagram.DDiagram;
import org.eclipse.sirius.diagram.business.api.helper.SiriusDiagramUtil;
import org.eclipse.sirius.diagram.description.DiagramDescription;
import org.eclipse.sirius.diagram.ui.edit.api.part.IDDiagramEditPart;
import org.eclipse.sirius.diagram.ui.tools.api.layout.provider.LayoutProvider;
import org.eclipse.sirius.ext.base.Option;
@@ -35,6 +36,7 @@ import org.eclipse.sirius.viewpoint.DMappingBased;
import org.eclipse.sirius.viewpoint.DSemanticDecorator;
import org.eclipse.sirius.viewpoint.DView;
import org.eclipse.sirius.viewpoint.description.RepresentationElementMapping;
import org.eclipse.sirius.viewpoint.description.Viewpoint;
import org.etsi.mts.tdl.ComponentInstance;
import org.etsi.mts.tdl.Connection;
import org.etsi.mts.tdl.GateReference;
@@ -59,10 +61,12 @@ public class SequenceDiagramFreeformLayoutProvider implements LayoutProvider {
		if (container instanceof IDDiagramEditPart) {
			Option<DDiagram> diagram = ((IDDiagramEditPart) container).resolveDDiagram();
			if (diagram.some()) {
				DView repContainer = SiriusDiagramUtil.findRepresentationContainer(diagram.get());
//				if (VIEWPOINT_ID.equals(repContainer.getViewpoint().getName()) &&
//						TEST_DESCRIPTION_DIAGRAM_ID.equals(diagram.get().getDescription().getName()))
//					return true;
				DDiagram d = diagram.get();
				DiagramDescription dDesc = d.getDescription();
				Viewpoint vp = (Viewpoint) dDesc.eContainer();
				if (VIEWPOINT_ID.equals(vp.getName()) &&
						TEST_DESCRIPTION_DIAGRAM_ID.equals(dDesc.getName()))
					return true;
			}
		}
		return false;