X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;ds=sidebyside;f=org.simantics.district.network.ui%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fui%2Fadapters%2FDistrictNetworkEdgeElement.java;fp=org.simantics.district.network.ui%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fui%2Fadapters%2FDistrictNetworkEdgeElement.java;h=20c86d8ddec6dbc4e83a88a5a630805bb8474e8f;hb=7ebf5a4d72468b19d55cb79c0494d0538081d2ff;hp=8af6b20a2b873bbea91d2e15fb4f7a4b2b9af43a;hpb=290f621728aa09db5a719f9e7a29d4e342d25425;p=simantics%2Fdistrict.git

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkEdgeElement.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkEdgeElement.java
index 8af6b20a..20c86d8d 100644
--- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkEdgeElement.java
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/adapters/DistrictNetworkEdgeElement.java
@@ -102,17 +102,13 @@ public class DistrictNetworkEdgeElement {
 
         public static final DNEdgeInternalSize INSTANCE = new DNEdgeInternalSize();
 
-        private ThreadLocal<Path2D> path = new ThreadLocal<Path2D>() {
-            protected Path2D initialValue() { return new Path2D.Double(); }
-        };
-
         @Override
         public Rectangle2D getBounds(IElement e, Rectangle2D size) {
-            DistrictNetworkEdge edge = e.getHint(KEY_DN_EDGE);
+            DistrictNetworkEdgeNode edgeNode = e.getHint(KEY_DN_EDGE_NODE);
             if (size == null)
                 size = new Rectangle2D.Double();
-            if (edge != null)
-                size.setFrame(DistrictNetworkEdgeNode.calculatePath(edge, path.get(), true).getBounds2D());
+            if (edgeNode != null)
+                size.setFrame(edgeNode.getBoundsInLocal());
             else
                 LOGGER.debug("Element {} does not have edge!", e);
 
@@ -121,9 +117,9 @@ public class DistrictNetworkEdgeElement {
 
         @Override
         public Shape getElementShape(IElement e) {
-            DistrictNetworkEdge edge = e.getHint(KEY_DN_EDGE);
-            if (edge != null) {
-                return DistrictNetworkEdgeNode.calculatePath(edge, null, true);
+            DistrictNetworkEdgeNode edgeNode = e.getHint(KEY_DN_EDGE_NODE);
+            if (edgeNode != null) {
+                return edgeNode.getPath();
             } else {
                 return getBounds(e, null);
             }