Show dynamic symbols when vertices and/or edges are hidden.

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

diff --git a/org.simantics.district.network/src/org/simantics/district/network/profile/HideStyle.java b/org.simantics.district.network/src/org/simantics/district/network/profile/HideStyle.java
index 8e053a065c6d88d923c29d1592586678e2316c46..d5f57c8296efcb157a5665ef9432bf45968643f8 100644 (file)
--- a/org.simantics.district.network/src/org/simantics/district/network/profile/HideStyle.java
+++ b/org.simantics.district.network/src/org/simantics/district/network/profile/HideStyle.java
@@ -7,6 +7,7 @@ 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
@@ -19,15 +20,17 @@ public class HideStyle extends StyleBase<Boolean> {
     }
 
     @Override
-    public void applyStyleForNode(EvaluationContext observer, INode node, Boolean result) {
+    public void applyStyleForNode(EvaluationContext evaluationContext, INode node, Boolean result) {
         SingleElementNode n = (SingleElementNode) node;
-        n.setVisible(false);
+        for (INode nn : n.getNodes())
+            ProfileVariables.claimNodeProperty(nn, "hidden", true, evaluationContext);
     }
-
+    
     @Override
     protected void cleanupStyleForNode(EvaluationContext evaluationContext, INode node) {
         SingleElementNode n = (SingleElementNode) node;
-        n.setVisible(true);
+        for (INode nn : n.getNodes())
+            ProfileVariables.claimNodeProperty(nn, "hidden", false, evaluationContext);
     }
 
     @Override