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;h=db730b1e6fafcec231b54ecdde6afe73d36e0d57;hb=refs%2Fchanges%2F70%2F1870%2F1;hp=cf10ffab7c85923d8824b94fcbaefb4363e17e1c;hpb=1bc60c2213f9b3fc7b4d935ba9afda2b767290e5;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 cf10ffab..db730b1e 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
@@ -1,6 +1,7 @@
 package org.simantics.district.network.ui.adapters;
 
 import java.awt.Color;
+import java.awt.geom.AffineTransform;
 import java.awt.geom.Rectangle2D;
 import java.util.Collection;
 import java.util.Collections;
@@ -59,6 +60,9 @@ public class DistrictNetworkEdgeElement {
                 node.setColor(ElementUtils.getAdditionalColor(edgeElement, Color.BLUE));
                 
                 node.setDNEdge(edge);
+                AffineTransform at = ElementUtils.getTransform(edgeElement);
+                if (at != null)
+                    node.setTransform(at);
             }
         }
 
@@ -83,13 +87,12 @@ public class DistrictNetworkEdgeElement {
             if (size == null)
                 size = new Rectangle2D.Double();
             if (edge != null)
-                edge.getBounds(size);
+                size.setFrame(DistrictNetworkEdgeNode.calculatePath(edge).getBounds2D());
             else
                 LOGGER.debug("Element {} does not have edge!", e);
 
             return size;
         }
-        
     }
     
     static class DNEdgeConnectionHandler implements ConnectionHandler {