def problem_3():
f = sy.symbols('f')
psi_hat = wv.battle_lemarie_wavelet_transform()
psi_hat_lambda = sy.lambdify(f, psi_hat, 'numpy')
freq = np.linspace(-2, 2, 100)
freq[0] = np.finfo(float).eps
psi_hat_signal = np.array(np.abs([psi_hat_lambda(v) for v in freq]))
plt.figure()
plt.title('Fourier Transform of Mother Wavelet')
plt.plot(freq, psi_hat_signal)
plt.show()
def problem_4():
# Convert scaling function to time domain
phi_hat = wv.battle_lemarie_scaling_transform()
Xs, ts = f_trans(phi_hat, 8.1, 128, 0)
# Plot scaling function
plt.figure()
plt.subplot(1, 2, 1)
plt.title('Scaling Function')
plt.plot(ts, Xs)
# Convert wavelet to time domain
psi_hat = wv.battle_lemarie_wavelet_transform()
Xs, ts = f_trans(psi_hat, 8.1, 128, 0)
# Plot wavelet
plt.subplot(1, 2, 2)
plt.title('Mother Wavelet')
plt.plot(ts, Xs)
plt.show()
