mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2025-01-12 18:35:35 +00:00
Code Issues 1 Projects Releases Wiki Activity

dirac: various fixes

Browse source 
- Work around BaseEncoder bug not tracking frame index early.
 - Fix conversion of UYVY, YUY2, AYUV to internal format.
This commit is contained in:
David Schleef 2009-10-05 17:04:28 -07:00
parent 76f489c595
commit 1beac5913d
1 changed files with 32 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
ext/dirac/gstdiracenc.cc
View file

@ -99,6 +99,8 @@ struct _GstDiracEnc
GstBuffer *buffer; GstBuffer *buffer;
GstCaps *srccaps; GstCaps *srccaps;
int pull_frame_num; int pull_frame_num;
int frame_index;
}; };
struct _GstDiracEncClass struct _GstDiracEncClass
@ -326,7 +328,8 @@ gst_dirac_enc_set_format (GstBaseVideoEncoder * base_video_encoder,
GstCaps *caps; GstCaps *caps;
GstStructure *structure; GstStructure *structure;
GST_DEBUG ("set_output_caps"); GST_DEBUG ("set_format");
caps = caps =
gst_pad_get_allowed_caps (GST_BASE_VIDEO_CODEC_SRC_PAD gst_pad_get_allowed_caps (GST_BASE_VIDEO_CODEC_SRC_PAD
(base_video_encoder)); (base_video_encoder));
@ -847,7 +850,11 @@ gst_dirac_enc_handle_frame (GstBaseVideoEncoder * base_video_encoder,
{ {
GstDiracEnc *dirac_enc = GST_DIRAC_ENC (base_video_encoder); GstDiracEnc *dirac_enc = GST_DIRAC_ENC (base_video_encoder);
GstFlowReturn ret; GstFlowReturn ret;
int r;
const GstVideoState *state; const GstVideoState *state;
uint8_t *data;
gboolean copied = FALSE;
int size;
state = gst_base_video_encoder_get_state (base_video_encoder); state = gst_base_video_encoder_get_state (base_video_encoder);
@ -860,17 +867,17 @@ gst_dirac_enc_handle_frame (GstBaseVideoEncoder * base_video_encoder,
switch (state->format) { switch (state->format) {
case GST_VIDEO_FORMAT_I420: case GST_VIDEO_FORMAT_I420:
dirac_encoder_load (dirac_enc->encoder, data = GST_BUFFER_DATA (frame->sink_buffer);
GST_BUFFER_DATA (frame->sink_buffer), size = GST_BUFFER_SIZE (frame->sink_buffer);
GST_BUFFER_SIZE (frame->sink_buffer));
break; break;
case GST_VIDEO_FORMAT_UYVY: case GST_VIDEO_FORMAT_YUY2:
{ {
uint8_t *data;
uint8_t *bufdata = GST_BUFFER_DATA (frame->sink_buffer); uint8_t *bufdata = GST_BUFFER_DATA (frame->sink_buffer);
int i, j; int i, j;
data = (uint8_t *) g_malloc (GST_BUFFER_SIZE (frame->sink_buffer)); data = (uint8_t *) g_malloc (GST_BUFFER_SIZE (frame->sink_buffer));
copied = TRUE;
size = GST_BUFFER_SIZE (frame->sink_buffer);
for (j = 0; j < state->height; j++) { for (j = 0; j < state->height; j++) {
for (i = 0; i < state->width; i++) { for (i = 0; i < state->width; i++) {
data[j * state->width + i] = bufdata[j * state->width * 2 + i * 2]; data[j * state->width + i] = bufdata[j * state->width * 2 + i * 2];
@ -885,18 +892,16 @@ gst_dirac_enc_handle_frame (GstBaseVideoEncoder * base_video_encoder,
bufdata[j * state->width * 2 + i * 4 + 3]; bufdata[j * state->width * 2 + i * 4 + 3];
} }
} }
dirac_encoder_load (dirac_enc->encoder, data,
GST_BUFFER_SIZE (frame->sink_buffer));
g_free (data);
} }
break; break;
case GST_VIDEO_FORMAT_YUY2: case GST_VIDEO_FORMAT_UYVY:
{ {
uint8_t *data;
uint8_t *bufdata = GST_BUFFER_DATA (frame->sink_buffer); uint8_t *bufdata = GST_BUFFER_DATA (frame->sink_buffer);
int i, j; int i, j;
data = (uint8_t *) g_malloc (GST_BUFFER_SIZE (frame->sink_buffer)); data = (uint8_t *) g_malloc (GST_BUFFER_SIZE (frame->sink_buffer));
copied = TRUE;
size = GST_BUFFER_SIZE (frame->sink_buffer);
for (j = 0; j < state->height; j++) { for (j = 0; j < state->height; j++) {
for (i = 0; i < state->width; i++) { for (i = 0; i < state->width; i++) {
data[j * state->width + i] = data[j * state->width + i] =
@ -912,24 +917,20 @@ gst_dirac_enc_handle_frame (GstBaseVideoEncoder * base_video_encoder,
bufdata[j * state->width * 2 + i * 4 + 2]; bufdata[j * state->width * 2 + i * 4 + 2];
} }
} }
dirac_encoder_load (dirac_enc->encoder, data,
GST_BUFFER_SIZE (frame->sink_buffer));
g_free (data);
} }
break; break;
case GST_VIDEO_FORMAT_AYUV: case GST_VIDEO_FORMAT_AYUV:
{ {
uint8_t *data;
uint8_t *bufdata = GST_BUFFER_DATA (frame->sink_buffer); uint8_t *bufdata = GST_BUFFER_DATA (frame->sink_buffer);
int i, j; int i, j;
data = (uint8_t *) g_malloc (GST_BUFFER_SIZE (frame->sink_buffer)); size = state->height * state->width * 3;
data = (uint8_t *) g_malloc (size);
copied = TRUE;
for (j = 0; j < state->height; j++) { for (j = 0; j < state->height; j++) {
for (i = 0; i < state->width; i++) { for (i = 0; i < state->width; i++) {
data[j * state->width + i] = data[j * state->width + i] =
bufdata[j * state->width * 4 + i * 4 + 1]; bufdata[j * state->width * 4 + i * 4 + 1];
}
for (i = 0; i < state->width; i++) {
data[state->height * state->width data[state->height * state->width
+ j * state->width + i] = + j * state->width + i] =
bufdata[j * state->width * 4 + i * 4 + 2]; bufdata[j * state->width * 4 + i * 4 + 2];
@ -938,20 +939,30 @@ gst_dirac_enc_handle_frame (GstBaseVideoEncoder * base_video_encoder,
bufdata[j * state->width * 4 + i * 4 + 3]; bufdata[j * state->width * 4 + i * 4 + 3];
} }
} }
dirac_encoder_load (dirac_enc->encoder, data,
GST_BUFFER_SIZE (frame->sink_buffer));
g_free (data);
} }
break; break;
default: default:
g_assert_not_reached (); g_assert_not_reached ();
} }
r = dirac_encoder_load (dirac_enc->encoder, data,
GST_BUFFER_SIZE (frame->sink_buffer));
if (copied) {
g_free (data);
}
if (r != (int) GST_BUFFER_SIZE (frame->sink_buffer)) {
GST_ERROR ("failed to push picture");
return GST_FLOW_ERROR;
}
GST_DEBUG ("handle frame"); GST_DEBUG ("handle frame");
gst_buffer_unref (frame->sink_buffer); gst_buffer_unref (frame->sink_buffer);
frame->sink_buffer = NULL; frame->sink_buffer = NULL;
frame->system_frame_number = dirac_enc->frame_index;
dirac_enc->frame_index++;
ret = gst_dirac_enc_process (dirac_enc, FALSE); ret = gst_dirac_enc_process (dirac_enc, FALSE);
return ret; return ret;