source: examples/utils.c @ 51b5f9c

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/timestretchfix/ffmpeg5
Last change on this file since 51b5f9c was fef012a, checked in by Paul Brossier <piem@piem.org>, 5 years ago

[examples] only delete sink if needed

  • Property mode set to 100644
File size: 6.0 KB
Line 
1/*
2  Copyright (C) 2003-2013 Paul Brossier <piem@aubio.org>
3
4  This file is part of aubio.
5
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*/
20
21/**
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
29#include "utils.h"
30#ifdef HAVE_JACK
31#include "jackio.h"
32#endif /* HAVE_JACK */
33
34int verbose = 0;
35int usejack = 0;
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;
42uint_t hop_size = 256;
43// onset stuff
44char_t * onset_method = "default";
45smpl_t onset_threshold = 0.0; // will be set if != 0.
46smpl_t onset_minioi = 0.0; // will be set if != 0.
47// pitch stuff
48char_t * pitch_unit = "default";
49char_t * pitch_method = "default";
50smpl_t pitch_tolerance = 0.0; // will be set if != 0.
51// time stuff
52uint_t time_format = 0; // for "seconds", 1 for "ms", 2 for "samples"
53// tempo stuff
54char_t * tempo_method = "default";
55// more general stuff
56smpl_t silence_threshold = -90.;
57smpl_t release_drop = 10.;
58uint_t mix_input = 0;
59
60uint_t force_overwrite = 0;
61
62//
63// internal memory stuff
64aubio_source_t *this_source = NULL;
65aubio_sink_t *this_sink = NULL;
66fvec_t *input_buffer;
67fvec_t *output_buffer;
68
69smpl_t miditap_note = 69.;
70smpl_t miditap_velo = 65.;
71
72/* settings */
73int blocks = 0;
74
75extern void usage (FILE * stream, int exit_code);
76extern int parse_args (int argc, char **argv);
77
78#if HAVE_JACK
79#define MAX_MIDI_EVENTS 128
80#define MAX_MIDI_EVENT_SIZE 3
81aubio_jack_t *jack_setup;
82jack_midi_event_t ev;
83jack_midi_data_t midi_data[MAX_MIDI_EVENTS * MAX_MIDI_EVENT_SIZE];
84size_t midi_event_count = 0;
85#endif /* HAVE_JACK */
86
87void examples_common_init (int argc, char **argv);
88void examples_common_del (void);
89void examples_common_process (aubio_process_func_t process_func,
90    aubio_print_func_t print);
91
92void examples_common_init (int argc, char **argv)
93{
94
95  /* parse command line arguments */
96  parse_args (argc, argv);
97
98  if (!usejack) {
99    debug ("Opening files ...\n");
100    this_source = new_aubio_source ((char_t*)source_uri, samplerate, hop_size);
101    if (this_source == NULL) {
102      errmsg ("Error: could not open input file %s\n", source_uri);
103      exit (1);
104    }
105    if (samplerate == 0) {
106      samplerate = aubio_source_get_samplerate(this_source);
107    }
108    if (sink_uri != NULL) {
109      uint_t sink_exists = (access(sink_uri, F_OK) == 0 );
110      if (!force_overwrite && sink_exists) {
111        errmsg ("Error: output file %s already exists, use -f to overwrite.\n",
112            sink_uri);
113        exit (1);
114      }
115      this_sink = new_aubio_sink ((char_t*)sink_uri, samplerate);
116      if (this_sink == NULL) {
117        errmsg ("Error: could not create output file %s\n", sink_uri);
118        exit (1);
119      }
120    }
121#ifdef HAVE_JACK
122  } else {
123    debug ("Jack init ...\n");
124    jack_setup = new_aubio_jack (hop_size, 1, 1, 0, 1);
125    samplerate = aubio_jack_get_samplerate (jack_setup);
126    source_uri = "jack";
127#endif /* HAVE_JACK */
128  }
129  input_buffer = new_fvec (hop_size);
130  output_buffer = new_fvec (hop_size);
131
132}
133
134void examples_common_del (void)
135{
136  del_fvec (input_buffer);
137  del_fvec (output_buffer);
138  aubio_cleanup ();
139  fflush(stderr);
140  fflush(stdout);
141}
142
143void examples_common_process (aubio_process_func_t process_func,
144    aubio_print_func_t print)
145{
146
147  uint_t read = 0;
148  if (usejack) {
149
150#ifdef HAVE_JACK
151    ev.size = MAX_MIDI_EVENT_SIZE;
152    ev.time = 0; // send it now
153    debug ("Jack activation ...\n");
154    aubio_jack_activate (jack_setup, process_func);
155    debug ("Processing (Ctrl+C to quit) ...\n");
156    pause ();
157    aubio_jack_close (jack_setup);
158#else /* HAVE_JACK */
159    usage (stderr, 1);
160    outmsg ("Compiled without jack output, exiting.\n");
161#endif /* HAVE_JACK */
162
163  } else {
164
165    uint_t total_read = 0;
166    blocks = 0;
167
168    do {
169      aubio_source_do (this_source, input_buffer, &read);
170      process_func (input_buffer, output_buffer);
171      // print to console if verbose or no output given
172      if (verbose || sink_uri == NULL) {
173        print();
174      }
175      if (this_sink) {
176        aubio_sink_do (this_sink, output_buffer, hop_size);
177      }
178      blocks++;
179      total_read += read;
180    } while (read == hop_size);
181
182    verbmsg ("read %.2fs (%d samples in %d blocks of %d) from %s at %dHz\n",
183        total_read * 1. / samplerate,
184        total_read, blocks, hop_size, source_uri, samplerate);
185
186    del_aubio_source (this_source);
187    if (this_sink)
188      del_aubio_sink   (this_sink);
189
190  }
191}
192
193void
194send_noteon (smpl_t pitch, smpl_t velo)
195{
196#ifdef HAVE_JACK
197  if (usejack) {
198    ev.buffer = midi_data + midi_event_count++ * MAX_MIDI_EVENT_SIZE;
199    if (midi_event_count >= MAX_MIDI_EVENTS) {
200      midi_event_count = 0;
201    }
202    ev.buffer[2] = velo;
203    ev.buffer[1] = pitch;
204    if (velo == 0) {
205      ev.buffer[0] = 0x80;      /* note off */
206    } else {
207      ev.buffer[0] = 0x90;      /* note on */
208    }
209    aubio_jack_midi_event_write (jack_setup, (jack_midi_event_t *) & ev);
210  } else
211#endif
212  if (velo == 0) {
213    print_time (blocks * hop_size);
214    outmsg ("\n");
215  } else {
216    outmsg ("%f\t", pitch);
217    print_time (blocks * hop_size);
218    outmsg ("\t");
219  }
220}
221
222void print_time (uint_t time_in_samples) {
223  /* output times in selected format */
224  if (time_format == 2) {
225    outmsg ("%d", time_in_samples);
226  } else if (time_format == 1) {
227    outmsg ("%f", 1000. * time_in_samples / (float) samplerate);
228  } else {
229    outmsg ("%f", time_in_samples / (float) samplerate);
230  }
231}
Note: See TracBrowser for help on using the repository browser.