mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2025-06-07 07:58:51 +00:00
Code Issues 1 Projects Releases Wiki Activity

tests: simple-encoder: add an option to set ROI

Browse source 
$ simple-encoder -r inputfile.y4m

And you'll got an output file in H264 with two regions of interest.

https://bugzilla.gnome.org/show_bug.cgi?id=768248

Signed-off-by: Víctor Manuel Jáquez Leal <vjaquez@igalia.com>
This commit is contained in:
Hyunjun Ko 2017-02-23 18:53:18 +09:00 committed by Víctor Manuel Jáquez Leal
parent f3302a0a79
commit c21345c478
1 changed files with 39 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
tests/simple-encoder.c
View file

@ -32,6 +32,7 @@ static guint g_bitrate = 0;
static gchar *g_codec_str; static gchar *g_codec_str;
static gchar *g_output_file_name; static gchar *g_output_file_name;
static char **g_input_files = NULL; static char **g_input_files = NULL;
static gboolean g_roi_enable = FALSE;
#define SURFACE_NUM 16 #define SURFACE_NUM 16
@ -42,6 +43,8 @@ static GOptionEntry g_options[] = {
"desired bitrate expressed in kbps", NULL}, "desired bitrate expressed in kbps", NULL},
{"output", 'o', 0, G_OPTION_ARG_FILENAME, &g_output_file_name, {"output", 'o', 0, G_OPTION_ARG_FILENAME, &g_output_file_name,
"output file name", NULL}, "output file name", NULL},
{"roi", 'r', 0, G_OPTION_ARG_NONE, &g_roi_enable,
"enable region of interest", NULL},
{G_OPTION_REMAINING, ' ', 0, G_OPTION_ARG_FILENAME_ARRAY, &g_input_files, {G_OPTION_REMAINING, ' ', 0, G_OPTION_ARG_FILENAME_ARRAY, &g_input_files,
"input file name", NULL}, "input file name", NULL},
{NULL} {NULL}
@ -58,6 +61,7 @@ typedef struct
FILE *output_file; FILE *output_file;
guint input_stopped:1; guint input_stopped:1;
guint encode_failed:1; guint encode_failed:1;
GstVaapiROI roi_region[2];
} App; } App;
static inline gchar * static inline gchar *
@ -180,6 +184,35 @@ set_format (GstVaapiEncoder * encoder, gint width, gint height, gint fps_n,
return (status == GST_VAAPI_ENCODER_STATUS_SUCCESS); return (status == GST_VAAPI_ENCODER_STATUS_SUCCESS);
} }
static void
add_roi (App * app)
{
guint i;
gint width, height;
width = app->parser->width;
height = app->parser->height;
for (i = 0; i < 2; i++) {
app->roi_region[i].roi_value = 4;
app->roi_region[i].rect.x = i * width / 2;
app->roi_region[i].rect.y = i * height / 2;
app->roi_region[i].rect.width = width / 4;
app->roi_region[i].rect.height = height / 4;
gst_vaapi_encoder_add_roi (app->encoder, &app->roi_region[i]);
}
}
static void
del_roi (App * app)
{
guint i;
for (i = 0; i < 2; i++)
gst_vaapi_encoder_del_roi (app->encoder, &app->roi_region[i]);
}
static GstBuffer * static GstBuffer *
allocate_buffer (GstVaapiCodedBuffer * vbuf) allocate_buffer (GstVaapiCodedBuffer * vbuf)
{ {
@ -293,6 +326,9 @@ app_free (App * app)
{ {
g_return_if_fail (app); g_return_if_fail (app);
if (g_roi_enable)
del_roi (app);
if (app->parser) if (app->parser)
y4m_reader_close (app->parser); y4m_reader_close (app->parser);
@ -348,6 +384,9 @@ app_new (const gchar * input_fn, const gchar * output_fn)
goto error; goto error;
} }
if (g_roi_enable)
add_roi (app);
return app; return app;
error: error: