33 #include <schroedinger/schro.h>
34 #include <schroedinger/schrodebug.h>
35 #include <schroedinger/schrovideoformat.h>
90 for (idx = 0; idx < num_formats; ++idx) {
93 p_schro_params->
format->chroma_format =
100 "This codec currently only supports planar YUV 4:2:0, 4:2:2"
101 " and 4:4:4 formats.\n");
109 SchroVideoFormatEnum preset;
115 p_schro_params->
encoder = schro_encoder_new();
117 if (!p_schro_params->
encoder) {
119 "Unrecoverable Error: schro_encoder_new failed. ");
126 schro_encoder_get_video_format(p_schro_params->
encoder);
127 schro_video_format_set_std_video_format(p_schro_params->
format, preset);
135 p_schro_params->
format->colour_primaries = SCHRO_COLOUR_PRIMARY_HDTV;
137 p_schro_params->
format->colour_primaries = SCHRO_COLOUR_PRIMARY_SDTV_625;
139 p_schro_params->
format->colour_primaries = SCHRO_COLOUR_PRIMARY_SDTV_525;
143 p_schro_params->
format->colour_matrix = SCHRO_COLOUR_MATRIX_HDTV;
145 p_schro_params->
format->colour_matrix = SCHRO_COLOUR_MATRIX_SDTV;
149 p_schro_params->
format->transfer_function = SCHRO_TRANSFER_CHAR_TV_GAMMA;
155 "This codec currently supports only planar YUV 4:2:0, 4:2:2"
156 " and 4:4:4 formats.\n");
170 schro_encoder_setting_set_double(p_schro_params->
encoder,
172 SCHRO_ENCODER_GOP_INTRA_ONLY);
175 schro_encoder_setting_set_double(p_schro_params->
encoder,
176 "enable_noarith", 1);
178 schro_encoder_setting_set_double(p_schro_params->
encoder,
179 "au_distance", avccontext->
gop_size);
181 p_schro_params->
dts = -1;
188 schro_encoder_setting_set_double(p_schro_params->
encoder,
190 SCHRO_ENCODER_RATE_CONTROL_LOSSLESS);
193 schro_encoder_setting_set_double(p_schro_params->
encoder,
195 SCHRO_ENCODER_RATE_CONTROL_CONSTANT_QUALITY);
200 schro_encoder_setting_set_double(p_schro_params->
encoder,
204 schro_encoder_setting_set_double(p_schro_params->
encoder,
206 SCHRO_ENCODER_RATE_CONTROL_CONSTANT_BITRATE);
208 schro_encoder_setting_set_double(p_schro_params->
encoder,
217 schro_encoder_setting_set_double(p_schro_params->
encoder,
218 "interlaced_coding", 1);
220 schro_encoder_setting_set_double(p_schro_params->
encoder,
"open_gop",
225 schro_video_format_set_std_signal_range(p_schro_params->
format,
226 SCHRO_SIGNAL_RANGE_8BIT_VIDEO);
229 schro_encoder_set_video_format(p_schro_params->
encoder,
233 schro_debug_set_level(avccontext->
debug);
235 schro_encoder_start(p_schro_params->
encoder);
246 SchroFrame *in_frame;
255 in_frame->components[0].data,
270 const AVFrame *frame,
int *got_packet)
274 SchroEncoder *encoder = p_schro_params->
encoder;
277 SchroBuffer *enc_buf;
278 int presentation_frame;
280 int last_frame_in_sequence = 0;
286 schro_encoder_end_of_stream(encoder);
294 schro_encoder_push_frame(encoder, in_frame);
302 SchroStateEnum
state;
303 state = schro_encoder_wait(encoder);
305 case SCHRO_STATE_HAVE_BUFFER:
306 case SCHRO_STATE_END_OF_STREAM:
307 enc_buf = schro_encoder_pull(encoder, &presentation_frame);
308 assert(enc_buf->length > 0);
309 assert(enc_buf->length <= buf_size);
310 parse_code = enc_buf->data[4];
320 enc_buf->data, enc_buf->length);
324 if (state == SCHRO_STATE_END_OF_STREAM) {
329 if (!SCHRO_PARSE_CODE_IS_PICTURE(parse_code)) {
330 schro_buffer_unref(enc_buf);
339 if (SCHRO_PARSE_CODE_IS_INTRA(parse_code) &&
340 SCHRO_PARSE_CODE_IS_REFERENCE(parse_code))
352 schro_buffer_unref(enc_buf);
356 case SCHRO_STATE_NEED_FRAME:
360 case SCHRO_STATE_AGAIN:
373 last_frame_in_sequence = 1;
380 pkt_size = p_frame_output->
size;
381 if (last_frame_in_sequence && p_schro_params->
enc_buf_size > 0)
384 av_log(avccontext,
AV_LOG_ERROR,
"Error getting output packet of size %d.\n", pkt_size);
395 pkt->
dts = p_schro_params->
dts++;
396 enc_size = p_frame_output->
size;
400 if (last_frame_in_sequence && p_schro_params->
enc_buf_size > 0) {
401 memcpy(pkt->
data + enc_size, p_schro_params->
enc_buf,
424 schro_encoder_free(p_schro_params->
encoder);
443 .
name =
"libschroedinger",