Mercurial > hg > aimc

changeset 266:35987061c93b

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
small bug in /swig/audio_example.py
author hamel.phil
date Thu, 13 Jan 2011 18:49:40 +0000
parents 3d5048725760
children 88b515b1a453
files swig/audio_example.py
diffstat 1 files changed, 27 insertions(+), 5 deletions(-) [+]
line wrap: on
line diff
--- a/swig/audio_example.py	Thu Jan 13 18:46:47 2011 +0000
+++ b/swig/audio_example.py	Thu Jan 13 18:49:40 2011 +0000
@@ -9,8 +9,18 @@
 
 import aimc
 import numpy
+from copy import copy
 
-do_plot=True
+def get_strobe_matrix(strobes,buf_len = 1024):
+	n_channel = len(strobes)
+	strobe_mat = numpy.zeros((n_channel,buf_len))
+	for ch in range(n_channel):
+		for s in strobes[ch]:
+			strobe_mat[ch,s] = 1.
+	return strobe_mat
+
+
+do_plot=False
 # Example wav file
 wavfile = '../test_data/short_example.wav'
 
@@ -26,7 +36,7 @@
 	x=x.astype('float')/float(pow(2,15) - 1)
 
 if x.ndim <= 1:
-	x=reshape(x,(x.shape[0],1))
+	x=numpy.reshape(x,(x.shape[0],1))
 sr = 1.*sr
 	
 nChannels = x.shape[1]
@@ -63,7 +73,8 @@
 pzfc.Initialize(sig,global_params)
 
 output_list = []
-bank_list =[]
+strobe_list = []
+centre_freq_list=[]
 
 for f in range(nFrames):
 	for i in range(nChannels):
@@ -71,13 +82,24 @@
 
 	pzfc.Process(sig)
 	output_bank = sai.GetOutputBank()
-	bank_list.append(output_bank)
 	n_channel = output_bank.channel_count()
 	sig_length = output_bank.buffer_length()
 	output_matrix = numpy.zeros((n_channel,sig_length))
+	strobes=[]
+	freqs=[]
 	for i in range(n_channel):
 		output_matrix[i] = numpy.array(output_bank.get_signal(i))
+		freqs.append(output_bank.centre_frequency(i))
+		channel_strobes = [] 
+		for j in range(output_bank.strobe_count(i)):
+			channel_strobes.append(output_bank.strobe(i,j))
+		strobes.append(channel_strobes)
+	centre_freq_list.append(freqs)
+	strobe_list.append(strobes)
 	output_list.append(output_matrix)
+#	P.figure()
+#	P.imshow(output_matrix, aspect='auto', origin='lower')
+
 
 print 'nFrames, nChannels, nSamples, sample_rate'
 print nFrames, n_channel, sig_length, sr
@@ -131,4 +153,4 @@
 #local_max_output = process_module(local_max, hcl_output, global_params)
 #
 #sai = aimc.ModuleSAI( aimc.Parameters())
-#sai_output = process_module(sai, local_max_output, global_params)
\ No newline at end of file
+#sai_output = process_module(sai, local_max_output, global_params)