Room.py

import pyroomacoustics as pra
import matplotlib.pyplot as plt
from Player import *

def makeroom(fs, data):
    """Place sources at the position in the room.
    Size of the room [x,y,z]
    Absorption of the wall (0 - 1)
    Max_order - number of reflection of the wall
    """
    for i in range(data.shape[0]):
        data[i].resize((len(data[i]),))

    room = pra.ShoeBox([7, 5, 3.2], fs=fs, absorption=0.35, max_order=15, sigma2_awgn=1e-8)
    '''Place the sources inside the room'''
    room.add_source([3., 2., 1.8], signal=data[0])
    room.add_source([6., 4., 1.8], signal=data[1])
    room.add_source([2., 4.5, 1.8], signal=data[2])

    #delays = [1., 0.]   #TODO: Check for delay
    '''Place the mic. array into the room'''
    R = np.c_[
        [3, 2.87, 1],  # microphone 1
        [3, 2.93, 1],  # microphone 2
        [3, 2.99, 1],  # microphone 3
    ]
    room.add_microphone_array(pra.MicrophoneArray(R, room.fs))
    '''Plot the sim room'''
    #fig, ax = room.plot()
    #ax.set_xlim([-1, 8])
    #ax.set_ylim([-1, 7])
    #ax.set_zlim([0, 3.5])

    #plt.savefig("Fig/Room Visualization.jpeg")
    #plt.show()
    #room.compute_rir()
    '''Plot the sim room with reflections'''
    #room.image_source_model(use_libroom=True)
    #fig, ax = room.plot(img_order=3)
    #fig.set_size_inches(10, 6, 3)
    #plt.savefig("Fig/Room Reflection.jpeg")
    '''Plot Impulse Response of the Room'''
    #room.plot_rir()
    #fig = plt.gcf()
    #fig.set_size_inches(12, 8)
    #plt.savefig("Fig/Room Impulse Response.jpeg")
    #plt.show()

    room.simulate()
    #play(room.mic_array.signals[0, :] * 5000)
    #play(room.mic_array.signals[1, :] * 5000)
    #play(room.mic_array.signals[2, :] * 5000)
    return room.mic_array.signals