source: python/aubio/aubioclass.py @ b4f5967

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5pitchshiftsamplertimestretchyinfft+
Last change on this file since b4f5967 was b4f5967, checked in by Paul Brossier <piem@piem.org>, 15 years ago

src/onset: use a string to set onset mode, keeping onset types enum private, update everywhere onsets are used

  • Property mode set to 100644
File size: 5.6 KB
Line 
1from aubiowrapper import *
2
3class fvec:
4    def __init__(self,size,chan):
5        self.vec = new_fvec(size,chan)
6    def __call__(self):
7        return self.vec
8    def __del__(self):
9        del_fvec(self())
10    def get(self,pos,chan):
11        return fvec_read_sample(self(),chan,pos)
12    def set(self,value,pos,chan):
13        return fvec_write_sample(self(),value,chan,pos)
14    def channel(self,chan):
15        return fvec_get_channel(self(),chan)
16    def data(self):
17        return fvec_get_data(self())
18
19class cvec:
20    def __init__(self,size,chan):
21        self.vec = new_cvec(size,chan)
22    def __call__(self):
23        return self.vec
24    def __del__(self):
25        del_cvec(self())
26    def get(self,pos,chan):
27        return self.get_norm(pos,chan)
28    def set(self,val,pos,chan):
29        self.set_norm(val,chan,pos)
30    def get_norm(self,pos,chan):
31        return cvec_read_norm(self(),chan,pos)
32    def set_norm(self,val,pos,chan):
33        cvec_write_norm(self(),val,chan,pos)
34    def get_phas(self,pos,chan):
35        return cvec_read_phas(self(),chan,pos)
36    def set_phas(self,val,pos,chan):
37        cvec_write_phas(self(),val,chan,pos)
38
39class sndfile:
40    def __init__(self,filename,model=None):
41        if (model!=None):
42            self.file = new_aubio_sndfile_wo(model.file,filename)
43        else:
44            self.file = new_aubio_sndfile_ro(filename)
45        if self.file == None:
46            raise IOError, "failed opening file %s" % filename
47    def __del__(self):
48        if self.file != None: del_aubio_sndfile(self.file)
49    def info(self):
50        aubio_sndfile_info(self.file)
51    def samplerate(self):
52        return aubio_sndfile_samplerate(self.file)
53    def channels(self):
54        return aubio_sndfile_channels(self.file)
55    def read(self,nfram,vecread):
56        return aubio_sndfile_read(self.file,nfram,vecread())
57    def write(self,nfram,vecwrite):
58        return aubio_sndfile_write(self.file,nfram,vecwrite())
59
60class pvoc:
61    def __init__(self,buf,hop,chan):
62        self.pv = new_aubio_pvoc(buf,hop,chan)
63    def __del__(self):
64        del_aubio_pvoc(self.pv)
65    def do(self,tf,tc):
66        aubio_pvoc_do(self.pv,tf(),tc())
67    def rdo(self,tc,tf):
68        aubio_pvoc_rdo(self.pv,tc(),tf())
69
70class onsetdetection:
71    """ class for aubio_onsetdetection """
72    def __init__(self,mode,buf,chan):
73        self.od = new_aubio_onsetdetection(mode,buf,chan)
74    def do(self,tc,tf):
75        aubio_onsetdetection_do(self.od,tc(),tf())
76    def __del__(self):
77        del_aubio_onsetdetection(self.od)
78
79class peakpick:
80    """ class for aubio_peakpicker """
81    def __init__(self,threshold=0.1):
82        self.pp = new_aubio_peakpicker(threshold)
83    def do(self,fv):
84        return aubio_peakpicker_do(self.pp, fv())
85    def getval(self):
86        return aubio_peakpicker_get_adaptive_threshold(self.pp)
87    def __del__(self):
88        del_aubio_peakpicker(self.pp)
89
90class onsetpick:
91    """ superclass for aubio_pvoc + aubio_onsetdetection + aubio_peakpicker """
92    def __init__(self,bufsize,hopsize,channels,myvec,threshold,mode='dual',derivate=False,dcthreshold=0):
93        self.myfft    = cvec(bufsize,channels)
94        self.pv       = pvoc(bufsize,hopsize,channels)
95        if mode in ['dual'] :
96                self.myod     = onsetdetection("hfc",bufsize,channels)
97                self.myod2    = onsetdetection("mkl",bufsize,channels)
98                self.myonset  = fvec(1,channels)
99                self.myonset2 = fvec(1,channels)
100        else: 
101                self.myod     = onsetdetection(mode,bufsize,channels)
102                self.myonset  = fvec(1,channels)
103        self.mode     = mode
104        self.pp       = peakpick(float(threshold))
105        self.derivate = derivate
106        self.dcthreshold = dcthreshold
107        self.oldval   = 0.
108
109    def do(self,myvec): 
110        self.pv.do(myvec,self.myfft)
111        self.myod.do(self.myfft,self.myonset)
112        if self.mode == 'dual':
113           self.myod2.do(self.myfft,self.myonset2)
114           self.myonset.set(self.myonset.get(0,0)*self.myonset2.get(0,0),0,0)
115        if self.derivate:
116           val         = self.myonset.get(0,0)
117           dval        = val - self.oldval
118           self.oldval = val
119           if dval > 0: self.myonset.set(dval,0,0)
120           else:  self.myonset.set(0.,0,0)
121        isonset, dval = self.pp.do(self.myonset),self.myonset.get(0,0)
122        if self.dcthreshold:
123           if dval < self.dcthreshold: isonset = 0 
124        return isonset, dval
125
126class pitchdetection:
127    def __init__(self,mode=aubio_pitch_mcomb,bufsize=2048,hopsize=1024,
128        channels=1,samplerate=44100.,omode=aubio_pitchm_freq,tolerance=0.1):
129        self.pitchp = new_aubio_pitchdetection(bufsize,hopsize,channels,
130                samplerate,mode,omode)
131        self.mypitch = fvec(1, channels)
132        aubio_pitchdetection_set_tolerance(self.pitchp,tolerance)
133        #self.filt     = filter(srate,"adsgn")
134    def __del__(self):
135        del_aubio_pitchdetection(self.pitchp)
136    def __call__(self,myvec): 
137        aubio_pitchdetection_do(self.pitchp,myvec(), self.mypitch())
138        return self.mypitch.get(0,0)
139
140class filter:
141    def __init__(self,srate,type=None):
142        if (type=="adsgn"):
143            self.filter = new_aubio_adsgn_filter(srate)
144    def __del__(self):
145        #del_aubio_filter(self.filter)
146        pass
147    def __call__(self,myvec):
148        aubio_filter_do(self.filter,myvec())
149
150class beattracking:
151    """ class for aubio_beattracking """
152    def __init__(self,winlen,channels):
153        self.p = new_aubio_beattracking(winlen,channels)
154    def do(self,dfframe,out):
155        return aubio_beattracking_do(self.p,dfframe(),out())
156    def __del__(self):
157        del_aubio_beattracking(self.p)
158
Note: See TracBrowser for help on using the repository browser.