1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154
|
/*
* 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 <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
#include <mpv/client.h>
// 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[])
{
ctx = mpv_create();
if (!ctx)
return 1;
atexit(exit_cleanup);
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;
}
|