mpv/test/libmpv_encode.c

/*
 * This file is part of mpv.
 *
 * mpv is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * mpv 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with mpv.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <inttypes.h>
#include <libmpv/client.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#ifdef _WIN32
#include <sys/types.h>
#include <io.h>
#else
#include <unistd.h>
#endif

// Stolen from osdep/compiler.h
#ifdef __GNUC__
#define PRINTF_ATTRIBUTE(a1, a2) __attribute__ ((format(printf, a1, a2)))
#define MP_NORETURN __attribute__((noreturn))
#else
#define PRINTF_ATTRIBUTE(a1, a2)
#define MP_NORETURN
#endif

// Broken crap with __USE_MINGW_ANSI_STDIO
#if defined(__MINGW32__) && defined(__GNUC__) && !defined(__clang__)
#undef PRINTF_ATTRIBUTE
#define PRINTF_ATTRIBUTE(a1, a2) __attribute__ ((format (gnu_printf, a1, a2)))
#endif

// Global handle
static mpv_handle *ctx;
// Temporary output file
static const char *out_path;

static void exit_cleanup(void)
{
    if (ctx)
        mpv_destroy(ctx);
    if (out_path && *out_path)
        unlink(out_path);
}

MP_NORETURN PRINTF_ATTRIBUTE(1, 2)
static void fail(const char *fmt, ...)
{
    if (fmt) {
        va_list va;
        va_start(va, fmt);
        vfprintf(stderr, fmt, va);
        va_end(va);
    }
    exit(1);
}

static void check_api_error(int status)
{
    if (status < 0)
        fail("libmpv error: %s\n", mpv_error_string(status));
}

static void wait_done(void)
{
    while (1) {
        mpv_event *ev = mpv_wait_event(ctx, -1.0);
        if (ev->event_id == MPV_EVENT_NONE)
            continue;
        printf("event: %s\n", mpv_event_name(ev->event_id));
        if (ev->event_id == MPV_EVENT_SHUTDOWN)
            return;
    }
}

static void check_output(FILE *fp)
{
    fseek(fp, 0, SEEK_END);
    long size = ftell(fp);
    if (size < 100)
        fail("did not encode anything\n");

    char magic[4];
    fseek(fp, 0, SEEK_SET);
    size_t ret = fread(magic, sizeof(magic), 1, fp);
    static const char ebml_magic[] = {26, 69, 223, 163};
    if (ret != 1 || memcmp(magic, ebml_magic, sizeof(magic)) != 0)
        fail("output was not Matroska\n");

    puts("output file ok");
}

int main(int argc, char *argv[])
{
    atexit(exit_cleanup);

    ctx = mpv_create();
    if (!ctx)
        return 1;

    static char path[] = "./testout.XXXXXX";

#ifdef _WIN32
    out_path = _mktemp(path);
    if (!out_path || !*out_path)
        fail("tmpfile failed\n");
#else
    int fd = mkstemp(path);
    if (fd == -1)
        fail("tmpfile failed\n");
    out_path = path;
#endif

    check_api_error(mpv_set_option_string(ctx, "o", out_path));
    check_api_error(mpv_set_option_string(ctx, "of", "matroska"));
    check_api_error(mpv_set_option_string(ctx, "end", "1.5"));
    check_api_error(mpv_set_option_string(ctx, "terminal", "yes"));
    check_api_error(mpv_set_option_string(ctx, "msg-level", "all=v"));

    if (mpv_initialize(ctx) != 0)
        return 1;

    check_api_error(mpv_set_option_string(ctx, "idle", "once"));

    const char *cmd[] = {"loadfile", "av://lavfi:testsrc", NULL};
    check_api_error(mpv_command(ctx, cmd));

    wait_done();
    mpv_destroy(ctx);
    ctx = NULL;

    FILE *output = fopen(out_path, "rb");
    if (!output)
        fail("output file doesn't exist\n");
    check_output(output);
    fclose(output);

    return 0;
}