79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
##| Copyright: (C) 2019-2020 Kevin Larke <contact AT larke DOT org>
|
|
##| License: GNU GPL version 3.0 or above. See the accompanying LICENSE file.
|
|
import types
|
|
|
|
import numpy as np
|
|
|
|
from rms_analysis import audio_harm_rms
|
|
from rms_analysis import audio_rms
|
|
from rms_analysis import locate_peak_indexes
|
|
from rms_analysis import measure_duration_ms
|
|
|
|
|
|
class RT_Analyzer:
|
|
def __init__(self):
|
|
self.td_dur_ms = 0
|
|
self.td_db = 0
|
|
self.hm_dur_ms = 0
|
|
self.hm_db = 0
|
|
|
|
def analyze_note( self, audioDev, midi_pitch, begTimeMs, endTimeMs, anlArgD ):
|
|
td_dur_ms = 0
|
|
td_db = 0
|
|
hm_dur_ms = 0
|
|
hm_db = 0
|
|
|
|
|
|
decay_pct = 50.0
|
|
|
|
result = audioDev.linear_buffer()
|
|
|
|
if result:
|
|
|
|
|
|
sigV = result.value
|
|
|
|
anlArgs = types.SimpleNamespace(**anlArgD)
|
|
|
|
rmsDbV, rms_srate, binHz = audio_harm_rms( audioDev.srate, np.squeeze(sigV), anlArgs.rmsWndMs, anlArgs.rmsHopMs, anlArgs.dbLinRef, midi_pitch, anlArgs.harmCandN, anlArgs.harmN )
|
|
|
|
pkIdxL = locate_peak_indexes( rmsDbV, rms_srate, [( begTimeMs, endTimeMs)] )
|
|
|
|
if len(pkIdxL) > 0:
|
|
|
|
end_idx = int(round(endTimeMs * rms_srate / 1000.0))
|
|
|
|
if end_idx > pkIdxL[0]:
|
|
hm_dur_ms = measure_duration_ms( rmsDbV, rms_srate, pkIdxL[0], end_idx, decay_pct )
|
|
|
|
hm_db = rmsDbV[ pkIdxL[0] ]
|
|
|
|
tdRmsDbV, rms0_srate = audio_rms( audioDev.srate, np.squeeze(sigV), anlArgs.rmsWndMs, anlArgs.rmsHopMs, anlArgs.dbLinRef )
|
|
|
|
tdPkIdxL = locate_peak_indexes( tdRmsDbV, rms0_srate, [( begTimeMs, endTimeMs)] )
|
|
|
|
if len(tdPkIdxL):
|
|
|
|
end_idx = int(round(endTimeMs * rms0_srate / 1000.0))
|
|
|
|
if end_idx > tdPkIdxL[0]:
|
|
td_dur_ms = measure_duration_ms( tdRmsDbV, rms0_srate, tdPkIdxL[0], end_idx, decay_pct )
|
|
|
|
td_db = tdRmsDbV[ tdPkIdxL[0] ]
|
|
|
|
|
|
td_d_ms = td_dur_ms - self.td_dur_ms
|
|
td_d_db = td_db - self.td_db
|
|
hm_d_ms = hm_dur_ms - self.hm_dur_ms
|
|
hm_d_db = hm_db - self.hm_db
|
|
|
|
|
|
print("DUR: %5.2f %5.2f d:%5.2f %5.2f dB | %i %i d:%i %i ms" % (hm_db, td_db, hm_d_db, td_d_db, hm_dur_ms, td_dur_ms, hm_d_ms, td_d_ms) )
|
|
|
|
self.td_dur_ms = td_dur_ms
|
|
self.td_db = td_db
|
|
self.hm_dur_ms = hm_dur_ms
|
|
self.hm_db = hm_db
|
|
|
|
|