X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.district.network%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fprofile%2FDNElementColorStyle.java;fp=org.simantics.district.network%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fprofile%2FDNElementColorStyle.java;h=682c91fb1d00782b53ddae99bba9e875b1d14bcf;hb=00803fe68b14b5dce310a5a88b9c2a55bd1c85e6;hp=0000000000000000000000000000000000000000;hpb=b654d9f368d601565303bcf646fc4fbdfbc37090;p=simantics%2Fdistrict.git

diff --git a/org.simantics.district.network/src/org/simantics/district/network/profile/DNElementColorStyle.java b/org.simantics.district.network/src/org/simantics/district/network/profile/DNElementColorStyle.java
new file mode 100644
index 00000000..682c91fb
--- /dev/null
+++ b/org.simantics.district.network/src/org/simantics/district/network/profile/DNElementColorStyle.java
@@ -0,0 +1,56 @@
+package org.simantics.district.network.profile;
+
+import java.awt.Color;
+
+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.diagram.profile.StyleBase;
+import org.simantics.scenegraph.INode;
+import org.simantics.scenegraph.g2d.nodes.SingleElementNode;
+import org.simantics.scenegraph.profile.EvaluationContext;
+import org.simantics.scenegraph.profile.common.ProfileVariables;
+
+/**
+ * @author Tuukka Lehtonen
+ * @since 1.35.0
+ */
+public class DNElementColorStyle extends StyleBase<Color> {
+
+	private static final boolean DEBUG = false;
+
+	@Override
+	public Color calculateStyle(ReadGraph graph, Resource runtimeDiagram, Resource entry, Resource groupItem) throws DatabaseException {
+		DiagramSettings ds = graph.syncRequest(new DiagramSettingsRequest(runtimeDiagram), TransientCacheAsyncListener.instance());
+		if (ds.elementColoringFunction.isPresent()) {
+			if (DEBUG)
+				System.out.print("elementColoringFunction: " + ds.elementColoringFunction + "(" + groupItem + "): ");
+			Double t = Simantics.applySCLRead(graph, ds.elementColoringFunction.get(), groupItem);
+			if (DEBUG)
+				System.out.print(t);
+			if (t != null) {
+				Color result = ds.coloringGradient.get(t);
+				//System.out.println("color(" + t + "): " + result);
+				return result;
+			}
+		}
+		return null;
+	}
+
+	@Override
+	public void applyStyleForNode(EvaluationContext observer, INode node, Color color) {
+		SingleElementNode n = (SingleElementNode) node;
+		for (INode nn : n.getNodes())
+			ProfileVariables.claimNodeProperty(nn, "dynamicColor", color, observer);
+	}
+
+	@Override
+	protected void cleanupStyleForNode(EvaluationContext evaluationContext, INode node) {
+		SingleElementNode n = (SingleElementNode) node;
+		for (INode nn : n.getNodes())
+			ProfileVariables.claimNodeProperty(nn, "dynamicColor", null, evaluationContext);
+	}
+
+}