Arrow length indicators for flow magnitude

[simantics/district.git] / org.simantics.district.network / src / org / simantics / district / network / profile / ArrowLengthStyle.java

diff --git a/org.simantics.district.network/src/org/simantics/district/network/profile/ArrowLengthStyle.java b/org.simantics.district.network/src/org/simantics/district/network/profile/ArrowLengthStyle.java
new file mode 100644 (file)
index 0000000..3b0524d
--- /dev/null
+++ b/org.simantics.district.network/src/org/simantics/district/network/profile/ArrowLengthStyle.java
@@ -0,0 +1,64 @@
+package org.simantics.district.network.profile;
+
+import org.simantics.Simantics;
+import org.simantics.db.ReadGraph;
+import org.simantics.db.Resource;
+import org.simantics.db.common.procedure.adapter.TransientCacheAsyncListener;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.scenegraph.INode;
+import org.simantics.scenegraph.g2d.G2DSceneGraph;
+import org.simantics.scenegraph.g2d.nodes.ConnectionNode;
+import org.simantics.scenegraph.profile.EvaluationContext;
+import org.simantics.scenegraph.profile.common.ProfileVariables;
+
+public class ArrowLengthStyle extends ThrottledStyleBase<Double> {
+
+       private static final Double PENDING = Double.NaN;
+       private static final Double ONE = 1.0;
+
+       @Override
+       public Double calculateThrottledStyle(ReadGraph graph, Resource runtimeDiagram, Resource entry, Resource groupItem) throws DatabaseException {
+               DiagramSettings ds = graph.syncRequest(new DiagramSettingsRequest(runtimeDiagram), TransientCacheAsyncListener.instance());
+               // Prevent PendingVariableExceptions from coming through
+               boolean wasSynchronous = graph.setSynchronous(true);
+               try {
+                       Double length = ONE;
+                       if (ds.arrowLengthProperty.isPresent()) {
+                               length = Simantics.applySCLRead(graph, ds.arrowLengthProperty.get(), groupItem);
+       //                      System.out.println("read thickness: " + thickness + " : " + ds.arrowLengthProperty);
+                               if (length == null) {
+                                       length = ONE;
+                               } else {
+                                       length = length * ds.arrowLengthGain + ds.arrowLengthBias;
+                               }
+                       }
+                       
+                       return length;
+               }
+               finally {
+                       graph.setSynchronous(wasSynchronous);
+               }
+       }
+
+       @Override
+       public void applyThrottledStyleForNode(EvaluationContext observer, INode node, Double value) {
+//             System.out.println("apply: " + node + " : " + value);
+               ConnectionNode n = (ConnectionNode) node;
+               if (value == PENDING) {
+                       ((G2DSceneGraph)node.getRootNode()).setPending(node);
+               } else {
+                       ((G2DSceneGraph)node.getRootNode()).clearPending(node);
+               }
+               for (INode nn : n.getNodes())
+                       ProfileVariables.claimNodeProperty(nn, "arrowLength", value, observer);
+       }
+
+       @Override
+       protected void cleanupStyleForNode(EvaluationContext evaluationContext, INode node) {
+               ((G2DSceneGraph)node.getRootNode()).clearPending(node);
+               ConnectionNode n = (ConnectionNode) node;
+               for (INode nn : n.getNodes())
+                       ProfileVariables.claimNodeProperty(nn, "arrowLength", null, evaluationContext);
+       }
+
+}