diff options
author | jaseg <git@jaseg.de> | 2020-12-21 16:22:11 +0100 |
---|---|---|
committer | jaseg <git@jaseg.de> | 2020-12-21 16:22:11 +0100 |
commit | d8b6d18d2fefee381eb6a26334a13e94126b6d8e (patch) | |
tree | ab3a005e80c7b3d5b5d727ec04a2cbeefc3d51e6 /fw/hid-dials/tools/cwt_wavelet_header_gen.py | |
parent | 26f5fefac6f15db1c8c81fb78a919d35211695c4 (diff) | |
download | minikbd-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.py | 29 |
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};') + |