tomwalters@277: #!/usr/bin/env python
tomwalters@277: # encoding: utf-8
tomwalters@277: #
tomwalters@277: # AIM-C: A C++ implementation of the Auditory Image Model
tomwalters@277: # http://www.acousticscale.org/AIMC
tomwalters@277: #
tomwalters@277: # This program is free software: you can redistribute it and/or modify
tomwalters@277: # it under the terms of the GNU General Public License as published by
tomwalters@277: # the Free Software Foundation, either version 3 of the License, or
tomwalters@277: # (at your option) any later version.
tomwalters@277: #
tomwalters@277: # This program is distributed in the hope that it will be useful,
tomwalters@277: # but WITHOUT ANY WARRANTY; without even the implied warranty of
tomwalters@277: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tomwalters@277: # GNU General Public License for more details.
tomwalters@277: #
tomwalters@277: # You should have received a copy of the GNU General Public License
tomwalters@277: # along with this program.  If not, see <http://www.gnu.org/licenses/>.
tomwalters@277: """
tomwalters@277: ModuleGammatone_test.py
tomwalters@277: 
tomwalters@277: Created by Thomas Walters on 2010-02-15.
tomwalters@277: Copyright 2010 Thomas Walters <tom@acousticscale.org>
tomwalters@277: Test for the Slaney IIR gammatone.
tomwalters@277: """
tomwalters@277: 
tomwalters@277: import aimc
tomwalters@277: from scipy import io
tomwalters@288: import wave
tomwalters@288: import scipy
tomwalters@277: 
tomwalters@277: def main():
tomwalters@277:   data_file = "src/Modules/BMM/testdata/gammatone.mat"
tomwalters@277:   data = io.loadmat(data_file)
tomwalters@277:   
tomwalters@277:   # The margin of error allowed between the returned values from AIM-C and
tomwalters@277:   # the stored MATLAB values.
tomwalters@277:   epsilon = 0.000001;
tomwalters@277:   
tomwalters@277:   input_wave = data["input_wave"]
tomwalters@277:   sample_rate = data["sample_rate"]
tomwalters@277:   centre_frequencies = data["centre_frequencies"]
tomwalters@277:   expected_output = data["expected_output"]
tomwalters@277:   
tomwalters@288:   (channel_count, frame_count) = expected_output.shape
tomwalters@288:   buffer_length = 20000;
tomwalters@277:   
tomwalters@277:   input_sig = aimc.SignalBank()
tomwalters@288:   input_sig.Initialize(1, buffer_length, 48000)
tomwalters@277:   parameters = aimc.Parameters()
tomwalters@288:   parameters.Load("src/Modules/BMM/testdata/gammatone.cfg")
tomwalters@277:   mod_gt = aimc.ModuleGammatone(parameters)
tomwalters@277:   mod_gt.Initialize(input_sig)
tomwalters@277:   
tomwalters@277:   correct_count = 0;
tomwalters@277:   incorrect_count  = 0;
tomwalters@288:   
tomwalters@288:   out = scipy.zeros((channel_count, buffer_length))
tomwalters@288:   
tomwalters@288:   cfs = scipy.zeros((channel_count))
tomwalters@288: 
tomwalters@288:   for i in range(0, buffer_length):
tomwalters@288:     input_sig.set_sample(0, i, input_wave[i][0])
tomwalters@288:   mod_gt.Process(input_sig)
tomwalters@288:   out_sig = mod_gt.GetOutputBank()
tomwalters@288:   for ch in range(0, out_sig.channel_count()):
tomwalters@288:     cfs[ch] = out_sig.centre_frequency(ch);
tomwalters@288:     for i in range(0, buffer_length):  
tomwalters@288:       out[ch, i] = out_sig.sample(ch, i)
tomwalters@288:   
tomwalters@288:   outmat = dict(filterbank_out=out, centre_frequencies_out=cfs)
tomwalters@288:   io.savemat("src/Modules/BMM/testdata/out_v2.mat", outmat)
tomwalters@277: 
tomwalters@277:   pass
tomwalters@277: 
tomwalters@277: 
tomwalters@277: if __name__ == '__main__':
tomwalters@277:   main()