X-Git-Url: http://git.tuebingen.mpg.de/?p=paraslash.git;a=blobdiff_plain;f=oggdec_filter.c;h=956af85223a1e101fd9b290d84487fc74acae46a;hp=c158df7e4df7777f1f92468c3ef0f3a333756237;hb=aa234b7afe223879a7bd7274ce05a3a315a2ec49;hpb=63c513602a6b6a97cdced053beea0b2d07a9f984 diff --git a/oggdec_filter.c b/oggdec_filter.c index c158df7e..956af852 100644 --- a/oggdec_filter.c +++ b/oggdec_filter.c @@ -8,12 +8,14 @@ #include #include +#include #include "para.h" #include "oggdec_filter.cmdline.h" #include "list.h" #include "sched.h" #include "ggo.h" +#include "buffer_tree.h" #include "filter.h" #include "error.h" #include "string.h" @@ -37,9 +39,13 @@ struct private_oggdec_data { size_t converted; /** When to start producing output. */ struct timeval stream_start; + /** The number of channels of the current stream. */ + unsigned int channels; + /** Current sample rate in Hz. */ + unsigned int samplerate; }; -static size_t cb_read(void *buf, size_t size, size_t nmemb, void *datasource) +static size_t cb_read_nobtr(void *buf, size_t size, size_t nmemb, void *datasource) { struct filter_node *fn = datasource; struct private_oggdec_data *pod = fn->private_data; @@ -60,6 +66,44 @@ static size_t cb_read(void *buf, size_t size, size_t nmemb, void *datasource) return ret; } +static size_t cb_read_btr(void *buf, size_t size, size_t nmemb, void *datasource) +{ + struct filter_node *fn = datasource; + struct btr_node *btrn = fn->btrn; + size_t copied; + + /** + * oggvorbis always uses size == 1. Other sizes would complicate the code + * for no real gain. So we simply don't support size != 1. + */ + assert(size == 1); + //PARA_DEBUG_LOG("vorbis requests %zu x %zu = %zu bytes\n", size, nmemb, size * nmemb); + copied = 0; + for (;;) { + char *btr_buf; + size_t nbytes = btr_next_buffer(btrn, &btr_buf); + if (nbytes == 0) + break; + nbytes = PARA_MIN(nmemb - copied, nbytes); + memcpy(buf + copied, btr_buf, nbytes); + copied += nbytes; + btr_consume(btrn, nbytes); + if (copied == nmemb) + break; + } + return copied; +} + +static size_t cb_read(void *buf, size_t size, size_t nmemb, void *datasource) +{ + struct filter_node *fn = datasource; + + if (fn->btrn) + return cb_read_btr(buf, size, nmemb, datasource); + else + return cb_read_nobtr(buf, size, nmemb, datasource); +} + /* * Custom data seeking function. * @@ -115,6 +159,111 @@ static void ogg_close(struct filter_node *fn) fn->private_data = NULL; } +#define OGGDEC_OUTPUT_CHUNK_SIZE (64 * 1024) + +static int oggdec_execute(struct btr_node *btrn, const char *cmd, char **result) +{ + struct filter_node *fn = btr_context(btrn); + struct private_oggdec_data *pod = fn->private_data; + + if (!strcmp(cmd, "samplerate")) { + if (pod->samplerate == 0) + return -ERRNO_TO_PARA_ERROR(ENAVAIL); + *result = make_message("%u", pod->samplerate); + return 1; + } + if (!strcmp(cmd, "channels")) { + if (pod->channels == 0) + return -ERRNO_TO_PARA_ERROR(ENAVAIL); + *result = make_message("%u", pod->channels); + return 1; + } + return -ERRNO_TO_PARA_ERROR(ENOTSUP); +} + +static void ogg_post_select(__a_unused struct sched *s, struct task *t) +{ + struct filter_node *fn = container_of(t, struct filter_node, task); + struct private_oggdec_data *pod = fn->private_data; + struct btr_node *btrn = fn->btrn; + size_t iqs, len; + int ret; + char *in; + + t->error = 0; + ret = btr_node_status(btrn, fn->min_iqs); + if (ret < 0) + goto err; + if (ret == 0) + return; + btr_merge(btrn, fn->min_iqs); + len = btr_next_buffer(btrn, &in); + iqs = btr_get_input_queue_size(btrn); + if (!pod->vf) { + int oret; + + pod->vf = para_malloc(sizeof(struct OggVorbis_File)); + PARA_NOTICE_LOG("input queue: %zu, opening ov callbacks\n", iqs); + oret = ov_open_callbacks(fn, pod->vf, + NULL, /* no initial buffer */ + 0, /* no initial bytes */ + ovc); /* the ov_open_callbacks */ + if (oret == OV_ENOTVORBIS || oret == OV_EBADHEADER) { + /* this might be due to the input buffer being too small */ + if (!btr_no_parent(btrn)) { + free(pod->vf); + pod->vf = NULL; + fn->min_iqs = iqs + 1; + return; + } + ret = (oret == OV_ENOTVORBIS)? + -E_OGGDEC_NOTVORBIS : -E_OGGDEC_BADHEADER; + goto err; + } + ret = -E_OGGDEC_READ; + if (oret == OV_EREAD) + goto err; + ret = -E_OGGDEC_VERSION; + if (oret == OV_EVERSION) + goto err; + ret = -E_OGGDEC_FAULT; + if (oret < 0) + goto err; + pod->channels = ov_info(pod->vf, 0)->channels; + pod->samplerate = ov_info(pod->vf, 0)->rate; + PARA_NOTICE_LOG("%d channels, %d Hz\n", pod->channels, + pod->samplerate); + ///* wait a bit to avoid buffer underruns */ + //tv_add(now, &(struct timeval){0, 500 * 1000}, &pod->stream_start); + return; + } + for (;;) { + char *out = para_malloc(OGGDEC_OUTPUT_CHUNK_SIZE); + ssize_t read_ret = ov_read(pod->vf, out, OGGDEC_OUTPUT_CHUNK_SIZE, + ENDIAN, 2 /* 16 bit */, 1 /* signed */, NULL); + if (read_ret <= 0) + free(out); + if (read_ret == 0) { + ret = -E_OGGDEC_EOF; + if (btr_no_parent(btrn)) + goto err; + return; + } + if (read_ret == OV_HOLE) + return; + if (read_ret < 0) { + ret = -E_OGGDEC_BADLINK; + goto err; + } + btr_add_output(out, read_ret, btrn); + } + +err: + assert(ret < 0); + t->error = ret; + btr_remove_node(btrn); +} + static ssize_t ogg_convert(char *inbuffer, size_t len, struct filter_node *fn) { ssize_t ret; @@ -225,7 +374,10 @@ void oggdec_filter_init(struct filter *f) f->open = ogg_open; f->close = ogg_close; f->convert = ogg_convert; + f->pre_select = generic_filter_pre_select; + f->post_select = ogg_post_select; f->parse_config = oggdec_parse_config; + f->execute = oggdec_execute; f->help = (struct ggo_help) { .short_help = oggdec_filter_args_info_help, .detailed_help = oggdec_filter_args_info_detailed_help