From be6a492ed50086a11caa0238aab3a237dc6ee715 Mon Sep 17 00:00:00 2001 From: Mark Nauwelaerts Date: Mon, 11 Aug 2008 17:24:58 +0000 Subject: ext/x264/gstx264enc.c: Allocate some buffers in more adaptive and economical fashion. Original commit message from CVS: * ext/x264/gstx264enc.c: (gst_x264_enc_init), (gst_x264_enc_header_buf), (gst_x264_enc_encode_frame): Allocate some buffers in more adaptive and economical fashion. --- ext/x264/gstx264enc.c | 28 ++++++++++++++-------------- 1 file changed, 14 insertions(+), 14 deletions(-) (limited to 'ext/x264/gstx264enc.c') diff --git a/ext/x264/gstx264enc.c b/ext/x264/gstx264enc.c index 3eae9129..5182e84e 100644 --- a/ext/x264/gstx264enc.c +++ b/ext/x264/gstx264enc.c @@ -424,7 +424,7 @@ gst_x264_enc_init (GstX264Enc * encoder, GstX264EncClass * klass) /* resources */ encoder->delay = g_queue_new (); - encoder->buffer_size = 1040000; + encoder->buffer_size = 100000; encoder->buffer = g_malloc (encoder->buffer_size); x264_param_default (&encoder->x264param); @@ -637,16 +637,18 @@ gst_x264_enc_header_buf (GstX264Enc * encoder) return NULL; } - /* This should be enough for a header buffer. */ - buffer_size = 100000; - buffer = g_malloc (buffer_size); - - if (nal[1].i_type != 7 || nal[2].i_type != 8) { - GST_ELEMENT_ERROR (encoder, STREAM, ENCODE, ("Unexpected x264 header."), - ("TODO avcC header construction for high profiles needs some work")); + /* x264 is expected to return an SEI (some identification info), + * followed by an SPS and PPS */ + if (i_nal != 3 || nal[1].i_type != 7 || nal[2].i_type != 8) { + GST_ELEMENT_ERROR (encoder, STREAM, ENCODE, NULL, + ("Unexpected x264 header.")); return NULL; } + /* nal payloads with emulation_prevention_three_byte, and some header data */ + buffer_size = (nal[1].i_payload + nal[2].i_payload) * 4 + 100; + buffer = g_malloc (buffer_size); + sps = nal[1].p_payload; buffer[0] = 1; /* AVC Decoder Configuration Record ver. 1 */ @@ -672,9 +674,7 @@ gst_x264_enc_header_buf (GstX264Enc * encoder) i_size += nal_size + 2; buf = gst_buffer_new_and_alloc (i_size); - memcpy (GST_BUFFER_DATA (buf), buffer, i_size); - g_free (buffer); return buf; @@ -906,12 +906,12 @@ gst_x264_enc_encode_frame (GstX264Enc * encoder, x264_picture_t * pic_in, i_size = 0; for (i = 0; i < *i_nal; i++) { - int i_data = encoder->buffer_size - i_size - 4; + gint i_data = encoder->buffer_size - i_size - 4; - if (i_data < encoder->buffer_size / 2) { - encoder->buffer_size *= 2; + if (i_data < nal[i].i_payload * 2) { + encoder->buffer_size += 2 * nal[i].i_payload; encoder->buffer = g_realloc (encoder->buffer, encoder->buffer_size); - i_data = encoder->buffer_size - i_size; + i_data = encoder->buffer_size - i_size - 4; } nal_size = -- cgit v1.2.1