51 lines
1.7 KiB
Python
Executable file
51 lines
1.7 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# coding: utf-8
|
|
|
|
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
from argparse import ArgumentParser
|
|
import os
|
|
|
|
parser = ArgumentParser(usage="Compare 2 files as outputted by sfizz_plot_lfo")
|
|
parser.add_argument("file", help="The file to test")
|
|
parser.add_argument("reference", help="The reference file")
|
|
args = parser.parse_args()
|
|
|
|
assert os.path.exists(args.file), "The file to test does not exist"
|
|
assert os.path.exists(args.reference), "The reference file does not exist"
|
|
|
|
reference_data = np.loadtxt(args.reference)
|
|
data = np.loadtxt(args.file)
|
|
|
|
assert reference_data.shape == data.shape, "The shapes of the data and reference are different"
|
|
|
|
n_samples, n_lfos = reference_data.shape
|
|
n_lfos -= 1 # The first column is the time
|
|
|
|
if n_lfos == 1:
|
|
fig, ax = plt.subplots(figsize=(20, 10))
|
|
ax.plot(reference_data[:, 0], reference_data[:, 1])
|
|
ax.plot(data[:, 0], data[:, 1])
|
|
ax.grid()
|
|
elif n_lfos > 4:
|
|
n_cols = 4
|
|
n_rows = n_lfos // n_cols
|
|
print(n_rows, n_cols)
|
|
fig, ax = plt.subplots(n_rows, n_cols, figsize=(20, 10))
|
|
for i in range(n_rows):
|
|
for j in range(n_cols):
|
|
lfo_index = 1 + i * 4 + j
|
|
ax[i, j].plot(reference_data[:, 0], reference_data[:, lfo_index])
|
|
ax[i, j].plot(data[:, 0], data[:, lfo_index])
|
|
ax[i, j].set_title("LFO {}".format(lfo_index))
|
|
ax[i, j].grid()
|
|
else:
|
|
fig, ax = plt.subplots(1, n_lfos, figsize=(20, 10))
|
|
for i in range(n_lfos):
|
|
lfo_index = i + 1
|
|
ax[i].plot(reference_data[:, 0], reference_data[:, lfo_index])
|
|
ax[i].plot(data[:, 0], data[:, lfo_index])
|
|
ax[i].set_title("LFO {}".format(lfo_index))
|
|
ax[i].grid()
|
|
|
|
plt.show()
|