mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-22 08:17:01 +00:00
encoder: add mpeg2 encoder.
Add initial support for MPEG-2 encoding. I/P/B frames are supported. Signed-off-by: Gwenole Beauchesne <gwenole.beauchesne@intel.com>
This commit is contained in:
XuGuangxin
Gwenole Beauchesne
parent
652245fb87
commit
d753c2226a
4 changed files with 1102 additions and 0 deletions
|
|
@ -137,12 +137,14 @@ endif
|
||||||
libgstvaapi_enc_source_c = \
|
libgstvaapi_enc_source_c = \
|
||||||
gstvaapiencoder.c \
|
gstvaapiencoder.c \
|
||||||
gstvaapiencoder_h264.c \
|
gstvaapiencoder_h264.c \
|
||||||
|
gstvaapiencoder_mpeg2.c \
|
||||||
gstvaapiencoder_objects.c \
|
gstvaapiencoder_objects.c \
|
||||||
$(NULL)
|
$(NULL)
|
||||||
|
|
||||||
libgstvaapi_enc_source_h = \
|
libgstvaapi_enc_source_h = \
|
||||||
gstvaapiencoder.h \
|
gstvaapiencoder.h \
|
||||||
gstvaapiencoder_h264.h \
|
gstvaapiencoder_h264.h \
|
||||||
|
gstvaapiencoder_mpeg2.h \
|
||||||
$(NULL)
|
$(NULL)
|
||||||
|
|
||||||
libgstvaapi_enc_source_priv_h = \
|
libgstvaapi_enc_source_priv_h = \
|
||||||
|
|
|
||||||
948
gst-libs/gst/vaapi/gstvaapiencoder_mpeg2.c
Normal file
948
gst-libs/gst/vaapi/gstvaapiencoder_mpeg2.c
Normal file
|
|
@ -0,0 +1,948 @@
|
||||||
|
/*
|
||||||
|
* gstvaapiencoder_mpeg2.c - MPEG-2 encoder
|
||||||
|
*
|
||||||
|
* Copyright (C) 2012 -2013 Intel Corporation
|
||||||
|
*
|
||||||
|
* This library is free software; you can redistribute it and/or
|
||||||
|
* modify it under the terms of the GNU Lesser General Public License
|
||||||
|
* as published by the Free Software Foundation; either version 2.1
|
||||||
|
* of the License, or (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This library is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||||
|
* Lesser General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU Lesser General Public
|
||||||
|
* License along with this library; if not, write to the Free
|
||||||
|
* Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
|
||||||
|
* Boston, MA 02110-1301 USA
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "sysdeps.h"
|
||||||
|
#include <gst/base/gstbitwriter.h>
|
||||||
|
#include "gstvaapicompat.h"
|
||||||
|
#include "gstvaapiencoder_mpeg2.h"
|
||||||
|
#include "gstvaapiencoder_mpeg2_priv.h"
|
||||||
|
#include "gstvaapiencoder_priv.h"
|
||||||
|
|
||||||
|
#include <va/va.h>
|
||||||
|
#include <va/va_enc_mpeg2.h>
|
||||||
|
|
||||||
|
#include "gstvaapicontext.h"
|
||||||
|
#include "gstvaapisurface.h"
|
||||||
|
#include "gstvaapidisplay_priv.h"
|
||||||
|
|
||||||
|
#define DEBUG 1
|
||||||
|
#include "gstvaapidebug.h"
|
||||||
|
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
gst_bit_writer_write_sps (GstBitWriter * bitwriter,
|
||||||
|
VAEncSequenceParameterBufferMPEG2 * seq, GstVaapiEncoderMpeg2 * encoder);
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
gst_bit_writer_write_pps (GstBitWriter * bitwriter,
|
||||||
|
VAEncPictureParameterBufferMPEG2 * pic);
|
||||||
|
|
||||||
|
|
||||||
|
static void clear_references (GstVaapiEncoderMpeg2 * encoder);
|
||||||
|
|
||||||
|
static void push_reference (GstVaapiEncoderMpeg2 * encoder,
|
||||||
|
GstVaapiSurfaceProxy * ref;);
|
||||||
|
|
||||||
|
static struct
|
||||||
|
{
|
||||||
|
int samplers_per_line;
|
||||||
|
int line_per_frame;
|
||||||
|
int frame_per_sec;
|
||||||
|
} mpeg2_upper_samplings[2][3] = {
|
||||||
|
{ { 0, 0, 0},
|
||||||
|
{ 720, 576, 30 },
|
||||||
|
{ 0, 0, 0 },
|
||||||
|
},
|
||||||
|
{ { 352, 288, 30 },
|
||||||
|
{ 720, 576, 30 },
|
||||||
|
{ 1920, 1152, 60 },
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
ensure_sampling_desity (GstVaapiEncoderMpeg2 * encoder)
|
||||||
|
{
|
||||||
|
guint p, l;
|
||||||
|
float fps;
|
||||||
|
p = encoder->profile;
|
||||||
|
l = encoder->level;
|
||||||
|
fps = GST_VAAPI_ENCODER_FPS_N (encoder) / GST_VAAPI_ENCODER_FPS_D (encoder);
|
||||||
|
if (mpeg2_upper_samplings[p][l].samplers_per_line <
|
||||||
|
GST_VAAPI_ENCODER_WIDTH (encoder)
|
||||||
|
|| mpeg2_upper_samplings[p][l].line_per_frame <
|
||||||
|
GST_VAAPI_ENCODER_HEIGHT (encoder)
|
||||||
|
|| mpeg2_upper_samplings[p][l].frame_per_sec < fps) {
|
||||||
|
GST_ERROR
|
||||||
|
("acording to slected profile(%d) and level(%d) the max resolution is %dx%d@%d",
|
||||||
|
p, l, mpeg2_upper_samplings[p][l].samplers_per_line,
|
||||||
|
mpeg2_upper_samplings[p][l].line_per_frame,
|
||||||
|
mpeg2_upper_samplings[p][l].frame_per_sec);
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
ensure_public_attributes (GstVaapiEncoderMpeg2 * encoder)
|
||||||
|
{
|
||||||
|
if (!GST_VAAPI_ENCODER_WIDTH (encoder) ||
|
||||||
|
!GST_VAAPI_ENCODER_HEIGHT (encoder) ||
|
||||||
|
!GST_VAAPI_ENCODER_FPS_N (encoder) ||
|
||||||
|
!GST_VAAPI_ENCODER_FPS_D (encoder)) {
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (encoder->ip_period > encoder->intra_period) {
|
||||||
|
encoder->ip_period = encoder->intra_period - 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (encoder->profile == GST_ENCODER_MPEG2_PROFILE_SIMPLE) {
|
||||||
|
/* no b frames */
|
||||||
|
encoder->ip_period = 0;
|
||||||
|
/* only main level is defined in mpeg2 */
|
||||||
|
encoder->level = GST_VAAPI_ENCODER_MPEG2_LEVEL_MAIN;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!ensure_sampling_desity (encoder))
|
||||||
|
return FALSE;
|
||||||
|
|
||||||
|
/* default compress ratio 1: (4*8*1.5) */
|
||||||
|
if (GST_VAAPI_RATECONTROL_CBR == GST_VAAPI_ENCODER_RATE_CONTROL (encoder)) {
|
||||||
|
if (!encoder->bitrate)
|
||||||
|
encoder->bitrate = GST_VAAPI_ENCODER_WIDTH (encoder) *
|
||||||
|
GST_VAAPI_ENCODER_HEIGHT (encoder) *
|
||||||
|
GST_VAAPI_ENCODER_FPS_N (encoder) /
|
||||||
|
GST_VAAPI_ENCODER_FPS_D (encoder) / 4 / 1024;
|
||||||
|
} else
|
||||||
|
encoder->bitrate = 0;
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static unsigned char
|
||||||
|
make_profile_and_level_indication (guint32 profile, guint32 level)
|
||||||
|
{
|
||||||
|
guint32 p = 4, l = 8;
|
||||||
|
switch (profile) {
|
||||||
|
case GST_ENCODER_MPEG2_PROFILE_SIMPLE:
|
||||||
|
p = 5;
|
||||||
|
break;
|
||||||
|
case GST_ENCODER_MPEG2_PROFILE_MAIN:
|
||||||
|
p = 4;
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
g_assert (0);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
switch (level) {
|
||||||
|
case GST_VAAPI_ENCODER_MPEG2_LEVEL_LOW:
|
||||||
|
l = 10;
|
||||||
|
break;
|
||||||
|
case GST_VAAPI_ENCODER_MPEG2_LEVEL_MAIN:
|
||||||
|
l = 8;
|
||||||
|
break;
|
||||||
|
case GST_VAAPI_ENCODER_MPEG2_LEVEL_HIGH:
|
||||||
|
l = 4;
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
g_assert (0);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
return p << 4 | l;
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
fill_sequence (GstVaapiEncoderMpeg2 * encoder, GstVaapiEncSequence * sequence)
|
||||||
|
{
|
||||||
|
VAEncSequenceParameterBufferMPEG2 *seq = sequence->param;
|
||||||
|
|
||||||
|
memset (seq, 0, sizeof (VAEncSequenceParameterBufferMPEG2));
|
||||||
|
|
||||||
|
seq->intra_period = encoder->intra_period;
|
||||||
|
seq->ip_period = encoder->ip_period;
|
||||||
|
seq->picture_width = GST_VAAPI_ENCODER_WIDTH (encoder);
|
||||||
|
seq->picture_height = GST_VAAPI_ENCODER_HEIGHT (encoder);
|
||||||
|
|
||||||
|
if (encoder->bitrate > 0)
|
||||||
|
seq->bits_per_second = encoder->bitrate * 1024;
|
||||||
|
else
|
||||||
|
seq->bits_per_second = 0;
|
||||||
|
|
||||||
|
if (GST_VAAPI_ENCODER_FPS_D (encoder))
|
||||||
|
seq->frame_rate =
|
||||||
|
GST_VAAPI_ENCODER_FPS_N (encoder) / GST_VAAPI_ENCODER_FPS_D (encoder);
|
||||||
|
else
|
||||||
|
seq->frame_rate = 0;
|
||||||
|
|
||||||
|
seq->aspect_ratio_information = 1;
|
||||||
|
seq->vbv_buffer_size = 3; /* B = 16 * 1024 * vbv_buffer_size */
|
||||||
|
|
||||||
|
seq->sequence_extension.bits.profile_and_level_indication =
|
||||||
|
make_profile_and_level_indication (encoder->profile, encoder->level);
|
||||||
|
seq->sequence_extension.bits.progressive_sequence = 1; /* progressive frame-pictures */
|
||||||
|
seq->sequence_extension.bits.chroma_format = CHROMA_FORMAT_420; /* 4:2:0 */
|
||||||
|
seq->sequence_extension.bits.low_delay = 0; /* FIXME */
|
||||||
|
seq->sequence_extension.bits.frame_rate_extension_n = 0; /*FIXME */
|
||||||
|
seq->sequence_extension.bits.frame_rate_extension_d = 0;
|
||||||
|
|
||||||
|
seq->gop_header.bits.time_code = (1 << 12); /* bit12: marker_bit */
|
||||||
|
seq->gop_header.bits.closed_gop = 0;
|
||||||
|
seq->gop_header.bits.broken_link = 0;
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static VAEncPictureType
|
||||||
|
get_va_enc_picture_type (GstVaapiPictureType type)
|
||||||
|
{
|
||||||
|
switch (type) {
|
||||||
|
case GST_VAAPI_PICTURE_TYPE_I:
|
||||||
|
return VAEncPictureTypeIntra;
|
||||||
|
case GST_VAAPI_PICTURE_TYPE_P:
|
||||||
|
return VAEncPictureTypePredictive;
|
||||||
|
case GST_VAAPI_PICTURE_TYPE_B:
|
||||||
|
return VAEncPictureTypeBidirectional;
|
||||||
|
default:
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
fill_picture (GstVaapiEncoderMpeg2 * encoder,
|
||||||
|
GstVaapiEncPicture * picture,
|
||||||
|
GstVaapiCodedBuffer * codedbuf, GstVaapiSurfaceProxy * surface)
|
||||||
|
{
|
||||||
|
VAEncPictureParameterBufferMPEG2 *pic = picture->param;
|
||||||
|
uint8_t f_code_x, f_code_y;
|
||||||
|
|
||||||
|
memset (pic, 0, sizeof (VAEncPictureParameterBufferMPEG2));
|
||||||
|
|
||||||
|
pic->reconstructed_picture = GST_VAAPI_SURFACE_PROXY_SURFACE_ID (surface);
|
||||||
|
pic->coded_buf = codedbuf->buf_id;
|
||||||
|
pic->picture_type = get_va_enc_picture_type (picture->type);
|
||||||
|
pic->temporal_reference = picture->frame_num & (1024 - 1);
|
||||||
|
pic->vbv_delay = 0xFFFF;
|
||||||
|
|
||||||
|
f_code_x = 0xf;
|
||||||
|
f_code_y = 0xf;
|
||||||
|
if (pic->picture_type != VAEncPictureTypeIntra) {
|
||||||
|
if (encoder->level == GST_VAAPI_ENCODER_MPEG2_LEVEL_LOW) {
|
||||||
|
f_code_x = 7;
|
||||||
|
f_code_y = 4;
|
||||||
|
} else if (encoder->level == GST_VAAPI_ENCODER_MPEG2_LEVEL_MAIN) {
|
||||||
|
f_code_x = 8;
|
||||||
|
f_code_y = 5;
|
||||||
|
} else {
|
||||||
|
f_code_x = 9;
|
||||||
|
f_code_y = 5;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pic->picture_type == VAEncPictureTypeIntra) {
|
||||||
|
pic->f_code[0][0] = 0xf;
|
||||||
|
pic->f_code[0][1] = 0xf;
|
||||||
|
pic->f_code[1][0] = 0xf;
|
||||||
|
pic->f_code[1][1] = 0xf;
|
||||||
|
pic->forward_reference_picture = VA_INVALID_SURFACE;
|
||||||
|
pic->backward_reference_picture = VA_INVALID_SURFACE;
|
||||||
|
|
||||||
|
} else if (pic->picture_type == VAEncPictureTypePredictive) {
|
||||||
|
pic->f_code[0][0] = f_code_x;
|
||||||
|
pic->f_code[0][1] = f_code_y;
|
||||||
|
pic->f_code[1][0] = 0xf;
|
||||||
|
pic->f_code[1][1] = 0xf;
|
||||||
|
pic->forward_reference_picture =
|
||||||
|
GST_VAAPI_SURFACE_PROXY_SURFACE_ID (encoder->forward);
|
||||||
|
pic->backward_reference_picture = VA_INVALID_SURFACE;
|
||||||
|
} else if (pic->picture_type == VAEncPictureTypeBidirectional) {
|
||||||
|
pic->f_code[0][0] = f_code_x;
|
||||||
|
pic->f_code[0][1] = f_code_y;
|
||||||
|
pic->f_code[1][0] = f_code_x;
|
||||||
|
pic->f_code[1][1] = f_code_y;
|
||||||
|
pic->forward_reference_picture =
|
||||||
|
GST_VAAPI_SURFACE_PROXY_SURFACE_ID (encoder->forward);;
|
||||||
|
pic->backward_reference_picture =
|
||||||
|
GST_VAAPI_SURFACE_PROXY_SURFACE_ID (encoder->backward);;
|
||||||
|
} else {
|
||||||
|
g_assert (0);
|
||||||
|
}
|
||||||
|
|
||||||
|
pic->picture_coding_extension.bits.intra_dc_precision = 0; /* 8bits */
|
||||||
|
pic->picture_coding_extension.bits.picture_structure = 3; /* frame picture */
|
||||||
|
pic->picture_coding_extension.bits.top_field_first = 0;
|
||||||
|
pic->picture_coding_extension.bits.frame_pred_frame_dct = 1; /* FIXME */
|
||||||
|
pic->picture_coding_extension.bits.concealment_motion_vectors = 0;
|
||||||
|
pic->picture_coding_extension.bits.q_scale_type = 0;
|
||||||
|
pic->picture_coding_extension.bits.intra_vlc_format = 0;
|
||||||
|
pic->picture_coding_extension.bits.alternate_scan = 0;
|
||||||
|
pic->picture_coding_extension.bits.repeat_first_field = 0;
|
||||||
|
pic->picture_coding_extension.bits.progressive_frame = 1;
|
||||||
|
pic->picture_coding_extension.bits.composite_display_flag = 0;
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
set_sequence_packed_header (GstVaapiEncoderMpeg2 * encoder,
|
||||||
|
GstVaapiEncPicture * picture, GstVaapiEncSequence * sequence)
|
||||||
|
{
|
||||||
|
GstVaapiEncPackedHeader *packed_seq;
|
||||||
|
GstBitWriter writer;
|
||||||
|
VAEncPackedHeaderParameterBuffer packed_header_param_buffer = { 0 };
|
||||||
|
VAEncSequenceParameterBufferMPEG2 *seq = sequence->param;
|
||||||
|
guint32 data_bit_size;
|
||||||
|
guint8 *data;
|
||||||
|
|
||||||
|
gst_bit_writer_init (&writer, 128 * 8);
|
||||||
|
gst_bit_writer_write_sps (&writer, seq, encoder);
|
||||||
|
g_assert (GST_BIT_WRITER_BIT_SIZE (&writer) % 8 == 0);
|
||||||
|
data_bit_size = GST_BIT_WRITER_BIT_SIZE (&writer);
|
||||||
|
data = GST_BIT_WRITER_DATA (&writer);
|
||||||
|
|
||||||
|
packed_header_param_buffer.type = VAEncPackedHeaderSequence;
|
||||||
|
packed_header_param_buffer.bit_length = data_bit_size;
|
||||||
|
packed_header_param_buffer.has_emulation_bytes = 0;
|
||||||
|
|
||||||
|
packed_seq = gst_vaapi_enc_packed_header_new (GST_VAAPI_ENCODER (encoder),
|
||||||
|
&packed_header_param_buffer, sizeof (packed_header_param_buffer),
|
||||||
|
data, (data_bit_size + 7) / 8);
|
||||||
|
g_assert (packed_seq);
|
||||||
|
|
||||||
|
gst_vaapi_enc_picture_add_packed_header (picture, packed_seq);
|
||||||
|
gst_vaapi_mini_object_replace ((GstVaapiMiniObject **) & packed_seq, NULL);
|
||||||
|
gst_bit_writer_clear (&writer, TRUE);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
set_picture_packed_header (GstVaapiEncoderMpeg2 * encoder,
|
||||||
|
GstVaapiEncPicture * picture)
|
||||||
|
{
|
||||||
|
GstVaapiEncPackedHeader *packed_pic;
|
||||||
|
GstBitWriter writer;
|
||||||
|
VAEncPackedHeaderParameterBuffer packed_header_param_buffer = { 0 };
|
||||||
|
VAEncPictureParameterBufferMPEG2 *pic = picture->param;
|
||||||
|
guint32 data_bit_size;
|
||||||
|
guint8 *data;
|
||||||
|
|
||||||
|
gst_bit_writer_init (&writer, 128 * 8);
|
||||||
|
gst_bit_writer_write_pps (&writer, pic);
|
||||||
|
g_assert (GST_BIT_WRITER_BIT_SIZE (&writer) % 8 == 0);
|
||||||
|
data_bit_size = GST_BIT_WRITER_BIT_SIZE (&writer);
|
||||||
|
data = GST_BIT_WRITER_DATA (&writer);
|
||||||
|
|
||||||
|
packed_header_param_buffer.type = VAEncPackedHeaderPicture;
|
||||||
|
packed_header_param_buffer.bit_length = data_bit_size;
|
||||||
|
packed_header_param_buffer.has_emulation_bytes = 0;
|
||||||
|
|
||||||
|
packed_pic = gst_vaapi_enc_packed_header_new (GST_VAAPI_ENCODER (encoder),
|
||||||
|
&packed_header_param_buffer, sizeof (packed_header_param_buffer),
|
||||||
|
data, (data_bit_size + 7) / 8);
|
||||||
|
g_assert (packed_pic);
|
||||||
|
|
||||||
|
gst_vaapi_enc_picture_add_packed_header (picture, packed_pic);
|
||||||
|
gst_vaapi_mini_object_replace ((GstVaapiMiniObject **) & packed_pic, NULL);
|
||||||
|
gst_bit_writer_clear (&writer, TRUE);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
ensure_sequence (GstVaapiEncoderMpeg2 * encoder, GstVaapiEncPicture * picture)
|
||||||
|
{
|
||||||
|
GstVaapiEncSequence *sequence;
|
||||||
|
|
||||||
|
g_assert (picture);
|
||||||
|
sequence = GST_VAAPI_ENC_SEQUENCE_NEW (MPEG2, encoder);
|
||||||
|
g_assert (sequence);
|
||||||
|
if (!sequence)
|
||||||
|
goto error;
|
||||||
|
|
||||||
|
if (!fill_sequence (encoder, sequence))
|
||||||
|
goto error;
|
||||||
|
|
||||||
|
if (picture->type == GST_VAAPI_PICTURE_TYPE_I &&
|
||||||
|
!set_sequence_packed_header (encoder, picture, sequence))
|
||||||
|
goto error;
|
||||||
|
gst_vaapi_enc_picture_set_sequence (picture, sequence);
|
||||||
|
gst_vaapi_mini_object_replace ((GstVaapiMiniObject **) (&sequence), NULL);
|
||||||
|
return TRUE;
|
||||||
|
|
||||||
|
error:
|
||||||
|
gst_vaapi_mini_object_replace ((GstVaapiMiniObject **) (&sequence), NULL);
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
ensure_picture (GstVaapiEncoderMpeg2 * encoder,
|
||||||
|
GstVaapiEncPicture * picture,
|
||||||
|
GstVaapiCodedBufferProxy * buf_proxy, GstVaapiSurfaceProxy * surface)
|
||||||
|
{
|
||||||
|
GstVaapiCodedBuffer *codedbuf = buf_proxy->buffer;
|
||||||
|
|
||||||
|
if (!fill_picture (encoder, picture, codedbuf, surface))
|
||||||
|
return FALSE;
|
||||||
|
|
||||||
|
if (!set_picture_packed_header (encoder, picture)) {
|
||||||
|
GST_ERROR ("set picture packed header failed");
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
set_misc_parameters (GstVaapiEncoderMpeg2 * encoder,
|
||||||
|
GstVaapiEncPicture * picture)
|
||||||
|
{
|
||||||
|
GstVaapiEncMiscParam *misc = NULL;
|
||||||
|
VAEncMiscParameterHRD *hrd;
|
||||||
|
VAEncMiscParameterRateControl *rate_control;
|
||||||
|
|
||||||
|
/* add hrd */
|
||||||
|
misc = GST_VAAPI_ENC_MISC_PARAM_NEW (HRD, encoder);
|
||||||
|
g_assert (misc);
|
||||||
|
if (!misc)
|
||||||
|
return FALSE;
|
||||||
|
gst_vaapi_enc_picture_add_misc_buffer (picture, misc);
|
||||||
|
hrd = misc->impl;
|
||||||
|
if (encoder->bitrate > 0) {
|
||||||
|
hrd->initial_buffer_fullness = encoder->bitrate * 1024 * 4;
|
||||||
|
hrd->buffer_size = encoder->bitrate * 1024 * 8;
|
||||||
|
} else {
|
||||||
|
hrd->initial_buffer_fullness = 0;
|
||||||
|
hrd->buffer_size = 0;
|
||||||
|
}
|
||||||
|
gst_vaapi_mini_object_replace ((GstVaapiMiniObject **) & misc, NULL);
|
||||||
|
|
||||||
|
/* add ratecontrol */
|
||||||
|
if (GST_VAAPI_ENCODER_RATE_CONTROL (encoder) == GST_VAAPI_RATECONTROL_CBR) {
|
||||||
|
misc = GST_VAAPI_ENC_MISC_PARAM_NEW (RateControl, encoder);
|
||||||
|
g_assert (misc);
|
||||||
|
if (!misc)
|
||||||
|
return FALSE;
|
||||||
|
gst_vaapi_enc_picture_add_misc_buffer (picture, misc);
|
||||||
|
rate_control = misc->impl;
|
||||||
|
memset (rate_control, 0, sizeof (VAEncMiscParameterRateControl));
|
||||||
|
if (encoder->bitrate)
|
||||||
|
rate_control->bits_per_second = encoder->bitrate * 1024;
|
||||||
|
else
|
||||||
|
rate_control->bits_per_second = 0;
|
||||||
|
rate_control->target_percentage = 70;
|
||||||
|
rate_control->window_size = 500;
|
||||||
|
rate_control->initial_qp = encoder->cqp;
|
||||||
|
rate_control->min_qp = 0;
|
||||||
|
rate_control->basic_unit_size = 0;
|
||||||
|
gst_vaapi_mini_object_replace ((GstVaapiMiniObject **) & misc, NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
fill_slices (GstVaapiEncoderMpeg2 * encoder, GstVaapiEncPicture * picture)
|
||||||
|
{
|
||||||
|
VAEncSliceParameterBufferMPEG2 *slice_param;
|
||||||
|
GstVaapiEncSlice *slice;
|
||||||
|
guint width_in_mbs, height_in_mbs;
|
||||||
|
guint i_slice;
|
||||||
|
|
||||||
|
g_assert (picture);
|
||||||
|
|
||||||
|
width_in_mbs = (GST_VAAPI_ENCODER_WIDTH (encoder) + 15) / 16;
|
||||||
|
height_in_mbs = (GST_VAAPI_ENCODER_HEIGHT (encoder) + 15) / 16;
|
||||||
|
|
||||||
|
for (i_slice = 0; i_slice < height_in_mbs; ++i_slice) {
|
||||||
|
slice = GST_VAAPI_ENC_SLICE_NEW (MPEG2, encoder);
|
||||||
|
g_assert (slice && slice->param_id != VA_INVALID_ID);
|
||||||
|
slice_param = slice->param;
|
||||||
|
|
||||||
|
memset (slice_param, 0, sizeof (VAEncSliceParameterBufferMPEG2));
|
||||||
|
|
||||||
|
slice_param->macroblock_address = i_slice * width_in_mbs;
|
||||||
|
slice_param->num_macroblocks = width_in_mbs;
|
||||||
|
slice_param->is_intra_slice = (picture->type == GST_VAAPI_PICTURE_TYPE_I);
|
||||||
|
slice_param->quantiser_scale_code = encoder->cqp / 2;
|
||||||
|
|
||||||
|
gst_vaapi_enc_picture_add_slice (picture, slice);
|
||||||
|
gst_vaapi_mini_object_replace ((GstVaapiMiniObject **) & slice, NULL);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
ensure_slices (GstVaapiEncoderMpeg2 * encoder, GstVaapiEncPicture * picture)
|
||||||
|
{
|
||||||
|
g_assert (picture);
|
||||||
|
|
||||||
|
if (!fill_slices (encoder, picture))
|
||||||
|
return FALSE;
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static GstVaapiEncoderStatus
|
||||||
|
gst_vaapi_encoder_mpeg2_encode (GstVaapiEncoder * base,
|
||||||
|
GstVaapiEncPicture * picture, GstVaapiCodedBufferProxy * codedbuf)
|
||||||
|
{
|
||||||
|
GstVaapiEncoderMpeg2 *encoder = GST_VAAPI_ENCODER_MPEG2_CAST (base);
|
||||||
|
GstVaapiEncoderStatus ret = GST_VAAPI_ENCODER_STATUS_UNKNOWN_ERR;
|
||||||
|
GstVaapiSurfaceProxy *reconstruct = NULL;
|
||||||
|
|
||||||
|
reconstruct = gst_vaapi_encoder_create_surface (base);
|
||||||
|
|
||||||
|
g_assert (GST_VAAPI_SURFACE_PROXY_SURFACE (reconstruct));
|
||||||
|
|
||||||
|
if (!ensure_sequence (encoder, picture))
|
||||||
|
goto error;
|
||||||
|
if (!ensure_picture (encoder, picture, codedbuf, reconstruct))
|
||||||
|
goto error;
|
||||||
|
if (!set_misc_parameters (encoder, picture))
|
||||||
|
goto error;
|
||||||
|
if (!ensure_slices (encoder, picture))
|
||||||
|
goto error;
|
||||||
|
if (!gst_vaapi_enc_picture_encode (picture))
|
||||||
|
goto error;
|
||||||
|
if (picture->type != GST_VAAPI_PICTURE_TYPE_B) {
|
||||||
|
if (encoder->new_gop)
|
||||||
|
clear_references (encoder);
|
||||||
|
push_reference (encoder, reconstruct);
|
||||||
|
} else if (reconstruct)
|
||||||
|
gst_vaapi_encoder_release_surface (GST_VAAPI_ENCODER (encoder),
|
||||||
|
reconstruct);
|
||||||
|
|
||||||
|
return GST_VAAPI_ENCODER_STATUS_SUCCESS;
|
||||||
|
error:
|
||||||
|
if (reconstruct)
|
||||||
|
gst_vaapi_encoder_release_surface (GST_VAAPI_ENCODER (encoder),
|
||||||
|
reconstruct);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
static GstVaapiEncoderStatus
|
||||||
|
gst_vaapi_encoder_mpeg2_flush (GstVaapiEncoder * base)
|
||||||
|
{
|
||||||
|
GstVaapiEncoderMpeg2 *encoder = GST_VAAPI_ENCODER_MPEG2_CAST (base);
|
||||||
|
GstVaapiEncPicture *pic;
|
||||||
|
|
||||||
|
while (!g_queue_is_empty (&encoder->b_frames)) {
|
||||||
|
pic = (GstVaapiEncPicture *) g_queue_pop_head (&encoder->b_frames);
|
||||||
|
gst_vaapi_enc_picture_unref (pic);
|
||||||
|
}
|
||||||
|
g_queue_clear (&encoder->b_frames);
|
||||||
|
|
||||||
|
return GST_VAAPI_ENCODER_STATUS_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
static GstVaapiEncoderStatus
|
||||||
|
gst_vaapi_encoder_mpeg2_reordering (GstVaapiEncoder * base,
|
||||||
|
GstVideoCodecFrame * frame, gboolean flush, GstVaapiEncPicture ** output)
|
||||||
|
{
|
||||||
|
GstVaapiEncoderMpeg2 *encoder = GST_VAAPI_ENCODER_MPEG2 (base);
|
||||||
|
GstVaapiEncPicture *picture = NULL;
|
||||||
|
GstVaapiEncoderStatus status = GST_VAAPI_ENCODER_STATUS_SUCCESS;
|
||||||
|
|
||||||
|
|
||||||
|
if (!frame) {
|
||||||
|
if (g_queue_is_empty (&encoder->b_frames) && encoder->dump_frames) {
|
||||||
|
push_reference (encoder, NULL);
|
||||||
|
encoder->dump_frames = FALSE;
|
||||||
|
}
|
||||||
|
if (!encoder->dump_frames) {
|
||||||
|
return GST_VAAPI_ENCODER_STATUS_FRAME_NOT_READY;
|
||||||
|
}
|
||||||
|
picture = g_queue_pop_head (&encoder->b_frames);
|
||||||
|
g_assert (picture);
|
||||||
|
goto end;
|
||||||
|
}
|
||||||
|
|
||||||
|
picture = GST_VAAPI_ENC_PICTURE_NEW (MPEG2, encoder, frame);
|
||||||
|
if (!picture) {
|
||||||
|
GST_WARNING ("create MPEG2 picture failed, frame timestamp:%"
|
||||||
|
GST_TIME_FORMAT, GST_TIME_ARGS (frame->pts));
|
||||||
|
return GST_VAAPI_ENCODER_STATUS_OBJECT_ERR;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (encoder->frame_num >= encoder->intra_period) {
|
||||||
|
encoder->frame_num = 0;
|
||||||
|
clear_references (encoder);
|
||||||
|
}
|
||||||
|
if (encoder->frame_num == 0) {
|
||||||
|
picture->type = GST_VAAPI_PICTURE_TYPE_I;
|
||||||
|
GST_VIDEO_CODEC_FRAME_SET_SYNC_POINT (frame);
|
||||||
|
encoder->new_gop = TRUE;
|
||||||
|
} else {
|
||||||
|
encoder->new_gop = FALSE;
|
||||||
|
if ((encoder->frame_num % (encoder->ip_period + 1)) == 0 ||
|
||||||
|
encoder->frame_num == encoder->intra_period - 1) {
|
||||||
|
picture->type = GST_VAAPI_PICTURE_TYPE_P;
|
||||||
|
encoder->dump_frames = TRUE;
|
||||||
|
} else {
|
||||||
|
picture->type = GST_VAAPI_PICTURE_TYPE_B;
|
||||||
|
status = GST_VAAPI_ENCODER_STATUS_FRAME_NOT_READY;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
picture->frame_num = encoder->frame_num++;
|
||||||
|
|
||||||
|
if (picture->type == GST_VAAPI_PICTURE_TYPE_B) {
|
||||||
|
g_queue_push_tail (&encoder->b_frames, picture);
|
||||||
|
picture = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
end:
|
||||||
|
*output = picture;
|
||||||
|
return status;
|
||||||
|
}
|
||||||
|
|
||||||
|
static GstVaapiProfile
|
||||||
|
to_vaapi_profile (guint32 profile)
|
||||||
|
{
|
||||||
|
GstVaapiProfile p;
|
||||||
|
switch (profile) {
|
||||||
|
case GST_ENCODER_MPEG2_PROFILE_SIMPLE:
|
||||||
|
p = GST_VAAPI_PROFILE_MPEG2_SIMPLE;
|
||||||
|
break;
|
||||||
|
case GST_ENCODER_MPEG2_PROFILE_MAIN:
|
||||||
|
p = GST_VAAPI_PROFILE_MPEG2_MAIN;
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
g_assert (0);
|
||||||
|
}
|
||||||
|
return p;
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
gst_vaapi_encoder_mpeg2_get_context_info (GstVaapiEncoder * base,
|
||||||
|
GstVaapiContextInfo * info)
|
||||||
|
{
|
||||||
|
GstVaapiEncoderMpeg2 *encoder = GST_VAAPI_ENCODER_MPEG2 (base);
|
||||||
|
|
||||||
|
g_return_val_if_fail (info, FALSE);
|
||||||
|
|
||||||
|
info->profile = to_vaapi_profile (encoder->profile);
|
||||||
|
info->entrypoint = GST_VAAPI_ENTRYPOINT_SLICE_ENCODE;
|
||||||
|
info->width = GST_VAAPI_ENCODER_WIDTH (encoder);
|
||||||
|
info->height = GST_VAAPI_ENCODER_HEIGHT (encoder);
|
||||||
|
info->ref_frames = 2;
|
||||||
|
info->rc_mode = GST_VAAPI_ENCODER_RATE_CONTROL (encoder);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
prepare_encoding (GstVaapiEncoderMpeg2 * encoder, GstCaps * caps)
|
||||||
|
{
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static GstCaps *
|
||||||
|
gst_vaapi_encoder_mpeg2_set_format (GstVaapiEncoder * base,
|
||||||
|
GstVideoCodecState * in_state, GstCaps * ref_caps)
|
||||||
|
{
|
||||||
|
GstVaapiEncoderMpeg2 *encoder;
|
||||||
|
GstCaps *result = NULL, *tmp;
|
||||||
|
encoder = GST_VAAPI_ENCODER_MPEG2 (base);
|
||||||
|
|
||||||
|
tmp = gst_caps_from_string ("video/mpeg");
|
||||||
|
gst_caps_set_simple (tmp,
|
||||||
|
"mpegversion", G_TYPE_INT, 2,
|
||||||
|
"systemstream", G_TYPE_BOOLEAN, FALSE,
|
||||||
|
"width", G_TYPE_INT, GST_VAAPI_ENCODER_WIDTH (encoder),
|
||||||
|
"height", G_TYPE_INT, GST_VAAPI_ENCODER_HEIGHT (encoder),
|
||||||
|
"framerate", GST_TYPE_FRACTION,
|
||||||
|
GST_VAAPI_ENCODER_FPS_N (encoder), GST_VAAPI_ENCODER_FPS_D (encoder),
|
||||||
|
NULL);
|
||||||
|
result = gst_caps_intersect (tmp, ref_caps);
|
||||||
|
gst_caps_unref (tmp);
|
||||||
|
|
||||||
|
result = gst_caps_fixate (result);
|
||||||
|
|
||||||
|
if (!ensure_public_attributes (encoder)) {
|
||||||
|
GST_WARNING ("encoder ensure public attributes failed ");
|
||||||
|
goto error;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!prepare_encoding (encoder, result)) {
|
||||||
|
GST_WARNING ("prepare encoding failed ");
|
||||||
|
goto error;
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
|
||||||
|
error:
|
||||||
|
gst_caps_unref (result);
|
||||||
|
return NULL;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
gst_vaapi_encoder_mpeg2_init (GstVaapiEncoder * base)
|
||||||
|
{
|
||||||
|
GstVaapiEncoderMpeg2 *encoder = GST_VAAPI_ENCODER_MPEG2 (base);
|
||||||
|
|
||||||
|
/* re-ordering */
|
||||||
|
g_queue_init (&encoder->b_frames);
|
||||||
|
encoder->dump_frames = FALSE;
|
||||||
|
|
||||||
|
encoder->forward = NULL;
|
||||||
|
encoder->backward = NULL;
|
||||||
|
|
||||||
|
encoder->frame_num = 0;
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
clear_ref (GstVaapiEncoderMpeg2 * encoder, GstVaapiSurfaceProxy ** ref)
|
||||||
|
{
|
||||||
|
if (*ref) {
|
||||||
|
gst_vaapi_encoder_release_surface (GST_VAAPI_ENCODER (encoder), *ref);
|
||||||
|
*ref = NULL;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
clear_references (GstVaapiEncoderMpeg2 * encoder)
|
||||||
|
{
|
||||||
|
clear_ref (encoder, &encoder->forward);
|
||||||
|
clear_ref (encoder, &encoder->backward);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
push_reference (GstVaapiEncoderMpeg2 * encoder, GstVaapiSurfaceProxy * ref)
|
||||||
|
{
|
||||||
|
if (encoder->backward) {
|
||||||
|
clear_ref (encoder, &encoder->forward);
|
||||||
|
encoder->forward = encoder->backward;
|
||||||
|
encoder->backward = NULL;
|
||||||
|
}
|
||||||
|
if (encoder->forward)
|
||||||
|
encoder->backward = ref;
|
||||||
|
else
|
||||||
|
encoder->forward = ref;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
gst_vaapi_encoder_mpeg2_destroy (GstVaapiEncoder * base)
|
||||||
|
{
|
||||||
|
/*free private buffers */
|
||||||
|
GstVaapiEncoderMpeg2 *encoder = GST_VAAPI_ENCODER_MPEG2 (base);
|
||||||
|
GstVaapiEncPicture *pic;
|
||||||
|
|
||||||
|
clear_references (encoder);
|
||||||
|
|
||||||
|
while (!g_queue_is_empty (&encoder->b_frames)) {
|
||||||
|
pic = (GstVaapiEncPicture *) g_queue_pop_head (&encoder->b_frames);
|
||||||
|
gst_vaapi_enc_picture_unref (pic);
|
||||||
|
}
|
||||||
|
g_queue_clear (&encoder->b_frames);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
gst_vaapi_encoder_mpeg2_class_init (GstVaapiEncoderMpeg2Class * klass)
|
||||||
|
{
|
||||||
|
GstVaapiMiniObjectClass *const object_class =
|
||||||
|
GST_VAAPI_MINI_OBJECT_CLASS (klass);
|
||||||
|
GstVaapiEncoderClass *const encoder_class = GST_VAAPI_ENCODER_CLASS (klass);
|
||||||
|
|
||||||
|
gst_vaapi_encoder_class_init (encoder_class);
|
||||||
|
|
||||||
|
object_class->size = sizeof (GstVaapiEncoderMpeg2);
|
||||||
|
|
||||||
|
encoder_class->init = gst_vaapi_encoder_mpeg2_init;
|
||||||
|
encoder_class->destroy = gst_vaapi_encoder_mpeg2_destroy;
|
||||||
|
encoder_class->set_format = gst_vaapi_encoder_mpeg2_set_format;
|
||||||
|
encoder_class->get_context_info = gst_vaapi_encoder_mpeg2_get_context_info;
|
||||||
|
encoder_class->reordering = gst_vaapi_encoder_mpeg2_reordering;
|
||||||
|
encoder_class->encode = gst_vaapi_encoder_mpeg2_encode;
|
||||||
|
encoder_class->flush = gst_vaapi_encoder_mpeg2_flush;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline const GstVaapiEncoderClass *
|
||||||
|
gst_vaapi_encoder_mpeg2_class ()
|
||||||
|
{
|
||||||
|
static GstVaapiEncoderMpeg2Class g_class;
|
||||||
|
static gsize g_class_init = FALSE;
|
||||||
|
|
||||||
|
if (g_once_init_enter (&g_class_init)) {
|
||||||
|
gst_vaapi_encoder_mpeg2_class_init (&g_class);
|
||||||
|
g_once_init_leave (&g_class_init, TRUE);
|
||||||
|
}
|
||||||
|
return GST_VAAPI_ENCODER_CLASS (&g_class);
|
||||||
|
}
|
||||||
|
|
||||||
|
GstVaapiEncoder *
|
||||||
|
gst_vaapi_encoder_mpeg2_new (GstVaapiDisplay * display)
|
||||||
|
{
|
||||||
|
return gst_vaapi_encoder_new (gst_vaapi_encoder_mpeg2_class (), display);
|
||||||
|
}
|
||||||
|
|
||||||
|
static struct
|
||||||
|
{
|
||||||
|
int code;
|
||||||
|
float value;
|
||||||
|
} frame_rate_tab[] = {
|
||||||
|
{ 1, 23.976 },
|
||||||
|
{ 2, 24.0 },
|
||||||
|
{ 3, 25.0 },
|
||||||
|
{ 4, 29.97 },
|
||||||
|
{ 5, 30 },
|
||||||
|
{ 6, 50 },
|
||||||
|
{ 7, 59.94 },
|
||||||
|
{ 8, 60 }
|
||||||
|
};
|
||||||
|
|
||||||
|
static int
|
||||||
|
find_frame_rate_code (const VAEncSequenceParameterBufferMPEG2 * seq_param)
|
||||||
|
{
|
||||||
|
unsigned int delta = -1;
|
||||||
|
int code = 1, i;
|
||||||
|
float frame_rate_value = seq_param->frame_rate *
|
||||||
|
(seq_param->sequence_extension.bits.frame_rate_extension_d + 1) /
|
||||||
|
(seq_param->sequence_extension.bits.frame_rate_extension_n + 1);
|
||||||
|
|
||||||
|
for (i = 0; i < sizeof (frame_rate_tab) / sizeof (frame_rate_tab[0]); i++) {
|
||||||
|
|
||||||
|
if (abs (1000 * frame_rate_tab[i].value - 1000 * frame_rate_value) < delta) {
|
||||||
|
code = frame_rate_tab[i].code;
|
||||||
|
delta = abs (1000 * frame_rate_tab[i].value - 1000 * frame_rate_value);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
gst_bit_writer_write_sps (GstBitWriter * bitwriter,
|
||||||
|
VAEncSequenceParameterBufferMPEG2 * seq, GstVaapiEncoderMpeg2 * encoder)
|
||||||
|
{
|
||||||
|
int frame_rate_code = find_frame_rate_code (seq);
|
||||||
|
|
||||||
|
if (encoder->new_gop) {
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, START_CODE_SEQ, 32);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, seq->picture_width, 12);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, seq->picture_height, 12);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, seq->aspect_ratio_information,
|
||||||
|
4);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, frame_rate_code, 4); /* frame_rate_code */
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, (seq->bits_per_second + 399) / 400, 18); /* the low 18 bits of bit_rate */
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, 1, 1); /* marker_bit */
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, seq->vbv_buffer_size, 10);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, 0, 1); /* constraint_parameter_flag, always 0 for MPEG-2 */
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, 0, 1); /* load_intra_quantiser_matrix */
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, 0, 1); /* load_non_intra_quantiser_matrix */
|
||||||
|
|
||||||
|
gst_bit_writer_align_bytes (bitwriter, 0);
|
||||||
|
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, START_CODE_EXT, 32);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, 1, 4); /* sequence_extension id */
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
seq->sequence_extension.bits.profile_and_level_indication, 8);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
seq->sequence_extension.bits.progressive_sequence, 1);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
seq->sequence_extension.bits.chroma_format, 2);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, seq->picture_width >> 12, 2);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, seq->picture_height >> 12, 2);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, ((seq->bits_per_second + 399) / 400) >> 18, 12); /* bit_rate_extension */
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, 1, 1); /* marker_bit */
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, seq->vbv_buffer_size >> 10, 8);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
seq->sequence_extension.bits.low_delay, 1);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
seq->sequence_extension.bits.frame_rate_extension_n, 2);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
seq->sequence_extension.bits.frame_rate_extension_d, 5);
|
||||||
|
|
||||||
|
gst_bit_writer_align_bytes (bitwriter, 0);
|
||||||
|
|
||||||
|
/* gop header */
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, START_CODE_GOP, 32);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, seq->gop_header.bits.time_code,
|
||||||
|
25);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, seq->gop_header.bits.closed_gop,
|
||||||
|
1);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, seq->gop_header.bits.broken_link,
|
||||||
|
1);
|
||||||
|
|
||||||
|
gst_bit_writer_align_bytes (bitwriter, 0);
|
||||||
|
}
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
static gboolean
|
||||||
|
gst_bit_writer_write_pps (GstBitWriter * bitwriter,
|
||||||
|
VAEncPictureParameterBufferMPEG2 * pic)
|
||||||
|
{
|
||||||
|
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, START_CODE_PICUTRE, 32);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, pic->temporal_reference, 10);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
pic->picture_type == VAEncPictureTypeIntra ? 1 :
|
||||||
|
pic->picture_type == VAEncPictureTypePredictive ? 2 : 3, 3);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, pic->vbv_delay, 16);
|
||||||
|
|
||||||
|
if (pic->picture_type == VAEncPictureTypePredictive ||
|
||||||
|
pic->picture_type == VAEncPictureTypeBidirectional) {
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, 0, 1); /* full_pel_forward_vector, always 0 for MPEG-2 */
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, 7, 3); /* forward_f_code, always 7 for MPEG-2 */
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pic->picture_type == VAEncPictureTypeBidirectional) {
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, 0, 1); /* full_pel_backward_vector, always 0 for MPEG-2 */
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, 7, 3); /* backward_f_code, always 7 for MPEG-2 */
|
||||||
|
}
|
||||||
|
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, 0, 1); /* extra_bit_picture, 0 */
|
||||||
|
|
||||||
|
gst_bit_writer_align_bytes (bitwriter, 0);
|
||||||
|
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, START_CODE_EXT, 32);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, 8, 4); /* Picture Coding Extension ID: 8 */
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, pic->f_code[0][0], 4);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, pic->f_code[0][1], 4);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, pic->f_code[1][0], 4);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, pic->f_code[1][1], 4);
|
||||||
|
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
pic->picture_coding_extension.bits.intra_dc_precision, 2);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
pic->picture_coding_extension.bits.picture_structure, 2);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
pic->picture_coding_extension.bits.top_field_first, 1);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
pic->picture_coding_extension.bits.frame_pred_frame_dct, 1);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
pic->picture_coding_extension.bits.concealment_motion_vectors, 1);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
pic->picture_coding_extension.bits.q_scale_type, 1);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
pic->picture_coding_extension.bits.intra_vlc_format, 1);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
pic->picture_coding_extension.bits.alternate_scan, 1);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
pic->picture_coding_extension.bits.repeat_first_field, 1);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter, 1, 1); /* always chroma 420 */
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
pic->picture_coding_extension.bits.progressive_frame, 1);
|
||||||
|
gst_bit_writer_put_bits_uint32 (bitwriter,
|
||||||
|
pic->picture_coding_extension.bits.composite_display_flag, 1);
|
||||||
|
|
||||||
|
gst_bit_writer_align_bytes (bitwriter, 0);
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
36
gst-libs/gst/vaapi/gstvaapiencoder_mpeg2.h
Normal file
36
gst-libs/gst/vaapi/gstvaapiencoder_mpeg2.h
Normal file
|
|
@ -0,0 +1,36 @@
|
||||||
|
/*
|
||||||
|
* gstvaapiencoder_mpeg2.h - MPEG-2 encoder
|
||||||
|
*
|
||||||
|
* Copyright (C) 2011-2013 Intel Corporation
|
||||||
|
*
|
||||||
|
* This library is free software; you can redistribute it and/or
|
||||||
|
* modify it under the terms of the GNU Lesser General Public License
|
||||||
|
* as published by the Free Software Foundation; either version 2.1
|
||||||
|
* of the License, or (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This library is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||||
|
* Lesser General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU Lesser General Public
|
||||||
|
* License along with this library; if not, write to the Free
|
||||||
|
* Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
|
||||||
|
* Boston, MA 02110-1301 USA
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef GST_VAAPI_ENCODER_MPEG2_H
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_H
|
||||||
|
|
||||||
|
#include <gst/vaapi/gstvaapiencoder.h>
|
||||||
|
|
||||||
|
G_BEGIN_DECLS
|
||||||
|
|
||||||
|
typedef struct _GstVaapiEncoderMpeg2 GstVaapiEncoderMpeg2;
|
||||||
|
|
||||||
|
GstVaapiEncoder *
|
||||||
|
gst_vaapi_encoder_mpeg2_new (GstVaapiDisplay * display);
|
||||||
|
|
||||||
|
G_END_DECLS
|
||||||
|
|
||||||
|
#endif /* GST_VAAPI_ENCODER_MPEG2_H */
|
||||||
116
gst-libs/gst/vaapi/gstvaapiencoder_mpeg2_priv.h
Normal file
116
gst-libs/gst/vaapi/gstvaapiencoder_mpeg2_priv.h
Normal file
|
|
@ -0,0 +1,116 @@
|
||||||
|
/*
|
||||||
|
* gstvaapiencoder_mpeg2_priv.h - MPEG-2 encoder (private definitions)
|
||||||
|
*
|
||||||
|
* Copyright (C) 2013 Intel Corporation
|
||||||
|
*
|
||||||
|
* This library is free software; you can redistribute it and/or
|
||||||
|
* modify it under the terms of the GNU Lesser General Public License
|
||||||
|
* as published by the Free Software Foundation; either version 2.1
|
||||||
|
* of the License, or (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This library is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||||
|
* Lesser General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU Lesser General Public
|
||||||
|
* License along with this library; if not, write to the Free
|
||||||
|
* Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
|
||||||
|
* Boston, MA 02110-1301 USA
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef GST_VAAPI_ENCODER_MPEG2_PRIV_H
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_PRIV_H
|
||||||
|
|
||||||
|
#include <glib.h>
|
||||||
|
#include <gst/vaapi/gstvaapiencoder.h>
|
||||||
|
#include <gst/vaapi/gstvaapiencoder_priv.h>
|
||||||
|
|
||||||
|
G_BEGIN_DECLS
|
||||||
|
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2(encoder) \
|
||||||
|
((GstVaapiEncoderMpeg2 *)(encoder))
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_CAST(encoder) \
|
||||||
|
((GstVaapiEncoderMpeg2 *)(encoder))
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_CLASS(klass) \
|
||||||
|
((GstVaapiEncoderMpeg2Class *)(klass))
|
||||||
|
#define GST_IS_VAAPI_ENCODER_MPEG2_CLASS(klass) \
|
||||||
|
((klass) != NULL)
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_GET_CLASS(obj) \
|
||||||
|
GST_VAAPI_ENCODER_MPEG2_CLASS(GST_VAAPI_MINI_OBJECT_GET_CLASS(obj))
|
||||||
|
|
||||||
|
typedef struct _GstVaapiEncoderMpeg2Class GstVaapiEncoderMpeg2Class;
|
||||||
|
|
||||||
|
typedef enum
|
||||||
|
{
|
||||||
|
GST_ENCODER_MPEG2_PROFILE_SIMPLE,
|
||||||
|
GST_ENCODER_MPEG2_PROFILE_MAIN,
|
||||||
|
} GstEncoderMpeg2Level;
|
||||||
|
|
||||||
|
typedef enum
|
||||||
|
{
|
||||||
|
GST_VAAPI_ENCODER_MPEG2_LEVEL_LOW,
|
||||||
|
GST_VAAPI_ENCODER_MPEG2_LEVEL_MAIN,
|
||||||
|
GST_VAAPI_ENCODER_MPEG2_LEVEL_HIGH
|
||||||
|
} GstVaapiEncoderMpeg2Level;
|
||||||
|
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_DEFAULT_PROFILE GST_ENCODER_MPEG2_PROFILE_MAIN
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_DEFAULT_LEVEL GST_VAAPI_ENCODER_MPEG2_LEVEL_HIGH
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_DEFAULT_RATE_CONTROL GST_VAAPI_RATECONTROL_CQP
|
||||||
|
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_MIN_CQP 2
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_MAX_CQP 62
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_DEFAULT_CQP 8
|
||||||
|
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_MAX_GOP_SIZE 512
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_DEFAULT_GOP_SIZE 30
|
||||||
|
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_MAX_MAX_BFRAMES 16
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_DEFAULT_MAX_BFRAMES 2
|
||||||
|
|
||||||
|
#define GST_VAAPI_ENCODER_MPEG2_MAX_BITRATE 100*1024
|
||||||
|
|
||||||
|
#define START_CODE_PICUTRE 0x00000100
|
||||||
|
#define START_CODE_SLICE 0x00000101
|
||||||
|
#define START_CODE_USER 0x000001B2
|
||||||
|
#define START_CODE_SEQ 0x000001B3
|
||||||
|
#define START_CODE_EXT 0x000001B5
|
||||||
|
#define START_CODE_GOP 0x000001B8
|
||||||
|
|
||||||
|
#define CHROMA_FORMAT_RESERVED 0
|
||||||
|
#define CHROMA_FORMAT_420 1
|
||||||
|
#define CHROMA_FORMAT_422 2
|
||||||
|
#define CHROMA_FORMAT_444 3
|
||||||
|
|
||||||
|
struct _GstVaapiEncoderMpeg2
|
||||||
|
{
|
||||||
|
GstVaapiEncoder parent;
|
||||||
|
|
||||||
|
/* public */
|
||||||
|
guint32 profile;
|
||||||
|
guint32 level;
|
||||||
|
guint32 bitrate; /*kbps */
|
||||||
|
guint32 cqp;
|
||||||
|
guint32 intra_period;
|
||||||
|
guint32 ip_period;
|
||||||
|
|
||||||
|
/* re-ordering */
|
||||||
|
GQueue b_frames;
|
||||||
|
gboolean dump_frames;
|
||||||
|
gboolean new_gop;
|
||||||
|
|
||||||
|
/* reference list */
|
||||||
|
GstVaapiSurfaceProxy *forward;
|
||||||
|
GstVaapiSurfaceProxy *backward;
|
||||||
|
guint32 frame_num; /* same value picture header, but it's not mod by 1024 */
|
||||||
|
};
|
||||||
|
|
||||||
|
struct _GstVaapiEncoderMpeg2Class
|
||||||
|
{
|
||||||
|
/*< private > */
|
||||||
|
GstVaapiEncoderClass parent_class;
|
||||||
|
};
|
||||||
|
|
||||||
|
G_END_DECLS
|
||||||
|
|
||||||
|
#endif /* GST_VAAPI_ENCODER_MPEG2_PRIV_H */
|
||||||
Loading…
Reference in a new issue