import numpy as np
import pyfar as pf
import pyrato as ra
from pyrato.analytic import rectangular_room_rigid_walls
L = np.array([8, 5, 3])/10
source_pos = np.array([5, 3, 1.2])/10
receiver_pos = np.array([1, 1, 1.2])/10
rir, _ = rectangular_room_rigid_walls(
    L, source_pos, receiver_pos,
    reverberation_time=1, max_freq=1e3, n_samples=2**16,
    speed_of_sound=343.9)
edc = ra.edc.schroeder_integration(rir)
pf.plot.time(rir/np.abs(rir.time).max(), dB=True, label='RIR')
ax = pf.plot.time(
    edc/edc.time[..., 0], dB=True, log_prefix=10, label='EDC')
ax.set_ylim(-65, 5)
ax.legend()