1 /*******************************************************************************
\r
2 * Copyright (c) 2014 Association for Decentralized Information Management in
\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
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
16 import gnu.trove.set.hash.THashSet;
\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
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
57 public class AssignDrawingTemplate implements ActionFactory, ActionFactory2 {
\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
65 resources.add((Resource) target);
\r
67 return (targets.isEmpty()) ? null : assign(resources);
\r
71 public Runnable create(Object target) {
\r
72 if(!(target instanceof Resource))
\r
74 return assign(Collections.singletonList((Resource) target));
\r
77 private Runnable assign(final Collection<Resource> composites) {
\r
78 return new Runnable() {
\r
81 Simantics.getSession().asyncRequest(new ReadRequest() {
\r
83 public void run(ReadGraph graph) throws DatabaseException {
\r
84 assign(graph, composites);
\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
96 Instances query = graph.adapt(TEMPLATE2D.DrawingTemplate, Instances.class);
\r
97 Resource root = null;
\r
98 Set<Resource> initialResources = new THashSet<Resource>();
\r
100 for (Resource composite : composites) {
\r
101 if (!graph.isInstanceOf(composite, STR.Composite))
\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
110 // Ensure each composite has a diagram.
\r
111 Resource diagram = graph.getPossibleObject(composite, MOD.CompositeToDiagram);
\r
112 if (diagram == null)
\r
115 Resource initial = graph.getPossibleObject(diagram, TEMPLATE2D.HasDrawingTemplate);
\r
116 if (initial != null)
\r
117 initialResources.add(initial);
\r
120 final List<Pair<Resource, String>> elements = new ArrayList<Pair<Resource, String>>();
\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
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
132 Display.getDefault().asyncExec(new Runnable() {
\r
134 public void run() {
\r
135 queryDrawingTemplate(composites, elements, initialSelection);
\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
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
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
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
165 protected IDialogSettings getDialogBoundsSettings() {
\r
166 return Activator.getDefault().getDialogSettings();
\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
180 public void perform(WriteGraph graph) throws DatabaseException {
\r
181 assignTemplateToDiagrams(graph, composites, template);
\r
188 private static class LabelProvider extends BaseLabelProvider implements ILabelProvider {
\r
191 public Image getImage(Object element) {
\r
195 @SuppressWarnings("unchecked")
\r
197 public String getText(Object element) {
\r
198 return ((Pair<?, String>) element).second;
\r
203 public static void assignTemplateToDiagrams(WriteGraph graph, Collection<Resource> composites, Resource template) throws DatabaseException {
\r
204 graph.markUndoPoint();
\r
206 ModelingResources MOD = ModelingResources.getInstance(graph);
\r
207 Template2dResource TEMPLATE2D = Template2dResource.getInstance(graph);
\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
213 comment.append("Removed diagram template from diagram");
\r
215 if (composites.size() > 1)
\r
216 comment.append('s');
\r
217 comment.append('\n');
\r
219 for (Resource composite : composites) {
\r
220 Resource diagram = graph.getSingleObject(composite, MOD.CompositeToDiagram);
\r
222 graph.deny(diagram, TEMPLATE2D.HasDrawingTemplate);
\r
223 if (template != null)
\r
224 graph.claim(diagram, TEMPLATE2D.HasDrawingTemplate, null, template);
\r
226 comment.append('\t').append(NameUtils.getSafeName(graph, composite, true));
\r
229 CommentMetadata cm = graph.getMetadata(CommentMetadata.class);
\r
230 graph.addMetadata( cm.add( comment.toString() ) );
\r