--- /dev/null
+/*******************************************************************************\r
+ * Copyright (c) 2014 Association for Decentralized Information Management in\r
+ * Industry THTH ry.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ * VTT Technical Research Centre of Finland - initial API and implementation\r
+ * Semantum Oy - enabled template unassignment (#5030)\r
+ * Semantum Oy - multi-selection support (#5425)\r
+ *******************************************************************************/\r
+package org.simantics.modeling.template2d.ui.actions;\r
+\r
+import gnu.trove.set.hash.THashSet;\r
+\r
+import java.util.ArrayList;\r
+import java.util.Collection;\r
+import java.util.Collections;\r
+import java.util.List;\r
+import java.util.Set;\r
+\r
+import org.eclipse.jface.dialogs.IDialogConstants;\r
+import org.eclipse.jface.dialogs.IDialogSettings;\r
+import org.eclipse.jface.viewers.BaseLabelProvider;\r
+import org.eclipse.jface.viewers.ILabelProvider;\r
+import org.eclipse.jface.window.Window;\r
+import org.eclipse.swt.graphics.Image;\r
+import org.eclipse.swt.widgets.Button;\r
+import org.eclipse.swt.widgets.Composite;\r
+import org.eclipse.swt.widgets.Display;\r
+import org.eclipse.swt.widgets.Shell;\r
+import org.eclipse.ui.PlatformUI;\r
+import org.simantics.NameLabelUtil;\r
+import org.simantics.Simantics;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.WriteGraph;\r
+import org.simantics.db.common.CommentMetadata;\r
+import org.simantics.db.common.request.PossibleIndexRoot;\r
+import org.simantics.db.common.request.ReadRequest;\r
+import org.simantics.db.common.request.WriteRequest;\r
+import org.simantics.db.common.utils.NameUtils;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.layer0.adapter.ActionFactory;\r
+import org.simantics.db.layer0.adapter.ActionFactory2;\r
+import org.simantics.db.layer0.adapter.Instances;\r
+import org.simantics.modeling.ModelingResources;\r
+import org.simantics.modeling.template2d.ontology.Template2dResource;\r
+import org.simantics.modeling.template2d.ui.Activator;\r
+import org.simantics.structural.stubs.StructuralResource2;\r
+import org.simantics.ui.workbench.dialogs.ResourceListDialog;\r
+import org.simantics.utils.ObjectUtils;\r
+import org.simantics.utils.datastructures.Pair;\r
+\r
+public class AssignDrawingTemplate implements ActionFactory, ActionFactory2 {\r
+\r
+ @Override\r
+ public Runnable create(Collection<?> targets) {\r
+ final List<Resource> resources = new ArrayList<Resource>();\r
+ for (Object target : targets) {\r
+ if (!(target instanceof Resource))\r
+ return null;\r
+ resources.add((Resource) target);\r
+ }\r
+ return (targets.isEmpty()) ? null : assign(resources);\r
+ }\r
+\r
+ @Override\r
+ public Runnable create(Object target) {\r
+ if(!(target instanceof Resource))\r
+ return null;\r
+ return assign(Collections.singletonList((Resource) target));\r
+ }\r
+\r
+ private Runnable assign(final Collection<Resource> composites) {\r
+ return new Runnable() {\r
+ @Override\r
+ public void run() {\r
+ Simantics.getSession().asyncRequest(new ReadRequest() {\r
+ @Override\r
+ public void run(ReadGraph graph) throws DatabaseException {\r
+ assign(graph, composites);\r
+ }\r
+ });\r
+ }\r
+ };\r
+ }\r
+\r
+ private void assign(ReadGraph graph, final Collection<Resource> composites) throws DatabaseException {\r
+ Template2dResource TEMPLATE2D = Template2dResource.getInstance(graph);\r
+ StructuralResource2 STR = StructuralResource2.getInstance(graph);\r
+ ModelingResources MOD = ModelingResources.getInstance(graph);\r
+\r
+ Instances query = graph.adapt(TEMPLATE2D.DrawingTemplate, Instances.class);\r
+ Resource root = null;\r
+ Set<Resource> initialResources = new THashSet<Resource>();\r
+\r
+ for (Resource composite : composites) {\r
+ if (!graph.isInstanceOf(composite, STR.Composite))\r
+ return;\r
+\r
+ // Ensure all composites are part of the same index root.\r
+ Resource r = graph.sync(new PossibleIndexRoot(composite));\r
+ if (r == null || (root != null && !r.equals(root)))\r
+ return;\r
+ root = r;\r
+\r
+ // Ensure each composite has a diagram.\r
+ Resource diagram = graph.getPossibleObject(composite, MOD.CompositeToDiagram);\r
+ if (diagram == null)\r
+ return;\r
+\r
+ Resource initial = graph.getPossibleObject(diagram, TEMPLATE2D.HasDrawingTemplate);\r
+ if (initial != null)\r
+ initialResources.add(initial);\r
+ }\r
+\r
+ final List<Pair<Resource, String>> elements = new ArrayList<Pair<Resource, String>>();\r
+\r
+ for (Resource template : query.find(graph, root)) {\r
+ String name = NameLabelUtil.modalName(graph, template);\r
+ elements.add(Pair.make(template, name));\r
+ }\r
+\r
+ Resource init = initialResources.size() == 1 ? initialResources.iterator().next() : null;\r
+ final Pair<Resource, String> initialSelection = init != null\r
+ ? Pair.make(init, NameLabelUtil.modalName(graph, init))\r
+ : null;\r
+\r
+ Display.getDefault().asyncExec(new Runnable() {\r
+ @Override\r
+ public void run() {\r
+ queryDrawingTemplate(composites, elements, initialSelection);\r
+ }\r
+ });\r
+ }\r
+\r
+ private static void queryDrawingTemplate(\r
+ final Collection<Resource> composites,\r
+ List<Pair<Resource, String>> elements,\r
+ Pair<Resource, String> initialSelection\r
+ )\r
+ {\r
+ Shell shell = PlatformUI.getWorkbench().getActiveWorkbenchWindow().getShell();\r
+ ResourceListDialog dialog = new ResourceListDialog(shell, "Select template from list", "Select diagram template to assign " + ResourceListDialog.DEFAULT_LABEL_SUFFIX, new LabelProvider()) {\r
+ @Override\r
+ protected void createButtonsForButtonBar(Composite parent) {\r
+ Button unassign = createButton(parent, IDialogConstants.CLIENT_ID, "Unassign", false);\r
+ unassign.setToolTipText("Remove Diagram Template Assignment");\r
+ createButton(parent, IDialogConstants.OK_ID, IDialogConstants.OK_LABEL, true);\r
+ createButton(parent, IDialogConstants.CANCEL_ID, IDialogConstants.CANCEL_LABEL, false);\r
+ }\r
+ @Override\r
+ protected void buttonPressed(int buttonId) {\r
+ super.buttonPressed(buttonId);\r
+ if (buttonId == IDialogConstants.CLIENT_ID) {\r
+ setSelectionResult(new Object[0]);\r
+ setReturnCode(OK);\r
+ close();\r
+ }\r
+ }\r
+ @Override\r
+ protected IDialogSettings getDialogBoundsSettings() {\r
+ return Activator.getDefault().getDialogSettings();\r
+ }\r
+ };\r
+ dialog.setElements(elements);\r
+ dialog.setInitialSelection(initialSelection);\r
+ dialog.setMultipleSelection(false);\r
+ if (dialog.open() == Window.OK) {\r
+ @SuppressWarnings("unchecked")\r
+ Pair<Resource, String> result = (Pair<Resource, String>) dialog.getSingleResult();\r
+ final Resource template = result != null ? result.first : null;\r
+ if (!ObjectUtils.objectEquals(template, initialSelection)) {\r
+ //System.out.println("setting template: " + template);\r
+ Simantics.getSession().async(new WriteRequest() {\r
+ @Override\r
+ public void perform(WriteGraph graph) throws DatabaseException {\r
+ assignTemplateToDiagrams(graph, composites, template);\r
+ }\r
+ });\r
+ }\r
+ }\r
+ }\r
+\r
+ private static class LabelProvider extends BaseLabelProvider implements ILabelProvider {\r
+\r
+ @Override\r
+ public Image getImage(Object element) {\r
+ return null;\r
+ }\r
+\r
+ @SuppressWarnings("unchecked")\r
+ @Override\r
+ public String getText(Object element) {\r
+ return ((Pair<?, String>) element).second;\r
+ }\r
+\r
+ }\r
+\r
+ public static void assignTemplateToDiagrams(WriteGraph graph, Collection<Resource> composites, Resource template) throws DatabaseException {\r
+ graph.markUndoPoint();\r
+ \r
+ ModelingResources MOD = ModelingResources.getInstance(graph);\r
+ Template2dResource TEMPLATE2D = Template2dResource.getInstance(graph);\r
+\r
+ StringBuilder comment = new StringBuilder();\r
+ if (template != null) {\r
+ comment.append("Assigned diagram template ").append(NameUtils.getSafeName(graph, template, true)).append(" to diagram");\r
+ } else {\r
+ comment.append("Removed diagram template from diagram");\r
+ }\r
+ if (composites.size() > 1)\r
+ comment.append('s');\r
+ comment.append('\n');\r
+\r
+ for (Resource composite : composites) {\r
+ Resource diagram = graph.getSingleObject(composite, MOD.CompositeToDiagram); \r
+\r
+ graph.deny(diagram, TEMPLATE2D.HasDrawingTemplate);\r
+ if (template != null)\r
+ graph.claim(diagram, TEMPLATE2D.HasDrawingTemplate, null, template);\r
+\r
+ comment.append('\t').append(NameUtils.getSafeName(graph, composite, true));\r
+ }\r
+\r
+ CommentMetadata cm = graph.getMetadata(CommentMetadata.class);\r
+ graph.addMetadata( cm.add( comment.toString() ) );\r
+ }\r
+\r
+}\r