/usr/share/doc/python-pywt-doc/examples/waveinfo.py is in python-pywt-doc 0.5.1-1.1ubuntu4.
This file is owned by root:root, with mode 0o755.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 | #!/usr/bin/env python
# -*- coding: utf-8 -*-
import sys
import matplotlib.pyplot as plt
import numpy as np
import pywt
usage = """
Usage:
python waveinfo.py waveletname
Example: python waveinfo.py 'sym5'
"""
try:
wavelet = pywt.Wavelet(sys.argv[1])
try:
level = int(sys.argv[2])
except IndexError as e:
level = 10
except ValueError as e:
print("Unknown wavelet")
raise SystemExit
except IndexError as e:
print(usage)
raise SystemExit
data = wavelet.wavefun(level)
if len(data) == 2:
x = data[1]
psi = data[0]
fig = plt.figure()
if wavelet.complex_cwt:
plt.subplot(211)
plt.title(wavelet.name+' real part')
mi, ma = np.real(psi).min(), np.real(psi).max()
margin = (ma - mi) * 0.05
plt.plot(x,np.real(psi))
plt.ylim(mi - margin, ma + margin)
plt.xlim(x[0], x[-1])
plt.subplot(212)
plt.title(wavelet.name+' imag part')
mi, ma = np.imag(psi).min(), np.imag(psi).max()
margin = (ma - mi) * 0.05
plt.plot(x,np.imag(psi))
plt.ylim(mi - margin, ma + margin)
plt.xlim(x[0], x[-1])
else:
mi, ma = psi.min(), psi.max()
margin = (ma - mi) * 0.05
plt.plot(x,psi)
plt.title(wavelet.name)
plt.ylim(mi - margin, ma + margin)
plt.xlim(x[0], x[-1])
else:
funcs, x = data[:-1], data[-1]
labels = ["scaling function (phi)", "wavelet function (psi)",
"r. scaling function (phi)", "r. wavelet function (psi)"]
colors = ("r", "g", "r", "g")
fig = plt.figure()
for i, (d, label, color) in enumerate(zip(funcs, labels, colors)):
mi, ma = d.min(), d.max()
margin = (ma - mi) * 0.05
ax = fig.add_subplot((len(data) - 1) // 2, 2, 1 + i)
ax.plot(x, d, color)
ax.set_title(label)
ax.set_ylim(mi - margin, ma + margin)
ax.set_xlim(x[0], x[-1])
plt.show()
|