Showing the changes to python (1 of 1 file). Show all
Unified · Split
| @@ -77,7 +77,7 @@ class peakpick: · 76 unmodified lines | ||
| 77 | 77 | |
| 78 | 78 | class onsetpick: |
| 79 | 79 | """ superclass for aubio_pvoc + aubio_onsetdetection + aubio_peakpicker """ |
| 80 | − def __init__(self,bufsize,hopsize,channels,myvec,threshold,mode='dual',derivate=False): | |
| 80 | + def __init__(self,bufsize,hopsize,channels,myvec,threshold,mode='dual',derivate=False,dcthreshold=0): | |
| 81 | 81 | self.myfft = cvec(bufsize,channels) |
| 82 | 82 | self.pv = pvoc(bufsize,hopsize,channels) |
| 83 | 83 | if mode in ['dual'] : |
| @@ -91,6 +91,7 @@ class onsetpick: · 7 unmodified lines | ||
| 91 | 91 | self.mode = mode |
| 92 | 92 | self.pp = peakpick(float(threshold)) |
| 93 | 93 | self.derivate = derivate |
| 94 | + self.dcthreshold = dcthreshold | |
| 94 | 95 | self.oldval = 0. |
| 95 | 96 | |
| 96 | 97 | def do(self,myvec): |
| @@ -105,6 +106,10 @@ class onsetpick: · 8 unmodified lines | ||
| 105 | 106 | self.oldval = val |
| 106 | 107 | if dval > 0: self.myonset.set(dval,0,0) |
| 107 | 108 | else: self.myonset.set(0.,0,0) |
| 109 | + if self.dcthreshold: | |
| 110 | + dval = self.myonset.get(0,0) - self.dcthreshold | |
| 111 | + if dval > 0: self.myonset.set(dval,0,0) | |
| 112 | + else: self.myonset.set(0.,0,0) | |
| 108 | 113 | return self.pp.do(self.myonset),self.myonset.get(0,0) |
| 109 | 114 | |
| 110 | 115 | class pitchdetection: |