Add getter for totalWeight to WeightedRandomSelection

matsim-org · Mar 1, 2024 · 97c94ba · 97c94ba
1 parent b78e7c7
commit 97c94ba
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/contribs/common/src/main/java/org/matsim/contrib/common/util/WeightedRandomSelection.java b/contribs/common/src/main/java/org/matsim/contrib/common/util/WeightedRandomSelection.java
@@ -67,6 +67,10 @@ public int size() {
 		return entryList.size();
 	}
 
+	public double getTotalWeight() {
+		return totalWeight;
+	}
+
 	private record Entry<E>(E e, double cumulativeWeight) implements Comparable<Entry<E>> {
 		public int compareTo(Entry<E> o) {
 			double diff = this.cumulativeWeight - o.cumulativeWeight;