1 /*****************************************************************************
2 * video.c: transcoding stream output module (video)
3 *****************************************************************************
4 * Copyright (C) 2003-2009 VLC authors and VideoLAN
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8 * Gildas Bazin <gbazin@videolan.org>
9 * Jean-Paul Saman <jpsaman #_at_# m2x dot nl>
10 * Antoine Cellerier <dionoea at videolan dot org>
12 * This program is free software; you can redistribute it and/or modify it
13 * under the terms of the GNU Lesser General Public License as published by
14 * the Free Software Foundation; either version 2.1 of the License, or
15 * (at your option) any later version.
17 * This program is distributed in the hope that it will be useful,
18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 * GNU Lesser General Public License for more details.
22 * You should have received a copy of the GNU Lesser General Public License
23 * along with this program; if not, write to the Free Software Foundation,
24 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
25 *****************************************************************************/
27 /*****************************************************************************
29 *****************************************************************************/
31 #include "transcode.h"
35 #include <vlc_modules.h>
37 #define ENC_FRAMERATE (25 * 1000)
38 #define ENC_FRAMERATE_BASE 1000
40 struct decoder_owner_sys_t
42 sout_stream_sys_t *p_sys;
45 static void video_del_buffer_decoder( decoder_t *p_decoder, picture_t *p_pic )
47 VLC_UNUSED(p_decoder);
48 picture_Release( p_pic );
51 static void video_link_picture_decoder( decoder_t *p_dec, picture_t *p_pic )
54 picture_Hold( p_pic );
57 static void video_unlink_picture_decoder( decoder_t *p_dec, picture_t *p_pic )
60 picture_Release( p_pic );
63 static picture_t *video_new_buffer_decoder( decoder_t *p_dec )
65 p_dec->fmt_out.video.i_chroma = p_dec->fmt_out.i_codec;
66 return picture_NewFromFormat( &p_dec->fmt_out.video );
69 static picture_t *video_new_buffer_encoder( encoder_t *p_enc )
71 p_enc->fmt_in.video.i_chroma = p_enc->fmt_in.i_codec;
72 return picture_NewFromFormat( &p_enc->fmt_in.video );
75 static picture_t *transcode_video_filter_buffer_new( filter_t *p_filter )
77 p_filter->fmt_out.video.i_chroma = p_filter->fmt_out.i_codec;
78 return picture_NewFromFormat( &p_filter->fmt_out.video );
80 static void transcode_video_filter_buffer_del( filter_t *p_filter, picture_t *p_pic )
83 picture_Release( p_pic );
86 static int transcode_video_filter_allocation_init( filter_t *p_filter,
90 p_filter->pf_video_buffer_new = transcode_video_filter_buffer_new;
91 p_filter->pf_video_buffer_del = transcode_video_filter_buffer_del;
95 static void transcode_video_filter_allocation_clear( filter_t *p_filter )
100 static void* EncoderThread( void *obj )
102 sout_stream_sys_t *p_sys = (sout_stream_sys_t*)obj;
103 sout_stream_id_t *id = p_sys->id_video;
104 picture_t *p_pic = NULL;
105 int canc = vlc_savecancel ();
106 block_t *p_block = NULL;
111 vlc_mutex_lock( &p_sys->lock_out );
112 while( !p_sys->b_abort &&
113 (p_pic = picture_fifo_Pop( p_sys->pp_pics )) == NULL )
114 vlc_cond_wait( &p_sys->cond, &p_sys->lock_out );
116 if( p_sys->b_abort && !p_pic )
118 vlc_mutex_unlock( &p_sys->lock_out );
121 vlc_mutex_unlock( &p_sys->lock_out );
125 p_block = id->p_encoder->pf_encode_video( id->p_encoder, p_pic );
127 vlc_mutex_lock( &p_sys->lock_out );
128 block_ChainAppend( &p_sys->p_buffers, p_block );
130 vlc_mutex_unlock( &p_sys->lock_out );
131 picture_Release( p_pic );
134 vlc_mutex_lock( &p_sys->lock_out );
137 vlc_mutex_unlock( &p_sys->lock_out );
140 vlc_mutex_unlock( &p_sys->lock_out );
143 /*Encode what we have in the buffer on closing*/
144 vlc_mutex_lock( &p_sys->lock_out );
145 while( (p_pic = picture_fifo_Pop( p_sys->pp_pics )) != NULL )
147 p_block = id->p_encoder->pf_encode_video( id->p_encoder, p_pic );
149 block_ChainAppend( &p_sys->p_buffers, p_block );
151 picture_Release( p_pic );
154 /*Now flush encoder*/
156 p_block = id->p_encoder->pf_encode_video(id->p_encoder, NULL );
157 block_ChainAppend( &p_sys->p_buffers, p_block );
160 vlc_mutex_unlock( &p_sys->lock_out );
163 vlc_restorecancel (canc);
167 int transcode_video_new( sout_stream_t *p_stream, sout_stream_id_t *id )
169 sout_stream_sys_t *p_sys = p_stream->p_sys;
172 * Initialization of decoder structures
174 id->p_decoder->fmt_out = id->p_decoder->fmt_in;
175 id->p_decoder->fmt_out.i_extra = 0;
176 id->p_decoder->fmt_out.p_extra = 0;
177 id->p_decoder->pf_decode_video = NULL;
178 id->p_decoder->pf_get_cc = NULL;
179 id->p_decoder->pf_get_cc = 0;
180 id->p_decoder->pf_vout_buffer_new = video_new_buffer_decoder;
181 id->p_decoder->pf_vout_buffer_del = video_del_buffer_decoder;
182 id->p_decoder->pf_picture_link = video_link_picture_decoder;
183 id->p_decoder->pf_picture_unlink = video_unlink_picture_decoder;
184 id->p_decoder->p_owner = malloc( sizeof(decoder_owner_sys_t) );
185 if( !id->p_decoder->p_owner )
188 id->p_decoder->p_owner->p_sys = p_sys;
189 /* id->p_decoder->p_cfg = p_sys->p_video_cfg; */
191 id->p_decoder->p_module =
192 module_need( id->p_decoder, "decoder", "$codec", false );
194 if( !id->p_decoder->p_module )
196 msg_Err( p_stream, "cannot find video decoder" );
197 free( id->p_decoder->p_owner );
203 * Because some info about the decoded input will only be available
204 * once the first frame is decoded, we actually only test the availability
205 * of the encoder here.
208 /* Initialization of encoder format structures */
209 es_format_Init( &id->p_encoder->fmt_in, id->p_decoder->fmt_in.i_cat,
210 id->p_decoder->fmt_out.i_codec );
211 id->p_encoder->fmt_in.video.i_chroma = id->p_decoder->fmt_out.i_codec;
213 /* The dimensions will be set properly later on.
214 * Just put sensible values so we can test an encoder is available. */
215 id->p_encoder->fmt_in.video.i_width =
216 id->p_encoder->fmt_out.video.i_width
217 ? id->p_encoder->fmt_out.video.i_width
218 : id->p_decoder->fmt_in.video.i_width
219 ? id->p_decoder->fmt_in.video.i_width : 16;
220 id->p_encoder->fmt_in.video.i_height =
221 id->p_encoder->fmt_out.video.i_height
222 ? id->p_encoder->fmt_out.video.i_height
223 : id->p_decoder->fmt_in.video.i_height
224 ? id->p_decoder->fmt_in.video.i_height : 16;
225 id->p_encoder->fmt_in.video.i_visible_width =
226 id->p_encoder->fmt_out.video.i_visible_width
227 ? id->p_encoder->fmt_out.video.i_visible_width
228 : id->p_decoder->fmt_in.video.i_visible_width
229 ? id->p_decoder->fmt_in.video.i_visible_width : id->p_encoder->fmt_in.video.i_width;
230 id->p_encoder->fmt_in.video.i_visible_height =
231 id->p_encoder->fmt_out.video.i_visible_height
232 ? id->p_encoder->fmt_out.video.i_visible_height
233 : id->p_decoder->fmt_in.video.i_visible_height
234 ? id->p_decoder->fmt_in.video.i_visible_height : id->p_encoder->fmt_in.video.i_height;
235 id->p_encoder->fmt_in.video.i_frame_rate = id->p_decoder->fmt_out.video.i_frame_rate;
236 id->p_encoder->fmt_in.video.i_frame_rate_base = id->p_decoder->fmt_out.video.i_frame_rate_base;
238 id->p_encoder->i_threads = p_sys->i_threads;
239 id->p_encoder->p_cfg = p_sys->p_video_cfg;
241 id->p_encoder->p_module =
242 module_need( id->p_encoder, "encoder", p_sys->psz_venc, true );
243 if( !id->p_encoder->p_module )
245 msg_Err( p_stream, "cannot find video encoder (module:%s fourcc:%4.4s). Take a look few lines earlier to see possible reason.",
246 p_sys->psz_venc ? p_sys->psz_venc : "any",
247 (char *)&p_sys->i_vcodec );
248 module_unneed( id->p_decoder, id->p_decoder->p_module );
249 id->p_decoder->p_module = 0;
250 free( id->p_decoder->p_owner );
254 /* Close the encoder.
255 * We'll open it only when we have the first frame. */
256 module_unneed( id->p_encoder, id->p_encoder->p_module );
257 if( id->p_encoder->fmt_out.p_extra )
259 free( id->p_encoder->fmt_out.p_extra );
260 id->p_encoder->fmt_out.p_extra = NULL;
261 id->p_encoder->fmt_out.i_extra = 0;
263 id->p_encoder->p_module = NULL;
265 if( p_sys->i_threads >= 1 )
267 int i_priority = p_sys->b_high_priority ? VLC_THREAD_PRIORITY_OUTPUT :
268 VLC_THREAD_PRIORITY_VIDEO;
269 p_sys->id_video = id;
270 vlc_mutex_init( &p_sys->lock_out );
271 vlc_cond_init( &p_sys->cond );
272 p_sys->pp_pics = picture_fifo_New();
273 if( p_sys->pp_pics == NULL )
275 msg_Err( p_stream, "cannot create picture fifo" );
276 vlc_mutex_destroy( &p_sys->lock_out );
277 vlc_cond_destroy( &p_sys->cond );
278 module_unneed( id->p_decoder, id->p_decoder->p_module );
279 id->p_decoder->p_module = NULL;
280 free( id->p_decoder->p_owner );
283 p_sys->p_buffers = NULL;
284 p_sys->b_abort = false;
285 if( vlc_clone( &p_sys->thread, EncoderThread, p_sys, i_priority ) )
287 msg_Err( p_stream, "cannot spawn encoder thread" );
288 vlc_mutex_destroy( &p_sys->lock_out );
289 vlc_cond_destroy( &p_sys->cond );
290 picture_fifo_Delete( p_sys->pp_pics );
291 module_unneed( id->p_decoder, id->p_decoder->p_module );
292 id->p_decoder->p_module = NULL;
293 free( id->p_decoder->p_owner );
300 static void transcode_video_filter_init( sout_stream_t *p_stream,
301 sout_stream_id_t *id )
303 es_format_t *p_fmt_out = &id->p_decoder->fmt_out;
304 id->p_encoder->fmt_in.video.i_chroma = id->p_encoder->fmt_in.i_codec;
306 id->p_f_chain = filter_chain_New( p_stream, "video filter2",
308 transcode_video_filter_allocation_init,
309 transcode_video_filter_allocation_clear,
311 filter_chain_Reset( id->p_f_chain, p_fmt_out, p_fmt_out );
314 if( p_stream->p_sys->b_deinterlace )
316 filter_chain_AppendFilter( id->p_f_chain,
317 p_stream->p_sys->psz_deinterlace,
318 p_stream->p_sys->p_deinterlace_cfg,
319 &id->p_decoder->fmt_out,
320 &id->p_decoder->fmt_out );
322 p_fmt_out = filter_chain_GetFmtOut( id->p_f_chain );
325 /* Check that we have visible_width/height*/
326 if( !p_fmt_out->video.i_visible_height )
327 p_fmt_out->video.i_visible_height = p_fmt_out->video.i_height;
328 if( !p_fmt_out->video.i_visible_width )
329 p_fmt_out->video.i_visible_width = p_fmt_out->video.i_width;
331 if( p_stream->p_sys->psz_vf2 )
333 id->p_uf_chain = filter_chain_New( p_stream, "video filter2",
335 transcode_video_filter_allocation_init,
336 transcode_video_filter_allocation_clear,
338 filter_chain_Reset( id->p_uf_chain, p_fmt_out,
339 &id->p_encoder->fmt_in );
340 if( p_fmt_out->video.i_chroma != id->p_encoder->fmt_in.video.i_chroma )
342 filter_chain_AppendFilter( id->p_uf_chain,
345 &id->p_encoder->fmt_in );
347 filter_chain_AppendFromString( id->p_uf_chain, p_stream->p_sys->psz_vf2 );
348 p_fmt_out = filter_chain_GetFmtOut( id->p_uf_chain );
349 es_format_Copy( &id->p_encoder->fmt_in, p_fmt_out );
350 id->p_encoder->fmt_out.video.i_width =
351 id->p_encoder->fmt_in.video.i_width;
352 id->p_encoder->fmt_out.video.i_height =
353 id->p_encoder->fmt_in.video.i_height;
354 id->p_encoder->fmt_out.video.i_sar_num =
355 id->p_encoder->fmt_in.video.i_sar_num;
356 id->p_encoder->fmt_out.video.i_sar_den =
357 id->p_encoder->fmt_in.video.i_sar_den;
362 /* Take care of the scaling and chroma conversions. */
363 static void conversion_video_filter_append( sout_stream_id_t *id )
365 const es_format_t *p_fmt_out = &id->p_decoder->fmt_out;
367 p_fmt_out = filter_chain_GetFmtOut( id->p_f_chain );
369 p_fmt_out = filter_chain_GetFmtOut( id->p_uf_chain );
371 if( ( p_fmt_out->video.i_chroma != id->p_encoder->fmt_in.video.i_chroma ) ||
372 ( p_fmt_out->video.i_width != id->p_encoder->fmt_in.video.i_width ) ||
373 ( p_fmt_out->video.i_height != id->p_encoder->fmt_in.video.i_height ) )
375 filter_chain_AppendFilter( id->p_f_chain,
378 &id->p_encoder->fmt_in );
382 static void transcode_video_encoder_init( sout_stream_t *p_stream,
383 sout_stream_id_t *id )
385 sout_stream_sys_t *p_sys = p_stream->p_sys;
387 const es_format_t *p_fmt_out = &id->p_decoder->fmt_out;
388 if( id->p_f_chain ) {
389 p_fmt_out = filter_chain_GetFmtOut( id->p_f_chain );
391 if( id->p_uf_chain ) {
392 p_fmt_out = filter_chain_GetFmtOut( id->p_uf_chain );
396 * width/height of source */
397 int i_src_visible_width = p_fmt_out->video.i_visible_width;
398 int i_src_visible_height = p_fmt_out->video.i_visible_height;
400 if (i_src_visible_width == 0)
401 i_src_visible_width = p_fmt_out->video.i_width;
402 if (i_src_visible_height == 0)
403 i_src_visible_height = p_fmt_out->video.i_height;
406 /* with/height scaling */
407 float f_scale_width = 1;
408 float f_scale_height = 1;
411 float f_aspect = (double)p_fmt_out->video.i_sar_num *
412 p_fmt_out->video.i_width /
413 p_fmt_out->video.i_sar_den /
414 p_fmt_out->video.i_height;
416 msg_Dbg( p_stream, "decoder aspect is %f:1", f_aspect );
418 /* Change f_aspect from source frame to source pixel */
419 f_aspect = f_aspect * i_src_visible_height / i_src_visible_width;
420 msg_Dbg( p_stream, "source pixel aspect is %f:1", f_aspect );
422 /* Calculate scaling factor for specified parameters */
423 if( id->p_encoder->fmt_out.video.i_visible_width <= 0 &&
424 id->p_encoder->fmt_out.video.i_visible_height <= 0 && p_sys->f_scale )
426 /* Global scaling. Make sure width will remain a factor of 16 */
429 int i_new_width = i_src_visible_width * p_sys->f_scale;
431 if( i_new_width % 16 <= 7 && i_new_width >= 16 )
432 i_new_width -= i_new_width % 16;
434 i_new_width += 16 - i_new_width % 16;
436 f_real_scale = (float)( i_new_width ) / (float) i_src_visible_width;
438 i_new_height = __MAX( 16, i_src_visible_height * (float)f_real_scale );
440 f_scale_width = f_real_scale;
441 f_scale_height = (float) i_new_height / (float) i_src_visible_height;
443 else if( id->p_encoder->fmt_out.video.i_visible_width > 0 &&
444 id->p_encoder->fmt_out.video.i_visible_height <= 0 )
446 /* Only width specified */
447 f_scale_width = (float)id->p_encoder->fmt_out.video.i_visible_width/i_src_visible_width;
448 f_scale_height = f_scale_width;
450 else if( id->p_encoder->fmt_out.video.i_visible_width <= 0 &&
451 id->p_encoder->fmt_out.video.i_visible_height > 0 )
453 /* Only height specified */
454 f_scale_height = (float)id->p_encoder->fmt_out.video.i_visible_height/i_src_visible_height;
455 f_scale_width = f_scale_height;
457 else if( id->p_encoder->fmt_out.video.i_visible_width > 0 &&
458 id->p_encoder->fmt_out.video.i_visible_height > 0 )
460 /* Width and height specified */
461 f_scale_width = (float)id->p_encoder->fmt_out.video.i_visible_width/i_src_visible_width;
462 f_scale_height = (float)id->p_encoder->fmt_out.video.i_visible_height/i_src_visible_height;
465 /* check maxwidth and maxheight */
466 if( p_sys->i_maxwidth && f_scale_width > (float)p_sys->i_maxwidth /
467 i_src_visible_width )
469 f_scale_width = (float)p_sys->i_maxwidth / i_src_visible_width;
472 if( p_sys->i_maxheight && f_scale_height > (float)p_sys->i_maxheight /
473 i_src_visible_height )
475 f_scale_height = (float)p_sys->i_maxheight / i_src_visible_height;
479 /* Change aspect ratio from source pixel to scaled pixel */
480 f_aspect = f_aspect * f_scale_height / f_scale_width;
481 msg_Dbg( p_stream, "scaled pixel aspect is %f:1", f_aspect );
483 /* f_scale_width and f_scale_height are now final */
484 /* Calculate width, height from scaling
485 * Make sure its multiple of 2
487 /* width/height of output stream */
488 int i_dst_visible_width = 2 * (int)(f_scale_width*i_src_visible_width/2+0.5);
489 int i_dst_visible_height = 2 * (int)(f_scale_height*i_src_visible_height/2+0.5);
490 int i_dst_width = 2 * (int)(f_scale_width*p_fmt_out->video.i_width/2+0.5);
491 int i_dst_height = 2 * (int)(f_scale_height*p_fmt_out->video.i_height/2+0.5);
493 /* Change aspect ratio from scaled pixel to output frame */
494 f_aspect = f_aspect * i_dst_visible_width / i_dst_visible_height;
496 /* Store calculated values */
497 id->p_encoder->fmt_out.video.i_width = i_dst_width;
498 id->p_encoder->fmt_out.video.i_visible_width = i_dst_visible_width;
499 id->p_encoder->fmt_out.video.i_height = i_dst_height;
500 id->p_encoder->fmt_out.video.i_visible_height = i_dst_visible_height;
502 id->p_encoder->fmt_in.video.i_width = i_dst_width;
503 id->p_encoder->fmt_in.video.i_visible_width = i_dst_visible_width;
504 id->p_encoder->fmt_in.video.i_height = i_dst_height;
505 id->p_encoder->fmt_in.video.i_visible_height = i_dst_visible_height;
507 msg_Dbg( p_stream, "source %ix%i, destination %ix%i",
508 i_src_visible_width, i_src_visible_height,
509 i_dst_visible_width, i_dst_visible_height
512 /* Handle frame rate conversion */
513 if( !id->p_encoder->fmt_out.video.i_frame_rate ||
514 !id->p_encoder->fmt_out.video.i_frame_rate_base )
516 if( p_fmt_out->video.i_frame_rate &&
517 p_fmt_out->video.i_frame_rate_base )
519 id->p_encoder->fmt_out.video.i_frame_rate =
520 p_fmt_out->video.i_frame_rate;
521 id->p_encoder->fmt_out.video.i_frame_rate_base =
522 p_fmt_out->video.i_frame_rate_base;
526 /* Pick a sensible default value */
527 id->p_encoder->fmt_out.video.i_frame_rate = ENC_FRAMERATE;
528 id->p_encoder->fmt_out.video.i_frame_rate_base = ENC_FRAMERATE_BASE;
532 id->p_encoder->fmt_in.video.i_frame_rate =
533 id->p_encoder->fmt_out.video.i_frame_rate;
534 id->p_encoder->fmt_in.video.i_frame_rate_base =
535 id->p_encoder->fmt_out.video.i_frame_rate_base;
537 vlc_ureduce( &id->p_encoder->fmt_in.video.i_frame_rate,
538 &id->p_encoder->fmt_in.video.i_frame_rate_base,
539 id->p_encoder->fmt_in.video.i_frame_rate,
540 id->p_encoder->fmt_in.video.i_frame_rate_base,
542 msg_Dbg( p_stream, "source fps %d/%d, destination %d/%d",
543 id->p_decoder->fmt_out.video.i_frame_rate,
544 id->p_decoder->fmt_out.video.i_frame_rate_base,
545 id->p_encoder->fmt_in.video.i_frame_rate,
546 id->p_encoder->fmt_in.video.i_frame_rate_base );
548 id->i_output_frame_interval = id->p_encoder->fmt_out.video.i_frame_rate_base * CLOCK_FREQ / id->p_encoder->fmt_out.video.i_frame_rate;
549 id->i_input_frame_interval = id->p_decoder->fmt_out.video.i_frame_rate_base * CLOCK_FREQ / id->p_decoder->fmt_out.video.i_frame_rate;
550 msg_Info( p_stream, "input interval %d (base %d) output interval %d (base %d)", id->i_input_frame_interval, id->p_decoder->fmt_out.video.i_frame_rate_base,
551 id->i_output_frame_interval, id->p_encoder->fmt_in.video.i_frame_rate_base );
553 date_Init( &id->interpolated_pts,
554 id->p_decoder->fmt_out.video.i_frame_rate,
556 date_Init( &id->next_output_pts,
557 id->p_encoder->fmt_in.video.i_frame_rate,
560 /* Check whether a particular aspect ratio was requested */
561 if( id->p_encoder->fmt_out.video.i_sar_num <= 0 ||
562 id->p_encoder->fmt_out.video.i_sar_den <= 0 )
564 vlc_ureduce( &id->p_encoder->fmt_out.video.i_sar_num,
565 &id->p_encoder->fmt_out.video.i_sar_den,
566 (uint64_t)p_fmt_out->video.i_sar_num * i_src_visible_width * i_dst_visible_height,
567 (uint64_t)p_fmt_out->video.i_sar_den * i_src_visible_height * i_dst_visible_width,
572 vlc_ureduce( &id->p_encoder->fmt_out.video.i_sar_num,
573 &id->p_encoder->fmt_out.video.i_sar_den,
574 id->p_encoder->fmt_out.video.i_sar_num,
575 id->p_encoder->fmt_out.video.i_sar_den,
579 id->p_encoder->fmt_in.video.i_sar_num =
580 id->p_encoder->fmt_out.video.i_sar_num;
581 id->p_encoder->fmt_in.video.i_sar_den =
582 id->p_encoder->fmt_out.video.i_sar_den;
584 msg_Dbg( p_stream, "encoder aspect is %i:%i",
585 id->p_encoder->fmt_out.video.i_sar_num * id->p_encoder->fmt_out.video.i_width,
586 id->p_encoder->fmt_out.video.i_sar_den * id->p_encoder->fmt_out.video.i_height );
590 static int transcode_video_encoder_open( sout_stream_t *p_stream,
591 sout_stream_id_t *id )
593 sout_stream_sys_t *p_sys = p_stream->p_sys;
596 msg_Dbg( p_stream, "destination (after video filters) %ix%i",
597 id->p_encoder->fmt_in.video.i_width,
598 id->p_encoder->fmt_in.video.i_height );
600 id->p_encoder->p_module =
601 module_need( id->p_encoder, "encoder", p_sys->psz_venc, true );
602 if( !id->p_encoder->p_module )
604 msg_Err( p_stream, "cannot find video encoder (module:%s fourcc:%4.4s)",
605 p_sys->psz_venc ? p_sys->psz_venc : "any",
606 (char *)&p_sys->i_vcodec );
610 id->p_encoder->fmt_in.video.i_chroma = id->p_encoder->fmt_in.i_codec;
613 id->p_encoder->fmt_out.i_codec =
614 vlc_fourcc_GetCodec( VIDEO_ES, id->p_encoder->fmt_out.i_codec );
616 id->id = sout_StreamIdAdd( p_stream->p_next, &id->p_encoder->fmt_out );
619 msg_Err( p_stream, "cannot add this stream" );
626 void transcode_video_close( sout_stream_t *p_stream,
627 sout_stream_id_t *id )
629 if( p_stream->p_sys->i_threads >= 1 )
631 vlc_mutex_lock( &p_stream->p_sys->lock_out );
632 p_stream->p_sys->b_abort = true;
633 vlc_cond_signal( &p_stream->p_sys->cond );
634 vlc_mutex_unlock( &p_stream->p_sys->lock_out );
636 vlc_join( p_stream->p_sys->thread, NULL );
637 vlc_mutex_destroy( &p_stream->p_sys->lock_out );
638 vlc_cond_destroy( &p_stream->p_sys->cond );
640 picture_fifo_Delete( p_stream->p_sys->pp_pics );
641 block_ChainRelease( p_stream->p_sys->p_buffers );
642 p_stream->p_sys->pp_pics = NULL;
646 if( id->p_decoder->p_module )
647 module_unneed( id->p_decoder, id->p_decoder->p_module );
648 if( id->p_decoder->p_description )
649 vlc_meta_Delete( id->p_decoder->p_description );
651 free( id->p_decoder->p_owner );
654 if( id->p_encoder->p_module )
655 module_unneed( id->p_encoder, id->p_encoder->p_module );
659 filter_chain_Delete( id->p_f_chain );
661 filter_chain_Delete( id->p_uf_chain );
664 static void OutputFrame( sout_stream_sys_t *p_sys, picture_t *p_pic, sout_stream_t *p_stream, sout_stream_id_t *id, block_t **out )
667 picture_t *p_pic2 = NULL;
668 bool b_need_duplicate=false;
669 /* If input pts + input_frame_interval is lower than next_output_pts - output_frame_interval
670 * Then the future input frame should fit better and we can drop this one
672 * Duplication need is checked in OutputFrame */
673 if( ( p_pic->date + (mtime_t)id->i_input_frame_interval ) <
674 ( date_Get( &id->next_output_pts ) ) )
677 msg_Dbg( p_stream, "dropping frame (%"PRId64" + %"PRId64" vs %"PRId64")",
678 p_pic->date, id->i_input_frame_interval, date_Get(&id->next_output_pts) );
680 picture_Release( p_pic );
687 /* Check if we have a subpicture to overlay */
690 video_format_t fmt = id->p_encoder->fmt_in.video;
691 if( fmt.i_visible_width <= 0 || fmt.i_visible_height <= 0 )
693 fmt.i_visible_width = fmt.i_width;
694 fmt.i_visible_height = fmt.i_height;
699 subpicture_t *p_subpic = spu_Render( p_sys->p_spu, NULL, &fmt, &fmt,
700 p_pic->date, p_pic->date, false );
702 /* Overlay subpicture */
705 if( picture_IsReferenced( p_pic ) && !filter_chain_GetLength( id->p_f_chain ) )
707 /* We can't modify the picture, we need to duplicate it,
708 * in this point the picture is already p_encoder->fmt.in format*/
709 picture_t *p_tmp = video_new_buffer_encoder( id->p_encoder );
710 if( likely( p_tmp ) )
712 picture_Copy( p_tmp, p_pic );
713 picture_Release( p_pic );
717 if( unlikely( !p_sys->p_spu_blend ) )
718 p_sys->p_spu_blend = filter_NewBlend( VLC_OBJECT( p_sys->p_spu ), &fmt );
719 if( likely( p_sys->p_spu_blend ) )
720 picture_BlendSubpicture( p_pic, p_sys->p_spu_blend, p_subpic );
721 subpicture_Delete( p_subpic );
724 /*This pts is handled, increase clock to next one*/
725 date_Increment( &id->next_output_pts, id->p_encoder->fmt_in.video.i_frame_rate_base );
727 if( p_sys->i_threads == 0 )
731 p_block = id->p_encoder->pf_encode_video( id->p_encoder, p_pic );
732 block_ChainAppend( out, p_block );
735 /* we need to duplicate while next_output_pts + output_frame_interval < input_pts (next input pts)*/
736 b_need_duplicate = ( date_Get( &id->next_output_pts ) + id->i_output_frame_interval ) <
737 ( date_Get( &id->interpolated_pts ) );
739 if( p_sys->i_threads )
741 if( p_sys->b_master_sync )
743 p_pic2 = video_new_buffer_encoder( id->p_encoder );
744 if( likely( p_pic2 != NULL ) )
745 picture_Copy( p_pic2, p_pic );
747 vlc_mutex_lock( &p_sys->lock_out );
748 picture_fifo_Push( p_sys->pp_pics, p_pic );
749 vlc_cond_signal( &p_sys->cond );
750 vlc_mutex_unlock( &p_sys->lock_out );
753 while( (p_sys->b_master_sync && b_need_duplicate ))
755 if( p_sys->i_threads >= 1 )
757 picture_t *p_tmp = NULL;
758 /* We can't modify the picture, we need to duplicate it */
759 p_tmp = video_new_buffer_encoder( id->p_encoder );
760 if( likely( p_tmp != NULL ) )
762 picture_Copy( p_tmp, p_pic2 );
763 p_tmp->date = date_Get( &id->next_output_pts );
764 vlc_mutex_lock( &p_sys->lock_out );
765 picture_fifo_Push( p_sys->pp_pics, p_tmp );
766 vlc_cond_signal( &p_sys->cond );
767 vlc_mutex_unlock( &p_sys->lock_out );
773 p_pic->date = date_Get( &id->next_output_pts );
774 p_block = id->p_encoder->pf_encode_video(id->p_encoder, p_pic);
775 block_ChainAppend( out, p_block );
778 msg_Dbg( p_stream, "duplicated frame");
780 date_Increment( &id->next_output_pts, id->p_encoder->fmt_in.video.i_frame_rate_base );
781 b_need_duplicate = ( date_Get( &id->next_output_pts ) + id->i_output_frame_interval ) <
782 ( date_Get( &id->interpolated_pts ) );
785 if( p_sys->i_threads && p_pic2 )
786 picture_Release( p_pic2 );
787 else if ( p_sys->i_threads == 0 )
788 picture_Release( p_pic );
791 int transcode_video_process( sout_stream_t *p_stream, sout_stream_id_t *id,
792 block_t *in, block_t **out )
794 sout_stream_sys_t *p_sys = p_stream->p_sys;
795 picture_t *p_pic = NULL;
798 if( unlikely( in == NULL ) )
800 if( p_sys->i_threads == 0 )
804 p_block = id->p_encoder->pf_encode_video(id->p_encoder, NULL );
805 block_ChainAppend( out, p_block );
810 msg_Dbg( p_stream, "Flushing thread and waiting that");
811 vlc_mutex_lock( &p_stream->p_sys->lock_out );
812 p_stream->p_sys->b_abort = true;
813 vlc_cond_signal( &p_stream->p_sys->cond );
814 vlc_mutex_unlock( &p_stream->p_sys->lock_out );
816 vlc_join( p_stream->p_sys->thread, NULL );
817 vlc_mutex_lock( &p_sys->lock_out );
818 *out = p_sys->p_buffers;
819 p_sys->p_buffers = NULL;
820 vlc_mutex_unlock( &p_sys->lock_out );
822 msg_Dbg( p_stream, "Flushing done");
828 while( (p_pic = id->p_decoder->pf_decode_video( id->p_decoder, &in )) )
831 if( p_stream->p_sout->i_out_pace_nocontrol && p_sys->b_hurry_up )
833 mtime_t current_date = mdate();
834 if( unlikely( (current_date - 50000) > p_pic->date ) )
836 msg_Dbg( p_stream, "late picture skipped (%"PRId64")",
837 current_date - 50000 - p_pic->date );
838 picture_Release( p_pic );
844 id->p_encoder->p_module &&
845 !video_format_IsSimilar( &p_sys->fmt_input_video, &id->p_decoder->fmt_out.video )
849 msg_Info( p_stream, "aspect-ratio changed, reiniting. %i -> %i : %i -> %i.",
850 p_sys->fmt_input_video.i_sar_num, id->p_decoder->fmt_out.video.i_sar_num,
851 p_sys->fmt_input_video.i_sar_den, id->p_decoder->fmt_out.video.i_sar_den
855 filter_chain_Delete( id->p_f_chain );
856 id->p_f_chain = NULL;
858 filter_chain_Delete( id->p_uf_chain );
859 id->p_uf_chain = NULL;
861 /* Reinitialize filters */
862 id->p_encoder->fmt_out.video.i_visible_width = p_sys->i_width & ~1;
863 id->p_encoder->fmt_out.video.i_visible_height = p_sys->i_height & ~1;
864 id->p_encoder->fmt_out.video.i_sar_num = id->p_encoder->fmt_out.video.i_sar_den = 0;
866 transcode_video_filter_init( p_stream, id );
867 transcode_video_encoder_init( p_stream, id );
868 conversion_video_filter_append( id );
869 memcpy( &p_sys->fmt_input_video, &id->p_decoder->fmt_out.video, sizeof(video_format_t));
873 if( unlikely( !id->p_encoder->p_module ) )
876 filter_chain_Delete( id->p_f_chain );
878 filter_chain_Delete( id->p_uf_chain );
879 id->p_f_chain = id->p_uf_chain = NULL;
881 transcode_video_filter_init( p_stream, id );
882 transcode_video_encoder_init( p_stream, id );
883 conversion_video_filter_append( id );
884 memcpy( &p_sys->fmt_input_video, &id->p_decoder->fmt_out.video, sizeof(video_format_t));
886 if( transcode_video_encoder_open( p_stream, id ) != VLC_SUCCESS )
888 picture_Release( p_pic );
889 transcode_video_close( p_stream, id );
890 id->b_transcode = false;
895 /*Input lipsync and drop check */
896 if( p_sys->b_master_sync )
898 /* How much audio has drifted */
899 mtime_t i_master_drift = p_sys->i_master_drift;
901 /* This is the pts input should have now with constant frame rate */
902 mtime_t i_pts = date_Get( &id->interpolated_pts );
904 /* How much video pts is ahead of calculated pts */
905 mtime_t i_video_drift = p_pic->date - i_pts;
907 /* Check that we are having lipsync with input here */
908 if( unlikely ( ( (i_video_drift - i_master_drift ) > MASTER_SYNC_MAX_DRIFT
909 || (i_video_drift + i_master_drift ) < -MASTER_SYNC_MAX_DRIFT ) ) )
912 "video drift too big, resetting sync %"PRId64" to %"PRId64,
913 (i_video_drift + i_master_drift),
916 date_Set( &id->interpolated_pts, p_pic->date );
917 date_Set( &id->next_output_pts, p_pic->date );
918 i_pts = date_Get( &id->interpolated_pts );
921 /* Set the pts of the frame being encoded */
924 /* now take next input pts, pts dates are only enabled if p_module is set*/
925 date_Increment( &id->interpolated_pts, id->p_decoder->fmt_out.video.i_frame_rate_base );
928 /* If input pts + input_frame_interval is lower than next_output_pts - output_frame_interval
929 * Then the future input frame should fit better and we can drop this one
931 * Duplication need is checked in OutputFrame */
932 if( ( p_pic->date + (mtime_t)id->i_input_frame_interval ) <
933 ( date_Get( &id->next_output_pts ) ) )
936 msg_Dbg( p_stream, "dropping frame (%"PRId64" + %"PRId64" vs %"PRId64")",
937 p_pic->date, id->i_input_frame_interval, date_Get(&id->next_output_pts) );
939 picture_Release( p_pic );
943 msg_Dbg( p_stream, "not dropping frame");
946 /* input calculated pts isn't necessary what pts output should be, so use output pts*/
947 p_pic->date = date_Get( &id->next_output_pts );
952 /* Run the filter and output chains; first with the picture,
953 * and then with NULL as many times as we need until they
954 * stop outputting frames.
957 picture_t *p_filtered_pic = p_pic;
959 /* Run filter chain */
961 p_filtered_pic = filter_chain_VideoFilter( id->p_f_chain, p_filtered_pic );
962 if( !p_filtered_pic )
966 picture_t *p_user_filtered_pic = p_filtered_pic;
968 /* Run user specified filter chain */
970 p_user_filtered_pic = filter_chain_VideoFilter( id->p_uf_chain, p_user_filtered_pic );
971 if( !p_user_filtered_pic )
974 OutputFrame( p_sys, p_user_filtered_pic, p_stream, id, out );
976 p_filtered_pic = NULL;
983 if( p_sys->i_threads >= 1 )
985 /* Pick up any return data the encoder thread wants to output. */
986 vlc_mutex_lock( &p_sys->lock_out );
987 *out = p_sys->p_buffers;
988 p_sys->p_buffers = NULL;
989 vlc_mutex_unlock( &p_sys->lock_out );
995 bool transcode_video_add( sout_stream_t *p_stream, es_format_t *p_fmt,
996 sout_stream_id_t *id )
998 sout_stream_sys_t *p_sys = p_stream->p_sys;
1001 "creating video transcoding from fcc=`%4.4s' to fcc=`%4.4s'",
1002 (char*)&p_fmt->i_codec, (char*)&p_sys->i_vcodec );
1004 /* Complete destination format */
1005 id->p_encoder->fmt_out.i_codec = p_sys->i_vcodec;
1006 id->p_encoder->fmt_out.video.i_visible_width = p_sys->i_width & ~1;
1007 id->p_encoder->fmt_out.video.i_visible_height = p_sys->i_height & ~1;
1008 id->p_encoder->fmt_out.i_bitrate = p_sys->i_vbitrate;
1010 /* Build decoder -> filter -> encoder chain */
1011 if( transcode_video_new( p_stream, id ) )
1013 msg_Err( p_stream, "cannot create video chain" );
1017 /* Stream will be added later on because we don't know
1018 * all the characteristics of the decoded stream yet */
1019 id->b_transcode = true;
1021 if( p_sys->f_fps > 0 )
1023 id->p_encoder->fmt_out.video.i_frame_rate = (p_sys->f_fps * ENC_FRAMERATE_BASE);
1024 id->p_encoder->fmt_out.video.i_frame_rate_base = ENC_FRAMERATE_BASE;