ref: 2689dfb5e8e79e22fb3723874893c08381e10afb
dir: /tools/input/input.c/
/* * Copyright © 2018, VideoLAN and dav1d authors * Copyright © 2018, Two Orioles, LLC * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * 1. Redistributions of source code must retain the above copyright notice, this * list of conditions and the following disclaimer. * * 2. Redistributions in binary form must reproduce the above copyright notice, * this list of conditions and the following disclaimer in the documentation * and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ #include "config.h" #include <assert.h> #include <errno.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include "input/input.h" #include "input/demuxer.h" struct DemuxerContext { DemuxerPriv *data; const Demuxer *impl; }; #define MAX_NUM_DEMUXERS 1 static const Demuxer *demuxers[MAX_NUM_DEMUXERS]; static int num_demuxers = 0; #define register_demuxer(impl) { \ extern const Demuxer impl; \ assert(num_demuxers < MAX_NUM_DEMUXERS); \ demuxers[num_demuxers++] = &impl; \ } void init_demuxers(void) { register_demuxer(ivf_demuxer); } static const char *find_extension(const char *const f) { const int l = strlen(f); if (l == 0) return NULL; const char *const end = &f[l - 1], *step = end; while ((*step >= 'a' && *step <= 'z') || (*step >= 'A' && *step <= 'Z') || (*step >= '0' && *step <= '9')) { step--; } return (step < end && step > f && *step == '.' && step[-1] != '/') ? &step[1] : NULL; } int input_open(DemuxerContext **const c_out, const char *const name, const char *const filename, unsigned fps[2], unsigned *const num_frames) { const Demuxer *impl; DemuxerContext *c; int res, i; if (name) { for (i = 0; i < num_demuxers; i++) { if (!strcmp(demuxers[i]->name, name)) { impl = demuxers[i]; break; } } if (i == num_demuxers) { fprintf(stderr, "Failed to find demuxer named \"%s\"\n", name); return -ENOPROTOOPT; } } else { const char *const ext = find_extension(filename); if (!ext) { fprintf(stderr, "No extension found for file %s\n", filename); return -1; } for (i = 0; i < num_demuxers; i++) { if (!strcmp(demuxers[i]->extension, ext)) { impl = demuxers[i]; break; } } if (i == num_demuxers) { fprintf(stderr, "Failed to find demuxer for file %s (\"%s\")\n", filename, ext); return -ENOPROTOOPT; } } if (!(c = malloc(sizeof(DemuxerContext) + impl->priv_data_size))) { fprintf(stderr, "Failed to allocate memory\n"); return -ENOMEM; } memset(c, 0, sizeof(DemuxerContext) + impl->priv_data_size); c->impl = impl; c->data = (DemuxerPriv *) &c[1]; if ((res = impl->open(c->data, filename, fps, num_frames)) < 0) { free(c); return res; } *c_out = c; return 0; } int input_read(DemuxerContext *const ctx, Dav1dData *const data) { return ctx->impl->read(ctx->data, data); } void input_close(DemuxerContext *const ctx) { ctx->impl->close(ctx->data); free(ctx); }