]> gerrit.simantics Code Review - simantics/platform.git/blobdiff - bundles/org.simantics.diagram/src/org/simantics/diagram/handler/TerminalConnectionDirectionsMenuContribution.java
Migrated source code from Simantics SVN
[simantics/platform.git] / bundles / org.simantics.diagram / src / org / simantics / diagram / handler / TerminalConnectionDirectionsMenuContribution.java
diff --git a/bundles/org.simantics.diagram/src/org/simantics/diagram/handler/TerminalConnectionDirectionsMenuContribution.java b/bundles/org.simantics.diagram/src/org/simantics/diagram/handler/TerminalConnectionDirectionsMenuContribution.java
new file mode 100644 (file)
index 0000000..74ba999
--- /dev/null
@@ -0,0 +1,160 @@
+/*******************************************************************************\r
+ * Copyright (c) 2013 Association for Decentralized Information Management\r
+ * in Industry THTH ry.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ *     Semantum Oy - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.diagram.handler;\r
+\r
+import java.util.ArrayList;\r
+import java.util.Collection;\r
+import java.util.Collections;\r
+import java.util.List;\r
+\r
+import org.eclipse.jface.action.Action;\r
+import org.eclipse.jface.action.ContributionItem;\r
+import org.eclipse.jface.action.IContributionItem;\r
+import org.eclipse.swt.SWT;\r
+import org.eclipse.swt.events.SelectionEvent;\r
+import org.eclipse.swt.events.SelectionListener;\r
+import org.eclipse.swt.widgets.Menu;\r
+import org.eclipse.swt.widgets.MenuItem;\r
+import org.simantics.databoard.Bindings;\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.diagram.stubs.DiagramResource;\r
+import org.simantics.ui.contribution.DynamicMenuContribution;\r
+import org.simantics.utils.datastructures.Callback;\r
+import org.simantics.utils.ui.AdaptionUtils;\r
+import org.simantics.utils.ui.ExceptionUtils;\r
+\r
+/**\r
+ * @author Tuukka Lehtonen\r
+ */\r
+public class TerminalConnectionDirectionsMenuContribution extends DynamicMenuContribution {\r
+\r
+       private static final IContributionItem[] NONE = {};\r
+\r
+       private static final int[] DIRECTION_MASK = { 8, 4, 1, 2 };\r
+       private static final String[] DIRECTION_NAME = { "Up", "Left", "Right", "Down" };\r
+\r
+       @Override\r
+       protected IContributionItem[] getContributionItems(ReadGraph graph, Object[] selection) throws DatabaseException {\r
+               return getTerminalDirectionItems(graph, selection).toArray(NONE);\r
+       }\r
+\r
+       public static Collection<IContributionItem> getTerminalDirectionItems(ReadGraph graph, Object[] selection) throws DatabaseException {\r
+               DiagramResource DIA = DiagramResource.getInstance(graph);\r
+\r
+               final List<Resource> terminals = new ArrayList<Resource>();\r
+               for (Object s : selection) {\r
+                       Resource connection = AdaptionUtils.adaptToSingle(s, Resource.class);\r
+                       if (connection == null || !graph.isInstanceOf(connection, DIA.Terminal))\r
+                               return Collections.emptyList();\r
+                       terminals.add(connection);\r
+               }\r
+               if (terminals.isEmpty())\r
+                       return Collections.emptyList();\r
+\r
+               int jointMask = 0;\r
+               for (Resource terminal : terminals) {\r
+                       Integer dirs = graph.getPossibleRelatedValue(terminal, DIA.Terminal_AllowedDirections, Bindings.INTEGER);\r
+                       int mask = dirs == null ? 0 : dirs;\r
+                       jointMask |= mask;\r
+               }\r
+\r
+               final List<ToggleDirectionAction> actions = new ArrayList<ToggleDirectionAction>(4);\r
+               for (int i = 0; i < DIRECTION_NAME.length; ++i) {\r
+                       String name = DIRECTION_NAME[i];\r
+                       int mask = DIRECTION_MASK[i];\r
+                       boolean wasChecked = (jointMask & mask) != 0;\r
+                       int orMask = wasChecked ? 0 : mask;\r
+                       int andMask = wasChecked ? ~mask : 0xffffffff;\r
+                       ToggleDirectionAction action = new ToggleDirectionAction(graph.getSession(), terminals, orMask, andMask, name);\r
+                       action.setChecked(wasChecked);\r
+                       actions.add(action);\r
+               }\r
+\r
+               return Collections.<IContributionItem>singleton(\r
+                               new ContributionItem() {\r
+                                       @Override\r
+                                       public void fill(Menu menu, int index) {\r
+                                               MenuItem directions = new MenuItem(menu, SWT.CASCADE, index);\r
+                                               directions.setText("Allowed Routing Directions");\r
+\r
+                                               Menu subMenu = new Menu(menu);\r
+                                               directions.setMenu(subMenu);\r
+\r
+                                               for (ToggleDirectionAction a : actions) {\r
+                                                       MenuItem item = new MenuItem(subMenu, SWT.CHECK);\r
+                                                       item.setText(a.getText());\r
+                                                       item.addSelectionListener(a);\r
+                                                       item.setSelection(a.isChecked());\r
+                                               }\r
+                                       }\r
+                               }\r
+                               );\r
+       }\r
+\r
+       static class ToggleDirectionAction extends Action implements SelectionListener {\r
+\r
+               Session session;\r
+               Collection<Resource> terminals;\r
+               int orMask;\r
+               int andMask;\r
+\r
+               public ToggleDirectionAction(Session session,\r
+                               List<Resource> terminals, int orMask, int andMask,\r
+                               String name) {\r
+                       super(name, Action.AS_CHECK_BOX);\r
+                       this.session = session;\r
+                       this.terminals = terminals;\r
+                       this.orMask = orMask;\r
+                       this.andMask = andMask;\r
+               }\r
+\r
+               @Override\r
+               public void run() {\r
+                       session.asyncRequest(new WriteRequest() {\r
+                               @Override\r
+                               public void perform(WriteGraph graph) throws DatabaseException {\r
+                                       DiagramResource DIA = DiagramResource.getInstance(graph);\r
+                                       for (Resource terminal : terminals) {\r
+                                               Integer dirs = graph.getPossibleRelatedValue(terminal, DIA.Terminal_AllowedDirections, Bindings.INTEGER);\r
+                                               int mask = dirs == null ? 0 : dirs;\r
+                                               mask &= andMask;\r
+                                               mask |= orMask;\r
+                                               graph.claimLiteral(terminal, DIA.Terminal_AllowedDirections, mask, Bindings.INTEGER);\r
+                                       }\r
+                               }\r
+                       }, new Callback<DatabaseException>() {\r
+                               @Override\r
+                               public void run(DatabaseException parameter) {\r
+                                       if (parameter != null)\r
+                                               ExceptionUtils.logError(parameter);\r
+                               }\r
+                       });\r
+               }\r
+\r
+               @Override\r
+               public void widgetDefaultSelected(SelectionEvent e) {\r
+                       widgetSelected(e);\r
+               }\r
+\r
+               @Override\r
+               public void widgetSelected(SelectionEvent e) {\r
+                       run();\r
+               }\r
+\r
+       }\r
+\r
+}\r