bundles/org.simantics.modeling.template2d.ui/src/org/simantics/modeling/template2d/ui/actions/AssignDrawingTemplate.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2014 Association for Decentralized Information Management in\r
   3  * Industry THTH ry.\r
   4  * All rights reserved. This program and the accompanying materials\r
   5  * are made available under the terms of the Eclipse Public License v1.0\r
   6  * which accompanies this distribution, and is available at\r
   7  * http://www.eclipse.org/legal/epl-v10.html\r
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *     Semantum Oy - enabled template unassignment (#5030)\r
  12  *     Semantum Oy - multi-selection support (#5425)\r
  13  *******************************************************************************/\r
  14 package org.simantics.modeling.template2d.ui.actions;\r
  15 \r
  16 import gnu.trove.set.hash.THashSet;\r
  17 \r
  18 import java.util.ArrayList;\r
  19 import java.util.Collection;\r
  20 import java.util.Collections;\r
  21 import java.util.List;\r
  22 import java.util.Set;\r
  23 \r
  24 import org.eclipse.jface.dialogs.IDialogConstants;\r
  25 import org.eclipse.jface.dialogs.IDialogSettings;\r
  26 import org.eclipse.jface.viewers.BaseLabelProvider;\r
  27 import org.eclipse.jface.viewers.ILabelProvider;\r
  28 import org.eclipse.jface.window.Window;\r
  29 import org.eclipse.swt.graphics.Image;\r
  30 import org.eclipse.swt.widgets.Button;\r
  31 import org.eclipse.swt.widgets.Composite;\r
  32 import org.eclipse.swt.widgets.Display;\r
  33 import org.eclipse.swt.widgets.Shell;\r
  34 import org.eclipse.ui.PlatformUI;\r
  35 import org.simantics.NameLabelUtil;\r
  36 import org.simantics.Simantics;\r
  37 import org.simantics.db.ReadGraph;\r
  38 import org.simantics.db.Resource;\r
  39 import org.simantics.db.WriteGraph;\r
  40 import org.simantics.db.common.CommentMetadata;\r
  41 import org.simantics.db.common.request.PossibleIndexRoot;\r
  42 import org.simantics.db.common.request.ReadRequest;\r
  43 import org.simantics.db.common.request.WriteRequest;\r
  44 import org.simantics.db.common.utils.NameUtils;\r
  45 import org.simantics.db.exception.DatabaseException;\r
  46 import org.simantics.db.layer0.adapter.ActionFactory;\r
  47 import org.simantics.db.layer0.adapter.ActionFactory2;\r
  48 import org.simantics.db.layer0.adapter.Instances;\r
  49 import org.simantics.modeling.ModelingResources;\r
  50 import org.simantics.modeling.template2d.ontology.Template2dResource;\r
  51 import org.simantics.modeling.template2d.ui.Activator;\r
  52 import org.simantics.structural.stubs.StructuralResource2;\r
  53 import org.simantics.ui.workbench.dialogs.ResourceListDialog;\r
  54 import org.simantics.utils.ObjectUtils;\r
  55 import org.simantics.utils.datastructures.Pair;\r
  56 \r
  57 public class AssignDrawingTemplate implements ActionFactory, ActionFactory2 {\r
  58 \r
  59     @Override\r
  60     public Runnable create(Collection<?> targets) {\r
  61         final List<Resource> resources = new ArrayList<Resource>();\r
  62         for (Object target : targets) {\r
  63             if (!(target instanceof Resource))\r
  64                 return null;\r
  65             resources.add((Resource) target);\r
  66         }\r
  67         return (targets.isEmpty()) ? null : assign(resources);\r
  68     }\r
  69 \r
  70     @Override\r
  71     public Runnable create(Object target) {\r
  72         if(!(target instanceof Resource))\r
  73             return null;\r
  74         return assign(Collections.singletonList((Resource) target));\r
  75     }\r
  76 \r
  77     private Runnable assign(final Collection<Resource> composites) {\r
  78         return new Runnable() {\r
  79             @Override\r
  80             public void run() {\r
  81                 Simantics.getSession().asyncRequest(new ReadRequest() {\r
  82                     @Override\r
  83                     public void run(ReadGraph graph) throws DatabaseException {\r
  84                         assign(graph, composites);\r
  85                     }\r
  86                 });\r
  87             }\r
  88         };\r
  89     }\r
  90 \r
  91     private void assign(ReadGraph graph, final Collection<Resource> composites) throws DatabaseException {\r
  92         Template2dResource TEMPLATE2D = Template2dResource.getInstance(graph);\r
  93         StructuralResource2 STR = StructuralResource2.getInstance(graph);\r
  94         ModelingResources MOD = ModelingResources.getInstance(graph);\r
  95 \r
  96         Instances query = graph.adapt(TEMPLATE2D.DrawingTemplate, Instances.class);\r
  97         Resource root = null;\r
  98         Set<Resource> initialResources = new THashSet<Resource>();\r
  99 \r
 100         for (Resource composite : composites) {\r
 101             if (!graph.isInstanceOf(composite, STR.Composite))\r
 102                 return;\r
 103 \r
 104             // Ensure all composites are part of the same index root.\r
 105             Resource r = graph.sync(new PossibleIndexRoot(composite));\r
 106             if (r == null || (root != null && !r.equals(root)))\r
 107                 return;\r
 108             root = r;\r
 109 \r
 110             // Ensure each composite has a diagram.\r
 111             Resource diagram = graph.getPossibleObject(composite, MOD.CompositeToDiagram);\r
 112             if (diagram == null)\r
 113                 return;\r
 114 \r
 115             Resource initial = graph.getPossibleObject(diagram, TEMPLATE2D.HasDrawingTemplate);\r
 116             if (initial != null)\r
 117                 initialResources.add(initial);\r
 118         }\r
 119 \r
 120         final List<Pair<Resource, String>> elements = new ArrayList<Pair<Resource, String>>();\r
 121 \r
 122         for (Resource template : query.find(graph, root)) {\r
 123             String name = NameLabelUtil.modalName(graph, template);\r
 124             elements.add(Pair.make(template, name));\r
 125         }\r
 126 \r
 127         Resource init = initialResources.size() == 1 ? initialResources.iterator().next() : null;\r
 128         final Pair<Resource, String> initialSelection = init != null\r
 129                 ? Pair.make(init, NameLabelUtil.modalName(graph, init))\r
 130                 : null;\r
 131 \r
 132         Display.getDefault().asyncExec(new Runnable() {\r
 133             @Override\r
 134             public void run() {\r
 135                 queryDrawingTemplate(composites, elements, initialSelection);\r
 136             }\r
 137         });\r
 138     }\r
 139 \r
 140     private static void queryDrawingTemplate(\r
 141             final Collection<Resource> composites,\r
 142             List<Pair<Resource, String>> elements,\r
 143             Pair<Resource, String> initialSelection\r
 144             )\r
 145     {\r
 146         Shell shell = PlatformUI.getWorkbench().getActiveWorkbenchWindow().getShell();\r
 147         ResourceListDialog dialog = new ResourceListDialog(shell, "Select template from list", "Select diagram template to assign " + ResourceListDialog.DEFAULT_LABEL_SUFFIX, new LabelProvider()) {\r
 148             @Override\r
 149             protected void createButtonsForButtonBar(Composite parent) {\r
 150                 Button unassign = createButton(parent, IDialogConstants.CLIENT_ID, "Unassign", false);\r
 151                 unassign.setToolTipText("Remove Diagram Template Assignment");\r
 152                 createButton(parent, IDialogConstants.OK_ID, IDialogConstants.OK_LABEL, true);\r
 153                 createButton(parent, IDialogConstants.CANCEL_ID, IDialogConstants.CANCEL_LABEL, false);\r
 154             }\r
 155             @Override\r
 156             protected void buttonPressed(int buttonId) {\r
 157                 super.buttonPressed(buttonId);\r
 158                 if (buttonId == IDialogConstants.CLIENT_ID) {\r
 159                     setSelectionResult(new Object[0]);\r
 160                     setReturnCode(OK);\r
 161                     close();\r
 162                 }\r
 163             }\r
 164             @Override\r
 165             protected IDialogSettings getDialogBoundsSettings() {\r
 166                 return Activator.getDefault().getDialogSettings();\r
 167             }\r
 168         };\r
 169         dialog.setElements(elements);\r
 170         dialog.setInitialSelection(initialSelection);\r
 171         dialog.setMultipleSelection(false);\r
 172         if (dialog.open() == Window.OK) {\r
 173             @SuppressWarnings("unchecked")\r
 174             Pair<Resource, String> result = (Pair<Resource, String>) dialog.getSingleResult();\r
 175             final Resource template = result != null ? result.first : null;\r
 176             if (!ObjectUtils.objectEquals(template, initialSelection)) {\r
 177                 //System.out.println("setting template: " + template);\r
 178                 Simantics.getSession().async(new WriteRequest() {\r
 179                     @Override\r
 180                     public void perform(WriteGraph graph) throws DatabaseException {\r
 181                         assignTemplateToDiagrams(graph, composites, template);\r
 182                     }\r
 183                 });\r
 184             }\r
 185         }\r
 186     }\r
 187 \r
 188     private static class LabelProvider extends BaseLabelProvider implements ILabelProvider {\r
 189 \r
 190         @Override\r
 191         public Image getImage(Object element) {\r
 192             return null;\r
 193         }\r
 194 \r
 195         @SuppressWarnings("unchecked")\r
 196         @Override\r
 197         public String getText(Object element) {\r
 198             return ((Pair<?, String>) element).second;\r
 199         }\r
 200 \r
 201     }\r
 202 \r
 203     public static void assignTemplateToDiagrams(WriteGraph graph, Collection<Resource> composites, Resource template) throws DatabaseException {\r
 204         graph.markUndoPoint();\r
 205         \r
 206         ModelingResources MOD = ModelingResources.getInstance(graph);\r
 207         Template2dResource TEMPLATE2D = Template2dResource.getInstance(graph);\r
 208 \r
 209         StringBuilder comment = new StringBuilder();\r
 210         if (template != null) {\r
 211             comment.append("Assigned diagram template ").append(NameUtils.getSafeName(graph, template, true)).append(" to diagram");\r
 212         } else {\r
 213             comment.append("Removed diagram template from diagram");\r
 214         }\r
 215         if (composites.size() > 1)\r
 216             comment.append('s');\r
 217         comment.append('\n');\r
 218 \r
 219         for (Resource composite : composites) {\r
 220             Resource diagram = graph.getSingleObject(composite, MOD.CompositeToDiagram); \r
 221 \r
 222             graph.deny(diagram, TEMPLATE2D.HasDrawingTemplate);\r
 223             if (template != null)\r
 224                 graph.claim(diagram, TEMPLATE2D.HasDrawingTemplate, null, template);\r
 225 \r
 226             comment.append('\t').append(NameUtils.getSafeName(graph, composite, true));\r
 227         }\r
 228 \r
 229         CommentMetadata cm = graph.getMetadata(CommentMetadata.class);\r
 230         graph.addMetadata( cm.add( comment.toString() ) );\r
 231     }\r
 232 \r
 233 }\r