1 files changed, 47 insertions, 0 deletions
diff --git a/Python/SyncRNG.py b/Python/SyncRNG.py
new file mode 100644
index 0000000..7ef6fd6
--- /dev/null
+++ b/Python/SyncRNG.py
@@ -0,0 +1,47 @@
+"""
+Simple interface to SyncRNG. This file defines a SyncRNG object which can be 
+used to seed and pull numbers from the RNG.
+
+"""
+
+from __future__ import division
+
+from warnings import warn as _warn
+
+import syncrng
+
+class SyncRNG(object):
+
+    def __init__(self, seed=0):
+        self.BPF = 32
+        self.seed = seed
+        self.state = syncrng.seed(seed)
+
+    def randi(self):
+        tmp = syncrng.rand(self.state)
+        self.state = tmp[:-1]
+        return(tmp[-1])
+
+    def rand(self):
+        return self.randi() * 2.3283064365387e-10
+
+    def randbelow(self, n):
+        maxsize = 1<<self.BPF
+        if n >= maxsize:
+            _warn("Underlying random generator does not supply \n"
+                    "enough bits to choose from a population range this "
+                    "large.\n")
+            return int(self.rand() * n)
+        rem = maxsize % n
+        limit = (maxsize - rem) / maxsize
+        r = self.rand()
+        while r >= limit:
+            r = self.rand()
+        return int(r*maxsize) % n
+
+    def shuffle(self, x):
+        y = x[:]
+        for i in reversed(range(1, len(y))):
+            j = self.randbelow(i+1)
+            y[i], y[j] = y[j], y[i]
+        return y