Added Layout information to StateTemplatesModification

plugins/org.yakindu.sct.ui.editor/src/org/yakindu/sct/ui/editor/editor/proposals/StateTemplatesModification.java

@@ -1,19 +1,44 @@
+/**
+ * Copyright (c) 2013 committers of YAKINDU and others.
+ * 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:
+ * 	committers of YAKINDU - initial API and implementation
+ * 
+ */
 package org.yakindu.sct.ui.editor.editor.proposals;
 
 import static org.eclipse.emf.ecore.util.EcoreUtil.getObjectByType;
-import static org.yakindu.sct.model.sgraph.SGraphPackage.Literals.STATECHART;
 
+import java.util.ArrayList;
+import java.util.List;
+
+import org.eclipse.emf.common.util.EList;
 import org.eclipse.emf.common.util.URI;
 import org.eclipse.emf.ecore.EObject;
 import org.eclipse.emf.ecore.resource.Resource;
 import org.eclipse.emf.ecore.resource.ResourceSet;
 import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
+import org.eclipse.gmf.runtime.diagram.core.util.ViewUtil;
+import org.eclipse.gmf.runtime.notation.Bounds;
+import org.eclipse.gmf.runtime.notation.Diagram;
+import org.eclipse.gmf.runtime.notation.Edge;
+import org.eclipse.gmf.runtime.notation.Node;
+import org.eclipse.gmf.runtime.notation.NotationPackage;
 import org.eclipse.gmf.runtime.notation.View;
+import org.yakindu.sct.model.sgraph.Region;
 import org.yakindu.sct.model.sgraph.State;
-import org.yakindu.sct.model.sgraph.Statechart;
+import org.yakindu.sct.ui.editor.providers.SemanticHints;
 
 import de.itemis.xtext.utils.gmf.proposals.AbstractSemanticModification;
 
+/**
+ * 
+ * @author andreas muelder - Initial contribution and API
+ * 
+ */
 public abstract class StateTemplatesModification extends AbstractSemanticModification {
 
 	public abstract String getTemplatePath();
@@ -22,21 +47,44 @@ public abstract class StateTemplatesModification extends AbstractSemanticModific
 		return object instanceof State;
 	}
 
+	@SuppressWarnings("unchecked")
 	@Override
 	protected void internalExecute(EObject semanticObject, View view) {
 		State state = (State) semanticObject;
 		Resource resource = loadTemplateResource(getTemplatePath());
-		// Semantic changes
-		Statechart statechart = (Statechart) getObjectByType(resource.getContents(), STATECHART);
-		state.getRegions().addAll(statechart.getRegions());
-		// Notation changes
-		// TODO: Implement me
-		resource.unload();
+		Diagram diagram = (Diagram) getObjectByType(resource.getContents(), NotationPackage.Literals.DIAGRAM);
+		EList<View> children = diagram.getChildren();
+
+		List<Node> regions = new ArrayList<Node>();
+		for (View child : children) {
+			if (child.getType().equals(SemanticHints.REGION)) {
+				regions.add((Node) child);
+			}
+		}
+		View figureCompartment = ViewUtil.getChildBySemanticHint(view, SemanticHints.STATE_FIGURE_COMPARTMENT);
+		while (regions.size() > 0) {
+			Node child = regions.get(0);
+			regions.remove(child);
+			figureCompartment.insertChild(child);
+			state.getRegions().add((Region) child.getElement());
+		}
+
+		EList<Edge> edges = diagram.getEdges();
+		while (edges.size() > 0) {
+			view.getDiagram().insertEdge(edges.get(0));
+		}
 
+		setPreferredSize(view);
 	}
 
 	protected Resource loadTemplateResource(String templatePath) {
 		ResourceSet set = new ResourceSetImpl();
 		return set.getResource(URI.createPlatformPluginURI(getTemplatePath(), true), true);
 	}
+
+	protected void setPreferredSize(View view) {
+		Bounds bounds = (Bounds) ((Node) view).getLayoutConstraint();
+		bounds.setWidth(-1);
+		bounds.setHeight(-1);
+	}
 }