X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.district.network.ui%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fnetwork%2Fui%2Fnodes%2FDistrictNetworkStaticInfoStyle.java;h=20e8ecbc4fa30727f0f4fa14a1029150350be62c;hb=4e9851013456ea1a8aa20eee5e33ccb97eed03ef;hp=49d524a4b2f9525b9fcde4e4f95c96f466a5e59b;hpb=4dd26a17d52de059617441f465e3bb5f288cf10a;p=simantics%2Fdistrict.git

diff --git a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkStaticInfoStyle.java b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkStaticInfoStyle.java
index 49d524a4..20e8ecbc 100644
--- a/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkStaticInfoStyle.java
+++ b/org.simantics.district.network.ui/src/org/simantics/district/network/ui/nodes/DistrictNetworkStaticInfoStyle.java
@@ -55,13 +55,17 @@ public class DistrictNetworkStaticInfoStyle extends StyleBase<DistrictNetworkSta
 			throws DatabaseException {
 		DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
 		
-		if (!graph.isInstanceOf(mapElement, DN.Edge))
+		boolean isEdge = graph.isInstanceOf(mapElement, DN.Edge);
+		boolean isVertex = graph.isInstanceOf(mapElement, DN.Vertex);
+		if (!isEdge && !isVertex)
 			return null;
 		
-		Resource diagram = graph.getSingleObject(mapElement, Layer0.getInstance(graph).PartOf);
-		Set<Resource> edgesToUse = graph.syncRequest(new MidBranchEdgeSetRequest(diagram), TransientCacheListener.instance());
-		if (!edgesToUse.contains(mapElement))
-			return null;
+		if (isEdge) {
+			Resource diagram = graph.getSingleObject(mapElement, Layer0.getInstance(graph).PartOf);
+			Set<Resource> edgesToUse = graph.syncRequest(new MidBranchEdgeSetRequest(diagram), TransientCacheListener.instance());
+			if (!edgesToUse.contains(mapElement))
+				return null;
+		}
 		
 		DiagramResource DIA = DiagramResource.getInstance(graph);
 		StructuralResource2 STR = StructuralResource2.getInstance(graph);
@@ -90,14 +94,23 @@ public class DistrictNetworkStaticInfoStyle extends StyleBase<DistrictNetworkSta
 			result = null;
 		}
 		
-		Resource v1 = graph.getSingleObject(mapElement, DN.HasStartVertex);
-		double[] coords1 = graph.getRelatedValue(v1, DIA.HasLocation);
-		Resource v2 = graph.getSingleObject(mapElement, DN.HasEndVertex);
-		double[] coords2 = graph.getRelatedValue(v2, DIA.HasLocation);
-		Point2D p1 = DistrictNetworkNodeUtils.calculatePoint2D(new Point2D.Double(coords1[0], coords1[1]), null);
-		Point2D p2 = DistrictNetworkNodeUtils.calculatePoint2D(new Point2D.Double(coords2[0], coords2[1]), null);
+		if (isVertex) {
+			double[] coords = graph.getRelatedValue(mapElement, DIA.HasLocation);
+			Point2D p = DistrictNetworkNodeUtils.calculatePoint2D(new Point2D.Double(coords[0], coords[1]), null);
+			return new StyleResult(p, p, result);
+		}
+		else if (isEdge) {
+			Resource v1 = graph.getSingleObject(mapElement, DN.HasStartVertex);
+			double[] coords1 = graph.getRelatedValue(v1, DIA.HasLocation);
+			Resource v2 = graph.getSingleObject(mapElement, DN.HasEndVertex);
+			double[] coords2 = graph.getRelatedValue(v2, DIA.HasLocation);
+			Point2D p1 = DistrictNetworkNodeUtils.calculatePoint2D(new Point2D.Double(coords1[0], coords1[1]), null);
+			Point2D p2 = DistrictNetworkNodeUtils.calculatePoint2D(new Point2D.Double(coords2[0], coords2[1]), null);
+			
+			return new StyleResult(p1, p2, result);
+		}
 		
-		return new StyleResult(p1, p2, result);
+		return null;
 	}
 	
 	@Override
@@ -114,11 +127,17 @@ public class DistrictNetworkStaticInfoStyle extends StyleBase<DistrictNetworkSta
 		Point2D p1 = result.p1;
 		Point2D p2 = result.p2;
 		
-		double sign = Math.signum(p1.getX() - p2.getX());
-		Point2D.Double origin = new Point2D.Double(0.5 * (p1.getX() + p2.getX()), 0.5 * (p1.getY() + p2.getY()));
-		Point2D direction = new Point2D.Double(0.5 * sign * (p1.getX() - p2.getX()), 0.5 * sign * (p1.getY() - p2.getY()));
+		if (p1.equals(p2)) {
+			node.setLocation(p1, new Point2D.Double(1.0, 0.0));
+		}
+		else {
+			double sign = Math.signum(p1.getX() - p2.getX());
+			Point2D.Double origin = new Point2D.Double(0.5 * (p1.getX() + p2.getX()), 0.5 * (p1.getY() + p2.getY()));
+			Point2D direction = new Point2D.Double(0.5 * sign * (p1.getX() - p2.getX()), 0.5 * sign * (p1.getY() - p2.getY()));
+			
+			node.setLocation(origin, direction);
+		}
 		
-		node.setLocation(origin, direction);
 		node.setInfo(result.info);
 	}
 	
@@ -155,6 +174,6 @@ public class DistrictNetworkStaticInfoStyle extends StyleBase<DistrictNetworkSta
 	
 	@Override
 	protected void cleanupStyleForNode(EvaluationContext evaluationContext, INode node) {
-		ProfileVariables.denyChild(node, DistrictNetworkStaticInfoNode.NODE_KEY);
+		ProfileVariables.denyChild(node, "*", DistrictNetworkStaticInfoNode.NODE_KEY);
 	}
 }