examples/dmabuf-capture: fix indentation
This commit is contained in:
parent
2681352e04
commit
2198fd5eed
|
@ -48,13 +48,13 @@ struct capture_context {
|
||||||
AVBufferRef *drm_device_ref;
|
AVBufferRef *drm_device_ref;
|
||||||
AVBufferRef *drm_frames_ref;
|
AVBufferRef *drm_frames_ref;
|
||||||
|
|
||||||
AVBufferRef *mapped_device_ref;
|
AVBufferRef *mapped_device_ref;
|
||||||
AVBufferRef *mapped_frames_ref;
|
AVBufferRef *mapped_frames_ref;
|
||||||
|
|
||||||
AVFormatContext *avf;
|
AVFormatContext *avf;
|
||||||
AVCodecContext *avctx;
|
AVCodecContext *avctx;
|
||||||
|
|
||||||
int64_t start_pts;
|
int64_t start_pts;
|
||||||
|
|
||||||
/* Config */
|
/* Config */
|
||||||
enum AVPixelFormat software_format;
|
enum AVPixelFormat software_format;
|
||||||
|
@ -136,7 +136,7 @@ static struct wayland_output *find_output(struct capture_context *ctx,
|
||||||
wl_list_for_each_safe(output, tmp, &ctx->output_list, link)
|
wl_list_for_each_safe(output, tmp, &ctx->output_list, link)
|
||||||
if ((output->output == out) || (output->id == id))
|
if ((output->output == out) || (output->id == id))
|
||||||
return output;
|
return output;
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void registry_handle_remove(void *data, struct wl_registry *reg,
|
static void registry_handle_remove(void *data, struct wl_registry *reg,
|
||||||
|
@ -328,7 +328,7 @@ static void frame_ready(void *data, struct zwlr_export_dmabuf_frame_v1 *frame,
|
||||||
|
|
||||||
AVHWFramesContext *mapped_hwfc;
|
AVHWFramesContext *mapped_hwfc;
|
||||||
mapped_hwfc = (AVHWFramesContext *)ctx->mapped_frames_ref->data;
|
mapped_hwfc = (AVHWFramesContext *)ctx->mapped_frames_ref->data;
|
||||||
mapped_frame->format = mapped_hwfc->format;
|
mapped_frame->format = mapped_hwfc->format;
|
||||||
|
|
||||||
/* Set frame hardware context referencce */
|
/* Set frame hardware context referencce */
|
||||||
mapped_frame->hw_frames_ctx = av_buffer_ref(ctx->mapped_frames_ref);
|
mapped_frame->hw_frames_ctx = av_buffer_ref(ctx->mapped_frames_ref);
|
||||||
|
@ -339,8 +339,8 @@ static void frame_ready(void *data, struct zwlr_export_dmabuf_frame_v1 *frame,
|
||||||
|
|
||||||
err = av_hwframe_map(mapped_frame, f, 0);
|
err = av_hwframe_map(mapped_frame, f, 0);
|
||||||
if (err) {
|
if (err) {
|
||||||
av_log(ctx, AV_LOG_ERROR, "Error mapping: %s!\n", av_err2str(err));
|
av_log(ctx, AV_LOG_ERROR, "Error mapping: %s!\n", av_err2str(err));
|
||||||
goto end;
|
goto end;
|
||||||
}
|
}
|
||||||
|
|
||||||
AVFrame *enc_input = mapped_frame;
|
AVFrame *enc_input = mapped_frame;
|
||||||
|
@ -369,7 +369,7 @@ static void frame_ready(void *data, struct zwlr_export_dmabuf_frame_v1 *frame,
|
||||||
do {
|
do {
|
||||||
err = avcodec_send_frame(ctx->avctx, enc_input);
|
err = avcodec_send_frame(ctx->avctx, enc_input);
|
||||||
|
|
||||||
av_frame_free(&enc_input);
|
av_frame_free(&enc_input);
|
||||||
|
|
||||||
if (err) {
|
if (err) {
|
||||||
av_log(ctx, AV_LOG_ERROR, "Error encoding: %s!\n", av_err2str(err));
|
av_log(ctx, AV_LOG_ERROR, "Error encoding: %s!\n", av_err2str(err));
|
||||||
|
@ -473,12 +473,12 @@ static int init_lavu_hwcontext(struct capture_context *ctx) {
|
||||||
static int set_hwframe_ctx(struct capture_context *ctx,
|
static int set_hwframe_ctx(struct capture_context *ctx,
|
||||||
AVBufferRef *hw_device_ctx)
|
AVBufferRef *hw_device_ctx)
|
||||||
{
|
{
|
||||||
AVHWFramesContext *frames_ctx = NULL;
|
AVHWFramesContext *frames_ctx = NULL;
|
||||||
int err = 0;
|
int err = 0;
|
||||||
|
|
||||||
if (!(ctx->mapped_frames_ref = av_hwframe_ctx_alloc(hw_device_ctx))) {
|
if (!(ctx->mapped_frames_ref = av_hwframe_ctx_alloc(hw_device_ctx))) {
|
||||||
return AVERROR(ENOMEM);
|
return AVERROR(ENOMEM);
|
||||||
}
|
}
|
||||||
|
|
||||||
AVHWFramesConstraints *cst =
|
AVHWFramesConstraints *cst =
|
||||||
av_hwdevice_get_hwframe_constraints(ctx->mapped_device_ref, NULL);
|
av_hwdevice_get_hwframe_constraints(ctx->mapped_device_ref, NULL);
|
||||||
|
@ -512,7 +512,7 @@ static int set_hwframe_ctx(struct capture_context *ctx,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return err;
|
return err;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int init_encoding(struct capture_context *ctx) {
|
static int init_encoding(struct capture_context *ctx) {
|
||||||
|
@ -521,79 +521,79 @@ static int init_encoding(struct capture_context *ctx) {
|
||||||
/* lavf init */
|
/* lavf init */
|
||||||
err = avformat_alloc_output_context2(&ctx->avf, NULL,
|
err = avformat_alloc_output_context2(&ctx->avf, NULL,
|
||||||
NULL, ctx->out_filename);
|
NULL, ctx->out_filename);
|
||||||
if (err) {
|
if (err) {
|
||||||
av_log(ctx, AV_LOG_ERROR, "Unable to init lavf context!\n");
|
av_log(ctx, AV_LOG_ERROR, "Unable to init lavf context!\n");
|
||||||
return err;
|
return err;
|
||||||
}
|
}
|
||||||
|
|
||||||
AVStream *st = avformat_new_stream(ctx->avf, NULL);
|
AVStream *st = avformat_new_stream(ctx->avf, NULL);
|
||||||
if (!st) {
|
if (!st) {
|
||||||
av_log(ctx, AV_LOG_ERROR, "Unable to alloc stream!\n");
|
av_log(ctx, AV_LOG_ERROR, "Unable to alloc stream!\n");
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Find encoder */
|
/* Find encoder */
|
||||||
AVCodec *out_codec = avcodec_find_encoder_by_name(ctx->encoder_name);
|
AVCodec *out_codec = avcodec_find_encoder_by_name(ctx->encoder_name);
|
||||||
if (!out_codec) {
|
if (!out_codec) {
|
||||||
av_log(ctx, AV_LOG_ERROR, "Codec not found (not compiled in lavc?)!\n");
|
av_log(ctx, AV_LOG_ERROR, "Codec not found (not compiled in lavc?)!\n");
|
||||||
return AVERROR(EINVAL);
|
return AVERROR(EINVAL);
|
||||||
}
|
}
|
||||||
ctx->avf->oformat->video_codec = out_codec->id;
|
ctx->avf->oformat->video_codec = out_codec->id;
|
||||||
ctx->is_software_encoder = !(out_codec->capabilities & AV_CODEC_CAP_HARDWARE);
|
ctx->is_software_encoder = !(out_codec->capabilities & AV_CODEC_CAP_HARDWARE);
|
||||||
|
|
||||||
ctx->avctx = avcodec_alloc_context3(out_codec);
|
ctx->avctx = avcodec_alloc_context3(out_codec);
|
||||||
if (!ctx->avctx)
|
if (!ctx->avctx)
|
||||||
return 1;
|
return 1;
|
||||||
|
|
||||||
ctx->avctx->opaque = ctx;
|
ctx->avctx->opaque = ctx;
|
||||||
ctx->avctx->bit_rate = (int)ctx->out_bitrate*1000000.0f;
|
ctx->avctx->bit_rate = (int)ctx->out_bitrate*1000000.0f;
|
||||||
ctx->avctx->pix_fmt = ctx->software_format;
|
ctx->avctx->pix_fmt = ctx->software_format;
|
||||||
ctx->avctx->time_base = (AVRational){ 1, 1000 };
|
ctx->avctx->time_base = (AVRational){ 1, 1000 };
|
||||||
ctx->avctx->compression_level = 7;
|
ctx->avctx->compression_level = 7;
|
||||||
ctx->avctx->width = find_output(ctx, ctx->target_output, 0)->width;
|
ctx->avctx->width = find_output(ctx, ctx->target_output, 0)->width;
|
||||||
ctx->avctx->height = find_output(ctx, ctx->target_output, 0)->height;
|
ctx->avctx->height = find_output(ctx, ctx->target_output, 0)->height;
|
||||||
|
|
||||||
if (ctx->avf->oformat->flags & AVFMT_GLOBALHEADER)
|
if (ctx->avf->oformat->flags & AVFMT_GLOBALHEADER)
|
||||||
ctx->avctx->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;
|
ctx->avctx->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;
|
||||||
|
|
||||||
st->id = 0;
|
st->id = 0;
|
||||||
st->time_base = ctx->avctx->time_base;
|
st->time_base = ctx->avctx->time_base;
|
||||||
st->avg_frame_rate = find_output(ctx, ctx->target_output, 0)->framerate;
|
st->avg_frame_rate = find_output(ctx, ctx->target_output, 0)->framerate;
|
||||||
|
|
||||||
/* Init hw frames context */
|
/* Init hw frames context */
|
||||||
err = set_hwframe_ctx(ctx, ctx->mapped_device_ref);
|
err = set_hwframe_ctx(ctx, ctx->mapped_device_ref);
|
||||||
if (err)
|
if (err)
|
||||||
return err;
|
return err;
|
||||||
|
|
||||||
err = avcodec_open2(ctx->avctx, out_codec, &ctx->encoder_opts);
|
err = avcodec_open2(ctx->avctx, out_codec, &ctx->encoder_opts);
|
||||||
if (err) {
|
if (err) {
|
||||||
av_log(ctx, AV_LOG_ERROR, "Cannot open encoder: %s!\n",
|
av_log(ctx, AV_LOG_ERROR, "Cannot open encoder: %s!\n",
|
||||||
av_err2str(err));
|
av_err2str(err));
|
||||||
return err;
|
return err;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (avcodec_parameters_from_context(st->codecpar, ctx->avctx) < 0) {
|
if (avcodec_parameters_from_context(st->codecpar, ctx->avctx) < 0) {
|
||||||
av_log(ctx, AV_LOG_ERROR, "Couldn't copy codec params: %s!\n",
|
av_log(ctx, AV_LOG_ERROR, "Couldn't copy codec params: %s!\n",
|
||||||
av_err2str(err));
|
av_err2str(err));
|
||||||
return err;
|
return err;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Debug print */
|
/* Debug print */
|
||||||
av_dump_format(ctx->avf, 0, ctx->out_filename, 1);
|
av_dump_format(ctx->avf, 0, ctx->out_filename, 1);
|
||||||
|
|
||||||
/* Open for writing */
|
/* Open for writing */
|
||||||
err = avio_open(&ctx->avf->pb, ctx->out_filename, AVIO_FLAG_WRITE);
|
err = avio_open(&ctx->avf->pb, ctx->out_filename, AVIO_FLAG_WRITE);
|
||||||
if (err) {
|
if (err) {
|
||||||
av_log(ctx, AV_LOG_ERROR, "Couldn't open %s: %s!\n", ctx->out_filename,
|
av_log(ctx, AV_LOG_ERROR, "Couldn't open %s: %s!\n", ctx->out_filename,
|
||||||
av_err2str(err));
|
av_err2str(err));
|
||||||
return err;
|
return err;
|
||||||
}
|
}
|
||||||
|
|
||||||
err = avformat_write_header(ctx->avf, NULL);
|
err = avformat_write_header(ctx->avf, NULL);
|
||||||
if (err) {
|
if (err) {
|
||||||
av_log(ctx, AV_LOG_ERROR, "Couldn't write header: %s!\n", av_err2str(err));
|
av_log(ctx, AV_LOG_ERROR, "Couldn't write header: %s!\n", av_err2str(err));
|
||||||
return err;
|
return err;
|
||||||
}
|
}
|
||||||
|
|
||||||
return err;
|
return err;
|
||||||
}
|
}
|
||||||
|
@ -608,12 +608,12 @@ void on_quit_signal(int signo) {
|
||||||
static int main_loop(struct capture_context *ctx) {
|
static int main_loop(struct capture_context *ctx) {
|
||||||
int err;
|
int err;
|
||||||
|
|
||||||
q_ctx = ctx;
|
q_ctx = ctx;
|
||||||
|
|
||||||
if (signal(SIGINT, on_quit_signal) == SIG_ERR) {
|
if (signal(SIGINT, on_quit_signal) == SIG_ERR) {
|
||||||
av_log(ctx, AV_LOG_ERROR, "Unable to install signal handler!\n");
|
av_log(ctx, AV_LOG_ERROR, "Unable to install signal handler!\n");
|
||||||
return AVERROR(EINVAL);
|
return AVERROR(EINVAL);
|
||||||
}
|
}
|
||||||
|
|
||||||
err = init_lavu_hwcontext(ctx);
|
err = init_lavu_hwcontext(ctx);
|
||||||
if (err)
|
if (err)
|
||||||
|
@ -627,11 +627,11 @@ static int main_loop(struct capture_context *ctx) {
|
||||||
register_cb(ctx);
|
register_cb(ctx);
|
||||||
|
|
||||||
while (!ctx->err && ctx->quit < 2) {
|
while (!ctx->err && ctx->quit < 2) {
|
||||||
while (wl_display_prepare_read(ctx->display) != 0) {
|
while (wl_display_prepare_read(ctx->display) != 0) {
|
||||||
wl_display_dispatch_pending(ctx->display);
|
wl_display_dispatch_pending(ctx->display);
|
||||||
}
|
}
|
||||||
|
|
||||||
wl_display_flush(ctx->display);
|
wl_display_flush(ctx->display);
|
||||||
|
|
||||||
struct pollfd fds[1] = {
|
struct pollfd fds[1] = {
|
||||||
{ .fd = wl_display_get_fd(ctx->display), .events = POLLIN },
|
{ .fd = wl_display_get_fd(ctx->display), .events = POLLIN },
|
||||||
|
@ -649,7 +649,7 @@ static int main_loop(struct capture_context *ctx) {
|
||||||
}
|
}
|
||||||
|
|
||||||
if (fds[0].revents & POLLIN) {
|
if (fds[0].revents & POLLIN) {
|
||||||
if (wl_display_read_events(ctx->display) < 0) {
|
if (wl_display_read_events(ctx->display) < 0) {
|
||||||
av_log(ctx, AV_LOG_ERROR, "Failed to read Wayland events!\n");
|
av_log(ctx, AV_LOG_ERROR, "Failed to read Wayland events!\n");
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
@ -658,13 +658,13 @@ static int main_loop(struct capture_context *ctx) {
|
||||||
}
|
}
|
||||||
|
|
||||||
err = av_write_trailer(ctx->avf);
|
err = av_write_trailer(ctx->avf);
|
||||||
if (err) {
|
if (err) {
|
||||||
av_log(ctx, AV_LOG_ERROR, "Error writing trailer: %s!\n",
|
av_log(ctx, AV_LOG_ERROR, "Error writing trailer: %s!\n",
|
||||||
av_err2str(err));
|
av_err2str(err));
|
||||||
return err;
|
return err;
|
||||||
}
|
}
|
||||||
|
|
||||||
av_log(ctx, AV_LOG_INFO, "Wrote trailer!\n");
|
av_log(ctx, AV_LOG_INFO, "Wrote trailer!\n");
|
||||||
|
|
||||||
return ctx->err;
|
return ctx->err;
|
||||||
}
|
}
|
||||||
|
@ -679,7 +679,7 @@ static int init(struct capture_context *ctx) {
|
||||||
wl_list_init(&ctx->output_list);
|
wl_list_init(&ctx->output_list);
|
||||||
|
|
||||||
ctx->registry = wl_display_get_registry(ctx->display);
|
ctx->registry = wl_display_get_registry(ctx->display);
|
||||||
wl_registry_add_listener(ctx->registry, ®istry_listener, ctx);
|
wl_registry_add_listener(ctx->registry, ®istry_listener, ctx);
|
||||||
|
|
||||||
wl_display_roundtrip(ctx->display);
|
wl_display_roundtrip(ctx->display);
|
||||||
wl_display_dispatch(ctx->display);
|
wl_display_dispatch(ctx->display);
|
||||||
|
@ -721,7 +721,7 @@ int main(int argc, char *argv[]) {
|
||||||
"./dmabuf-capture 0 vaapi /dev/dri/renderD129 libx264 nv12 12 "
|
"./dmabuf-capture 0 vaapi /dev/dri/renderD129 libx264 nv12 12 "
|
||||||
"dmabuf_recording_01.mkv\n");
|
"dmabuf_recording_01.mkv\n");
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
const int o_id = strtol(argv[1], NULL, 10);
|
const int o_id = strtol(argv[1], NULL, 10);
|
||||||
o = find_output(&ctx, NULL, o_id);
|
o = find_output(&ctx, NULL, o_id);
|
||||||
|
@ -759,9 +759,9 @@ static void uninit(struct capture_context *ctx) {
|
||||||
zwlr_export_dmabuf_manager_v1_destroy(ctx->export_manager);
|
zwlr_export_dmabuf_manager_v1_destroy(ctx->export_manager);
|
||||||
|
|
||||||
av_buffer_unref(&ctx->drm_frames_ref);
|
av_buffer_unref(&ctx->drm_frames_ref);
|
||||||
av_buffer_unref(&ctx->drm_device_ref);
|
av_buffer_unref(&ctx->drm_device_ref);
|
||||||
av_buffer_unref(&ctx->mapped_frames_ref);
|
av_buffer_unref(&ctx->mapped_frames_ref);
|
||||||
av_buffer_unref(&ctx->mapped_device_ref);
|
av_buffer_unref(&ctx->mapped_device_ref);
|
||||||
|
|
||||||
av_dict_free(&ctx->encoder_opts);
|
av_dict_free(&ctx->encoder_opts);
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue