X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.district.network%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fprofile%2FHideStyle.java;h=d5f57c8296efcb157a5665ef9432bf45968643f8;hb=9b9ead4280aeb4981d3294230b17abc3c16b4b10;hp=d5f2620fc27732f8cc09cfbc4b231649e6e6d47f;hpb=cf69153ffe21dfc3b771fa9a6e10ca7621c2dab5;p=simantics%2Fdistrict.git

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 d5f2620f..d5f57c82 100644
--- 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,17 +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);
-        n.setHidden(true);
+        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);
-        n.setHidden(false);
+        for (INode nn : n.getNodes())
+            ProfileVariables.claimNodeProperty(nn, "hidden", false, evaluationContext);
     }
 
     @Override