1import cmsisdsp as dsp
2import numpy as np
3from scipy import signal
4from scipy.fftpack import dct
5import cmsisdsp.fixedpoint as f
6from pyquaternion import Quaternion
7
8import colorama
9from colorama import init,Fore, Back, Style
10import statsmodels.tsa.stattools
11
12import scipy.spatial
13
14
15init()
16
17def printTitle(s):
18    print("\n" + Fore.GREEN + Style.BRIGHT +  s + Style.RESET_ALL)
19
20def printSubTitle(s):
21    print("\n" + Style.BRIGHT + s + Style.RESET_ALL)
22
23printTitle("Max and AbsMax")
24a=np.array([1.,-3.,4.,0.,-10.,8.])
25i=dsp.arm_absmax_f32(a)
26
27printSubTitle("Fixed point tests")
28# Normalize for fixed point tests
29a = a / i[0]
30
31a7 = f.toQ7(a)
32a15 = f.toQ15(a)
33a31 = f.toQ31(a)
34
35print(a31)
36print(dsp.arm_absmax_no_idx_q31(a31))
37print(dsp.arm_max_no_idx_q31(a31))
38
39print(a15)
40print(dsp.arm_absmax_no_idx_q15(a15))
41print(dsp.arm_max_no_idx_q15(a15))
42
43print(a7)
44print(dsp.arm_absmax_no_idx_q7(a7))
45print(dsp.arm_max_no_idx_q7(a7))
46
47printTitle("Min and AbsMin")
48a=np.array([1.,-3.,4.,0.5,-10.,8.])
49i=dsp.arm_absmax_f32(a)
50
51printSubTitle("Fixed point tests")
52
53# Normalize for fixed point tests
54a = a / i[0]
55
56
57a7 = f.toQ7(a)
58a15 = f.toQ15(a)
59a31 = f.toQ31(a)
60
61print(a31)
62print(dsp.arm_absmin_no_idx_q31(a31))
63print(dsp.arm_min_no_idx_q31(a31))
64
65print(a15)
66print(dsp.arm_absmin_no_idx_q15(a15))
67print(dsp.arm_min_no_idx_q15(a15))
68
69print(a7)
70print(dsp.arm_absmin_no_idx_q7(a7))
71print(dsp.arm_min_no_idx_q7(a7))