[dc46037b] | 1 | /* |
---|
| 2 | Copyright (C) 2018 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 | |
---|
[7e93013] | 21 | /* |
---|
| 22 | This file is largely inspired by `examples/encoder_example.c` in the |
---|
| 23 | libvorbis source package (versions 1.3.5 and later) available online at |
---|
| 24 | https://xiph.org/vorbis/ |
---|
| 25 | */ |
---|
| 26 | |
---|
[dc46037b] | 27 | #include "aubio_priv.h" |
---|
| 28 | |
---|
| 29 | #ifdef HAVE_VORBISENC |
---|
| 30 | |
---|
| 31 | #include "fmat.h" |
---|
[b8fa393] | 32 | #include "io/ioutils.h" |
---|
[dc46037b] | 33 | |
---|
| 34 | #include <vorbis/vorbisenc.h> |
---|
| 35 | #include <string.h> // strerror |
---|
| 36 | #include <errno.h> // errno |
---|
| 37 | #include <time.h> // time |
---|
| 38 | |
---|
[5e9bdca] | 39 | #define MAX_SIZE 4096 |
---|
[dc72476] | 40 | |
---|
[dc46037b] | 41 | struct _aubio_sink_vorbis_t { |
---|
| 42 | FILE *fid; // file id |
---|
| 43 | ogg_stream_state os; // stream |
---|
| 44 | ogg_page og; // page |
---|
| 45 | ogg_packet op; // data packet |
---|
| 46 | vorbis_info vi; // vorbis bitstream settings |
---|
| 47 | vorbis_comment vc; // user comment |
---|
| 48 | vorbis_dsp_state vd; // working state |
---|
| 49 | vorbis_block vb; // working space |
---|
| 50 | |
---|
| 51 | uint_t samplerate; |
---|
| 52 | uint_t channels; |
---|
| 53 | char_t *path; |
---|
| 54 | }; |
---|
| 55 | |
---|
| 56 | typedef struct _aubio_sink_vorbis_t aubio_sink_vorbis_t; |
---|
| 57 | |
---|
| 58 | uint_t aubio_sink_vorbis_preset_channels(aubio_sink_vorbis_t *s, |
---|
| 59 | uint_t channels); |
---|
| 60 | uint_t aubio_sink_vorbis_preset_samplerate(aubio_sink_vorbis_t *s, |
---|
| 61 | uint_t samplerate); |
---|
| 62 | uint_t aubio_sink_vorbis_open(aubio_sink_vorbis_t *s); |
---|
| 63 | uint_t aubio_sink_vorbis_close (aubio_sink_vorbis_t *s); |
---|
| 64 | void del_aubio_sink_vorbis (aubio_sink_vorbis_t *s); |
---|
| 65 | |
---|
[5e9bdca] | 66 | static uint_t aubio_sink_vorbis_write_page(aubio_sink_vorbis_t *s); |
---|
| 67 | |
---|
[dc46037b] | 68 | aubio_sink_vorbis_t * new_aubio_sink_vorbis (const char_t *uri, |
---|
| 69 | uint_t samplerate) |
---|
| 70 | { |
---|
| 71 | aubio_sink_vorbis_t * s = AUBIO_NEW(aubio_sink_vorbis_t); |
---|
| 72 | |
---|
[4c37e87] | 73 | if (!uri) { |
---|
[0e6ad10] | 74 | AUBIO_ERROR("sink_vorbis: Aborted opening null path\n"); |
---|
[4c37e87] | 75 | goto failure; |
---|
| 76 | } |
---|
| 77 | |
---|
[dc46037b] | 78 | s->path = AUBIO_ARRAY(char_t, strnlen(uri, PATH_MAX) + 1); |
---|
| 79 | strncpy(s->path, uri, strnlen(uri, PATH_MAX) + 1); |
---|
[be63940] | 80 | s->path[strnlen(uri, PATH_MAX)] = '\0'; |
---|
[dc46037b] | 81 | |
---|
| 82 | s->channels = 0; |
---|
| 83 | |
---|
| 84 | if ((sint_t)samplerate == 0) |
---|
| 85 | return s; |
---|
| 86 | |
---|
| 87 | aubio_sink_vorbis_preset_samplerate(s, samplerate); |
---|
| 88 | s->channels = 1; |
---|
| 89 | |
---|
| 90 | if (aubio_sink_vorbis_open(s) != AUBIO_OK) |
---|
| 91 | goto failure; |
---|
| 92 | |
---|
| 93 | return s; |
---|
| 94 | |
---|
| 95 | failure: |
---|
| 96 | del_aubio_sink_vorbis(s); |
---|
| 97 | return NULL; |
---|
| 98 | } |
---|
| 99 | |
---|
| 100 | void del_aubio_sink_vorbis (aubio_sink_vorbis_t *s) |
---|
| 101 | { |
---|
| 102 | if (s->fid) aubio_sink_vorbis_close(s); |
---|
| 103 | // clean up |
---|
| 104 | ogg_stream_clear(&s->os); |
---|
| 105 | vorbis_block_clear(&s->vb); |
---|
| 106 | vorbis_dsp_clear(&s->vd); |
---|
| 107 | vorbis_comment_clear(&s->vc); |
---|
| 108 | vorbis_info_clear(&s->vi); |
---|
| 109 | |
---|
[be63940] | 110 | if (s->path) AUBIO_FREE(s->path); |
---|
[dc46037b] | 111 | AUBIO_FREE(s); |
---|
| 112 | } |
---|
| 113 | |
---|
| 114 | uint_t aubio_sink_vorbis_open(aubio_sink_vorbis_t *s) |
---|
| 115 | { |
---|
| 116 | float quality_mode = .9; |
---|
| 117 | |
---|
[6031419] | 118 | if (s->samplerate == 0 || s->channels == 0) return AUBIO_FAIL; |
---|
[dc46037b] | 119 | |
---|
| 120 | s->fid = fopen((const char *)s->path, "wb"); |
---|
[0850e54] | 121 | if (!s->fid) { |
---|
[59109e9] | 122 | AUBIO_STRERR("sink_vorbis: Error opening file \'%s\' (%s)\n", |
---|
[5e9bdca] | 123 | s->path, errorstr); |
---|
[0850e54] | 124 | return AUBIO_FAIL; |
---|
| 125 | } |
---|
[dc46037b] | 126 | |
---|
| 127 | vorbis_info_init(&s->vi); |
---|
| 128 | if (vorbis_encode_init_vbr(&s->vi, s->channels, s->samplerate, quality_mode)) |
---|
| 129 | { |
---|
| 130 | AUBIO_ERR("sink_vorbis: vorbis_encode_init_vbr failed\n"); |
---|
| 131 | return AUBIO_FAIL; |
---|
| 132 | } |
---|
| 133 | |
---|
| 134 | // add comment |
---|
| 135 | vorbis_comment_init(&s->vc); |
---|
| 136 | vorbis_comment_add_tag(&s->vc, "ENCODER", "aubio"); |
---|
| 137 | |
---|
| 138 | // initalise analysis and block |
---|
| 139 | vorbis_analysis_init(&s->vd, &s->vi); |
---|
| 140 | vorbis_block_init(&s->vd, &s->vb); |
---|
| 141 | |
---|
| 142 | // pick randome serial number |
---|
| 143 | srand(time(NULL)); |
---|
| 144 | ogg_stream_init(&s->os, rand()); |
---|
| 145 | |
---|
| 146 | // write header |
---|
| 147 | { |
---|
| 148 | ogg_packet header; |
---|
| 149 | ogg_packet header_comm; |
---|
| 150 | ogg_packet header_code; |
---|
| 151 | |
---|
| 152 | vorbis_analysis_headerout(&s->vd, &s->vc, &header, &header_comm, |
---|
| 153 | &header_code); |
---|
| 154 | |
---|
| 155 | ogg_stream_packetin(&s->os, &header); |
---|
| 156 | ogg_stream_packetin(&s->os, &header_comm); |
---|
| 157 | ogg_stream_packetin(&s->os, &header_code); |
---|
| 158 | |
---|
| 159 | // make sure audio data will start on a new page |
---|
| 160 | while (1) |
---|
| 161 | { |
---|
[5e9bdca] | 162 | if (!ogg_stream_flush(&s->os, &s->og)) break; |
---|
| 163 | if (aubio_sink_vorbis_write_page(s)) return AUBIO_FAIL; |
---|
[dc46037b] | 164 | } |
---|
| 165 | } |
---|
| 166 | |
---|
| 167 | return AUBIO_OK; |
---|
| 168 | } |
---|
| 169 | |
---|
| 170 | uint_t aubio_sink_vorbis_preset_samplerate(aubio_sink_vorbis_t *s, |
---|
| 171 | uint_t samplerate) |
---|
| 172 | { |
---|
| 173 | if (aubio_io_validate_samplerate("sink_vorbis", s->path, samplerate)) |
---|
| 174 | return AUBIO_FAIL; |
---|
| 175 | s->samplerate = samplerate; |
---|
[252f585] | 176 | if (/* s->samplerate != 0 && */ s->channels != 0) |
---|
[dc46037b] | 177 | return aubio_sink_vorbis_open(s); |
---|
| 178 | return AUBIO_OK; |
---|
| 179 | } |
---|
| 180 | |
---|
| 181 | uint_t aubio_sink_vorbis_preset_channels(aubio_sink_vorbis_t *s, |
---|
| 182 | uint_t channels) |
---|
| 183 | { |
---|
| 184 | if (aubio_io_validate_channels("sink_vorbis", s->path, channels)) { |
---|
| 185 | return AUBIO_FAIL; |
---|
| 186 | } |
---|
| 187 | s->channels = channels; |
---|
| 188 | // automatically open when both samplerate and channels have been set |
---|
[252f585] | 189 | if (s->samplerate != 0 /* && s->channels != 0 */) { |
---|
[dc46037b] | 190 | return aubio_sink_vorbis_open(s); |
---|
| 191 | } |
---|
| 192 | return AUBIO_OK; |
---|
| 193 | } |
---|
| 194 | |
---|
| 195 | uint_t aubio_sink_vorbis_get_samplerate(const aubio_sink_vorbis_t *s) |
---|
| 196 | { |
---|
| 197 | return s->samplerate; |
---|
| 198 | } |
---|
| 199 | |
---|
| 200 | uint_t aubio_sink_vorbis_get_channels(const aubio_sink_vorbis_t *s) |
---|
| 201 | { |
---|
| 202 | return s->channels; |
---|
| 203 | } |
---|
| 204 | |
---|
[5e9bdca] | 205 | static |
---|
| 206 | uint_t aubio_sink_vorbis_write_page(aubio_sink_vorbis_t *s) { |
---|
[0850e54] | 207 | int result; |
---|
| 208 | size_t wrote; |
---|
[5e9bdca] | 209 | wrote = fwrite(s->og.header, 1, s->og.header_len, s->fid); |
---|
| 210 | result = (wrote == (unsigned)s->og.header_len); |
---|
| 211 | wrote = fwrite(s->og.body, 1, s->og.body_len, s->fid); |
---|
| 212 | result &= (wrote == (unsigned)s->og.body_len); |
---|
| 213 | if (result == 0) { |
---|
[59109e9] | 214 | AUBIO_STRERR("sink_vorbis: failed writing \'%s\' to disk (%s)\n", |
---|
[5e9bdca] | 215 | s->path, errorstr); |
---|
| 216 | return AUBIO_FAIL; |
---|
| 217 | } |
---|
| 218 | return AUBIO_OK; |
---|
| 219 | } |
---|
| 220 | |
---|
| 221 | static |
---|
| 222 | void aubio_sink_vorbis_write(aubio_sink_vorbis_t *s) |
---|
| 223 | { |
---|
[dc46037b] | 224 | // pre-analysis |
---|
| 225 | while (vorbis_analysis_blockout(&s->vd, &s->vb) == 1) { |
---|
| 226 | |
---|
| 227 | vorbis_analysis(&s->vb, NULL); |
---|
| 228 | vorbis_bitrate_addblock(&s->vb); |
---|
| 229 | |
---|
| 230 | while (vorbis_bitrate_flushpacket(&s->vd, &s->op)) |
---|
| 231 | { |
---|
| 232 | ogg_stream_packetin(&s->os, &s->op); |
---|
| 233 | |
---|
| 234 | while (1) { |
---|
[5e9bdca] | 235 | if (!ogg_stream_pageout (&s->os, &s->og)) break; |
---|
| 236 | aubio_sink_vorbis_write_page(s); |
---|
[dc46037b] | 237 | if (ogg_page_eos(&s->og)) break; |
---|
| 238 | } |
---|
| 239 | } |
---|
| 240 | } |
---|
| 241 | } |
---|
| 242 | |
---|
| 243 | void aubio_sink_vorbis_do(aubio_sink_vorbis_t *s, fvec_t *write_data, |
---|
| 244 | uint_t write) |
---|
| 245 | { |
---|
| 246 | uint_t c, v; |
---|
[dc72476] | 247 | uint_t length = aubio_sink_validate_input_length("sink_vorbis", s->path, |
---|
| 248 | MAX_SIZE, write_data->length, write); |
---|
| 249 | float **buffer = vorbis_analysis_buffer(&s->vd, (long)length); |
---|
[dc46037b] | 250 | // fill buffer |
---|
| 251 | if (!write) { |
---|
| 252 | return; |
---|
| 253 | } else if (!buffer) { |
---|
| 254 | AUBIO_WRN("sink_vorbis: failed fetching buffer of size %d\n", write); |
---|
[4b9481f9] | 255 | return; |
---|
[dc46037b] | 256 | } else { |
---|
| 257 | for (c = 0; c < s->channels; c++) { |
---|
[dc72476] | 258 | for (v = 0; v < length; v++) { |
---|
[dc46037b] | 259 | buffer[c][v] = write_data->data[v]; |
---|
| 260 | } |
---|
| 261 | } |
---|
| 262 | // tell vorbis how many frames were written |
---|
[dc72476] | 263 | vorbis_analysis_wrote(&s->vd, (long)length); |
---|
[dc46037b] | 264 | } |
---|
| 265 | // write to file |
---|
| 266 | aubio_sink_vorbis_write(s); |
---|
| 267 | } |
---|
| 268 | |
---|
| 269 | void aubio_sink_vorbis_do_multi(aubio_sink_vorbis_t *s, fmat_t *write_data, |
---|
| 270 | uint_t write) |
---|
| 271 | { |
---|
| 272 | uint_t c, v; |
---|
[dc72476] | 273 | uint_t channels = aubio_sink_validate_input_channels("sink_vorbis", s->path, |
---|
| 274 | s->channels, write_data->height); |
---|
| 275 | uint_t length = aubio_sink_validate_input_length("sink_vorbis", s->path, |
---|
| 276 | MAX_SIZE, write_data->length, write); |
---|
| 277 | float **buffer = vorbis_analysis_buffer(&s->vd, (long)length); |
---|
[dc46037b] | 278 | // fill buffer |
---|
| 279 | if (!write) { |
---|
| 280 | return; |
---|
| 281 | } else if (!buffer) { |
---|
| 282 | AUBIO_WRN("sink_vorbis: failed fetching buffer of size %d\n", write); |
---|
[4b9481f9] | 283 | return; |
---|
[dc46037b] | 284 | } else { |
---|
[6031419] | 285 | for (c = 0; c < channels; c++) { |
---|
[dc72476] | 286 | for (v = 0; v < length; v++) { |
---|
[dc46037b] | 287 | buffer[c][v] = write_data->data[c][v]; |
---|
| 288 | } |
---|
| 289 | } |
---|
| 290 | // tell vorbis how many frames were written |
---|
[5e9bdca] | 291 | vorbis_analysis_wrote(&s->vd, (long)length); |
---|
[dc46037b] | 292 | } |
---|
| 293 | |
---|
| 294 | aubio_sink_vorbis_write(s); |
---|
| 295 | } |
---|
| 296 | |
---|
| 297 | uint_t aubio_sink_vorbis_close (aubio_sink_vorbis_t *s) |
---|
| 298 | { |
---|
[56fa1e0] | 299 | if (!s->fid) return AUBIO_FAIL; |
---|
[dc46037b] | 300 | //mark the end of stream |
---|
| 301 | vorbis_analysis_wrote(&s->vd, 0); |
---|
| 302 | |
---|
| 303 | aubio_sink_vorbis_write(s); |
---|
| 304 | |
---|
[7107ed9] | 305 | if (s->fid && fclose(s->fid)) { |
---|
[59109e9] | 306 | AUBIO_STRERR("sink_vorbis: Error closing file \'%s\' (%s)\n", |
---|
[5e9bdca] | 307 | s->path, errorstr); |
---|
[dc46037b] | 308 | return AUBIO_FAIL; |
---|
| 309 | } |
---|
[7107ed9] | 310 | s->fid = NULL; |
---|
[dc46037b] | 311 | return AUBIO_OK; |
---|
| 312 | } |
---|
| 313 | |
---|
| 314 | #endif /* HAVE_VORBISENC */ |
---|