aboutsummaryrefslogtreecommitdiff
path: root/fw/datagen.py
diff options
context:
space:
mode:
Diffstat (limited to 'fw/datagen.py')
-rw-r--r--fw/datagen.py39
1 files changed, 39 insertions, 0 deletions
diff --git a/fw/datagen.py b/fw/datagen.py
new file mode 100644
index 0000000..63402dc
--- /dev/null
+++ b/fw/datagen.py
@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+
+import statistics
+
+if __name__ == '__main__':
+ import argparse
+ parser = argparse.ArgumentParser()
+ parser.add_argument('infile')
+
+ parser.add_argument('-b', '--bitlen', type=float, default=250, help='Average length of a bit in fractional samples')
+ parser.add_argument('-j', '--jitter', type=float, default=0, help='Bit length jitter standard deviation in fractional samples')
+
+ parser.add_argument('-v', '--high', type=float, default=12.0, help='High level in V')
+ parser.add_argument('-l', '--low', type=float, default=0.2, help='Low level in V')
+ parser.add_argument('-n', '--noise', type=float, default=0, help='Gaussian white voltage noise amplitude on top of signal')
+
+ parser.add_argument('-r', '--risetime', type=int, default=1, help='Rise time in sp/V')
+ parser.add_argument('-f', '--falltime', type=int, default=1, help='Fall time in sp/V')
+ #parser.add_argument('-g', '--glitchfreq', type=float, default=0, help='Add glitches every {g} samples on average (set to 0 to turn off)')
+ #parser.add_argument('--glitchlen', type=float, default=0, help='Length of glitches to add in samples')
+ #parser.add_argument('-t', '--glitch-constant', type=float, default=0, help='When the signal stays constant between bits add a glitch with probability {t}')
+ args = paresr.parse_args()
+
+ with open(args.infile, 'r') as f:
+ bits = f.read().replace(' ', '').splitlines()
+
+
+ def join_transitions(chunks):
+ for l, r in zip(chunks, chunks[1:]):
+ t = args.risetime if l[-1] < r[0] else args.falltime
+ yield from (l[-1] + (r[0] - l[-1] * i/t) for i in range(t))
+ yield from c[args.risetime:-args.falltime]
+
+ def add_noise(vals):
+ yield from (val + random.gauss(0, args.noise) for val in vals)
+
+ def encode_bits(bitstring):
+ return add_noise(join_transitions([ args.high if bit == '1' else args.low for _ in range(random.gauss(args.bitlen, args.jitter)) ]))
+