Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.modeling.ui / src / org / simantics / modeling / ui / modelBrowser2 / contributions / SetSubscriptionEnabledContribution.java

diff --git a/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/modelBrowser2/contributions/SetSubscriptionEnabledContribution.java b/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/modelBrowser2/contributions/SetSubscriptionEnabledContribution.java
new file mode 100644 (file)
index 0000000..e83a670
--- /dev/null
+++ b/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/modelBrowser2/contributions/SetSubscriptionEnabledContribution.java
@@ -0,0 +1,112 @@
+package org.simantics.modeling.ui.modelBrowser2.contributions;\r
+\r
+import java.util.Arrays;\r
+import java.util.Collection;\r
+import java.util.HashSet;\r
+import java.util.Map;\r
+import java.util.Set;\r
+\r
+import org.eclipse.core.runtime.CoreException;\r
+import org.eclipse.core.runtime.IConfigurationElement;\r
+import org.eclipse.core.runtime.IExecutableExtension;\r
+import org.eclipse.jface.action.Action;\r
+import org.eclipse.jface.action.ActionContributionItem;\r
+import org.eclipse.jface.action.IContributionItem;\r
+import org.eclipse.jface.resource.ImageDescriptor;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.Session;\r
+import org.simantics.db.WriteGraph;\r
+import org.simantics.db.common.request.WriteRequest;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.modeling.ModelingResources;\r
+import org.simantics.modeling.requests.SetSubscriptionEnabled;\r
+import org.simantics.modeling.ui.Activator;\r
+import org.simantics.ui.SimanticsUI;\r
+import org.simantics.ui.contribution.DynamicMenuContribution;\r
+import org.simantics.ui.utils.ResourceAdaptionUtils;\r
+\r
+/**\r
+ * @author Tuukka Lehtonen\r
+ */\r
+public class SetSubscriptionEnabledContribution extends DynamicMenuContribution implements IExecutableExtension {\r
+\r
+    String          name    = "";\r
+    ImageDescriptor image   = null;\r
+    boolean         include = true;\r
+\r
+    @Override\r
+    public void setInitializationData(IConfigurationElement config, String propertyName, Object data)\r
+    throws CoreException {\r
+        if (data instanceof Map<?,?>) {\r
+            @SuppressWarnings("unchecked")\r
+            Map<String, String> args = (Map<String, String>) data;\r
+            name = args.get("name");\r
+            String imageId = args.get("image");\r
+            image = Activator.getDefault().getImageRegistry().getDescriptor(imageId);\r
+            include = Boolean.parseBoolean(args.get("include"));\r
+        }\r
+    }\r
+\r
+    private Collection<Resource> toSubscriptions(ReadGraph graph, Collection<Resource> resources) throws DatabaseException {\r
+        ModelingResources mod = ModelingResources.getInstance(graph);\r
+\r
+        Set<Resource> result = new HashSet<Resource>(resources.size());\r
+        for (Resource o : resources) {\r
+            if (graph.isInstanceOf(o, mod.Subscription))\r
+                result.add(o);\r
+        }\r
+\r
+        return result;\r
+    }\r
+\r
+    @Override\r
+    protected Object[] getSelectedObjects() {\r
+        return ResourceAdaptionUtils.toResources(getSelection());\r
+    }\r
+\r
+    @Override\r
+    protected IContributionItem[] getContributionItems(ReadGraph graph, Object[] elements) throws DatabaseException {\r
+        final Resource[] resources = (Resource[]) elements;\r
+        if (resources.length == 0)\r
+            return new IContributionItem[0];\r
+\r
+        // Are there any components in the selection whose inclusion state\r
+        // is different from the argument 'included'?\r
+        final Collection<Resource> subscriptions = toSubscriptions(graph, Arrays.asList(resources));\r
+        if (subscriptions.isEmpty())\r
+            return new IContributionItem[0];\r
+\r
+        ModelingResources mod = ModelingResources.getInstance(graph);\r
+\r
+        boolean foundNonMatchingEnableStates = false;\r
+        for (Resource subscription : subscriptions) {\r
+            Boolean isIncluded = graph.getPossibleRelatedValue(subscription, mod.Subscription_Enabled);\r
+            if (isIncluded == null)\r
+                isIncluded = Boolean.TRUE;\r
+            if (isIncluded != include)\r
+                foundNonMatchingEnableStates = true;\r
+        }\r
+        if (!foundNonMatchingEnableStates)\r
+            return new IContributionItem[0];\r
+\r
+        return new IContributionItem[] {\r
+                new ActionContributionItem(new Action(name, image) {\r
+                    @Override\r
+                    public void run() {\r
+                        Session session = SimanticsUI.peekSession();\r
+                        if (session == null)\r
+                            return;\r
+\r
+                        SimanticsUI.getSession().asyncRequest(new WriteRequest() {\r
+                            @Override\r
+                            public void perform(WriteGraph graph) throws DatabaseException {\r
+                                new SetSubscriptionEnabled(subscriptions, include).perform(graph);\r
+                            }\r
+                        });\r
+                    }\r
+                })\r
+        };\r
+    }\r
+\r
+}\r