bundles/org.simantics.diagram/src/org/simantics/diagram/ui/WorkbenchSelectionProvider.java

   1 /*******************************************************************************
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management
   3  * in Industry THTH ry.
   4  * All rights reserved. This program and the accompanying materials
   5  * are made available under the terms of the Eclipse Public License v1.0
   6  * which accompanies this distribution, and is available at
   7  * http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  * Contributors:
  10  *     VTT Technical Research Centre of Finland - initial API and implementation
  11  *******************************************************************************/
  12 package org.simantics.diagram.ui;
  13
  14 import java.util.ArrayList;
  15 import java.util.Collections;
  16 import java.util.Optional;
  17 import java.util.concurrent.CopyOnWriteArrayList;
  18 import java.util.concurrent.TimeUnit;
  19
  20 import org.eclipse.jface.viewers.IPostSelectionProvider;
  21 import org.eclipse.jface.viewers.ISelection;
  22 import org.eclipse.jface.viewers.ISelectionChangedListener;
  23 import org.eclipse.jface.viewers.SelectionChangedEvent;
  24 import org.eclipse.jface.viewers.StructuredSelection;
  25 import org.eclipse.ui.IWorkbenchPartSite;
  26 import org.simantics.db.exception.DatabaseException;
  27 import org.simantics.diagram.elements.AdaptableImmutableProxyElement;
  28 import org.simantics.g2d.canvas.ICanvasContext;
  29 import org.simantics.g2d.canvas.impl.AbstractCanvasParticipant;
  30 import org.simantics.g2d.canvas.impl.DependencyReflection.Dependency;
  31 import org.simantics.g2d.canvas.impl.HintReflection.HintListener;
  32 import org.simantics.g2d.diagram.participant.ElementJSON;
  33 import org.simantics.g2d.diagram.participant.Selection;
  34 import org.simantics.g2d.element.ElementHints;
  35 import org.simantics.g2d.element.IElement;
  36 import org.simantics.ui.selection.WorkbenchSelectionUtils;
  37 import org.simantics.utils.datastructures.hints.IHintContext.Key;
  38 import org.simantics.utils.datastructures.hints.IHintObservable;
  39 import org.simantics.utils.threads.IThreadWorkQueue;
  40 import org.simantics.utils.threads.ThreadUtils;
  41 import org.slf4j.Logger;
  42 import org.slf4j.LoggerFactory;
  43
  44 /**
  45  * A canvas participant that listens to the #0 mouse selection and provides it
  46  * forward through the {@link IPostSelectionProvider} interface.
  47  *
  48  * @author Tuukka Lehtonen
  49  */
  50 public class WorkbenchSelectionProvider extends AbstractCanvasParticipant implements IPostSelectionProvider, ElementJSON {
  51
  52     private static final Logger LOGGER = LoggerFactory.getLogger(WorkbenchSelectionProvider.class);
  53
  54     private static final long                                 POST_SELECTION_DELAY = 300;
  55
  56     @Dependency protected Selection selection;
  57
  58     protected IThreadWorkQueue                                swt;
  59     protected IWorkbenchPartSite                              site;
  60     protected ISelection                                      currentSelection = StructuredSelection.EMPTY;
  61     protected CopyOnWriteArrayList<ISelectionChangedListener> listeners        = new CopyOnWriteArrayList<ISelectionChangedListener>();
  62     protected CopyOnWriteArrayList<ISelectionChangedListener> postListeners    = new CopyOnWriteArrayList<ISelectionChangedListener>();
  63
  64     public WorkbenchSelectionProvider(IThreadWorkQueue swt) {
  65         this(swt, null);
  66     }
  67
  68     public WorkbenchSelectionProvider(IThreadWorkQueue swt, IWorkbenchPartSite site) {
  69         this.swt = swt;
  70         this.site = site;
  71         if (site != null)
  72             site.setSelectionProvider(this);
  73     }
  74
  75     @Override
  76     public void addedToContext(ICanvasContext ctx) {
  77         super.addedToContext(ctx);
  78         if (site != null && site.getSelectionProvider() != this) {
  79             swt.asyncExec(new Runnable() {
  80                 @Override
  81                 public void run() {
  82                     if (site.getSelectionProvider() != WorkbenchSelectionProvider.this)
  83                         site.setSelectionProvider(WorkbenchSelectionProvider.this);
  84                 }
  85             });
  86         }
  87     }
  88
  89     @Override
  90     public void removedFromContext(ICanvasContext ctx) {
  91         if (site != null && site.getSelectionProvider() == this) {
  92             swt.asyncExec(new Runnable() {
  93                 @Override
  94                 public void run() {
  95                     if (site.getSelectionProvider() == WorkbenchSelectionProvider.this)
  96                         site.setSelectionProvider(null);
  97                 }
  98             });
  99         }
 100         super.removedFromContext(ctx);
 101     }
 102
 103     @HintListener(Class = Selection.class, Field = "SELECTION0")
 104     public void hintChanged(IHintObservable sender, Key key, Object oldValue, Object newValue) {
 105         Iterable<?> selection = (Iterable<?>) newValue;
 106         final ISelection s = constructAdaptableSelection(selection);
 107         swt.asyncExec(new Runnable() {
 108             @Override
 109             public void run() {
 110                 currentSelection = s;
 111                 fireSelectionChanged(s);
 112                 schedulePostSelectionChanged(s);
 113             }
 114         });
 115     }
 116
 117     @HintListener(Class = Selection.class, Field = "SELECTION0")
 118     public void hintRemoved(IHintObservable sender, Key key, Object oldValue) {
 119         //System.out.println("selection removed: " + oldValue);
 120         final ISelection s = constructAdaptableSelection(Collections.emptyList());
 121         swt.asyncExec(new Runnable() {
 122             @Override
 123             public void run() {
 124                 currentSelection = s;
 125                 fireSelectionChanged(s);
 126                 schedulePostSelectionChanged(s);
 127             }
 128         });
 129     }
 130
 131     // Post selection changed scheduling helper.
 132     private int modCount = 0;
 133
 134     protected void schedulePostSelectionChanged(final ISelection s) {
 135         final int count = ++modCount;
 136         ThreadUtils.getNonBlockingWorkExecutor().schedule(new Runnable() {
 137             @Override
 138             public void run() {
 139                 int newCount = modCount;
 140                 if (count != newCount)
 141                     return;
 142                 if (isRemoved())
 143                     return;
 144                 swt.asyncExec(new Runnable() {
 145                     @Override
 146                     public void run() {
 147                         if (isRemoved())
 148                             return;
 149                         firePostSelectionChanged(s);
 150                     }
 151                 });
 152             }
 153         }, POST_SELECTION_DELAY, TimeUnit.MILLISECONDS);
 154     }
 155
 156     protected ISelection constructAdaptableSelection(Iterable<?> selection) {
 157         ArrayList<Object> objects = new ArrayList<Object>();
 158         for (Object o : selection) {
 159             if (o instanceof IElement) {
 160                 IElement e = (IElement) o;
 161                 Object object = e.getHint(ElementHints.KEY_OBJECT);
 162                 if (object != null) {
 163                     objects.add(new AdaptableImmutableProxyElement(e));
 164                 } else {
 165                     System.out.println("  discarding element from selection, null object for " + e);
 166                 }
 167             } else {
 168                 System.out.println("  unrecognized selection: " + o.getClass() + ": " + o);
 169             }
 170         }
 171         return new StructuredSelection(objects);
 172     }
 173
 174     void fireSelectionChanged(ISelection selection) {
 175         SelectionChangedEvent e = new SelectionChangedEvent(this, selection);
 176         for (ISelectionChangedListener l : listeners)
 177             l.selectionChanged(e);
 178     }
 179
 180     void firePostSelectionChanged(ISelection selection) {
 181         SelectionChangedEvent e = new SelectionChangedEvent(this, selection);
 182         for (ISelectionChangedListener l : postListeners)
 183             l.selectionChanged(e);
 184     }
 185
 186     @Override
 187     public void addPostSelectionChangedListener(ISelectionChangedListener listener) {
 188         postListeners.add(listener);
 189     }
 190
 191     @Override
 192     public void removePostSelectionChangedListener(ISelectionChangedListener listener) {
 193         postListeners.remove(listener);
 194     }
 195
 196     @Override
 197     public void addSelectionChangedListener(ISelectionChangedListener listener) {
 198         listeners.add(listener);
 199     }
 200
 201     @Override
 202     public void removeSelectionChangedListener(ISelectionChangedListener listener) {
 203         listeners.remove(listener);
 204     }
 205
 206     @Override
 207     public ISelection getSelection() {
 208         return currentSelection;
 209     }
 210
 211     @Override
 212     public void setSelection(ISelection selection) {
 213         System.out.println("WorkbenchSelectionProvider: TODO: set selection: " + selection);
 214     }
 215
 216     @Override
 217     public Optional<String> getJSON(IElement element) {
 218         ISelection sel = constructAdaptableSelection(Collections.singleton(element));
 219         try {
 220             return Optional.ofNullable( WorkbenchSelectionUtils.getPossibleJSON(sel) );
 221         } catch (DatabaseException e) {
 222             LOGGER.error("Unexpected failure while constructing JSON from element " + element, e);
 223             return Optional.empty();
 224         }
 225     }
 226
 227 }