diff options
author | John Glover <glover.john@gmail.com> | 2011-06-24 18:21:52 +0100 |
---|---|---|
committer | John Glover <glover.john@gmail.com> | 2011-06-24 18:21:52 +0100 |
commit | 0ecf55c21718ce97384f0c44d797549fce29db12 (patch) | |
tree | 3f70f0d82cbabd06b05b9427849bd4a5e224c004 /tests/sndobj.py | |
parent | 416bd737074a287ea47106c73ea6bcfde40a75a8 (diff) | |
download | simpl-0ecf55c21718ce97384f0c44d797549fce29db12.tar.gz simpl-0ecf55c21718ce97384f0c44d797549fce29db12.tar.bz2 simpl-0ecf55c21718ce97384f0c44d797549fce29db12.zip |
Update test file names
Diffstat (limited to 'tests/sndobj.py')
-rw-r--r-- | tests/sndobj.py | 389 |
1 files changed, 0 insertions, 389 deletions
diff --git a/tests/sndobj.py b/tests/sndobj.py deleted file mode 100644 index 527a9d7..0000000 --- a/tests/sndobj.py +++ /dev/null @@ -1,389 +0,0 @@ -# Copyright (c) 2009 John Glover, National University of Ireland, Maynooth -# -# This program is free software; you can redistribute it and/or modify -# it under the terms of the GNU General Public License as published by -# the Free Software Foundation; either version 2 of the License, or -# (at your option) any later version. -# -# This program is distributed in the hope that it will be useful, -# but WITHOUT ANY WARRANTY; without even the implied warranty of -# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -# GNU General Public License for more details. -# -# You should have received a copy of the GNU General Public License -# along with this program; if not, write to the Free Software -# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA - -import unittest -import sndobj -import simpl -from simpl import pysndobj -from scipy.io.wavfile import read -import numpy as np - -FLOAT_PRECISION = 2 # number of decimal places to check for accuracy - -class TestSimplSndObj(unittest.TestCase): - def setUp(self): - self.input_file = 'audio/flute.wav' - - def test_sndobj_numpy(self): - "Test reading and writing numpy data in SndObjs" - frame_size = 2048 - hop_size = 2048 - num_frames = 4 - num_samples = frame_size + (num_frames * hop_size) - - audio_in_data = read(self.input_file) - audio_in = simpl.asarray(audio_in_data[1]) / 32768.0 - audio_in = audio_in[0:num_samples] - frame = simpl.zeros(frame_size) - frame_out = simpl.zeros(hop_size) - audio_out = simpl.array([]) - - input = pysndobj.SndObj() - input.SetVectorSize(frame_size) - - i = 0 - while (i + frame_size) <= num_samples: - frame = audio_in[i:i+frame_size] - input.PushIn(frame) - input.PopOut(frame_out) - audio_out = np.hstack((audio_out, frame_out)) - i += hop_size - - self.assertEqual(audio_in.size, audio_out.size) - for i in range(audio_out.size): - self.assertEqual(audio_in[i], audio_out[i]) - - def test_hammingtable(self): - "Compare sndobj.HammingTable with simplsndobj.HammingTable" - frame_size = 2048 - window1 = sndobj.HammingTable(frame_size, 0.5) - window2 = pysndobj.HammingTable(frame_size, 0.5) - for i in range(frame_size): - self.assertEqual(window1.Lookup(i), window2.Lookup(i)) - - def test_ifgram(self): - "Compare sndobj.IFGram with simplsndobj.IFGram" - frame_size = 2048 - hop_size = 512 - num_frames = 4 - num_samples = frame_size + ((num_frames - 1) * hop_size) - - audio_in_data = read(self.input_file) - audio_in = simpl.asarray(audio_in_data[1]) / 32768.0 - audio_in = audio_in[0:num_samples] - frame = simpl.zeros(frame_size) - - sndobj_input = sndobj.SndObj() - simpl_input = pysndobj.SndObj() - sndobj_input.SetVectorSize(frame_size) - simpl_input.SetVectorSize(frame_size) - sndobj_window = sndobj.HammingTable(frame_size, 0.5) - simpl_window = pysndobj.HammingTable(frame_size, 0.5) - sndobj_ifgram = sndobj.IFGram(sndobj_window, sndobj_input, 1, frame_size, hop_size) - simpl_ifgram = pysndobj.IFGram(simpl_window, simpl_input, 1, frame_size, hop_size) - - i = 0 - while (i + frame_size) <= num_samples: - frame = audio_in[i:i+frame_size] - sndobj_input.PushIn(frame) - simpl_input.PushIn(frame) - sndobj_ifgram.DoProcess() - simpl_ifgram.DoProcess() - for j in range(frame_size): - self.assertAlmostEquals(sndobj_ifgram.Output(j), simpl_ifgram.Output(j), - places = FLOAT_PRECISION) - i += hop_size - - def test_sinanal(self): - "Compare sndobj.SinAnal with simplsndobj.SinAnal" - frame_size = 2048 - hop_size = 512 - max_tracks = 20 - num_frames = 4 - num_samples = frame_size + ((num_frames - 1) * hop_size) - - audio_in_data = read(self.input_file) - audio_in = simpl.asarray(audio_in_data[1]) / 32768.0 - audio_in = audio_in[0:num_samples] - frame = simpl.zeros(frame_size) - - sndobj_input = sndobj.SndObj() - simpl_input = pysndobj.SndObj() - sndobj_input.SetVectorSize(frame_size) - simpl_input.SetVectorSize(frame_size) - sndobj_window = sndobj.HammingTable(frame_size, 0.5) - simpl_window = pysndobj.HammingTable(frame_size, 0.5) - sndobj_ifgram = sndobj.IFGram(sndobj_window, sndobj_input, 1, frame_size, hop_size) - simpl_ifgram = pysndobj.IFGram(simpl_window, simpl_input, 1, frame_size, hop_size) - sndobj_sinmod = sndobj.SinAnal(sndobj_ifgram, 0.003, max_tracks, 1, 3) - simpl_sinmod = pysndobj.SinAnal(simpl_ifgram, 0.003, max_tracks, 1, 3) - - i = 0 - while (i + frame_size) <= num_samples: - frame = audio_in[i:i+frame_size] - sndobj_input.PushIn(frame) - simpl_input.PushIn(frame) - sndobj_ifgram.DoProcess() - simpl_ifgram.DoProcess() - sndobj_sinmod.DoProcess() - simpl_sinmod.DoProcess() - for j in range(max_tracks * 3): - self.assertAlmostEquals(sndobj_sinmod.Output(j), simpl_sinmod.Output(j), - places = FLOAT_PRECISION) - i += hop_size - - def test_adsyn_doprocess(self): - "Compare sndobj.AdSyn with simplsndobj.AdSyn" - frame_size = 2048 - hop_size = 512 - max_tracks = 20 - num_frames = 4 - num_samples = frame_size + ((num_frames - 1) * hop_size) - - audio_in_data = read(self.input_file) - audio_in = simpl.asarray(audio_in_data[1]) / 32768.0 - audio_in = audio_in[0:num_samples] - frame = simpl.zeros(frame_size) - sndobj_frame_out = simpl.zeros(hop_size) - simpl_frame_out = simpl.zeros(hop_size) - sndobj_audio_out = simpl.array([]) - simpl_audio_out = simpl.array([]) - - sndobj_input = sndobj.SndObj() - simpl_input = pysndobj.SndObj() - sndobj_input.SetVectorSize(frame_size) - simpl_input.SetVectorSize(frame_size) - sndobj_window = sndobj.HammingTable(frame_size, 0.5) - simpl_window = pysndobj.HammingTable(frame_size, 0.5) - sndobj_ifgram = sndobj.IFGram(sndobj_window, sndobj_input, 1, frame_size, hop_size) - simpl_ifgram = pysndobj.IFGram(simpl_window, simpl_input, 1, frame_size, hop_size) - sndobj_sinmod = sndobj.SinAnal(sndobj_ifgram, 0.003, max_tracks, 1, 3) - simpl_sinmod = pysndobj.SinAnal(simpl_ifgram, 0.003, max_tracks, 1, 3) - sndobj_table = sndobj.HarmTable(10000, 1, 1, 0.25) - simpl_table = pysndobj.HarmTable(10000, 1, 1, 0.25) - sndobj_synth = sndobj.AdSyn(sndobj_sinmod, max_tracks, sndobj_table, 1, 1, hop_size) - simpl_synth = pysndobj.AdSyn(simpl_sinmod, max_tracks, simpl_table, 1, 1, hop_size) - - i = 0 - while (i + frame_size) <= num_samples: - frame = audio_in[i:i+frame_size] - sndobj_input.PushIn(frame) - simpl_input.PushIn(frame) - sndobj_ifgram.DoProcess() - simpl_ifgram.DoProcess() - sndobj_sinmod.DoProcess() - simpl_sinmod.DoProcess() - sndobj_synth.DoProcess() - simpl_synth.DoProcess() - sndobj_synth.PopOut(sndobj_frame_out) - simpl_synth.PopOut(simpl_frame_out) - sndobj_audio_out = np.hstack((sndobj_audio_out, sndobj_frame_out)) - simpl_audio_out = np.hstack((simpl_audio_out, simpl_frame_out)) - i += hop_size - - self.assertEqual(sndobj_audio_out.size, simpl_audio_out.size) - for i in range(sndobj_audio_out.size): - self.assertAlmostEquals(sndobj_audio_out[i], simpl_audio_out[i], - places = FLOAT_PRECISION) - - def test_sinsyn_doprocess(self): - "Compare sndobj.SinSyn with pysndobj.SinSyn" - frame_size = 2048 - hop_size = 512 - max_tracks = 20 - num_frames = 4 - num_samples = frame_size + ((num_frames - 1) * hop_size) - - audio_in_data = read(self.input_file) - audio_in = simpl.asarray(audio_in_data[1]) / 32768.0 - audio_in = audio_in[0:num_samples] - frame = simpl.zeros(frame_size) - sndobj_frame_out = simpl.zeros(hop_size) - simpl_frame_out = simpl.zeros(hop_size) - sndobj_audio_out = simpl.array([]) - simpl_audio_out = simpl.array([]) - - sndobj_input = sndobj.SndObj() - simpl_input = pysndobj.SndObj() - sndobj_input.SetVectorSize(frame_size) - simpl_input.SetVectorSize(frame_size) - sndobj_window = sndobj.HammingTable(frame_size, 0.5) - simpl_window = pysndobj.HammingTable(frame_size, 0.5) - sndobj_ifgram = sndobj.IFGram(sndobj_window, sndobj_input, 1, frame_size, hop_size) - simpl_ifgram = pysndobj.IFGram(simpl_window, simpl_input, 1, frame_size, hop_size) - sndobj_sinmod = sndobj.SinAnal(sndobj_ifgram, 0.003, max_tracks, 1, 3) - simpl_sinmod = pysndobj.SinAnal(simpl_ifgram, 0.003, max_tracks, 1, 3) - sndobj_table = sndobj.HarmTable(10000, 1, 1, 0.25) - simpl_table = pysndobj.HarmTable(10000, 1, 1, 0.25) - sndobj_synth = sndobj.SinSyn(sndobj_sinmod, max_tracks, sndobj_table, 1, hop_size) - simpl_synth = pysndobj.SinSyn(simpl_sinmod, max_tracks, simpl_table, 1, hop_size) - - i = 0 - while (i + frame_size) <= num_samples: - frame = audio_in[i:i+frame_size] - sndobj_input.PushIn(frame) - simpl_input.PushIn(frame) - sndobj_ifgram.DoProcess() - simpl_ifgram.DoProcess() - sndobj_sinmod.DoProcess() - simpl_sinmod.DoProcess() - sndobj_synth.DoProcess() - simpl_synth.DoProcess() - sndobj_synth.PopOut(sndobj_frame_out) - simpl_synth.PopOut(simpl_frame_out) - sndobj_audio_out = np.hstack((sndobj_audio_out, sndobj_frame_out)) - simpl_audio_out = np.hstack((simpl_audio_out, simpl_frame_out)) - i += hop_size - - self.assertEqual(sndobj_audio_out.size, simpl_audio_out.size) - for i in range(sndobj_audio_out.size): - self.assertAlmostEquals(sndobj_audio_out[i], simpl_audio_out[i], - places = FLOAT_PRECISION) - - def test_peak_detection(self): - "Compare pysndobj Peaks with SndObj peaks" - frame_size = 2048 - hop_size = 512 - num_frames = 4 - max_peaks = 20 - max_tracks = 20 - num_samples = frame_size + ((num_frames - 1) * hop_size) - - audio_in_data = read(self.input_file) - audio_in = simpl.asarray(audio_in_data[1]) / 32768.0 - audio_in = audio_in[0:num_samples] - frame = simpl.zeros(frame_size) - - sndobj_input = sndobj.SndObj() - sndobj_input.SetVectorSize(frame_size) - sndobj_window = sndobj.HammingTable(frame_size, 0.5) - sndobj_ifgram = sndobj.IFGram(sndobj_window, sndobj_input, 1, frame_size, hop_size) - simpl_sinmod = pysndobj.SinAnal(sndobj_ifgram, 0.003, max_tracks, 1, 3) - - pd = simpl.SndObjPeakDetection() - pd.max_peaks = max_peaks - - i = 0 - while (i + frame_size) <= num_samples: - frame = audio_in[i:i+frame_size] - sndobj_input.PushIn(frame) - sndobj_ifgram.DoProcess() - num_peaks_found = simpl_sinmod.FindPeaks() - - frame = audio_in[i:i+frame_size] - peaks = pd.find_peaks(frame) - self.assertEquals(len(peaks), 1) - peaks = peaks[0] - - self.assertEquals(num_peaks_found, len(peaks)) - for j in range(num_peaks_found): - self.assertAlmostEquals(peaks[j].amplitude, simpl_sinmod.Output(j*3), - places=FLOAT_PRECISION) - self.assertAlmostEquals(peaks[j].frequency, simpl_sinmod.Output((j*3)+1), - places=FLOAT_PRECISION) - self.assertAlmostEquals(peaks[j].phase, simpl_sinmod.Output((j*3)+2), - places=FLOAT_PRECISION) - i += hop_size - - def test_partial_tracking(self): - "Compare pysndobj Partials with SndObj tracks" - frame_size = 2048 - hop_size = 512 - num_frames = 4 - max_tracks = 20 - max_peaks = 20 - num_samples = frame_size + ((num_frames - 1) * hop_size) - - audio_in_data = read(self.input_file) - audio_in = simpl.asarray(audio_in_data[1]) / 32768.0 - audio_in = audio_in[0:num_samples] - frame = simpl.zeros(frame_size) - - sndobj_input = sndobj.SndObj() - sndobj_input.SetVectorSize(frame_size) - sndobj_window = sndobj.HammingTable(frame_size, 0.5) - sndobj_ifgram = sndobj.IFGram(sndobj_window, sndobj_input, 1, frame_size, hop_size) - sndobj_sinmod = pysndobj.SinAnal(sndobj_ifgram, 0.003, max_tracks, 1, 3) - - pd = simpl.SndObjPeakDetection() - pd.max_peaks = max_peaks - pt = simpl.SndObjPartialTracking() - pt.max_partials = max_tracks - - i = 0 - while (i + frame_size) <= num_samples: - frame = audio_in[i:i+frame_size] - sndobj_input.PushIn(frame) - sndobj_ifgram.DoProcess() - sndobj_sinmod.DoProcess() - - frame = audio_in[i:i+frame_size] - peaks = pd.find_peaks(frame)[0] - partials = pt.update_partials(peaks, i/hop_size) - - num_sndobj_partials = sndobj_sinmod.GetTracks() - num_simpl_partials = len(partials) - self.assertEquals(num_sndobj_partials, num_simpl_partials) - - for j in range(num_simpl_partials): - self.assertAlmostEquals(partials[j].amplitude, sndobj_sinmod.Output(j*3), - places=FLOAT_PRECISION) - self.assertAlmostEquals(partials[j].frequency, sndobj_sinmod.Output((j*3)+1), - places=FLOAT_PRECISION) - self.assertAlmostEquals(partials[j].phase, sndobj_sinmod.Output((j*3)+2), - places=FLOAT_PRECISION) - i += hop_size - - def test_synthesis(self): - "Compare pysndobj synthesised audio with SndObj synthesised audio" - frame_size = 2048 - hop_size = 512 - num_frames = 4 - max_tracks = 20 - max_peaks = 20 - num_samples = frame_size + ((num_frames - 1) * hop_size) - - audio_in_data = read(self.input_file) - audio_in = simpl.asarray(audio_in_data[1]) / 32768.0 - audio_in = audio_in[0:num_samples] - frame = simpl.zeros(frame_size) - sndobj_frame_out = simpl.zeros(hop_size) - sndobj_audio_out = simpl.array([]) - - sndobj_input = sndobj.SndObj() - sndobj_input.SetVectorSize(frame_size) - sndobj_window = sndobj.HammingTable(frame_size, 0.5) - sndobj_ifgram = sndobj.IFGram(sndobj_window, sndobj_input, 1, frame_size, hop_size) - sndobj_sinmod = pysndobj.SinAnal(sndobj_ifgram, 0.003, max_tracks, 1, 3) - sndobj_table = sndobj.HarmTable(10000, 1, 1, 0.25) - sndobj_synth = sndobj.AdSyn(sndobj_sinmod, max_tracks, sndobj_table, 1, 1, hop_size) - - i = 0 - while (i + frame_size) <= num_samples: - frame = audio_in[i:i+frame_size] - sndobj_input.PushIn(frame) - sndobj_ifgram.DoProcess() - sndobj_sinmod.DoProcess() - sndobj_synth.DoProcess() - sndobj_synth.PopOut(sndobj_frame_out) - sndobj_audio_out = np.hstack((sndobj_audio_out, sndobj_frame_out)) - i += hop_size - - pd = simpl.SndObjPeakDetection() - pd.max_peaks = max_peaks - pt = simpl.SndObjPartialTracking() - pt.max_partials = max_tracks - peaks = pd.find_peaks(audio_in) - partials = pt.find_partials(peaks) - synth = simpl.SndObjSynthesis() - simpl_audio_out = synth.synth(partials) - self.assertEquals(sndobj_audio_out.size, simpl_audio_out.size) - for i in range(simpl_audio_out.size): - self.assertAlmostEquals(sndobj_audio_out[i], simpl_audio_out[i], - places = FLOAT_PRECISION) - -if __name__ == '__main__': - unittest.main() |