86 lines
2.1 KiB
Python
86 lines
2.1 KiB
Python
|
|
||
|
|
import sys,os
|
||
|
|
import numpy as np
|
||
|
|
import common
|
||
|
|
import rms_analysis
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def fit_points_to_reference( usL, dbL, usRefL, dbRefL ):
|
||
|
|
|
||
|
|
dbV = None
|
||
|
|
|
||
|
|
yyL = [ (db,dbRefL[ usRefL.index(us)]) for i,(us,db) in enumerate(zip(usL,dbL)) if us in usRefL ]
|
||
|
|
|
||
|
|
if len(yyL) < 10:
|
||
|
|
print("NO FIT")
|
||
|
|
else:
|
||
|
|
|
||
|
|
y0L,yrL = zip(*yyL)
|
||
|
|
yN = len(y0L)
|
||
|
|
|
||
|
|
A = np.vstack([np.ones(yN),y0L]).T
|
||
|
|
c,m = np.linalg.lstsq(A,yrL,rcond=None)[0]
|
||
|
|
|
||
|
|
dbV = (np.array(dbL) * m) + c
|
||
|
|
|
||
|
|
|
||
|
|
return dbV
|
||
|
|
|
||
|
|
def find_elbow( usL, dbL, pointsPerLine=10 ):
|
||
|
|
|
||
|
|
ppl_2 = int(pointsPerLine/2)
|
||
|
|
dL = []
|
||
|
|
|
||
|
|
|
||
|
|
i = pointsPerLine
|
||
|
|
|
||
|
|
# for each consecutive set of 'pointsPerLine' points in usL and dbL
|
||
|
|
while i < len(usL):
|
||
|
|
|
||
|
|
# find the x,y coordinates of the first 'ppl_2' coordinates
|
||
|
|
x0L = np.array([ (us,1.0) for us in usL[i-pointsPerLine:i-ppl_2] ])
|
||
|
|
y0L = np.array(usL[i-pointsPerLine:i-ppl_2])
|
||
|
|
|
||
|
|
# find the x,y coordinates of the second 'ppl_2' coordinates
|
||
|
|
x1L = np.array([ (us,1.0) for us in usL[i-ppl_2:i]])
|
||
|
|
y1L = np.array(dbL[i-ppl_2:i])
|
||
|
|
|
||
|
|
|
||
|
|
m0,c0 = np.linalg.lstsq(x0L,y0L,rcond=None)[0] # fit a line through the first set of points
|
||
|
|
m1,c1 = np.linalg.lstsq(x1L,y1L,rcond=None)[0] # fit a line through the second set of points
|
||
|
|
|
||
|
|
# store the angle between the two lines
|
||
|
|
dL.append(m1-m0)
|
||
|
|
|
||
|
|
i += 1
|
||
|
|
|
||
|
|
# find the max angle
|
||
|
|
i = np.argmax( dL )
|
||
|
|
|
||
|
|
# return the x,y coordinate of the first data point of the second line
|
||
|
|
return (usL[i+ppl_2],dbL[i+ppl_2])
|
||
|
|
|
||
|
|
def find_elbow_main( cfg, inDir, midi_pitch, takeId ):
|
||
|
|
|
||
|
|
inDir = os.path.join(inDir,str(pitch),str(takeId))
|
||
|
|
analysisArgsD = cfg.analysisArgs['rmsAnalysArgs']
|
||
|
|
|
||
|
|
r = rms_analysis_main( inDir, int(midi_pitch), **analysisD )
|
||
|
|
|
||
|
|
usL = r.pkUsL
|
||
|
|
dbL = r.pkDbL
|
||
|
|
|
||
|
|
return find_elbow(r.pkUsL,r.pkDbL)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
|
||
|
|
inDir = sys.argv[1]
|
||
|
|
cfgFn = sys.argv[2]
|
||
|
|
pitch = sys.argv[3]
|
||
|
|
|
||
|
|
cfg = common.parse_yaml_cfg(cfgFn)
|
||
|
|
|
||
|
|
find_elbow( cfg, inDir, pitch, 0 )
|