examples/dmabuf-capture: fix indentation

This commit is contained in:
emersion 2018-05-28 08:20:22 +01:00
parent 2681352e04
commit 2198fd5eed
No known key found for this signature in database
GPG Key ID: 0FDE7BE0E88F5E48
1 changed files with 86 additions and 86 deletions

View File

@ -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, &registry_listener, ctx); wl_registry_add_listener(ctx->registry, &registry_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);