def analyze(audio_file, frame_rate=300, window_size=1001, window_type=pysms.SMS_WIN_HAMMING,
num_stoc_coeffs=128, default_fundamental=100, highest_freq=12000,
env_type=0, env_order=0):
pysms.sms_init()
analysis_params = pysms.SMS_AnalParams()
pysms.sms_initAnalParams(analysis_params)
analysis_params.fDefaultFundamental = default_fundamental
analysis_params.fHighestFreq = highest_freq
analysis_params.specEnvParams.iType = env_type
analysis_params.specEnvParams.iOrder = env_order
analysis_params.iFrameRate = frame_rate
analysis_params.nStochasticCoeff = num_stoc_coeffs
# Try to open the input file
snd_header = pysms.SMS_SndHeader()
if(pysms.sms_openSF(audio_file, snd_header)):
raise NameError("error opening sound file: " + pysms.sms_errorString())
# initialize memory for analysis
if pysms.sms_initAnalysis(analysis_params, snd_header) != 0:
raise Exception("Error allocating memory for analysis_params")
# copy data into the sms header
sms_header = pysms.SMS_Header()
pysms.sms_fillHeader(sms_header, analysis_params, "analysis.py")
analysis_frames = []
num_samples = snd_header.nSamples
num_frames = 0
sample_offset = 0
size_new_data = 0
do_analysis = True
# Analysis loop
while do_analysis:
sample_offset += size_new_data
if((sample_offset + analysis_params.sizeNextRead) < num_samples):
size_new_data = analysis_params.sizeNextRead
else:
size_new_data = num_samples - sample_offset
frame = np.zeros(size_new_data, dtype=np.float32)
if pysms.sms_getSound(snd_header, frame, sample_offset, analysis_params):
raise NameError("error opening sound file: " + pysms.sms_errorString())
data = pysms.SMS_Data()
pysms.sms_allocFrameH(sms_header, data)
status = pysms.sms_analyze(frame, data, analysis_params)
if status == 1:
analysis_frames.append(data)
num_frames += 1
elif status == -1:
do_analysis = False
sms_header.nFrames = num_frames
# free memory and return
pysms.sms_freeAnalysis(analysis_params)
pysms.sms_closeSF()
pysms.sms_free()
return analysis_frames, sms_header, snd_header
def _pysms_analysis_params(sampling_rate):
analysis_params = pysms.SMS_AnalParams()
pysms.sms_initAnalParams(analysis_params)
analysis_params.iSamplingRate = sampling_rate
analysis_params.iFrameRate = sampling_rate / hop_size
analysis_params.iWindowType = pysms.SMS_WIN_HAMMING
analysis_params.fHighestFreq = 20000
analysis_params.iFormat = pysms.SMS_FORMAT_IHP
analysis_params.nTracks = max_peaks
analysis_params.peakParams.iMaxPeaks = max_peaks
analysis_params.nGuides = max_peaks
analysis_params.iMaxDelayFrames = 4
analysis_params.analDelay = 0
analysis_params.minGoodFrames = 1
analysis_params.iCleanTracks = 0
analysis_params.iStochasticType = pysms.SMS_STOC_NONE
analysis_params.preEmphasis = 0
return analysis_params