|
|
|
@ -40,8 +40,7 @@ |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/** libschroedinger encoder private data */ |
|
|
|
|
typedef struct FfmpegSchroEncoderParams |
|
|
|
|
{ |
|
|
|
|
typedef struct FfmpegSchroEncoderParams { |
|
|
|
|
/** Schroedinger video format */ |
|
|
|
|
SchroVideoFormat *format; |
|
|
|
|
|
|
|
|
@ -86,16 +85,16 @@ static int SetSchroChromaFormat(AVCodecContext *avccontext) |
|
|
|
|
|
|
|
|
|
for (idx = 0; idx < num_formats; ++idx) { |
|
|
|
|
if (ffmpeg_schro_pixel_format_map[idx].ff_pix_fmt == |
|
|
|
|
avccontext->pix_fmt) { |
|
|
|
|
avccontext->pix_fmt) { |
|
|
|
|
p_schro_params->format->chroma_format = |
|
|
|
|
ffmpeg_schro_pixel_format_map[idx].schro_pix_fmt; |
|
|
|
|
return 0; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
av_log (avccontext, AV_LOG_ERROR, |
|
|
|
|
"This codec currently only supports planar YUV 4:2:0, 4:2:2" |
|
|
|
|
" and 4:4:4 formats.\n"); |
|
|
|
|
av_log(avccontext, AV_LOG_ERROR, |
|
|
|
|
"This codec currently only supports planar YUV 4:2:0, 4:2:2" |
|
|
|
|
" and 4:4:4 formats.\n"); |
|
|
|
|
|
|
|
|
|
return -1; |
|
|
|
|
} |
|
|
|
@ -121,8 +120,8 @@ static int libschroedinger_encode_init(AVCodecContext *avccontext) |
|
|
|
|
preset = ff_get_schro_video_format_preset(avccontext); |
|
|
|
|
p_schro_params->format = |
|
|
|
|
schro_encoder_get_video_format(p_schro_params->encoder); |
|
|
|
|
schro_video_format_set_std_video_format (p_schro_params->format, preset); |
|
|
|
|
p_schro_params->format->width = avccontext->width; |
|
|
|
|
schro_video_format_set_std_video_format(p_schro_params->format, preset); |
|
|
|
|
p_schro_params->format->width = avccontext->width; |
|
|
|
|
p_schro_params->format->height = avccontext->height; |
|
|
|
|
|
|
|
|
|
if (SetSchroChromaFormat(avccontext) == -1) |
|
|
|
@ -130,9 +129,9 @@ static int libschroedinger_encode_init(AVCodecContext *avccontext) |
|
|
|
|
|
|
|
|
|
if (ff_get_schro_frame_format(p_schro_params->format->chroma_format, |
|
|
|
|
&p_schro_params->frame_format) == -1) { |
|
|
|
|
av_log (avccontext, AV_LOG_ERROR, |
|
|
|
|
"This codec currently supports only planar YUV 4:2:0, 4:2:2" |
|
|
|
|
" and 4:4:4 formats.\n"); |
|
|
|
|
av_log(avccontext, AV_LOG_ERROR, |
|
|
|
|
"This codec currently supports only planar YUV 4:2:0, 4:2:2" |
|
|
|
|
" and 4:4:4 formats.\n"); |
|
|
|
|
return -1; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -146,18 +145,17 @@ static int libschroedinger_encode_init(AVCodecContext *avccontext) |
|
|
|
|
avccontext->coded_frame = &p_schro_params->picture; |
|
|
|
|
|
|
|
|
|
if (!avccontext->gop_size) { |
|
|
|
|
schro_encoder_setting_set_double (p_schro_params->encoder, |
|
|
|
|
"gop_structure", |
|
|
|
|
SCHRO_ENCODER_GOP_INTRA_ONLY); |
|
|
|
|
schro_encoder_setting_set_double(p_schro_params->encoder, |
|
|
|
|
"gop_structure", |
|
|
|
|
SCHRO_ENCODER_GOP_INTRA_ONLY); |
|
|
|
|
|
|
|
|
|
if (avccontext->coder_type == FF_CODER_TYPE_VLC) |
|
|
|
|
schro_encoder_setting_set_double (p_schro_params->encoder, |
|
|
|
|
"enable_noarith", 1); |
|
|
|
|
} |
|
|
|
|
else { |
|
|
|
|
schro_encoder_setting_set_double (p_schro_params->encoder, |
|
|
|
|
"gop_structure", |
|
|
|
|
SCHRO_ENCODER_GOP_BIREF); |
|
|
|
|
schro_encoder_setting_set_double(p_schro_params->encoder, |
|
|
|
|
"enable_noarith", 1); |
|
|
|
|
} else { |
|
|
|
|
schro_encoder_setting_set_double(p_schro_params->encoder, |
|
|
|
|
"gop_structure", |
|
|
|
|
SCHRO_ENCODER_GOP_BIREF); |
|
|
|
|
avccontext->has_b_frames = 1; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -165,39 +163,38 @@ static int libschroedinger_encode_init(AVCodecContext *avccontext) |
|
|
|
|
if (avccontext->flags & CODEC_FLAG_QSCALE) { |
|
|
|
|
if (!avccontext->global_quality) { |
|
|
|
|
/* lossless coding */ |
|
|
|
|
schro_encoder_setting_set_double (p_schro_params->encoder, |
|
|
|
|
"rate_control", |
|
|
|
|
SCHRO_ENCODER_RATE_CONTROL_LOSSLESS); |
|
|
|
|
schro_encoder_setting_set_double(p_schro_params->encoder, |
|
|
|
|
"rate_control", |
|
|
|
|
SCHRO_ENCODER_RATE_CONTROL_LOSSLESS); |
|
|
|
|
} else { |
|
|
|
|
int noise_threshold; |
|
|
|
|
schro_encoder_setting_set_double (p_schro_params->encoder, |
|
|
|
|
"rate_control", |
|
|
|
|
SCHRO_ENCODER_RATE_CONTROL_CONSTANT_NOISE_THRESHOLD); |
|
|
|
|
schro_encoder_setting_set_double(p_schro_params->encoder, |
|
|
|
|
"rate_control", |
|
|
|
|
SCHRO_ENCODER_RATE_CONTROL_CONSTANT_NOISE_THRESHOLD); |
|
|
|
|
|
|
|
|
|
noise_threshold = avccontext->global_quality/FF_QP2LAMBDA; |
|
|
|
|
noise_threshold = avccontext->global_quality / FF_QP2LAMBDA; |
|
|
|
|
if (noise_threshold > 100) |
|
|
|
|
noise_threshold = 100; |
|
|
|
|
schro_encoder_setting_set_double (p_schro_params->encoder, |
|
|
|
|
"noise_threshold", |
|
|
|
|
noise_threshold); |
|
|
|
|
schro_encoder_setting_set_double(p_schro_params->encoder, |
|
|
|
|
"noise_threshold", |
|
|
|
|
noise_threshold); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
else { |
|
|
|
|
schro_encoder_setting_set_double ( p_schro_params->encoder, |
|
|
|
|
"rate_control", |
|
|
|
|
SCHRO_ENCODER_RATE_CONTROL_CONSTANT_BITRATE); |
|
|
|
|
} else { |
|
|
|
|
schro_encoder_setting_set_double(p_schro_params->encoder, |
|
|
|
|
"rate_control", |
|
|
|
|
SCHRO_ENCODER_RATE_CONTROL_CONSTANT_BITRATE); |
|
|
|
|
|
|
|
|
|
schro_encoder_setting_set_double (p_schro_params->encoder, |
|
|
|
|
"bitrate", |
|
|
|
|
avccontext->bit_rate); |
|
|
|
|
schro_encoder_setting_set_double(p_schro_params->encoder, |
|
|
|
|
"bitrate", |
|
|
|
|
avccontext->bit_rate); |
|
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (avccontext->flags & CODEC_FLAG_INTERLACED_ME) |
|
|
|
|
/* All material can be coded as interlaced or progressive
|
|
|
|
|
irrespective of the type of source material. */ |
|
|
|
|
schro_encoder_setting_set_double (p_schro_params->encoder, |
|
|
|
|
"interlaced_coding", 1); |
|
|
|
|
schro_encoder_setting_set_double(p_schro_params->encoder, |
|
|
|
|
"interlaced_coding", 1); |
|
|
|
|
|
|
|
|
|
/* FIXME: Signal range hardcoded to 8-bit data until both libschroedinger
|
|
|
|
|
* and libdirac support other bit-depth data. */ |
|
|
|
@ -209,32 +206,32 @@ static int libschroedinger_encode_init(AVCodecContext *avccontext) |
|
|
|
|
p_schro_params->format); |
|
|
|
|
|
|
|
|
|
/* Set the debug level. */ |
|
|
|
|
schro_debug_set_level (avccontext->debug); |
|
|
|
|
schro_debug_set_level(avccontext->debug); |
|
|
|
|
|
|
|
|
|
schro_encoder_start (p_schro_params->encoder); |
|
|
|
|
schro_encoder_start(p_schro_params->encoder); |
|
|
|
|
|
|
|
|
|
/* Initialize the encoded frame queue. */ |
|
|
|
|
ff_dirac_schro_queue_init (&p_schro_params->enc_frame_queue); |
|
|
|
|
return 0 ; |
|
|
|
|
ff_dirac_schro_queue_init(&p_schro_params->enc_frame_queue); |
|
|
|
|
return 0; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
static SchroFrame *libschroedinger_frame_from_data (AVCodecContext *avccontext, |
|
|
|
|
void *in_data) |
|
|
|
|
static SchroFrame *libschroedinger_frame_from_data(AVCodecContext *avccontext, |
|
|
|
|
void *in_data) |
|
|
|
|
{ |
|
|
|
|
FfmpegSchroEncoderParams* p_schro_params = avccontext->priv_data; |
|
|
|
|
SchroFrame *in_frame; |
|
|
|
|
/* Input line size may differ from what the codec supports. Especially
|
|
|
|
|
* when transcoding from one format to another. So use avpicture_layout |
|
|
|
|
* to copy the frame. */ |
|
|
|
|
in_frame = schro_frame_new_and_alloc (NULL, |
|
|
|
|
p_schro_params->frame_format, |
|
|
|
|
p_schro_params->format->width, |
|
|
|
|
p_schro_params->format->height); |
|
|
|
|
in_frame = schro_frame_new_and_alloc(NULL, |
|
|
|
|
p_schro_params->frame_format, |
|
|
|
|
p_schro_params->format->width, |
|
|
|
|
p_schro_params->format->height); |
|
|
|
|
|
|
|
|
|
avpicture_layout ((AVPicture *)in_data, avccontext->pix_fmt, |
|
|
|
|
avccontext->width, avccontext->height, |
|
|
|
|
in_frame->components[0].data, |
|
|
|
|
p_schro_params->frame_size); |
|
|
|
|
avpicture_layout((AVPicture *)in_data, avccontext->pix_fmt, |
|
|
|
|
avccontext->width, avccontext->height, |
|
|
|
|
in_frame->components[0].data, |
|
|
|
|
p_schro_params->frame_size); |
|
|
|
|
|
|
|
|
|
return in_frame; |
|
|
|
|
} |
|
|
|
@ -243,7 +240,7 @@ static void SchroedingerFreeFrame(void *data) |
|
|
|
|
{ |
|
|
|
|
FfmpegDiracSchroEncodedFrame *enc_frame = data; |
|
|
|
|
|
|
|
|
|
av_freep (&(enc_frame->p_encbuf)); |
|
|
|
|
av_freep(&(enc_frame->p_encbuf)); |
|
|
|
|
av_free(enc_frame); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -269,8 +266,8 @@ static int libschroedinger_encode_frame(AVCodecContext *avccontext, |
|
|
|
|
} |
|
|
|
|
} else { |
|
|
|
|
/* Allocate frame data to schro input buffer. */ |
|
|
|
|
SchroFrame *in_frame = libschroedinger_frame_from_data (avccontext, |
|
|
|
|
data); |
|
|
|
|
SchroFrame *in_frame = libschroedinger_frame_from_data(avccontext, |
|
|
|
|
data); |
|
|
|
|
/* Load next frame. */ |
|
|
|
|
schro_encoder_push_frame(encoder, in_frame); |
|
|
|
|
} |
|
|
|
@ -280,28 +277,24 @@ static int libschroedinger_encode_frame(AVCodecContext *avccontext, |
|
|
|
|
|
|
|
|
|
/* Now check to see if we have any output from the encoder. */ |
|
|
|
|
while (go) { |
|
|
|
|
SchroStateEnum state; |
|
|
|
|
SchroStateEnum state; |
|
|
|
|
state = schro_encoder_wait(encoder); |
|
|
|
|
switch (state) |
|
|
|
|
{ |
|
|
|
|
switch (state) { |
|
|
|
|
case SCHRO_STATE_HAVE_BUFFER: |
|
|
|
|
case SCHRO_STATE_END_OF_STREAM: |
|
|
|
|
enc_buf = schro_encoder_pull (encoder, |
|
|
|
|
&presentation_frame); |
|
|
|
|
assert (enc_buf->length > 0); |
|
|
|
|
assert (enc_buf->length <= buf_size); |
|
|
|
|
enc_buf = schro_encoder_pull(encoder, &presentation_frame); |
|
|
|
|
assert(enc_buf->length > 0); |
|
|
|
|
assert(enc_buf->length <= buf_size); |
|
|
|
|
parse_code = enc_buf->data[4]; |
|
|
|
|
|
|
|
|
|
/* All non-frame data is prepended to actual frame data to
|
|
|
|
|
* be able to set the pts correctly. So we don't write data |
|
|
|
|
* to the frame output queue until we actually have a frame |
|
|
|
|
*/ |
|
|
|
|
p_schro_params->enc_buf = av_realloc ( |
|
|
|
|
p_schro_params->enc_buf, |
|
|
|
|
p_schro_params->enc_buf_size + enc_buf->length |
|
|
|
|
); |
|
|
|
|
p_schro_params->enc_buf = av_realloc(p_schro_params->enc_buf, |
|
|
|
|
p_schro_params->enc_buf_size + enc_buf->length); |
|
|
|
|
|
|
|
|
|
memcpy(p_schro_params->enc_buf+p_schro_params->enc_buf_size, |
|
|
|
|
memcpy(p_schro_params->enc_buf + p_schro_params->enc_buf_size, |
|
|
|
|
enc_buf->data, enc_buf->length); |
|
|
|
|
p_schro_params->enc_buf_size += enc_buf->length; |
|
|
|
|
|
|
|
|
@ -312,7 +305,7 @@ static int libschroedinger_encode_frame(AVCodecContext *avccontext, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (!SCHRO_PARSE_CODE_IS_PICTURE(parse_code)) { |
|
|
|
|
schro_buffer_unref (enc_buf); |
|
|
|
|
schro_buffer_unref(enc_buf); |
|
|
|
|
break; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -332,12 +325,12 @@ static int libschroedinger_encode_frame(AVCodecContext *avccontext, |
|
|
|
|
(enc_buf->data[15] << 8) + |
|
|
|
|
enc_buf->data[16]; |
|
|
|
|
|
|
|
|
|
ff_dirac_schro_queue_push_back (&p_schro_params->enc_frame_queue, |
|
|
|
|
p_frame_output); |
|
|
|
|
ff_dirac_schro_queue_push_back(&p_schro_params->enc_frame_queue, |
|
|
|
|
p_frame_output); |
|
|
|
|
p_schro_params->enc_buf_size = 0; |
|
|
|
|
p_schro_params->enc_buf = NULL; |
|
|
|
|
|
|
|
|
|
schro_buffer_unref (enc_buf); |
|
|
|
|
schro_buffer_unref(enc_buf); |
|
|
|
|
|
|
|
|
|
break; |
|
|
|
|
|
|
|
|
@ -360,8 +353,7 @@ static int libschroedinger_encode_frame(AVCodecContext *avccontext, |
|
|
|
|
p_schro_params->eos_pulled) |
|
|
|
|
last_frame_in_sequence = 1; |
|
|
|
|
|
|
|
|
|
p_frame_output = |
|
|
|
|
ff_dirac_schro_queue_pop (&p_schro_params->enc_frame_queue); |
|
|
|
|
p_frame_output = ff_dirac_schro_queue_pop(&p_schro_params->enc_frame_queue); |
|
|
|
|
|
|
|
|
|
if (!p_frame_output) |
|
|
|
|
return 0; |
|
|
|
@ -376,17 +368,16 @@ static int libschroedinger_encode_frame(AVCodecContext *avccontext, |
|
|
|
|
|
|
|
|
|
/* Append the end of sequence information to the last frame in the
|
|
|
|
|
* sequence. */ |
|
|
|
|
if (last_frame_in_sequence && p_schro_params->enc_buf_size > 0) |
|
|
|
|
{ |
|
|
|
|
memcpy (frame + enc_size, p_schro_params->enc_buf, |
|
|
|
|
p_schro_params->enc_buf_size); |
|
|
|
|
if (last_frame_in_sequence && p_schro_params->enc_buf_size > 0) { |
|
|
|
|
memcpy(frame + enc_size, p_schro_params->enc_buf, |
|
|
|
|
p_schro_params->enc_buf_size); |
|
|
|
|
enc_size += p_schro_params->enc_buf_size; |
|
|
|
|
av_freep (&p_schro_params->enc_buf); |
|
|
|
|
av_freep(&p_schro_params->enc_buf); |
|
|
|
|
p_schro_params->enc_buf_size = 0; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/* free frame */ |
|
|
|
|
SchroedingerFreeFrame (p_frame_output); |
|
|
|
|
SchroedingerFreeFrame(p_frame_output); |
|
|
|
|
|
|
|
|
|
return enc_size; |
|
|
|
|
} |
|
|
|
@ -397,12 +388,12 @@ static int libschroedinger_encode_close(AVCodecContext *avccontext) |
|
|
|
|
|
|
|
|
|
FfmpegSchroEncoderParams* p_schro_params = avccontext->priv_data; |
|
|
|
|
|
|
|
|
|
/* Close the encoder. */ |
|
|
|
|
/* Close the encoder. */ |
|
|
|
|
schro_encoder_free(p_schro_params->encoder); |
|
|
|
|
|
|
|
|
|
/* Free data in the output frame queue. */ |
|
|
|
|
ff_dirac_schro_queue_free (&p_schro_params->enc_frame_queue, |
|
|
|
|
SchroedingerFreeFrame); |
|
|
|
|
ff_dirac_schro_queue_free(&p_schro_params->enc_frame_queue, |
|
|
|
|
SchroedingerFreeFrame); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Free the encoder buffer. */ |
|
|
|
@ -412,7 +403,7 @@ static int libschroedinger_encode_close(AVCodecContext *avccontext) |
|
|
|
|
/* Free the video format structure. */ |
|
|
|
|
av_freep(&p_schro_params->format); |
|
|
|
|
|
|
|
|
|
return 0 ; |
|
|
|
|
return 0; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -424,7 +415,7 @@ AVCodec libschroedinger_encoder = { |
|
|
|
|
libschroedinger_encode_init, |
|
|
|
|
libschroedinger_encode_frame, |
|
|
|
|
libschroedinger_encode_close, |
|
|
|
|
.capabilities= CODEC_CAP_DELAY, |
|
|
|
|
.pix_fmts= (enum PixelFormat[]){PIX_FMT_YUV420P, PIX_FMT_YUV422P, PIX_FMT_YUV444P, PIX_FMT_NONE}, |
|
|
|
|
.long_name= NULL_IF_CONFIG_SMALL("libschroedinger Dirac 2.2"), |
|
|
|
|
.capabilities = CODEC_CAP_DELAY, |
|
|
|
|
.pix_fmts = (enum PixelFormat[]){PIX_FMT_YUV420P, PIX_FMT_YUV422P, PIX_FMT_YUV444P, PIX_FMT_NONE}, |
|
|
|
|
.long_name = NULL_IF_CONFIG_SMALL("libschroedinger Dirac 2.2"), |
|
|
|
|
}; |
|
|
|
|