mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-19 16:21:17 +00:00
omxvideoenc: adjust stride and slice height from input
Use the stride and slice height information from the first buffer meta data to adjust the settings of the input port. This will ensure that the OMX input buffers match the GStreamer ones and so will save us from having to copy line-by-line each one. This is also the first step to allow the OMX encoder to receive dmabuf. Tested on rpi and zynqultrascaleplus. https://bugzilla.gnome.org/show_bug.cgi?id=785967
This commit is contained in:
parent
07e80c9425
commit
0ee8213ab2
1 changed files with 35 additions and 8 deletions
|
@ -1003,24 +1003,51 @@ gst_omx_video_enc_disable (GstOMXVideoEnc * self)
|
|||
}
|
||||
|
||||
static gboolean
|
||||
gst_omx_video_enc_configure_input_buffer (GstOMXVideoEnc * self)
|
||||
gst_omx_video_enc_configure_input_buffer (GstOMXVideoEnc * self,
|
||||
GstBuffer * input)
|
||||
{
|
||||
GstOMXVideoEncClass *klass = GST_OMX_VIDEO_ENC_GET_CLASS (self);
|
||||
GstVideoInfo *info = &self->input_state->info;
|
||||
OMX_PARAM_PORTDEFINITIONTYPE port_def;
|
||||
GstVideoMeta *meta;
|
||||
guint stride, slice_height;
|
||||
|
||||
gst_omx_port_get_port_definition (self->enc_in_port, &port_def);
|
||||
|
||||
meta = gst_buffer_get_video_meta (input);
|
||||
if (meta) {
|
||||
/* Use the stride and slice height of the first plane */
|
||||
stride = meta->stride[0];
|
||||
g_assert (stride != 0);
|
||||
slice_height = (meta->offset[1] - meta->offset[0]) / stride;
|
||||
|
||||
GST_DEBUG_OBJECT (self,
|
||||
"adjusting stride (%d) and slice-height (%d) using input buffer meta",
|
||||
stride, slice_height);
|
||||
} else {
|
||||
GST_WARNING_OBJECT (self,
|
||||
"input buffer doesn't provide video meta, can't adjust stride and slice height");
|
||||
|
||||
stride = info->stride[0];
|
||||
slice_height = info->height;
|
||||
}
|
||||
|
||||
if (port_def.nBufferAlignment)
|
||||
port_def.format.video.nStride =
|
||||
GST_ROUND_UP_N (info->width, port_def.nBufferAlignment);
|
||||
GST_ROUND_UP_N (stride, port_def.nBufferAlignment);
|
||||
else
|
||||
port_def.format.video.nStride = GST_ROUND_UP_4 (info->width); /* safe (?) default */
|
||||
port_def.format.video.nStride = GST_ROUND_UP_4 (stride); /* safe (?) default */
|
||||
|
||||
if (klass->cdata.hacks & GST_OMX_HACK_HEIGHT_MULTIPLE_16)
|
||||
port_def.format.video.nSliceHeight = GST_ROUND_UP_16 (info->height);
|
||||
port_def.format.video.nSliceHeight = GST_ROUND_UP_16 (slice_height);
|
||||
else
|
||||
port_def.format.video.nSliceHeight = info->height;
|
||||
port_def.format.video.nSliceHeight = slice_height;
|
||||
|
||||
GST_DEBUG_OBJECT (self,
|
||||
"setting input nStride=%d and nSliceHeight=%d (nBufferAlignment=%d)",
|
||||
(guint) port_def.format.video.nStride,
|
||||
(guint) port_def.format.video.nSliceHeight,
|
||||
(guint) port_def.nBufferAlignment);
|
||||
|
||||
switch (port_def.format.video.eColorFormat) {
|
||||
case OMX_COLOR_FormatYUV420Planar:
|
||||
|
@ -1053,13 +1080,13 @@ gst_omx_video_enc_configure_input_buffer (GstOMXVideoEnc * self)
|
|||
}
|
||||
|
||||
static gboolean
|
||||
gst_omx_video_enc_enable (GstOMXVideoEnc * self)
|
||||
gst_omx_video_enc_enable (GstOMXVideoEnc * self, GstBuffer * input)
|
||||
{
|
||||
GstOMXVideoEncClass *klass;
|
||||
|
||||
klass = GST_OMX_VIDEO_ENC_GET_CLASS (self);
|
||||
|
||||
if (!gst_omx_video_enc_configure_input_buffer (self))
|
||||
if (!gst_omx_video_enc_configure_input_buffer (self, input))
|
||||
return FALSE;
|
||||
|
||||
GST_DEBUG_OBJECT (self, "Enabling component");
|
||||
|
@ -1536,7 +1563,7 @@ gst_omx_video_enc_handle_frame (GstVideoEncoder * encoder,
|
|||
|
||||
if (!self->started) {
|
||||
if (gst_omx_port_is_flushing (self->enc_out_port)) {
|
||||
if (!gst_omx_video_enc_enable (self))
|
||||
if (!gst_omx_video_enc_enable (self, frame->input_buffer))
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in a new issue