| import os, fnmatch |
| import numpy as np |
| import random |
| import soundfile as sf |
| from scipy.io.wavfile import write |
| |
| import RT60 |
|
|
| folder_path = "/cephfs/anton/room-impulse-responses/AIR/RWCP_REVERB_AACHEN/real_rirs_isotropic_noises/" |
| final_path = "/cephfs/anton/room-impulse-responses/AIR/RWCP_REVERB_AACHEN/AACHEN/" |
| tfs =16000 |
| file_label = open("RT60.txt","w") |
|
|
| for root, dirnames, filenames in os.walk(folder_path): |
| for filename in filenames: |
| if filename.endswith(".wav"): |
| ACE_Path = os.path.join(root, filename) |
| wave,fs = sf.read(ACE_Path) |
| channel = int(wave.size/len(wave)) |
|
|
| if(channel == 1): |
| wave_single = wave |
| max_loc = np.where(wave_single == np.amax(wave_single)) |
| min_loc = np.where(wave_single == np.amin(wave_single)) |
| start = min(max_loc[0][0],min_loc[0][0]) |
| wave_single =wave_single[start:len(wave_single)] |
| T60_val = RT60.t60_impulse(wave_single,tfs) |
| |
| if(T60_val<1): |
| file_label.write(str(T60_val)+"\n") |
| save_path = final_path+ filename |
| write(save_path,tfs,wave_single.astype(np.float32)) |
| else: |
| for n in range(channel): |
| wave_single = wave[:,n] |
| max_loc = np.where(wave_single == np.amax(wave_single)) |
| min_loc = np.where(wave_single == np.amin(wave_single)) |
| start = min(max_loc[0][0],min_loc[0][0]) |
| wave_single =wave_single[start:len(wave_single)] |
| T60_val = RT60.t60_impulse(wave_single,tfs) |
| |
| if(T60_val<1): |
| file_label.write(str(T60_val)+"\n") |
| save_path = final_path+filename+str(n)+".wav" |
| write(save_path,tfs,wave_single.astype(np.float32)) |
| |
