source: examples/utils.c @ 0a7f424

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

examples/: check if sink_uri exists, add -f to force overwrite

  • Property mode set to 100644
File size: 4.8 KB
RevLine 
[50bc5f2]1/*
[466dff3]2  Copyright (C) 2003-2013 Paul Brossier <piem@aubio.org>
[96fb8ad]3
[50bc5f2]4  This file is part of aubio.
[96fb8ad]5
[50bc5f2]6  aubio is free software: you can redistribute it and/or modify
7  it under the terms of the GNU General Public License as published by
8  the Free Software Foundation, either version 3 of the License, or
9  (at your option) any later version.
10
11  aubio is distributed in the hope that it will be useful,
12  but WITHOUT ANY WARRANTY; without even the implied warranty of
13  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  GNU General Public License for more details.
15
16  You should have received a copy of the GNU General Public License
17  along with aubio.  If not, see <http://www.gnu.org/licenses/>.
18
19*/
[96fb8ad]20
[9430dfd]21/**
[d4c5de7]22
23  This file includes some tools common to all examples. Code specific to the
24  algorithm performed by each program should go in the source file of that
25  program instead.
26
27*/
28
[96fb8ad]29#include "utils.h"
[1b25a70]30#ifdef HAVE_JACK
31#include "jackio.h"
32#endif /* HAVE_JACK */
[96fb8ad]33
[bd2f2ab]34int verbose = 0;
[466dff3]35int usejack = 0;
[1b25a70]36// input / output
37char_t *sink_uri = NULL;
38char_t *source_uri = NULL;
39// general stuff
40uint_t samplerate = 0;
41uint_t buffer_size = 512;
[466dff3]42uint_t hop_size = 256;
[1b25a70]43// onset stuff
44char_t * onset_method = "default";
45smpl_t onset_threshold = 0.0; // will be set if != 0.
46// pitch stuff
[d4c5de7]47char_t * pitch_unit = "default";
[1b25a70]48char_t * pitch_method = "default";
49smpl_t pitch_tolerance = 0.0; // will be set if != 0.
50// tempo stuff
51char_t * tempo_method = "default";
52// more general stuff
[ce6186a]53smpl_t silence_threshold = -90.;
[1b25a70]54uint_t mix_input = 0;
[50bc5f2]55
[0a509c6]56uint_t force_overwrite = 0;
57
[1b25a70]58//
59// internal memory stuff
[5a66677]60aubio_source_t *this_source = NULL;
61aubio_sink_t *this_sink = NULL;
[50bc5f2]62fvec_t *ibuf;
63fvec_t *obuf;
[96fb8ad]64
[1b25a70]65/* settings */
[466dff3]66int blocks = 0;
[50bc5f2]67
[1b25a70]68extern void usage (FILE * stream, int exit_code);
69extern int parse_args (int argc, char **argv);
[96fb8ad]70
[466dff3]71#if HAVE_JACK
72aubio_jack_t *jack_setup;
73#endif
74
[50bc5f2]75void
76examples_common_init (int argc, char **argv)
77{
[bd2f2ab]78
79  /* parse command line arguments */
[50bc5f2]80  parse_args (argc, argv);
[bd2f2ab]81
[50bc5f2]82  if (!usejack) {
83    debug ("Opening files ...\n");
[466dff3]84    this_source = new_aubio_source ((char_t*)source_uri, samplerate, hop_size);
[5a66677]85    if (this_source == NULL) {
[912f343]86      outmsg ("Could not open input file %s\n", source_uri);
[50bc5f2]87      exit (1);
[9af07aa]88    }
[466dff3]89    if (samplerate == 0) {
90      samplerate = aubio_source_get_samplerate(this_source);
91    }
[5a66677]92    if (sink_uri != NULL) {
[0a509c6]93      uint_t sink_exists = (access(sink_uri, F_OK) == 0 );
94      if (!force_overwrite && sink_exists) {
95        outmsg ("Output file %s already exists, use -f to overwrite.\n",
96            sink_uri);
97        exit (1);
98      }
[5a66677]99      this_sink = new_aubio_sink ((char_t*)sink_uri, samplerate);
[88fc249]100      if (this_sink == NULL) {
[912f343]101        outmsg ("Could not open output file %s\n", sink_uri);
[88fc249]102        exit (1);
103      }
[5a66677]104    }
[466dff3]105#ifdef HAVE_JACK
106  } else {
107    debug ("Jack init ...\n");
108    jack_setup = new_aubio_jack (hop_size, 1, 1, 0, 1);
109    samplerate = aubio_jack_get_samplerate (jack_setup);
110    source_uri = "jack";
111#endif
[bd2f2ab]112  }
[466dff3]113  ibuf = new_fvec (hop_size);
114  obuf = new_fvec (hop_size);
[bd2f2ab]115
116}
117
[50bc5f2]118void
119examples_common_del (void)
120{
121  del_fvec (ibuf);
[d4c5de7]122  del_fvec (obuf);
[50bc5f2]123  aubio_cleanup ();
[466dff3]124  fflush(stderr);
125  fflush(stdout);
[bd2f2ab]126}
127
[50bc5f2]128void
129examples_common_process (aubio_process_func_t process_func,
130    aubio_print_func_t print)
131{
[5a66677]132
133  uint_t read = 0;
[50bc5f2]134  if (usejack) {
[ebbf5a0]135
[b511fa9]136#if HAVE_JACK
[50bc5f2]137    debug ("Jack activation ...\n");
[466dff3]138    aubio_jack_activate (jack_setup, process_func);
[50bc5f2]139    debug ("Processing (Ctrl+C to quit) ...\n");
140    pause ();
141    aubio_jack_close (jack_setup);
[bd2f2ab]142#else
[50bc5f2]143    usage (stderr, 1);
144    outmsg ("Compiled without jack output, exiting.\n");
[bd2f2ab]145#endif
146
147  } else {
148    /* phasevoc */
[466dff3]149    blocks = 0;
150    uint_t total_read = 0;
[bd2f2ab]151
[5a66677]152    do {
153      aubio_source_do (this_source, ibuf, &read);
[466dff3]154      process_func (ibuf, obuf);
155      // print to console if verbose or no output given
156      if (verbose || sink_uri == NULL) {
157        print();
158      }
[5a66677]159      if (this_sink) {
[466dff3]160        aubio_sink_do (this_sink, obuf, hop_size);
[bd2f2ab]161      }
[466dff3]162      blocks++;
163      total_read += read;
164    } while (read == hop_size);
[bd2f2ab]165
[466dff3]166    verbmsg ("read %d samples (%d blocks of %d) from %s at %dHz\n",
167        total_read, blocks, hop_size, source_uri, samplerate);
[e24378a]168
[5a66677]169    del_aubio_source (this_source);
170    del_aubio_sink   (this_sink);
[bd2f2ab]171
172  }
173}
174
[50bc5f2]175void
176send_noteon (int pitch, int velo)
[bd2f2ab]177{
[50bc5f2]178  smpl_t mpitch = floor (aubio_freqtomidi (pitch) + .5);
[ebbf5a0]179#if HAVE_JACK
180  jack_midi_event_t ev;
181  ev.size = 3;
182  ev.buffer = malloc (3 * sizeof (jack_midi_data_t)); // FIXME
183  ev.time = 0;
[50bc5f2]184  if (usejack) {
[ebbf5a0]185    ev.buffer[2] = velo;
186    ev.buffer[1] = mpitch;
[50bc5f2]187    if (velo == 0) {
[ebbf5a0]188      ev.buffer[0] = 0x80;      /* note off */
[50bc5f2]189    } else {
[ebbf5a0]190      ev.buffer[0] = 0x90;      /* note on */
[50bc5f2]191    }
[ebbf5a0]192    aubio_jack_midi_event_write (jack_setup, (jack_midi_event_t *) & ev);
[50bc5f2]193  } else
[bd2f2ab]194#endif
[466dff3]195  if (velo == 0) {
[3da8187]196    verbmsg ("%f\n", blocks * hop_size / (float) samplerate);
[466dff3]197  } else {
[3da8187]198    verbmsg ("%f\t%f\t", mpitch, blocks * hop_size / (float) samplerate);
[50bc5f2]199  }
[bd2f2ab]200}
201
Note: See TracBrowser for help on using the repository browser.