Something went wrong. Try again.
🎥 Command line media player
Something went wrong. Try again.
C
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172/* * 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 <stdio.h>#include <stdlib.h>#include <string.h>#include <math.h>#include <stdbool.h>#include <sys/stat.h>
#include <libswscale/swscale.h>
#include "misc/bstr.h"#include "osdep/io.h"#include "options/m_config.h"#include "options/path.h"#include "mpv_talloc.h"#include "common/common.h"#include "common/msg.h"#include "video/out/vo.h"#include "video/csputils.h"#include "video/mp_image.h"#include "video/fmt-conversion.h"#include "video/image_writer.h"#include "video/sws_utils.h"#include "sub/osd.h"#include "options/m_option.h"
static const struct m_sub_options image_writer_conf = { .opts = image_writer_opts, .size = sizeof(struct image_writer_opts), .defaults = &image_writer_opts_defaults,};
struct vo_image_opts { struct image_writer_opts *opts; char *outdir;};
#define OPT_BASE_STRUCT struct vo_image_opts
static const struct m_sub_options vo_image_conf = { .opts = (const struct m_option[]) { {"vo-image", OPT_SUBSTRUCT(opts, image_writer_conf)}, {"vo-image-outdir", OPT_STRING(outdir), .flags = M_OPT_FILE}, {0}, }, .size = sizeof(struct vo_image_opts),};
struct priv { struct vo_image_opts *opts;
struct mp_image *current; char *dir; int frame;};
static bool checked_mkdir(struct vo *vo, const char *buf){ struct priv *p = vo->priv; p->dir = mp_get_user_path(vo, vo->global, buf); MP_INFO(vo, "Creating output directory '%s'...\n", p->dir); if (mkdir(p->dir, 0755) < 0) { char *errstr = mp_strerror(errno); if (errno == EEXIST) { struct stat stat_p; if (stat(buf, &stat_p ) == 0 && S_ISDIR(stat_p.st_mode)) return true; } MP_ERR(vo, "Error creating output directory: %s\n", errstr); return false; } return true;}
static int reconfig(struct vo *vo, struct mp_image_params *params){ return 0;}
static bool draw_frame(struct vo *vo, struct vo_frame *frame){ struct priv *p = vo->priv; if (!frame->current) goto done;
p->current = frame->current;
struct mp_osd_res dim = osd_res_from_image_params(vo->params); osd_draw_on_image(vo->osd, dim, frame->current->pts, OSD_DRAW_SUB_ONLY, p->current);
done: return VO_TRUE;}
static void flip_page(struct vo *vo){ struct priv *p = vo->priv; if (!p->current) return;
(p->frame)++;
void *t = talloc_new(NULL); char *filename = talloc_asprintf(t, "%08d.%s", p->frame, image_writer_file_ext(p->opts->opts));
if (p->dir && strlen(p->dir)) filename = mp_path_join(t, p->dir, filename);
MP_INFO(vo, "Saving %s\n", filename); write_image(p->current, p->opts->opts, filename, vo->global, vo->log, true);
talloc_free(t);}
static int query_format(struct vo *vo, int fmt){ if (mp_sws_supported_format(fmt)) return 1; return 0;}
static void uninit(struct vo *vo){}
static int preinit(struct vo *vo){ struct priv *p = vo->priv; p->opts = mp_get_config_group(vo, vo->global, &vo_image_conf); if (p->opts->outdir && !checked_mkdir(vo, p->opts->outdir)) return -1; return 0;}
static int control(struct vo *vo, uint32_t request, void *data){ return VO_NOTIMPL;}
const struct vo_driver video_out_image ={ .description = "Write video frames to image files", .name = "image", .untimed = true, .priv_size = sizeof(struct priv), .preinit = preinit, .query_format = query_format, .reconfig = reconfig, .control = control, .draw_frame = draw_frame, .flip_page = flip_page, .uninit = uninit, .global_opts = &vo_image_conf,};