X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.district.network.ui%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fui%2Fadapters%2FDistrictNetworkEdgeElement.java;h=a53c02d2e0ba98708b8661cf64379f4f6a232545;hb=8034693a5c86d690649a70a62709d29c3fa197aa;hp=8ec1430b83c4e0de2d1bbbb5206505f8323c6b9c;hpb=31bcbe46c93a576cd6ade673a745ea9cb28672da;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 8ec1430b..a53c02d2 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
@@ -10,7 +10,7 @@ import org.simantics.g2d.element.IElement;
 import org.simantics.g2d.element.SceneGraphNodeKey;
 import org.simantics.g2d.element.handler.InternalSize;
 import org.simantics.g2d.element.handler.SceneGraph;
-import org.simantics.g2d.elementclass.connection.EdgeClass.FixedTransform;
+import org.simantics.g2d.element.handler.impl.DefaultTransform;
 import org.simantics.scenegraph.g2d.G2DParentNode;
 import org.simantics.utils.datastructures.hints.IHintContext.Key;
 import org.simantics.utils.datastructures.hints.IHintContext.KeyOf;
@@ -22,7 +22,7 @@ public class DistrictNetworkEdgeElement {
     
     public static final ElementClass CLASS =
             ElementClass.compile(
-                    FixedTransform.INSTANCE,
+                    DefaultTransform.INSTANCE,
                     DNEdgeInternalSize.INSTANCE,
                     DNEdgeSceneGraph.INSTANCE
             ).setId(DistrictNetworkEdgeElement.class.getSimpleName());
@@ -64,7 +64,11 @@ public class DistrictNetworkEdgeElement {
         @Override
         public Rectangle2D getBounds(IElement e, Rectangle2D size) {
             DistrictNetworkEdge edge = e.getHint(KEY_DN_EDGE);
-            return edge.getBounds(size);
+            if (size == null)
+                size = new Rectangle2D.Double();
+            edge.getBounds(size);
+
+            return size;
         }
         
     }