add methods to sum up the values in a dict, and return percentages for each

benfry · benfry · commit 6405faaacd9d · 2013-07-24T10:53:21.000-04:00
diff --git a/core/src/processing/data/FloatDict.java b/core/src/processing/data/FloatDict.java
@@ -680,6 +680,25 @@ public void swap(int a, int b) {
   }
 
 
+  /**
+   * Sum all of the values in this dictionary, then return a new FloatDict of
+   * each key, divided by the total sum. The total for all values will be ~1.0.
+   * @return a Dict with the original keys, mapped to their pct of the total
+   */
+  public FloatDict getPercentages() {
+    double sum = 0;
+    for (float value : valueArray()) {
+      sum += value;
+    }
+    FloatDict outgoing = new FloatDict();
+    for (int i = 0; i < size(); i++) {
+      double percent = value(i) / sum;
+      outgoing.set(key(i), (float) percent);
+    }
+    return outgoing;
+  }
+
+
   /** Returns a duplicate copy of this object. */
   public FloatDict copy() {
     FloatDict outgoing = new FloatDict(count);
diff --git a/core/src/processing/data/IntDict.java b/core/src/processing/data/IntDict.java
@@ -621,6 +621,25 @@ public void swap(int a, int b) {
   }
 
 
+  /**
+   * Sum all of the values in this dictionary, then return a new FloatDict of
+   * each key, divided by the total sum. The total for all values will be ~1.0.
+   * @return a Dict with the original keys, mapped to their pct of the total
+   */
+  public FloatDict getPercentages() {
+    double sum = 0;
+    for (int value : valueArray()) {
+      sum += value;
+    }
+    FloatDict outgoing = new FloatDict();
+    for (int i = 0; i < size(); i++) {
+      double percent = value(i) / sum;
+      outgoing.set(key(i), (float) percent);
+    }
+    return outgoing;
+  }
+
+
   /** Returns a duplicate copy of this object. */
   public IntDict copy() {
     IntDict outgoing = new IntDict(count);
