mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2025-06-05 06:58:56 +00:00
theora: Split out buffer resize in its own function
This commit is contained in:
Benjamin Otte
Sebastian Dröge
parent
9d4f72f2fa
commit
c1ead16f71
1 changed files with 142 additions and 134 deletions
|
|
@ -752,6 +752,141 @@ theora_enc_is_discontinuous (GstTheoraEnc * enc, GstClockTime timestamp,
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static GstBuffer *
|
||||||
|
theora_enc_resize_buffer (GstTheoraEnc * enc, GstBuffer * buffer)
|
||||||
|
{
|
||||||
|
GstBuffer *newbuf;
|
||||||
|
gint i;
|
||||||
|
guchar *dest_y, *src_y;
|
||||||
|
guchar *dest_u, *src_u;
|
||||||
|
guchar *dest_v, *src_v;
|
||||||
|
gint src_y_stride, src_uv_stride;
|
||||||
|
gint dst_y_stride, dst_uv_stride;
|
||||||
|
gint width, height;
|
||||||
|
gint cwidth, cheight;
|
||||||
|
gint offset_x, right_x, right_border;
|
||||||
|
gint y_size;
|
||||||
|
|
||||||
|
if (enc->width == enc->info_width && enc->height == enc->info_height) {
|
||||||
|
GST_LOG_OBJECT (enc, "no cropping/conversion needed");
|
||||||
|
return buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
GST_LOG_OBJECT (enc, "cropping/conversion needed for strides");
|
||||||
|
/* source width/height */
|
||||||
|
width = enc->width;
|
||||||
|
height = enc->height;
|
||||||
|
/* soucre chroma width/height */
|
||||||
|
cwidth = width / 2;
|
||||||
|
cheight = height / 2;
|
||||||
|
|
||||||
|
/* source strides as defined in videotestsrc */
|
||||||
|
src_y_stride = GST_ROUND_UP_4 (width);
|
||||||
|
src_uv_stride = GST_ROUND_UP_8 (width) / 2;
|
||||||
|
|
||||||
|
/* destination strides from the real picture width */
|
||||||
|
dst_y_stride = enc->info_width;
|
||||||
|
dst_uv_stride = enc->info_width / 2;
|
||||||
|
|
||||||
|
y_size = enc->info_width * enc->info_height;
|
||||||
|
|
||||||
|
newbuf = gst_buffer_new_and_alloc (y_size * 3 / 2);
|
||||||
|
if (!newbuf) {
|
||||||
|
gst_buffer_unref (buffer);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
GST_BUFFER_OFFSET (newbuf) = GST_BUFFER_OFFSET_NONE;
|
||||||
|
gst_buffer_set_caps (newbuf, GST_PAD_CAPS (enc->srcpad));
|
||||||
|
|
||||||
|
dest_y = GST_BUFFER_DATA (newbuf);
|
||||||
|
dest_u = dest_y + y_size;
|
||||||
|
dest_v = dest_u + y_size / 4;
|
||||||
|
|
||||||
|
src_y = GST_BUFFER_DATA (buffer);
|
||||||
|
src_u = src_y + src_y_stride * GST_ROUND_UP_2 (height);
|
||||||
|
src_v = src_u + src_uv_stride * GST_ROUND_UP_2 (height) / 2;
|
||||||
|
|
||||||
|
if (enc->border != BORDER_NONE) {
|
||||||
|
/* fill top border */
|
||||||
|
for (i = 0; i < enc->offset_y; i++) {
|
||||||
|
memset (dest_y, 0, dst_y_stride);
|
||||||
|
dest_y += dst_y_stride;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
dest_y += dst_y_stride * enc->offset_y;
|
||||||
|
}
|
||||||
|
|
||||||
|
offset_x = enc->offset_x;
|
||||||
|
right_x = width + enc->offset_x;
|
||||||
|
right_border = dst_y_stride - right_x;
|
||||||
|
|
||||||
|
/* copy Y plane */
|
||||||
|
for (i = 0; i < height; i++) {
|
||||||
|
memcpy (dest_y + offset_x, src_y, width);
|
||||||
|
if (enc->border != BORDER_NONE) {
|
||||||
|
memset (dest_y, 0, offset_x);
|
||||||
|
memset (dest_y + right_x, 0, right_border);
|
||||||
|
}
|
||||||
|
|
||||||
|
dest_y += dst_y_stride;
|
||||||
|
src_y += src_y_stride;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (enc->border != BORDER_NONE) {
|
||||||
|
/* fill bottom border */
|
||||||
|
for (i = height + enc->offset_y; i < enc->info.height; i++) {
|
||||||
|
memset (dest_y, 0, dst_y_stride);
|
||||||
|
dest_y += dst_y_stride;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* fill top border chroma */
|
||||||
|
for (i = 0; i < enc->offset_y / 2; i++) {
|
||||||
|
memset (dest_u, 128, dst_uv_stride);
|
||||||
|
memset (dest_v, 128, dst_uv_stride);
|
||||||
|
dest_u += dst_uv_stride;
|
||||||
|
dest_v += dst_uv_stride;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
dest_u += dst_uv_stride * enc->offset_y / 2;
|
||||||
|
dest_v += dst_uv_stride * enc->offset_y / 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
offset_x = enc->offset_x / 2;
|
||||||
|
right_x = cwidth + offset_x;
|
||||||
|
right_border = dst_uv_stride - right_x;
|
||||||
|
|
||||||
|
/* copy UV planes */
|
||||||
|
for (i = 0; i < cheight; i++) {
|
||||||
|
memcpy (dest_v + offset_x, src_v, cwidth);
|
||||||
|
memcpy (dest_u + offset_x, src_u, cwidth);
|
||||||
|
|
||||||
|
if (enc->border != BORDER_NONE) {
|
||||||
|
memset (dest_u, 128, offset_x);
|
||||||
|
memset (dest_u + right_x, 128, right_border);
|
||||||
|
memset (dest_v, 128, offset_x);
|
||||||
|
memset (dest_v + right_x, 128, right_border);
|
||||||
|
}
|
||||||
|
|
||||||
|
dest_u += dst_uv_stride;
|
||||||
|
dest_v += dst_uv_stride;
|
||||||
|
src_u += src_uv_stride;
|
||||||
|
src_v += src_uv_stride;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (enc->border != BORDER_NONE) {
|
||||||
|
/* fill bottom border */
|
||||||
|
for (i = cheight + enc->offset_y / 2; i < enc->info_height / 2; i++) {
|
||||||
|
memset (dest_u, 128, dst_uv_stride);
|
||||||
|
memset (dest_v, 128, dst_uv_stride);
|
||||||
|
dest_u += dst_uv_stride;
|
||||||
|
dest_v += dst_uv_stride;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
gst_buffer_unref (buffer);
|
||||||
|
return newbuf;
|
||||||
|
}
|
||||||
|
|
||||||
static GstFlowReturn
|
static GstFlowReturn
|
||||||
theora_enc_chain (GstPad * pad, GstBuffer * buffer)
|
theora_enc_chain (GstPad * pad, GstBuffer * buffer)
|
||||||
{
|
{
|
||||||
|
|
@ -922,138 +1057,15 @@ theora_enc_chain (GstPad * pad, GstBuffer * buffer)
|
||||||
|
|
||||||
y_size = enc->info_width * enc->info_height;
|
y_size = enc->info_width * enc->info_height;
|
||||||
|
|
||||||
if (enc->width == enc->info_width && enc->height == enc->info_height) {
|
buffer = theora_enc_resize_buffer (enc, buffer);
|
||||||
GST_LOG_OBJECT (enc, "no cropping/conversion needed");
|
if (buffer == NULL)
|
||||||
/* easy case, no cropping/conversion needed */
|
return GST_FLOW_ERROR;
|
||||||
|
|
||||||
pixels = GST_BUFFER_DATA (buffer);
|
pixels = GST_BUFFER_DATA (buffer);
|
||||||
|
|
||||||
yuv.y = pixels;
|
yuv.y = pixels;
|
||||||
yuv.u = yuv.y + y_size;
|
yuv.u = yuv.y + y_size;
|
||||||
yuv.v = yuv.u + y_size / 4;
|
yuv.v = yuv.u + y_size / 4;
|
||||||
} else {
|
|
||||||
GstBuffer *newbuf;
|
|
||||||
gint i;
|
|
||||||
guchar *dest_y, *src_y;
|
|
||||||
guchar *dest_u, *src_u;
|
|
||||||
guchar *dest_v, *src_v;
|
|
||||||
gint src_y_stride, src_uv_stride;
|
|
||||||
gint dst_y_stride, dst_uv_stride;
|
|
||||||
gint width, height;
|
|
||||||
gint cwidth, cheight;
|
|
||||||
gint offset_x, right_x, right_border;
|
|
||||||
|
|
||||||
GST_LOG_OBJECT (enc, "cropping/conversion needed for strides");
|
|
||||||
/* source width/height */
|
|
||||||
width = enc->width;
|
|
||||||
height = enc->height;
|
|
||||||
/* soucre chroma width/height */
|
|
||||||
cwidth = width / 2;
|
|
||||||
cheight = height / 2;
|
|
||||||
|
|
||||||
/* source strides as defined in videotestsrc */
|
|
||||||
src_y_stride = GST_ROUND_UP_4 (width);
|
|
||||||
src_uv_stride = GST_ROUND_UP_8 (width) / 2;
|
|
||||||
|
|
||||||
/* destination strides from the real picture width */
|
|
||||||
dst_y_stride = enc->info_width;
|
|
||||||
dst_uv_stride = enc->info_width / 2;
|
|
||||||
|
|
||||||
newbuf = gst_buffer_new_and_alloc (y_size * 3 / 2);
|
|
||||||
if (!newbuf) {
|
|
||||||
ret = GST_FLOW_ERROR;
|
|
||||||
goto no_buffer;
|
|
||||||
}
|
|
||||||
GST_BUFFER_OFFSET (newbuf) = GST_BUFFER_OFFSET_NONE;
|
|
||||||
gst_buffer_set_caps (newbuf, GST_PAD_CAPS (enc->srcpad));
|
|
||||||
|
|
||||||
dest_y = yuv.y = GST_BUFFER_DATA (newbuf);
|
|
||||||
dest_u = yuv.u = yuv.y + y_size;
|
|
||||||
dest_v = yuv.v = yuv.u + y_size / 4;
|
|
||||||
|
|
||||||
src_y = GST_BUFFER_DATA (buffer);
|
|
||||||
src_u = src_y + src_y_stride * GST_ROUND_UP_2 (height);
|
|
||||||
src_v = src_u + src_uv_stride * GST_ROUND_UP_2 (height) / 2;
|
|
||||||
|
|
||||||
if (enc->border != BORDER_NONE) {
|
|
||||||
/* fill top border */
|
|
||||||
for (i = 0; i < enc->offset_y; i++) {
|
|
||||||
memset (dest_y, 0, dst_y_stride);
|
|
||||||
dest_y += dst_y_stride;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
dest_y += dst_y_stride * enc->offset_y;
|
|
||||||
}
|
|
||||||
|
|
||||||
offset_x = enc->offset_x;
|
|
||||||
right_x = width + enc->offset_x;
|
|
||||||
right_border = dst_y_stride - right_x;
|
|
||||||
|
|
||||||
/* copy Y plane */
|
|
||||||
for (i = 0; i < height; i++) {
|
|
||||||
memcpy (dest_y + offset_x, src_y, width);
|
|
||||||
if (enc->border != BORDER_NONE) {
|
|
||||||
memset (dest_y, 0, offset_x);
|
|
||||||
memset (dest_y + right_x, 0, right_border);
|
|
||||||
}
|
|
||||||
|
|
||||||
dest_y += dst_y_stride;
|
|
||||||
src_y += src_y_stride;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (enc->border != BORDER_NONE) {
|
|
||||||
/* fill bottom border */
|
|
||||||
for (i = height + enc->offset_y; i < enc->info.height; i++) {
|
|
||||||
memset (dest_y, 0, dst_y_stride);
|
|
||||||
dest_y += dst_y_stride;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* fill top border chroma */
|
|
||||||
for (i = 0; i < enc->offset_y / 2; i++) {
|
|
||||||
memset (dest_u, 128, dst_uv_stride);
|
|
||||||
memset (dest_v, 128, dst_uv_stride);
|
|
||||||
dest_u += dst_uv_stride;
|
|
||||||
dest_v += dst_uv_stride;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
dest_u += dst_uv_stride * enc->offset_y / 2;
|
|
||||||
dest_v += dst_uv_stride * enc->offset_y / 2;
|
|
||||||
}
|
|
||||||
|
|
||||||
offset_x = enc->offset_x / 2;
|
|
||||||
right_x = cwidth + offset_x;
|
|
||||||
right_border = dst_uv_stride - right_x;
|
|
||||||
|
|
||||||
/* copy UV planes */
|
|
||||||
for (i = 0; i < cheight; i++) {
|
|
||||||
memcpy (dest_v + offset_x, src_v, cwidth);
|
|
||||||
memcpy (dest_u + offset_x, src_u, cwidth);
|
|
||||||
|
|
||||||
if (enc->border != BORDER_NONE) {
|
|
||||||
memset (dest_u, 128, offset_x);
|
|
||||||
memset (dest_u + right_x, 128, right_border);
|
|
||||||
memset (dest_v, 128, offset_x);
|
|
||||||
memset (dest_v + right_x, 128, right_border);
|
|
||||||
}
|
|
||||||
|
|
||||||
dest_u += dst_uv_stride;
|
|
||||||
dest_v += dst_uv_stride;
|
|
||||||
src_u += src_uv_stride;
|
|
||||||
src_v += src_uv_stride;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (enc->border != BORDER_NONE) {
|
|
||||||
/* fill bottom border */
|
|
||||||
for (i = cheight + enc->offset_y / 2; i < enc->info_height / 2; i++) {
|
|
||||||
memset (dest_u, 128, dst_uv_stride);
|
|
||||||
memset (dest_v, 128, dst_uv_stride);
|
|
||||||
dest_u += dst_uv_stride;
|
|
||||||
dest_v += dst_uv_stride;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
gst_buffer_unref (buffer);
|
|
||||||
buffer = newbuf;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (theora_enc_is_discontinuous (enc, running_time, duration)) {
|
if (theora_enc_is_discontinuous (enc, running_time, duration)) {
|
||||||
theora_enc_reset (enc);
|
theora_enc_reset (enc);
|
||||||
|
|
@ -1099,10 +1111,6 @@ header_push:
|
||||||
gst_buffer_unref (buffer);
|
gst_buffer_unref (buffer);
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
no_buffer:
|
|
||||||
{
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
data_push:
|
data_push:
|
||||||
{
|
{
|
||||||
gst_buffer_unref (buffer);
|
gst_buffer_unref (buffer);
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue