Make plots easily

Author: DrPeaboss

.gitignore

@@ -2,3 +2,4 @@
 /output
 /perf*
 /flamegraph.svg
+/__pycache__

README.md

@@ -43,13 +43,20 @@ For multi-channel example see [examples/two_channels.rs](/examples/two_channels.
 
 The *linear* Converter is not recommended unless performance is really important.
 
-Use [plots.py](/plots.py) to show the result of tests.
+Use [plots.py](/plots.py) to show the result of tests. It need *numpy*, *scipy*
+and *matplotlib*.
 
-```python
+```
+$ cargo test -r --test testwav -- --ignored --exact --show-output generate
+$ cargo test -r --test sinc -- --ignored --exact --show-output ta120_2_96k_down
+$ python
+>>> import plots
 >>> import os
 >>> os.chdir('output')
->>> import plots
->>> plots.show_wav_spectrogram('sweep_96k_44k_xxx.wav')
+>>> plots.spectrum('beep_96k_44k_s_a120_2.wav')
+>>> plots.spectrogram('sweep_96k_44k_s_a120_2.wav')
+>>> plots.impulse('impulse_96k_44k_s_a120_2.wav')
+>>> plots.impulse('impulse_96k_44k_s_a120_2.wav', True)
 ```
 
 See [tests](/tests/) for more details.

plots.py

@@ -4,7 +4,7 @@
 from scipy.signal import ShortTimeFFT
 from scipy.signal.windows import kaiser
 
-def _show_spectrum(fs, data, name, impulse: None | str = None):
+def _spectrum(fs, data, name, impulse: None | str = None):
     passband = impulse == 'passband'
     N = len(data)
     half_N = N // 2
@@ -27,19 +27,19 @@ def _show_spectrum(fs, data, name, impulse: None | str = None):
     plt.title(prefix + name)
     plt.show()
 
-def show_wav_spectrum(filename):
+def spectrum(filename):
     fs, data = wavfile.read(filename)
-    _show_spectrum(fs, data, filename)
+    _spectrum(fs, data, filename)
 
-def show_wav_impulse(filename, passband=False):
+def impulse(filename, passband=False):
     fs, data = wavfile.read(filename)
-    _show_spectrum(fs, data, filename, impulse='passband' if passband else '')
+    _spectrum(fs, data, filename, impulse='passband' if passband else '')
 
-def show_raw_impulse(filename, fs, passband=False):
+def raw_impulse(filename, fs, passband=False):
     data = np.fromfile(filename, np.float64)
-    _show_spectrum(fs, data, filename, impulse='passband' if passband else '')
+    _spectrum(fs, data, filename, impulse='passband' if passband else '')
 
-def show_wav_spectrogram(filename):
+def spectrogram(filename):
     fs, data = wavfile.read(filename)
     N = len(data)
     window_size = 2048

tests/sinc.rs

@@ -296,11 +296,13 @@ fn ta120_2_96k_down() {
     let trans_width = 2050.0 / 22050.0;
     let src = Src::new_by_trans_width(96000, 44100, 120.0, 512, trans_width);
     println!("order of 96 to 44k {remark} is {}", src.manager.order());
+    convert("beep", &src, remark);
     convert("sweep", &src, remark);
     impulse(&src, remark);
     let trans_width = 4000.0 / 24000.0;
     let src = Src::new_by_trans_width(96000, 48000, 120.0, 512, trans_width);
     println!("order of 96k to 48k {remark} is {}", src.manager.order());
+    convert("beep", &src, remark);
     convert("sweep", &src, remark);
     impulse(&src, remark);
 }

tests/testwav.rs

@@ -79,7 +79,10 @@ fn generate() {
     std::env::set_current_dir("output").unwrap();
     gen_beep(44100);
     gen_beep(48000);
+    gen_beep(96000);
+    gen_beep(192000);
     gen_sweep(44100);
     gen_sweep(48000);
     gen_sweep(96000);
+    gen_sweep(192000);
 }