ref: dc46037b6fb103f803afb6d9aea12e4a75178428
parent: 3efb63178cb218bc56dfec9bc7e808aed8982f4c
author: Paul Brossier <piem@piem.org>
date: Wed Dec 12 09:51:18 EST 2018
[io] add first sink_vorbis draft
--- /dev/null
+++ b/src/io/sink_vorbis.c
@@ -1,0 +1,278 @@
+/*
+ Copyright (C) 2018 Paul Brossier <piem@aubio.org>
+
+ This file is part of aubio.
+
+ aubio is free software: you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
+
+ aubio is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with aubio. If not, see <http://www.gnu.org/licenses/>.
+
+*/
+
+#include "aubio_priv.h"
+
+#ifdef HAVE_VORBISENC
+
+#include "io/ioutils.h"
+#include "fmat.h"
+
+#include <vorbis/vorbisenc.h>
+#include <string.h> // strerror
+#include <errno.h> // errno
+#include <time.h> // time
+
+#define VORBIS_READSIZE 1024
+
+struct _aubio_sink_vorbis_t {+ FILE *fid; // file id
+ ogg_stream_state os; // stream
+ ogg_page og; // page
+ ogg_packet op; // data packet
+ vorbis_info vi; // vorbis bitstream settings
+ vorbis_comment vc; // user comment
+ vorbis_dsp_state vd; // working state
+ vorbis_block vb; // working space
+
+ uint_t samplerate;
+ uint_t channels;
+ char_t *path;
+};
+
+typedef struct _aubio_sink_vorbis_t aubio_sink_vorbis_t;
+
+uint_t aubio_sink_vorbis_preset_channels(aubio_sink_vorbis_t *s,
+ uint_t channels);
+uint_t aubio_sink_vorbis_preset_samplerate(aubio_sink_vorbis_t *s,
+ uint_t samplerate);
+uint_t aubio_sink_vorbis_open(aubio_sink_vorbis_t *s);
+uint_t aubio_sink_vorbis_close (aubio_sink_vorbis_t *s);
+void del_aubio_sink_vorbis (aubio_sink_vorbis_t *s);
+
+aubio_sink_vorbis_t * new_aubio_sink_vorbis (const char_t *uri,
+ uint_t samplerate)
+{+ aubio_sink_vorbis_t * s = AUBIO_NEW(aubio_sink_vorbis_t);
+
+ s->path = AUBIO_ARRAY(char_t, strnlen(uri, PATH_MAX) + 1);
+ strncpy(s->path, uri, strnlen(uri, PATH_MAX) + 1);
+ s->path[PATH_MAX - 1] = '\0';
+
+ s->channels = 0;
+
+ if ((sint_t)samplerate == 0)
+ return s;
+
+ aubio_sink_vorbis_preset_samplerate(s, samplerate);
+ s->channels = 1;
+
+ if (aubio_sink_vorbis_open(s) != AUBIO_OK)
+ goto failure;
+
+ return s;
+
+failure:
+ del_aubio_sink_vorbis(s);
+ return NULL;
+}
+
+void del_aubio_sink_vorbis (aubio_sink_vorbis_t *s)
+{+ if (s->fid) aubio_sink_vorbis_close(s);
+ // clean up
+ ogg_stream_clear(&s->os);
+ vorbis_block_clear(&s->vb);
+ vorbis_dsp_clear(&s->vd);
+ vorbis_comment_clear(&s->vc);
+ vorbis_info_clear(&s->vi);
+
+ AUBIO_FREE(s);
+}
+
+uint_t aubio_sink_vorbis_open(aubio_sink_vorbis_t *s)
+{+ float quality_mode = .9;
+
+ if (s->samplerate == 0 || s->channels == 0)
+ {+ AUBIO_ERR("sink_vorbis: vorbis_encode_init_vbr failed\n");+ return AUBIO_FAIL;
+ }
+
+ s->fid = fopen((const char *)s->path, "wb");
+ if (!s->fid) return AUBIO_FAIL;
+
+ vorbis_info_init(&s->vi);
+ if (vorbis_encode_init_vbr(&s->vi, s->channels, s->samplerate, quality_mode))
+ {+ AUBIO_ERR("sink_vorbis: vorbis_encode_init_vbr failed\n");+ return AUBIO_FAIL;
+ }
+
+ // add comment
+ vorbis_comment_init(&s->vc);
+ vorbis_comment_add_tag(&s->vc, "ENCODER", "aubio");
+
+ // initalise analysis and block
+ vorbis_analysis_init(&s->vd, &s->vi);
+ vorbis_block_init(&s->vd, &s->vb);
+
+ // pick randome serial number
+ srand(time(NULL));
+ ogg_stream_init(&s->os, rand());
+
+ // write header
+ {+ int ret = 0;
+ ogg_packet header;
+ ogg_packet header_comm;
+ ogg_packet header_code;
+
+ vorbis_analysis_headerout(&s->vd, &s->vc, &header, &header_comm,
+ &header_code);
+
+ ogg_stream_packetin(&s->os, &header);
+ ogg_stream_packetin(&s->os, &header_comm);
+ ogg_stream_packetin(&s->os, &header_code);
+
+ // make sure audio data will start on a new page
+ while (1)
+ {+ ret = ogg_stream_flush(&s->os, &s->og);
+ if (ret==0) break;
+ fwrite(s->og.header, 1, s->og.header_len, s->fid);
+ fwrite(s->og.body, 1, s->og.body_len, s->fid);
+ }
+ }
+
+ return AUBIO_OK;
+}
+
+uint_t aubio_sink_vorbis_preset_samplerate(aubio_sink_vorbis_t *s,
+ uint_t samplerate)
+{+ if (aubio_io_validate_samplerate("sink_vorbis", s->path, samplerate))+ return AUBIO_FAIL;
+ s->samplerate = samplerate;
+ if (s->samplerate != 0 && s->channels != 0)
+ return aubio_sink_vorbis_open(s);
+ return AUBIO_OK;
+}
+
+uint_t aubio_sink_vorbis_preset_channels(aubio_sink_vorbis_t *s,
+ uint_t channels)
+{+ if (aubio_io_validate_channels("sink_vorbis", s->path, channels)) {+ return AUBIO_FAIL;
+ }
+ s->channels = channels;
+ // automatically open when both samplerate and channels have been set
+ if (s->samplerate != 0 && s->channels != 0) {+ return aubio_sink_vorbis_open(s);
+ }
+ return AUBIO_OK;
+}
+
+uint_t aubio_sink_vorbis_get_samplerate(const aubio_sink_vorbis_t *s)
+{+ return s->samplerate;
+}
+
+uint_t aubio_sink_vorbis_get_channels(const aubio_sink_vorbis_t *s)
+{+ return s->channels;
+}
+
+void aubio_sink_vorbis_write(aubio_sink_vorbis_t *s)
+{+ // pre-analysis
+ while (vorbis_analysis_blockout(&s->vd, &s->vb) == 1) {+
+ vorbis_analysis(&s->vb, NULL);
+ vorbis_bitrate_addblock(&s->vb);
+
+ while (vorbis_bitrate_flushpacket(&s->vd, &s->op))
+ {+ ogg_stream_packetin(&s->os, &s->op);
+
+ while (1) {+ int result = ogg_stream_pageout (&s->os, &s->og);
+ if (result == 0) break;
+ fwrite(s->og.header, 1, s->og.header_len, s->fid);
+ fwrite(s->og.body, 1, s->og.body_len, s->fid);
+ if (ogg_page_eos(&s->og)) break;
+ }
+ }
+ }
+}
+
+void aubio_sink_vorbis_do(aubio_sink_vorbis_t *s, fvec_t *write_data,
+ uint_t write)
+{+ uint_t c, v;
+ float **buffer = vorbis_analysis_buffer(&s->vd, (long)write);
+ // fill buffer
+ if (!write) {+ return;
+ } else if (!buffer) {+ AUBIO_WRN("sink_vorbis: failed fetching buffer of size %d\n", write);+ } else {+ for (c = 0; c < s->channels; c++) {+ for (v = 0; v < write; v++) {+ buffer[c][v] = write_data->data[v];
+ }
+ }
+ // tell vorbis how many frames were written
+ vorbis_analysis_wrote(&s->vd, (long)write);
+ }
+ // write to file
+ aubio_sink_vorbis_write(s);
+}
+
+void aubio_sink_vorbis_do_multi(aubio_sink_vorbis_t *s, fmat_t *write_data,
+ uint_t write)
+{+ uint_t c, v;
+ float **buffer = vorbis_analysis_buffer(&s->vd, (long)write);
+ // fill buffer
+ if (!write) {+ return;
+ } else if (!buffer) {+ AUBIO_WRN("sink_vorbis: failed fetching buffer of size %d\n", write);+ } else {+ for (c = 0; c < s->channels; c++) {+ for (v = 0; v < write; v++) {+ buffer[c][v] = write_data->data[c][v];
+ }
+ }
+ // tell vorbis how many frames were written
+ vorbis_analysis_wrote(&s->vd, (long)write);
+ }
+
+ aubio_sink_vorbis_write(s);
+}
+
+uint_t aubio_sink_vorbis_close (aubio_sink_vorbis_t *s)
+{+ //mark the end of stream
+ vorbis_analysis_wrote(&s->vd, 0);
+
+ aubio_sink_vorbis_write(s);
+
+ if (fclose(s->fid)) {+ AUBIO_ERR("sink_vorbis: Error closing file %s (%s)\n",+ s->path, strerror(errno));
+ return AUBIO_FAIL;
+ }
+ return AUBIO_OK;
+}
+
+#endif /* HAVE_VORBISENC */