python/aubiocut

   1 #!/usr/bin/python
   2
   3 """ this file was written by Paul Brossier
   4   it is released under the GNU/GPL license.
   5 """
   6
   7 from aubio.aubioclass import *
   8 import sys
   9
  10 bufsize   = 1024
  11 hopsize   = bufsize/2
  12
  13 def getonsets(filein,threshold):
  14         frameread = 0
  15         filei     = sndfile(filein)
  16         srate     = filei.samplerate()
  17         channels  = filei.channels()
  18         myvec     = fvec(hopsize,channels)
  19         readsize  = filei.read(hopsize,myvec)
  20         opick     = onsetpick(bufsize,hopsize,channels,myvec,threshold)
  21         #newname   = "%s%.8f%s" % ("/tmp/",0.0000000,filein[-4:])
  22         #fileo     = sndfile(newname,model=filei)
  23         mylist    = list()
  24         ovalist   = [0., 0., 0., 0., 0., 0.]
  25         while(readsize==hopsize):
  26                 readsize = filei.read(hopsize,myvec)
  27                 isonset,val = opick.do(myvec)
  28                 ovalist.append(val)
  29                 ovalist.pop(0)
  30                 if (isonset == 1):
  31                         print frameread
  32                         i=len(ovalist)-1
  33                         # find local minima
  34                         while ovalist[i-1] < ovalist[i] and i > 0:
  35                                 i -= 1
  36                         now = (frameread-i+1)*hopsize/(srate+0.)
  37                         #del fileo
  38                         #fileo = sndfile("%s%f%s" % ("/tmp/",now,filein[-4:]),model=filei)
  39                         mylist.append(now)
  40                 #writesize = fileo.write(readsize,myoldvec)
  41                 frameread += 1
  42         return mylist
  43
  44 def cutfile(filein,onsets):
  45         frameread = 0
  46         readsize  = hopsize
  47         filei     = sndfile(filein)
  48         srate     = filei.samplerate()
  49         channels  = filei.channels()
  50         newname   = "%s%f%s" % ("/tmp/",0.0000000,filein[-4:])
  51         fileo     = sndfile(newname,model=filei)
  52         myvec     = fvec(hopsize,channels)
  53         while(readsize==hopsize):
  54                 readsize = filei.read(hopsize,myvec)
  55                 now = (frameread)*hopsize/(srate+0.)
  56                 writesize = fileo.write(readsize,myvec)
  57                 if len(onsets) and now == onsets[0]:
  58                     onsets.pop(0)
  59                     del fileo
  60                     fileo = sndfile("%s%f%s%s" % ("/tmp/",now,".",filein.split(".")[-1]),model=filei)
  61                 frameread += 1
  62         del fileo
  63
  64 filename  = sys.argv[1]
  65 threshold = sys.argv[2]
  66 onsets    = getonsets(filename,threshold)
  67 cutfile(filename,onsets)