Added gen_exps.py script.

1 files changed, 33 insertions, 0 deletions

diff --git a/gen/dp.py b/gen/dp.py
new file mode 100644
index 0000000..0ac8cce
--- /dev/null
+++ b/gen/dp.py
@@ -0,0 +1,33 @@
+from __future__ import division
+
+class DesignPointGenerator(object):
+    '''Iterates over all combinations of values specified in options.
+    Shamelessly stolen (and simplified) from bcw.'''
+    def __init__(self, options):
+        self.point_idx = 0  # Current point
+        self.options = options
+        self.total = 1
+        for x in options.itervalues():
+            self.total *= len(x)
+
+    def __iter__(self):
+        return self
+
+    def next(self):
+        while True:
+            if self.point_idx == self.total:
+                raise StopIteration
+            else:
+                point = {}
+
+                divisor = 1
+                for key in sorted(self.options.keys()):
+                    size = len(self.options[key])
+
+                    option_idx = int(self.point_idx / divisor) % size
+                    point[key] = self.options[key][option_idx]
+
+                    divisor *= size
+                self.point_idx += 1
+
+                return point