Functions
	computeNoveltyFunction (cNoveltyName, x, f_s, afWindow=None, iBlockLength=4096, iHopLength=512)
	computes the novelty function for onset detection supported novelty measures are: 'Flux', 'Laroche', 'Hainsworth'

	computeNoveltyFunctionCl (cPath, cNoveltyName)
	main

Variables
	parser = argparse.ArgumentParser(description='Compute key of wav file')

	metavar

	required

	help

	args = parser.parse_args()

	cInPath = args.infile

	cNoveltyName = args.noveltyname

	bPlotOutput = args.plotoutput

Function Documentation

◆ computeNoveltyFunction()

computeNoveltyFunction	(	cNoveltyName,
		x,
		f_s,
		afWindow = None,
		iBlockLength = 4096,
		iHopLength = 512 )

computes the novelty function for onset detection supported novelty measures are: 'Flux', 'Laroche', 'Hainsworth'

Parameters

cNoveltyName	name of the novelty measure
x	array with floating point audio data (dimension samples x channels)
f_s	sample rate of audio data
afWindow	FFT window of length iBlockLength (default: hann), can be [] empty
iBlockLength	internal block length (default: 4096 samples)
iHopLength	internal hop length (default: 2048 samples)

Returns: d: novelty function; t: time stamps; iPeaks: indices of picked onset times

Definition at line 32 of file computeNoveltyFunction.py.

def computeNoveltyFunction(cNoveltyName, x, f_s, afWindow=None, iBlockLength=4096, iHopLength=512):
 
    # compute window function for FFT
    if afWindow is None:
        afWindow = ToolComputeHann(iBlockLength)
 
    assert(afWindow.shape[0] == iBlockLength), "parameter error: invalid window dimension"
 
    hNoveltyFunc = getattr(pyACA, "Novelty" + cNoveltyName)
 
    # lp initialization
    fLenSmoothLpInS = 0.07
    fLenThreshLpInS = 0.14
    iLenSmoothLp = np.max([2, math.ceil(fLenSmoothLpInS * f_s / iHopLength)])
    iLenThreshLp = np.max([2, math.ceil(fLenThreshLpInS * f_s / iHopLength)])
 
    # pre-processing
    x = ToolPreprocAudio(x)
 
    # in the real world, we would do this block by block...
    [X, f, t] = computeSpectrogram(x, f_s, None, iBlockLength, iHopLength)
 
    # novelty function
    d = hNoveltyFunc(X, f_s)
 
    # smooth novelty function
    b = np.ones(iLenSmoothLp) / iLenSmoothLp
    d = filtfilt(b, 1, d)
    d[d < 0] = 0
 
    # compute threshold
    iLenThreshLp = min(iLenThreshLp, np.floor(len(d)/3))
    b = np.ones(iLenThreshLp) / iLenThreshLp
    G_T = .4 * np.mean(d[np.arange(1, d.shape[0])]) + filtfilt(b, 1, d)
 
    # find local maxima above the threshold
    iPeaks = find_peaks(d - G_T, height=0)
 
    return d, t, iPeaks[0]
 
 

◆ computeNoveltyFunctionCl()

computeNoveltyFunctionCl	(		cPath,
			cNoveltyName )

main

Definition at line 75 of file computeNoveltyFunction.py.

def computeNoveltyFunctionCl(cPath, cNoveltyName):
    
    [f_s, x] = ToolReadAudio(cPath)
    # afAudioData = np.sin(2*np.pi * np.arange(f_s*1)*440./f_s)
    [d, t, iPeaks] = computeNoveltyFunction(cNoveltyName, x, f_s)
    
    # plot feature output
    if bPlotOutput:
        plt.plot(t, d)
    return d, t, iPeaks
 
 

Variable Documentation

◆ args

args = parser.parse_args()

Definition at line 99 of file computeNoveltyFunction.py.

◆ bPlotOutput

bool bPlotOutput = args.plotoutput

Definition at line 102 of file computeNoveltyFunction.py.

◆ cInPath

str cInPath = args.infile

Definition at line 100 of file computeNoveltyFunction.py.

◆ cNoveltyName

str cNoveltyName = args.noveltyname

Definition at line 101 of file computeNoveltyFunction.py.

◆ help

help

Definition at line 92 of file computeNoveltyFunction.py.

◆ metavar

metavar

Definition at line 91 of file computeNoveltyFunction.py.

◆ parser

parser = argparse.ArgumentParser(description='Compute key of wav file')

Definition at line 90 of file computeNoveltyFunction.py.

◆ required

required