X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.district.selection%2Fsrc%2Forg%2Fsimantics%2Fdistrict%2Fselection%2FElementSelector.java;h=bf051b4c9828e3f9d4d0b08a3e8409a3d4d3c091;hb=50306ed44f39142ec59265c7329e8e5930b9ff38;hp=74eaa227096a06b011e230d47089328c210bacab;hpb=59a2dd367b6cb9d02f22e473386d63ab59a857b3;p=simantics%2Fdistrict.git

diff --git a/org.simantics.district.selection/src/org/simantics/district/selection/ElementSelector.java b/org.simantics.district.selection/src/org/simantics/district/selection/ElementSelector.java
index 74eaa227..bf051b4c 100644
--- a/org.simantics.district.selection/src/org/simantics/district/selection/ElementSelector.java
+++ b/org.simantics.district.selection/src/org/simantics/district/selection/ElementSelector.java
@@ -543,13 +543,12 @@ public class ElementSelector {
 		@SuppressWarnings("unchecked")
 		@Override
 		Collection<Resource> select(ReadGraph graph, Collection<Resource> elements) {
-			List<Resource> result = new ArrayList<>(elements);
 			List<Tuple2> result2 = Lists.map(new FunctionImpl1<Resource, Tuple2>() {
 				@Override
 				public Tuple2 apply(Resource r) {
 					return new Tuple2(r, getPropertyValue(graph, r, propertyName));
 				}
-			}, result);
+			}, new ArrayList<>(elements));
 			
 			result2 = Lists.filter(new FunctionImpl1<Tuple2, Boolean>() {
 				@Override
@@ -560,17 +559,21 @@ public class ElementSelector {
 			
 			result2.sort((t1, t2) -> smallest ? Double.compare((Double) t1.c1, (Double) t2.c1) : Double.compare((Double) t2.c1, (Double) t1.c1));
 			
-			if (resultCount < result2.size())
-				result2 = result2.subList(0, resultCount);
+			if (resultCount < result2.size()) {
+				double limitValue = (double) result2.get(resultCount-1).c1;
+				
+				// Expand selection to contain all items with the same value as the nth one
+				int count = resultCount;
+				while (count < result2.size() && (double)result2.get(count).c1 == limitValue) count++;
+				result2 = result2.subList(0, count);
+			}
 			
-			result = Lists.map(new FunctionImpl1<Tuple2, Resource>() {
+			return (List<Resource>) Lists.map(new FunctionImpl1<Tuple2, Resource>() {
 				@Override
 				public Resource apply(Tuple2 p0) {
 					return (Resource) p0.c0;
 				}
 			}, result2);
-			
-			return result;
 		}
 	}