source: python/aubiopitch @ 9bec8fe

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

update params and pitch
update params and pitch

  • Property mode set to 100755
File size: 5.1 KB
RevLine 
[96fb8ad]1#!/usr/bin/python
2
[d4a0cc4]3""" this file was written by Paul Brossier
4  it is released under the GNU/GPL license.
5"""
6
7import sys
[13c3fba]8from aubio.task import *
[d4a0cc4]9
10usage = "usage: %s [options] -i soundfile" % sys.argv[0]
11
12
13def parse_args():
14        from optparse import OptionParser
15        parser = OptionParser(usage=usage)
16        parser.add_option("-i","--input",
17                          action="store", dest="filename", 
18                          help="input sound file")
[d9101a5]19        parser.add_option("-m","--mode", 
20                          action="store", dest="mode", default='mcomb',
[97b8c3d]21                          help="pitch detection mode [default=mcomb] \
[d4a0cc4]22                          mcomb|yin|fcomb|schmitt")
[aa17581]23        parser.add_option("-u","--units", action="callback", 
24                          callback=check_pitchm_mode, dest="omode",
25                          default=aubio_pitchm_freq,
26                          help="output pitch in units [default=Hz] \
27                          freq|midi|cent|bin")
[d4a0cc4]28        parser.add_option("-B","--bufsize",
[aa17581]29                          action="store", dest="bufsize", default=None, 
[d4a0cc4]30                          help="buffer size [default=1024]")
31        parser.add_option("-H","--hopsize",
[aa17581]32                          action="store", dest="hopsize", default=None, 
[d4a0cc4]33                          help="overlap size [default=512]")
34        parser.add_option("-t","--threshold",
35                          action="store", dest="threshold", default=0.1, 
36                          help="pitch threshold (for yin) [default=0.1]")
37        parser.add_option("-s","--silence",
38                          action="store", dest="silence", default=-70, 
39                          help="silence threshold [default=-70]")
40        parser.add_option("-D","--delay",
41                          action="store", dest="delay", 
[4798fdf]42                          help="number of seconds frames to take back [default=0]")
43        parser.add_option("-S","--smoothing",
44                          action="store", dest="smoothing", default=False, 
45                          help="use a median filter of N frames [default=0]")
46        parser.add_option("-M","--maximum",
47                          action="store", dest="pitchmax", default=False, 
48                          help="maximum pitch value to look for (Hz) [default=20000]")
49        parser.add_option("-l","--minimum",
50                          action="store", dest="pitchmin", default=False, 
51                          help="minimum pitch value to look for (Hz) [default=20]")
[d4a0cc4]52        # to be implemented
53        parser.add_option("-n","--note",
54                          action="store_true", dest="note", default=False,
55                          help="NOT IMPLEMENTED output notes")
56        # plotting functions
57        parser.add_option("-p","--plot",
58                          action="store_true", dest="plot", default=False, 
[4798fdf]59                          help="draw plot of the pitch track")
[d4a0cc4]60        parser.add_option("-O","--outplot",
61                          action="store", dest="outplot", default=None, 
[4798fdf]62                          help="save the plot to output.{ps,png,svg} instead of displaying it")
[d4a0cc4]63        parser.add_option("-v","--verbose",
64                          action="store_true", dest="verbose", default=False,
[4798fdf]65                          help="make lots of noise")
[d4a0cc4]66        parser.add_option("-q","--quiet",
67                          action="store_false", dest="verbose", default=False, 
68                          help="be quiet")
69        (options, args) = parser.parse_args()
[aa17581]70        if not options.bufsize:
71                if options.mode == aubio_pitch_yin:     options.bufsize = 1024
72                if options.mode == aubio_pitch_schmitt: options.bufsize = 2048
73                if options.mode == aubio_pitch_mcomb:   options.bufsize = 4096
74                if options.mode == aubio_pitch_fcomb:   options.bufsize = 4096 
[5e491b3b]75                else: options.bufsize = 2048
[aa17581]76        if not options.hopsize:
77                options.hopsize = float(options.bufsize) / 2
[d4a0cc4]78        if not options.filename: 
[aa17581]79                print "no file name given\n", usage
80                sys.exit(1)
[d4a0cc4]81        return options, args
82
83options, args = parse_args()
84
[aa17581]85#print options.bufsize, options.hopsize
86
[d4a0cc4]87filename   = options.filename
[d9101a5]88params = taskparams()
[0fe9aab]89params.samplerate = float(sndfile(filename).samplerate())
[d9101a5]90params.hopsize    = int(options.hopsize)
91params.bufsize    = int(options.bufsize)
[0fe9aab]92params.step       = params.samplerate/float(params.hopsize)
[650e39b]93params.yinthresh  = float(options.threshold)
[d9101a5]94params.silence    = float(options.silence)
[0fe9aab]95params.verbose    = options.verbose
[4798fdf]96if options.smoothing: params.pitchsmooth = int(options.smoothing)
97if options.pitchmax:  params.pitchmax    = int(options.pitchmax)
98if options.pitchmin:  params.pitchmin    = int(options.pitchmin)
[d4a0cc4]99#mintol     = float(options.mintol)*step
100# default take back system delay
[4798fdf]101if options.delay: params.pitchdelay = float(options.delay)
[d4a0cc4]102
103if options.note:
104        exit("not implemented yet")
105
[0fe9aab]106wplot,oplots = [],[]
[d9101a5]107modes = options.mode.split(',')
108for i in range(len(modes)):
[0fe9aab]109        pitch = []
[d9101a5]110        params.pitchmode  = modes[i]
[0fe9aab]111        filetask = taskpitch(filename,params=params)
112        pitch = filetask.compute_all()
113        #print filetask.eval(pitch[i])
[9610f0a]114        if options.plot: filetask.plot(pitch,wplot,oplots)
[d4a0cc4]115
[9610f0a]116if options.plot: 
117        filetask.plotplot(wplot,oplots,outplot=options.outplot)
Note: See TracBrowser for help on using the repository browser.