| 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 | def getonsets(filein,threshold): |
|---|
| 11 | bufsize = 1024 |
|---|
| 12 | hopsize = bufsize/2 |
|---|
| 13 | frameread = 0 |
|---|
| 14 | filei = sndfile(filein) |
|---|
| 15 | srate = filei.samplerate() |
|---|
| 16 | channels = filei.channels() |
|---|
| 17 | myvec = fvec(hopsize,channels) |
|---|
| 18 | readsize = filei.read(hopsize,myvec) |
|---|
| 19 | opick = onsetpick(bufsize,hopsize,channels,myvec,threshold) |
|---|
| 20 | #newname = "%s%.8f%s" % ("/tmp/",0.0000000,filein[-4:]) |
|---|
| 21 | #fileo = sndfile(newname,model=filei) |
|---|
| 22 | mylist = list() |
|---|
| 23 | while(readsize==hopsize): |
|---|
| 24 | readsize = filei.read(hopsize,myvec) |
|---|
| 25 | isonset,val = opick.do(myvec) |
|---|
| 26 | if (isonset == 1): |
|---|
| 27 | now = (frameread-4)*hopsize/(srate+0.) |
|---|
| 28 | #del fileo |
|---|
| 29 | #fileo = sndfile("%s%f%s" % ("/tmp/",now,filein[-4:]),model=filei) |
|---|
| 30 | mylist.append(now) |
|---|
| 31 | #writesize = fileo.write(readsize,myoldvec) |
|---|
| 32 | frameread += 1 |
|---|
| 33 | return mylist |
|---|
| 34 | |
|---|
| 35 | def cutfile(filein,onsets): |
|---|
| 36 | hopsize = 512 |
|---|
| 37 | frameread = 0 |
|---|
| 38 | readsize = hopsize |
|---|
| 39 | filei = sndfile(filein) |
|---|
| 40 | srate = filei.samplerate() |
|---|
| 41 | channels = filei.channels() |
|---|
| 42 | newname = "%s%f%s" % ("/tmp/",0.0000000,filein[-4:]) |
|---|
| 43 | fileo = sndfile(newname,model=filei) |
|---|
| 44 | myvec = fvec(hopsize,channels) |
|---|
| 45 | while(readsize==hopsize): |
|---|
| 46 | readsize = filei.read(hopsize,myvec) |
|---|
| 47 | now = (frameread)*hopsize/(srate+0.) |
|---|
| 48 | writesize = fileo.write(readsize,myvec) |
|---|
| 49 | if len(onsets) and now == onsets[0]: |
|---|
| 50 | onsets.pop(0) |
|---|
| 51 | del fileo |
|---|
| 52 | fileo = sndfile("%s%f%s%s" % ("/tmp/",now,".",filein.split(".")[-1]),model=filei) |
|---|
| 53 | frameread += 1 |
|---|
| 54 | del fileo |
|---|
| 55 | |
|---|
| 56 | filename = sys.argv[1] |
|---|
| 57 | threshold = sys.argv[2] |
|---|
| 58 | onsets = getonsets(filename,threshold) |
|---|
| 59 | cutfile(filename,onsets) |
|---|
