bundles/org.simantics.issues.ui/src/org/simantics/issues/ui/handler/MenuActions.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2011 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.issues.ui.handler;\r
  13 \r
  14 import java.util.ArrayList;\r
  15 import java.util.Arrays;\r
  16 import java.util.List;\r
  17 \r
  18 import org.eclipse.jface.action.ActionContributionItem;\r
  19 import org.eclipse.jface.action.ContributionItem;\r
  20 import org.eclipse.jface.action.IAction;\r
  21 import org.eclipse.jface.action.IContributionItem;\r
  22 import org.eclipse.jface.resource.ImageDescriptor;\r
  23 import org.eclipse.jface.viewers.ISelection;\r
  24 import org.eclipse.swt.SWT;\r
  25 import org.eclipse.swt.events.SelectionAdapter;\r
  26 import org.eclipse.swt.events.SelectionEvent;\r
  27 import org.eclipse.swt.widgets.Menu;\r
  28 import org.eclipse.swt.widgets.MenuItem;\r
  29 import org.simantics.Simantics;\r
  30 import org.simantics.db.ReadGraph;\r
  31 import org.simantics.db.Resource;\r
  32 import org.simantics.db.WriteGraph;\r
  33 import org.simantics.db.common.request.WriteRequest;\r
  34 import org.simantics.db.exception.DatabaseException;\r
  35 import org.simantics.db.layer0.SelectionHints;\r
  36 import org.simantics.issues.Severity;\r
  37 import org.simantics.issues.common.IssueConstants;\r
  38 import org.simantics.issues.common.IssueUtils;\r
  39 import org.simantics.issues.ontology.IssueResource;\r
  40 import org.simantics.issues.ui.internal.Activator;\r
  41 import org.simantics.ui.contribution.DynamicMenuContribution;\r
  42 import org.simantics.utils.datastructures.Callback;\r
  43 import org.simantics.utils.ui.ErrorLogger;\r
  44 import org.simantics.utils.ui.ISelectionUtils;\r
  45 \r
  46 /**\r
  47  * @author Tuukka Lehtonen\r
  48  */\r
  49 public class MenuActions extends DynamicMenuContribution {\r
  50 \r
  51     @Override\r
  52     protected Object[] getSelectedObjects() {\r
  53         ISelection sel = getSelection();\r
  54         List<Resource> resources = ISelectionUtils.getPossibleKeys(sel, SelectionHints.KEY_MAIN, Resource.class);\r
  55         return resources.toArray();\r
  56     }\r
  57 \r
  58     List<Resource> toResources(Object[] array) {\r
  59         Resource[] a = new Resource[array.length];\r
  60         for (int i = 0; i < array.length; ++i)\r
  61             a[i] = (Resource) array[i];\r
  62         return Arrays.asList(a);\r
  63     }\r
  64 \r
  65     @Override\r
  66     protected IContributionItem[] getContributionItems(ReadGraph graph, Object[] selection) throws DatabaseException {\r
  67         if (selection.length == 0)\r
  68             return new IContributionItem[0];\r
  69 \r
  70         List<Resource> input = toResources(selection);\r
  71 \r
  72         @SuppressWarnings("unused")\r
  73         boolean issues = false;\r
  74         @SuppressWarnings("unused")\r
  75         boolean userIssues = false;\r
  76 \r
  77         IssueResource ISSUE = IssueResource.getInstance(graph);\r
  78         int issueCount = 0;\r
  79         int hiddenCount = 0; \r
  80         int resolvedCount = 0;\r
  81         int userIssueCount = 0;\r
  82         for (Resource r : input) {\r
  83             boolean isIssue = graph.isInstanceOf(r, ISSUE.Issue);\r
  84             issues |= isIssue;\r
  85             if (isIssue) {\r
  86                 ++issueCount;\r
  87                 if (graph.hasStatement(r, ISSUE.Hidden))\r
  88                     ++hiddenCount;\r
  89                 if (graph.hasStatement(r, ISSUE.Resolved))\r
  90                     ++resolvedCount;\r
  91             }\r
  92 \r
  93             boolean userIssue = graph.hasStatement(r, ISSUE.UserIssue);\r
  94             userIssues |= userIssue;\r
  95             if (userIssue) {\r
  96                 ++userIssueCount;\r
  97             }\r
  98         }\r
  99         boolean allHidden = hiddenCount == selection.length;\r
 100         boolean allResolved = resolvedCount == selection.length;\r
 101         boolean allUserIssues = userIssueCount == selection.length;\r
 102         boolean allIssueSelection = issueCount == selection.length;\r
 103 \r
 104         List<IContributionItem> actions = new ArrayList<IContributionItem>(2); \r
 105         if (!allHidden && allIssueSelection)\r
 106             actions.add(new ActionContributionItem(hideAction(input)));\r
 107         if (hiddenCount > 0 || allHidden && allIssueSelection)\r
 108             actions.add(new ActionContributionItem(unhideAction(input)));\r
 109         if (!allResolved && allIssueSelection)\r
 110             actions.add(new ActionContributionItem(resolveAction(input)));\r
 111         if (resolvedCount > 0 || allResolved && allIssueSelection)\r
 112             actions.add(new ActionContributionItem(unresolveAction(input)));\r
 113         if (allUserIssues)\r
 114             actions.add(setSeverityAction(input));\r
 115 \r
 116         return actions.toArray(new IContributionItem[actions.size()]);\r
 117     }\r
 118 \r
 119     private IAction unhideAction(List<Resource> input) {\r
 120         return tagAction("Unhide", Activator.UNHIDE_ICON, IssueResource.URIs.Hidden, false, input);\r
 121     }\r
 122 \r
 123     private IAction hideAction(List<Resource> input) {\r
 124         return tagAction("Hide", Activator.HIDE_ICON, IssueResource.URIs.Hidden, true, input);\r
 125     }\r
 126 \r
 127     private IAction resolveAction(List<Resource> input) {\r
 128         return tagAction("Mark Resolved", Activator.RESOLVE_ICON, IssueResource.URIs.Resolved, true, input);\r
 129     }\r
 130 \r
 131     private IAction unresolveAction(List<Resource> input) {\r
 132         return tagAction("Mark Unresolved", Activator.UNRESOLVE_ICON, IssueResource.URIs.Resolved, false, input);\r
 133     }\r
 134 \r
 135     private IAction tagAction(String label, ImageDescriptor image, String tagURI, boolean tag, List<Resource> input) {\r
 136         return new TagAction(label, image, IssueConstants.ISSUE_VG, tagURI, tag, input);\r
 137         //return new TagAction(label, image, null, tagURI, tag, input);\r
 138     }\r
 139 \r
 140     private IContributionItem setSeverityAction(final List<Resource> userIssues) {\r
 141         return new ContributionItem() {\r
 142             @Override\r
 143             public void fill(Menu menu, int index) {\r
 144                 MenuItem setSeverityItem = new MenuItem(menu, SWT.CASCADE);\r
 145                 setSeverityItem.setText("Set Severity");\r
 146                 Menu setSeverity = new Menu(menu);\r
 147                 setSeverityItem.setMenu(setSeverity);\r
 148                 for (final Severity sev : Severity.values()) {\r
 149                     if (sev.equals(Severity.UNKNOWN))\r
 150                         continue;\r
 151 \r
 152                     MenuItem item = new MenuItem(setSeverity, SWT.PUSH);\r
 153                     item.setText(sev.toString().toLowerCase());\r
 154                     item.setImage(Activator.getDefault().getImageRegistry().get(sev.toString() + "-full"));\r
 155                     item.addSelectionListener(new SelectionAdapter() {\r
 156                         @Override\r
 157                         public void widgetSelected(SelectionEvent e) {\r
 158                             setSeverity(userIssues, sev);\r
 159                         }\r
 160                     });\r
 161                 }\r
 162             }\r
 163         };\r
 164     }\r
 165 \r
 166     private void setSeverity(final List<Resource> userIssues, final Severity severity) {\r
 167         Simantics.getSession().asyncRequest(new WriteRequest() {\r
 168             @Override\r
 169             public void perform(WriteGraph graph) throws DatabaseException {\r
 170                 IssueResource ISSUE = IssueResource.getInstance(graph);\r
 171                 Resource sev = IssueUtils.toSeverityResource(ISSUE, severity);\r
 172                 if (sev != null) {\r
 173                     for (Resource issue : userIssues) {\r
 174                         if (!graph.hasStatement(issue, ISSUE.Issue_HasSeverity, sev)) {\r
 175                             graph.deny(issue, ISSUE.Issue_HasSeverity);\r
 176                             graph.claim(issue, ISSUE.Issue_HasSeverity, null, sev);\r
 177                         }\r
 178                     }\r
 179                 } else {\r
 180                     for (Resource issue : userIssues)\r
 181                         graph.deny(issue, ISSUE.Issue_HasSeverity);\r
 182                 }\r
 183             }\r
 184         }, new Callback<DatabaseException>() {\r
 185             @Override\r
 186             public void run(DatabaseException e) {\r
 187                 if (e != null)\r
 188                     ErrorLogger.defaultLogError(e);\r
 189             }\r
 190         });\r
 191     }\r
 192 \r
 193 }