summaryrefslogtreecommitdiff
path: root/fw/hid-dials/tools/cwt_wavelet_header_gen.py
diff options
context:
space:
mode:
authorjaseg <git@jaseg.de>2020-12-21 16:22:11 +0100
committerjaseg <git@jaseg.de>2020-12-21 16:22:11 +0100
commitd8b6d18d2fefee381eb6a26334a13e94126b6d8e (patch)
treeab3a005e80c7b3d5b5d727ec04a2cbeefc3d51e6 /fw/hid-dials/tools/cwt_wavelet_header_gen.py
parent26f5fefac6f15db1c8c81fb78a919d35211695c4 (diff)
downloadminikbd-d8b6d18d2fefee381eb6a26334a13e94126b6d8e.tar.gz
minikbd-d8b6d18d2fefee381eb6a26334a13e94126b6d8e.tar.bz2
minikbd-d8b6d18d2fefee381eb6a26334a13e94126b6d8e.zip
Remove obsolete template code from fw
Diffstat (limited to 'fw/hid-dials/tools/cwt_wavelet_header_gen.py')
-rw-r--r--fw/hid-dials/tools/cwt_wavelet_header_gen.py29
1 files changed, 29 insertions, 0 deletions
diff --git a/fw/hid-dials/tools/cwt_wavelet_header_gen.py b/fw/hid-dials/tools/cwt_wavelet_header_gen.py
new file mode 100644
index 0000000..8be785b
--- /dev/null
+++ b/fw/hid-dials/tools/cwt_wavelet_header_gen.py
@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+
+import textwrap
+
+import scipy.signal as sig
+import numpy as np
+
+if __name__ == '__main__':
+ import argparse
+ parser = argparse.ArgumentParser()
+ parser.add_argument('n', type=int, help='Window size')
+ parser.add_argument('w', type=float, help='Wavelet width')
+ parser.add_argument('-v', '--variable', default='cwt_ricker_table', help='Name for alias variable pointing to generated wavelet LUT')
+ args = parser.parse_args()
+
+ print(f'/* CWT Ricker wavelet LUT for {args.n} sample window of width {args.w}. */')
+ varname = f'cwt_ricker_{args.n}_window_{str(args.w).replace(".", "F")}'
+ print(f'const float {varname}[{args.n}] = {{')
+
+ win = sig.ricker(args.n, args.w)
+ par = ' '.join(f'{f:>015.12e}f,' for f in win)
+ print(textwrap.fill(par,
+ initial_indent=' '*4, subsequent_indent=' '*4,
+ width=120,
+ replace_whitespace=False, drop_whitespace=False))
+ print('};')
+ print()
+ print(f'const float * const {args.variable} __attribute__((weak)) = {varname};')
+