diff options
| author | Jon Nordby <jononor@gmail.com> | 2025-08-08 22:22:47 +0200 |
|---|---|---|
| committer | Jon Nordby <jononor@gmail.com> | 2025-08-08 22:22:47 +0200 |
| commit | 549688cc3b1175538788361e379719dbd18306cd (patch) | |
| tree | 13e7779f37ae5f455395f0013648f0bb4a71f8e1 /tools | |
| parent | 23ef09d5dd337049458b48eaf9ed7ece3f887592 (diff) | |
pdm: Support optional markers in plot
Diffstat (limited to 'tools')
| -rw-r--r-- | tools/test_pdm.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/tools/test_pdm.py b/tools/test_pdm.py index 3bac00b..69cd119 100644 --- a/tools/test_pdm.py +++ b/tools/test_pdm.py @@ -8,7 +8,11 @@ from pcm2pdm import convert, convert_file, save_pdm_bin from pdm2pcm import pdm_to_pcm, load_pdm_file from testsignal import generate_test_tone -def plot_reconstruct(pcm_in, pdm, pcm_reconstructed, sr, aspect = 4.0, height = 3.0): +def plot_reconstruct(pcm_in, pdm, pcm_reconstructed, sr, + aspect = 4.0, + height = 3.0, + pcm_marker = None, + ): # --- Plotting using OO API --- width = height * aspect @@ -20,7 +24,7 @@ def plot_reconstruct(pcm_in, pdm, pcm_reconstructed, sr, aspect = 4.0, height = t = np.linspace(0, duration, int(sr * duration), endpoint=False) # Plot original PCM - axs[0].plot(t, pcm_in, label="PCM_in", color='blue') + axs[0].plot(t, pcm_in, label="PCM_in", color='blue', marker=pcm_marker) axs[0].set_title("Original PCM") axs[0].set_ylabel("Amplitude") axs[0].set_ylim(-1.2, 1.2) @@ -35,7 +39,7 @@ def plot_reconstruct(pcm_in, pdm, pcm_reconstructed, sr, aspect = 4.0, height = axs[1].grid(True) # Plot reconstructed PCM - axs[2].plot(t, pcm_reconstructed[:len(t)], label="PCM_rec", color='green') + axs[2].plot(t, pcm_reconstructed[:len(t)], label="PCM_rec", color='green', marker=pcm_marker) axs[2].set_title("Reconstructed PCM") axs[2].set_ylabel("Amplitude") axs[2].set_xlabel("Time [s]") |