bundles/org.simantics.diagram/src/org/simantics/diagram/contribution/DiagramDynamicMenuContribution.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
   3  * in 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  *******************************************************************************/\r
  12 package org.simantics.diagram.contribution;\r
  13 \r
  14 import org.eclipse.core.runtime.Platform;\r
  15 import org.eclipse.jface.action.Action;\r
  16 import org.eclipse.jface.action.ActionContributionItem;\r
  17 import org.eclipse.jface.action.IContributionItem;\r
  18 import org.eclipse.jface.resource.ImageDescriptor;\r
  19 import org.eclipse.swt.widgets.Menu;\r
  20 import org.eclipse.ui.IWorkbenchPart;\r
  21 import org.simantics.db.ReadGraph;\r
  22 import org.simantics.db.Session;\r
  23 import org.simantics.db.exception.DatabaseException;\r
  24 import org.simantics.g2d.canvas.ICanvasContext;\r
  25 import org.simantics.g2d.diagram.participant.Selection;\r
  26 import org.simantics.ui.contribution.DynamicMenuContribution;\r
  27 import org.simantics.utils.threads.ThreadUtils;\r
  28 import org.simantics.utils.ui.BundleUtils;\r
  29 import org.simantics.utils.ui.workbench.WorkbenchUtils;\r
  30 \r
  31 \r
  32 /**\r
  33  * @author Antti Villberg\r
  34  */\r
  35 abstract public class DiagramDynamicMenuContribution<T> extends DynamicMenuContribution {\r
  36 \r
  37         private static final IContributionItem[] NONE = {};\r
  38 \r
  39         private ICanvasContext canvas;\r
  40 \r
  41         @Override\r
  42         public void fill(Menu menu, int index) {\r
  43                 // Need to grab active part here, we're still in the SWT thread.\r
  44                 IWorkbenchPart activePart = WorkbenchUtils.getActiveWorkbenchPart();\r
  45                 if (activePart == null)\r
  46                         return;\r
  47                 ICanvasContext ctx = (ICanvasContext) activePart.getAdapter(ICanvasContext.class);\r
  48                 if (ctx == null)\r
  49                         return;\r
  50 \r
  51                 this.canvas = ctx;\r
  52                 try {\r
  53                         super.fill(menu, index);\r
  54                 } finally {\r
  55                         this.canvas = null;\r
  56                 }\r
  57         }\r
  58 \r
  59         @Override\r
  60         protected IContributionItem[] getContributionItems(ReadGraph graph, Object[] selection) throws DatabaseException {\r
  61 \r
  62                 T input = computeInput(graph, selection);\r
  63 \r
  64                 if(input == null) return NONE;\r
  65 \r
  66                 return new IContributionItem[] {\r
  67                                 new ActionContributionItem(new Helper(graph.getSession(), canvas, input))\r
  68                 };\r
  69 \r
  70         }\r
  71 \r
  72         public class Helper extends Action {\r
  73 \r
  74                 private final T input;\r
  75                 protected final Session        session;\r
  76                 protected final ICanvasContext context;\r
  77 \r
  78                 public Helper(Session session, ICanvasContext context, T input) {\r
  79                         super(getName(), getImage());\r
  80                         this.session = session;\r
  81                         this.context = context;\r
  82                         this.input = input;\r
  83                 }\r
  84 \r
  85                 @Override\r
  86                 public void run() {\r
  87                         perform(session, context, input);\r
  88                         ThreadUtils.asyncExec(context.getThreadAccess(), new Runnable() {\r
  89                                 @Override\r
  90                                 public void run() {\r
  91                                         if (context.isDisposed())\r
  92                                                 return;\r
  93                                         Selection selection = context.getAtMostOneItemOfClass(Selection.class);\r
  94                                         if (selection != null) {\r
  95                                                 // This prevents workbench selection from being left over.\r
  96                                                 // Also prevents scene graph crap from being left on the screen.\r
  97                                                 selection.clear(0);\r
  98                                         }\r
  99                                 }\r
 100                         });\r
 101                 }\r
 102 \r
 103         }\r
 104 \r
 105         protected ImageDescriptor silk(String name) {\r
 106                 return BundleUtils.getImageDescriptorFromBundle(Platform.getBundle("com.famfamfam.silk"), "/icons/" + name);\r
 107         }\r
 108 \r
 109         abstract protected T computeInput(ReadGraph graph, Object[] selection) throws DatabaseException;\r
 110         abstract protected void perform(Session session, ICanvasContext context, T input);\r
 111         abstract protected String getName();\r
 112         abstract protected ImageDescriptor getImage();\r
 113 \r
 114 }\r