Mercurial > hg > vampy

annotate Example VamPy plugins/obsolete/PySpectralCentroid.py @ 46:af9c4cee95a8

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
VC++ fixes. Much of this is changing "and" to "&&". I had never realised until today that "and" is in fact a keyword in C++, albeit not one that has been there since the start, so this should compile (I eventually looked this up having been puzzled by how this code could ever build with any other compiler). However, despite its keywordness, "and" still doesn't seem to be acceptable to VC++. Possibly there's an option to change this, or one could use a macro -- but why not just stick with the token that compilers are known to like?
author cannam
date Mon, 05 Oct 2009 16:14:25 +0000
parents 27bab3a16c9a
children
rev   line source
fazekasgy@37 1 '''PySpectralCentroid.py - Example plugin demonstrates
fazekasgy@37 2 how to write a C style plugin using VamPy.
fazekasgy@37 3
fazekasgy@37 4 Obsolete warning: this plugin will no longer be supported
fazekasgy@37 5 since the legacy interface should pass a list of complex
fazekasgy@37 6 numbers for frequency domain plugins and a list of floats
fazekasgy@37 7 for time domin plugins.
fazekasgy@37 8
fazekasgy@37 9 '''
fazekasgy@37 10
fazekasgy@37 11 from numpy import *
fazekasgy@37 12
fazekasgy@37 13 class PySpectralCentroid:
fazekasgy@37 14
fazekasgy@37 15 def __init__(self,inputSampleRate):
fazekasgy@37 16 self.m_imputSampleRate = 0.0
fazekasgy@37 17 self.m_stepSize = 0
fazekasgy@37 18 self.m_blockSize = 0
fazekasgy@37 19 self.m_channels = 0
fazekasgy@37 20 self.previousSample = 0.0
fazekasgy@37 21 self.m_inputSampleRate = inputSampleRate
fazekasgy@37 22 self.threshold = 0.00
fazekasgy@37 23
fazekasgy@37 24 def initialise(self,channels,stepSize,blockSize):
fazekasgy@37 25 self.m_channels = channels
fazekasgy@37 26 self.m_stepSize = stepSize
fazekasgy@37 27 self.m_blockSize = blockSize
fazekasgy@37 28 return True
fazekasgy@37 29
fazekasgy@37 30 def getMaker(self):
fazekasgy@37 31 return 'Vampy Example Plugins'
fazekasgy@37 32
fazekasgy@37 33 def getName(self):
fazekasgy@37 34 return 'Spectral Centroid (legacy process interface)'
fazekasgy@37 35
fazekasgy@37 36 def getIdentifier(self):
fazekasgy@37 37 return 'vampy-sc2'
fazekasgy@37 38
fazekasgy@37 39 def getMaxChannelCount(self):
fazekasgy@37 40 return 1
fazekasgy@37 41
fazekasgy@37 42 def getInputDomain(self):
fazekasgy@37 43 return 'FrequencyDomain'
fazekasgy@37 44
fazekasgy@37 45 def getOutputDescriptors(self):
fazekasgy@37 46
fazekasgy@37 47 output0={
fazekasgy@37 48 'identifier':'vampy-sf1',
fazekasgy@37 49 'name':'Spectral Centroid',
fazekasgy@37 50 'description':'Spectral Centroid (Brightness)',
fazekasgy@37 51 'unit':' ',
fazekasgy@37 52 'hasFixedBinCount':True,
fazekasgy@37 53 'binCount':1,
fazekasgy@37 54 'hasKnownExtents':False,
fazekasgy@37 55 'isQuantized':True,
fazekasgy@37 56 'quantizeStep':1.0,
fazekasgy@37 57 'sampleType':'OneSamplePerStep'
fazekasgy@37 58 }
fazekasgy@37 59
fazekasgy@37 60 return [output0]
fazekasgy@37 61
fazekasgy@37 62 def getParameterDescriptors(self):
fazekasgy@37 63 paramlist1={
fazekasgy@37 64 'identifier':'threshold',
fazekasgy@37 65 'name':'Noise threshold: ',
fazekasgy@37 66 'description':'Return null or delete this function if not needed.',
fazekasgy@37 67 'unit':'v',
fazekasgy@37 68 'minValue':0.0,
fazekasgy@37 69 'maxValue':0.5,
fazekasgy@37 70 'defaultValue':0.05,
fazekasgy@37 71 'isQuantized':False
fazekasgy@37 72 }
fazekasgy@37 73 return [paramlist1]
fazekasgy@37 74
fazekasgy@37 75 def setParameter(self,paramid,newval):
fazekasgy@37 76 if paramid == 'threshold' :
fazekasgy@37 77 self.threshold = newval
fazekasgy@37 78 return
fazekasgy@37 79
fazekasgy@37 80 def getParameter(self,paramid):
fazekasgy@37 81 if paramid == 'threshold' :
fazekasgy@37 82 return self.threshold
fazekasgy@37 83 else:
fazekasgy@37 84 return 0.0
fazekasgy@37 85
fazekasgy@37 86 def process(self,inbuf,timestamp):
fazekasgy@37 87
fazekasgy@37 88 inArray = array(inbuf[0])
fazekasgy@37 89 crossing = False
fazekasgy@37 90 prev = self.previousSample
fazekasgy@37 91 count = 0.0
fazekasgy@37 92 numLin = 0.0
fazekasgy@37 93 denom = 0.0
fazekasgy@37 94 centroid = 0.0
fazekasgy@37 95
fazekasgy@37 96 # re = array(inbuf[2:len(inArray):2])
fazekasgy@37 97 # im = array(inbuf[3:len(inArray):2])
fazekasgy@37 98
fazekasgy@37 99 output0=[]
fazekasgy@37 100 output1=[]
fazekasgy@37 101
fazekasgy@37 102 # pw = 0
fazekasgy@37 103 # for i in xrange(1,len(inbuf[0])) :
fazekasgy@37 104 # pw = pw + abs(inbuf[0][i])
fazekasgy@37 105
fazekasgy@37 106 if sum(abs(inArray)) > self.threshold :
fazekasgy@37 107 for i in range(1,(len(inArray)/2)) :
fazekasgy@37 108 # for i in range(1,len(inbuf[0])) :
fazekasgy@37 109
fazekasgy@37 110 re = inArray[i*2]
fazekasgy@37 111 im = inArray[i*2+1]
fazekasgy@37 112 # re = inbuf[0][i].real
fazekasgy@37 113 # im = inbuf[0][i].imag
fazekasgy@37 114 freq = i * self.m_inputSampleRate / self.m_blockSize
fazekasgy@37 115 power = sqrt (re*re + im*im) / (self.m_blockSize/2)
fazekasgy@37 116 denom = denom + power
fazekasgy@37 117 numLin = numLin + freq * power
fazekasgy@37 118
fazekasgy@37 119 if denom != 0 :
fazekasgy@37 120 centroid = numLin / denom
fazekasgy@37 121
fazekasgy@37 122 else :
fazekasgy@37 123 centroid = 0.0
fazekasgy@37 124
fazekasgy@37 125 feature0={
fazekasgy@37 126 'hasTimestamp':False,
fazekasgy@37 127 'values':[centroid], #strictly must be a list
fazekasgy@37 128 'label':str(centroid)
fazekasgy@37 129 }
fazekasgy@37 130 output0.append(feature0)
fazekasgy@37 131
fazekasgy@37 132 return [output0]