]> git.sesse.net Git - vlc/blob - modules/stream_out/transcode.c
* modules/stream_out/transcode.c, src/input/decoder.c, src/input/control.c: fixed...
[vlc] / modules / stream_out / transcode.c
1 /*****************************************************************************
2  * transcode.c: transcoding stream output module
3  *****************************************************************************
4  * Copyright (C) 2003-2004 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8  *          Gildas Bazin <gbazin@videolan.org>
9  *          Jean-Paul Saman <jpsaman #_at_# m2x dot nl> 
10  *
11  * This program is free software; you can redistribute it and/or modify
12  * it under the terms of the GNU General Public License as published by
13  * the Free Software Foundation; either version 2 of the License, or
14  * (at your option) any later version.
15  *
16  * This program is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19  * GNU General Public License for more details.
20  *
21  * You should have received a copy of the GNU General Public License
22  * along with this program; if not, write to the Free Software
23  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
24  *****************************************************************************/
25
26 /*****************************************************************************
27  * Preamble
28  *****************************************************************************/
29 #include <stdlib.h>
30 #include <string.h>
31 #include <math.h>
32
33 #include <vlc/vlc.h>
34 #include <vlc/input.h>
35 #include <vlc/sout.h>
36 #include <vlc/vout.h>
37 #include <vlc/decoder.h>
38 #include "vlc_filter.h"
39 #include "vlc_osd.h"
40
41 #define MASTER_SYNC_MAX_DRIFT 100000
42
43 /*****************************************************************************
44  * Module descriptor
45  *****************************************************************************/
46 #define VENC_TEXT N_("Video encoder")
47 #define VENC_LONGTEXT N_( \
48     "Allows you to specify the video encoder to use and its associated " \
49     "options." )
50 #define VCODEC_TEXT N_("Destination video codec")
51 #define VCODEC_LONGTEXT N_( \
52     "Allows you to specify the destination video codec used for the " \
53     "streaming output." )
54 #define VB_TEXT N_("Video bitrate")
55 #define VB_LONGTEXT N_( \
56     "Allows you to specify the video bitrate used for the streaming " \
57     "output." )
58 #define SCALE_TEXT N_("Video scaling")
59 #define SCALE_LONGTEXT N_( \
60     "Allows you to scale the video before encoding." )
61 #define FPS_TEXT N_("Video frame-rate")
62 #define FPS_LONGTEXT N_( \
63     "Allows you to specify an output frame rate for the video." )
64 #define DEINTERLACE_TEXT N_("Deinterlace video")
65 #define DEINTERLACE_LONGTEXT N_( \
66     "Allows you to deinterlace the video before encoding." )
67 #define DEINTERLACE_MODULE_TEXT N_("Deinterlace module")
68 #define DEINTERLACE_MODULE_LONGTEXT N_( \
69     "Specifies the deinterlace module to use." )
70 #define WIDTH_TEXT N_("Video width")
71 #define WIDTH_LONGTEXT N_( \
72     "Allows you to specify the output video width." )
73 #define HEIGHT_TEXT N_("Video height")
74 #define HEIGHT_LONGTEXT N_( \
75     "Allows you to specify the output video height." )
76 #define MAXWIDTH_TEXT N_("Maximum video width")
77 #define MAXWIDTH_LONGTEXT N_( \
78     "Allows you to specify a maximum output video width." )
79 #define MAXHEIGHT_TEXT N_("Maximum video height")
80 #define MAXHEIGHT_LONGTEXT N_( \
81     "Allows you to specify a maximum output video height." )
82 #define VFILTER_TEXT N_("Video filter")
83 #define VFILTER_LONGTEXT N_( \
84     "Allows you to specify video filters used after the video " \
85     "transcoding and subpictures overlaying." )
86
87 #define CROPTOP_TEXT N_("Video crop top")
88 #define CROPTOP_LONGTEXT N_( \
89     "Allows you to specify the top coordinate for the video cropping." )
90 #define CROPLEFT_TEXT N_("Video crop left")
91 #define CROPLEFT_LONGTEXT N_( \
92     "Allows you to specify the left coordinate for the video cropping." )
93 #define CROPBOTTOM_TEXT N_("Video crop bottom")
94 #define CROPBOTTOM_LONGTEXT N_( \
95     "Allows you to specify the bottom coordinate for the video cropping." )
96 #define CROPRIGHT_TEXT N_("Video crop right")
97 #define CROPRIGHT_LONGTEXT N_( \
98     "Allows you to specify the right coordinate for the video cropping." )
99
100 #define AENC_TEXT N_("Audio encoder")
101 #define AENC_LONGTEXT N_( \
102     "Allows you to specify the audio encoder to use and its associated " \
103     "options." )
104 #define ACODEC_TEXT N_("Destination audio codec")
105 #define ACODEC_LONGTEXT N_( \
106     "Allows you to specify the destination audio codec used for the " \
107     "streaming output." )
108 #define AB_TEXT N_("Audio bitrate")
109 #define AB_LONGTEXT N_( \
110     "Allows you to specify the audio bitrate used for the streaming " \
111     "output." )
112 #define ARATE_TEXT N_("Audio sample rate")
113 #define ARATE_LONGTEXT N_( \
114     "Allows you to specify the audio sample rate used for the streaming " \
115     "output." )
116 #define ACHANS_TEXT N_("Audio channels")
117 #define ACHANS_LONGTEXT N_( \
118     "Allows you to specify the number of audio channels used for the " \
119     "streaming output." )
120
121 #define SENC_TEXT N_("Subtitles encoder")
122 #define SENC_LONGTEXT N_( \
123     "Allows you to specify the subtitles encoder to use and its associated " \
124     "options." )
125 #define SCODEC_TEXT N_("Destination subtitles codec")
126 #define SCODEC_LONGTEXT N_( \
127     "Allows you to specify the destination subtitles codec used for the " \
128     "streaming output." )
129 #define SFILTER_TEXT N_("Subpictures filter")
130 #define SFILTER_LONGTEXT N_( \
131     "Allows you to specify subpictures filters used during the video " \
132     "transcoding. The subpictures produced by the filters will be overlayed " \
133     "directly onto the video." )
134
135 #define OSD_TEXT N_("OSD menu")
136 #define OSD_LONGTEXT N_(\
137     "Enable streaming of the On Screen Display. It uses the osdmenu subfilter." )
138                   
139 #define THREADS_TEXT N_("Number of threads")
140 #define THREADS_LONGTEXT N_( \
141     "Allows you to specify the number of threads used for the transcoding." )
142 #define HP_TEXT N_("High priority")
143 #define HP_LONGTEXT N_( \
144     "Runs the optional encoder thread at the OUTPUT priority instead of " \
145     "VIDEO." )
146
147 #define ASYNC_TEXT N_("Synchronise on audio track")
148 #define ASYNC_LONGTEXT N_( \
149     "This option will drop/duplicate video frames to synchronise the video " \
150     "track on the audio track." )
151
152 #define HURRYUP_TEXT N_( "Hurry up" )
153 #define HURRYUP_LONGTEXT N_( "Allows you to specify if the transcoder " \
154   "should drop frames if your CPU can't keep up with the encoding rate." )
155
156 static char *ppsz_deinterlace_type[] =
157 {
158     "deinterlace", "ffmpeg-deinterlace"
159 };
160
161 static int  Open ( vlc_object_t * );
162 static void Close( vlc_object_t * );
163
164 #define SOUT_CFG_PREFIX "sout-transcode-"
165
166 vlc_module_begin();
167     set_shortname( _("Transcode"));
168     set_description( _("Transcode stream output") );
169     set_capability( "sout stream", 50 );
170     add_shortcut( "transcode" );
171     set_callbacks( Open, Close );
172     set_category( CAT_SOUT );
173     set_subcategory( SUBCAT_SOUT_STREAM );
174     set_section( N_("Video"), NULL );
175     add_string( SOUT_CFG_PREFIX "venc", NULL, NULL, VENC_TEXT,
176                 VENC_LONGTEXT, VLC_FALSE );
177     add_string( SOUT_CFG_PREFIX "vcodec", NULL, NULL, VCODEC_TEXT,
178                 VCODEC_LONGTEXT, VLC_FALSE );
179     add_integer( SOUT_CFG_PREFIX "vb", 800 * 1000, NULL, VB_TEXT,
180                  VB_LONGTEXT, VLC_FALSE );
181     add_float( SOUT_CFG_PREFIX "scale", 1, NULL, SCALE_TEXT,
182                SCALE_LONGTEXT, VLC_FALSE );
183     add_float( SOUT_CFG_PREFIX "fps", 0, NULL, FPS_TEXT,
184                FPS_LONGTEXT, VLC_FALSE );
185     add_bool( SOUT_CFG_PREFIX "hurry-up", VLC_TRUE, NULL, HURRYUP_TEXT,
186                HURRYUP_LONGTEXT, VLC_FALSE );
187     add_bool( SOUT_CFG_PREFIX "deinterlace", 0, NULL, DEINTERLACE_TEXT,
188               DEINTERLACE_LONGTEXT, VLC_FALSE );
189     add_string( SOUT_CFG_PREFIX "deinterlace-module", "deinterlace", NULL,
190                 DEINTERLACE_MODULE_TEXT, DEINTERLACE_MODULE_LONGTEXT,
191                 VLC_FALSE );
192         change_string_list( ppsz_deinterlace_type, 0, 0 );
193     add_integer( SOUT_CFG_PREFIX "width", 0, NULL, WIDTH_TEXT,
194                  WIDTH_LONGTEXT, VLC_TRUE );
195     add_integer( SOUT_CFG_PREFIX "height", 0, NULL, HEIGHT_TEXT,
196                  HEIGHT_LONGTEXT, VLC_TRUE );
197     add_integer( SOUT_CFG_PREFIX "maxwidth", 0, NULL, MAXWIDTH_TEXT,
198                  MAXWIDTH_LONGTEXT, VLC_TRUE );
199     add_integer( SOUT_CFG_PREFIX "maxheight", 0, NULL, MAXHEIGHT_TEXT,
200                  MAXHEIGHT_LONGTEXT, VLC_TRUE );
201     add_module_list_cat( SOUT_CFG_PREFIX "vfilter", SUBCAT_VIDEO_VFILTER,
202                      NULL, NULL,
203                      VFILTER_TEXT, VFILTER_LONGTEXT, VLC_FALSE );
204
205     add_integer( SOUT_CFG_PREFIX "croptop", 0, NULL, CROPTOP_TEXT,
206                  CROPTOP_LONGTEXT, VLC_TRUE );
207     add_integer( SOUT_CFG_PREFIX "cropleft", 0, NULL, CROPLEFT_TEXT,
208                  CROPLEFT_LONGTEXT, VLC_TRUE );
209     add_integer( SOUT_CFG_PREFIX "cropbottom", 0, NULL, CROPBOTTOM_TEXT,
210                  CROPBOTTOM_LONGTEXT, VLC_TRUE );
211     add_integer( SOUT_CFG_PREFIX "cropright", 0, NULL, CROPRIGHT_TEXT,
212                  CROPRIGHT_LONGTEXT, VLC_TRUE );
213
214     set_section( N_("Audio"), NULL );
215     add_string( SOUT_CFG_PREFIX "aenc", NULL, NULL, AENC_TEXT,
216                 AENC_LONGTEXT, VLC_FALSE );
217     add_string( SOUT_CFG_PREFIX "acodec", NULL, NULL, ACODEC_TEXT,
218                 ACODEC_LONGTEXT, VLC_FALSE );
219     add_integer( SOUT_CFG_PREFIX "ab", 64000, NULL, AB_TEXT,
220                  AB_LONGTEXT, VLC_FALSE );
221     add_integer( SOUT_CFG_PREFIX "channels", 0, NULL, ACHANS_TEXT,
222                  ACHANS_LONGTEXT, VLC_FALSE );
223     add_integer( SOUT_CFG_PREFIX "samplerate", 0, NULL, ARATE_TEXT,
224                  ARATE_LONGTEXT, VLC_TRUE );
225     add_bool( SOUT_CFG_PREFIX "audio-sync", 0, NULL, ASYNC_TEXT,
226               ASYNC_LONGTEXT, VLC_FALSE );
227
228     set_section( N_("Overlays/Subtitles"), NULL );
229     add_string( SOUT_CFG_PREFIX "senc", NULL, NULL, SENC_TEXT,
230                 SENC_LONGTEXT, VLC_FALSE );
231     add_string( SOUT_CFG_PREFIX "scodec", NULL, NULL, SCODEC_TEXT,
232                 SCODEC_LONGTEXT, VLC_FALSE );
233     add_bool( SOUT_CFG_PREFIX "soverlay", 0, NULL, SCODEC_TEXT,
234                SCODEC_LONGTEXT, VLC_FALSE );
235     add_module_list_cat( SOUT_CFG_PREFIX "sfilter", SUBCAT_VIDEO_SUBPIC,
236                      NULL, NULL,
237                      SFILTER_TEXT, SFILTER_LONGTEXT, VLC_FALSE );
238
239     set_section( N_("On Screen Display"), NULL );
240     add_bool( SOUT_CFG_PREFIX "osd", 0, NULL, OSD_TEXT,
241               OSD_LONGTEXT, VLC_FALSE );
242     
243     set_section( N_("Miscellaneous"), NULL );
244     add_integer( SOUT_CFG_PREFIX "threads", 0, NULL, THREADS_TEXT,
245                  THREADS_LONGTEXT, VLC_TRUE );
246     add_bool( SOUT_CFG_PREFIX "high-priority", 0, NULL, HP_TEXT, HP_LONGTEXT,
247               VLC_TRUE );
248
249 vlc_module_end();
250
251 static const char *ppsz_sout_options[] = {
252     "venc", "vcodec", "vb", "croptop", "cropbottom", "cropleft", "cropright",
253     "scale", "fps", "width", "height", "vfilter", "deinterlace",
254     "deinterlace-module", "threads", "hurry-up", "aenc", "acodec", "ab",
255     "samplerate", "channels", "senc", "scodec", "soverlay", "sfilter",
256     "osd", "audio-sync", "high-priority", "maxwidth", "maxheight", NULL
257 };
258
259 /*****************************************************************************
260  * Exported prototypes
261  *****************************************************************************/
262 static sout_stream_id_t *Add ( sout_stream_t *, es_format_t * );
263 static int               Del ( sout_stream_t *, sout_stream_id_t * );
264 static int               Send( sout_stream_t *, sout_stream_id_t *, block_t* );
265
266 static int  transcode_audio_new    ( sout_stream_t *, sout_stream_id_t * );
267 static void transcode_audio_close  ( sout_stream_t *, sout_stream_id_t * );
268 static int  transcode_audio_process( sout_stream_t *, sout_stream_id_t *,
269                                      block_t *, block_t ** );
270
271 static aout_buffer_t *audio_new_buffer( decoder_t *, int );
272 static void audio_del_buffer( decoder_t *, aout_buffer_t * );
273
274 static int  transcode_video_new    ( sout_stream_t *, sout_stream_id_t * );
275 static void transcode_video_close  ( sout_stream_t *, sout_stream_id_t * );
276 static int  transcode_video_encoder_open( sout_stream_t *, sout_stream_id_t *);
277 static int  transcode_video_process( sout_stream_t *, sout_stream_id_t *,
278                                      block_t *, block_t ** );
279
280 static void video_del_buffer( vlc_object_t *, picture_t * );
281 static picture_t *video_new_buffer_decoder( decoder_t * );
282 static void video_del_buffer_decoder( decoder_t *, picture_t * );
283 static void video_link_picture_decoder( decoder_t *, picture_t * );
284 static void video_unlink_picture_decoder( decoder_t *, picture_t * );
285 static picture_t *video_new_buffer_filter( filter_t * );
286 static void video_del_buffer_filter( filter_t *, picture_t * );
287
288 static int  transcode_spu_new    ( sout_stream_t *, sout_stream_id_t * );
289 static void transcode_spu_close  ( sout_stream_t *, sout_stream_id_t * );
290 static int  transcode_spu_process( sout_stream_t *, sout_stream_id_t *,
291                                    block_t *, block_t ** );
292
293 static int  transcode_osd_new    ( sout_stream_t *, sout_stream_id_t * );
294 static void transcode_osd_close  ( sout_stream_t *, sout_stream_id_t * );
295 static int  transcode_osd_process( sout_stream_t *, sout_stream_id_t *,
296                                    block_t *, block_t ** );
297                                    
298 static int  EncoderThread( struct sout_stream_sys_t * p_sys );
299
300 static int pi_channels_maps[6] =
301 {
302     0,
303     AOUT_CHAN_CENTER,   AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
304     AOUT_CHAN_CENTER | AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
305     AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_REARLEFT
306      | AOUT_CHAN_REARRIGHT,
307     AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER
308      | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT
309 };
310
311 #define PICTURE_RING_SIZE 64
312 #define SUBPICTURE_RING_SIZE 20
313
314 struct sout_stream_sys_t
315 {
316     VLC_COMMON_MEMBERS
317
318     sout_stream_t   *p_out;
319     sout_stream_id_t *id_video;
320     block_t         *p_buffers;
321     vlc_mutex_t     lock_out;
322     vlc_cond_t      cond;
323     picture_t *     pp_pics[PICTURE_RING_SIZE];
324     int             i_first_pic, i_last_pic;
325
326     /* Audio */
327     vlc_fourcc_t    i_acodec;   /* codec audio (0 if not transcode) */
328     char            *psz_aenc;
329     sout_cfg_t      *p_audio_cfg;
330     int             i_sample_rate;
331     int             i_channels;
332     int             i_abitrate;
333
334     /* Video */
335     vlc_fourcc_t    i_vcodec;   /* codec video (0 if not transcode) */
336     char            *psz_venc;
337     sout_cfg_t      *p_video_cfg;
338     int             i_vbitrate;
339     double          f_scale;
340     double          f_fps;
341     unsigned int    i_width, i_maxwidth;
342     unsigned int    i_height, i_maxheight;
343     vlc_bool_t      b_deinterlace;
344     char            *psz_deinterlace;
345     sout_cfg_t      *p_deinterlace_cfg;
346     int             i_threads;
347     vlc_bool_t      b_high_priority;
348     vlc_bool_t      b_hurry_up;
349     char            *psz_vfilters[10];
350     sout_cfg_t      *p_vfilters_cfg[10];
351     int             i_vfilters;
352
353     int             i_crop_top;
354     int             i_crop_bottom;
355     int             i_crop_right;
356     int             i_crop_left;
357
358     /* SPU */
359     vlc_fourcc_t    i_scodec;   /* codec spu (0 if not transcode) */
360     char            *psz_senc;
361     vlc_bool_t      b_soverlay;
362     sout_cfg_t      *p_spu_cfg;
363     spu_t           *p_spu;
364
365     /* OSD Menu */
366     sout_stream_id_t *id_osd;   /* extension for streaming OSD menus */
367     vlc_fourcc_t    i_osdcodec; /* codec osd menu (0 if not transcode) */
368     char            *psz_osdenc;
369     sout_cfg_t      *p_osd_cfg;
370     vlc_bool_t      b_es_osd;      /* VLC_TRUE when osd es is registered */
371     vlc_bool_t      b_sout_osd;
372         
373     /* Sync */
374     vlc_bool_t      b_master_sync;
375     mtime_t         i_master_drift;
376 };
377
378 struct decoder_owner_sys_t
379 {
380     picture_t *pp_pics[PICTURE_RING_SIZE];
381     sout_stream_sys_t *p_sys;
382 };
383 struct filter_owner_sys_t
384 {
385     picture_t *pp_pics[PICTURE_RING_SIZE];
386     sout_stream_sys_t *p_sys;
387 };
388
389 /*****************************************************************************
390  * Open:
391  *****************************************************************************/
392 static int Open( vlc_object_t *p_this )
393 {
394     sout_stream_t     *p_stream = (sout_stream_t*)p_this;
395     sout_stream_sys_t *p_sys;
396     vlc_value_t       val;
397
398     p_sys = vlc_object_create( p_this, sizeof( sout_stream_sys_t ) );
399
400     p_sys->p_out = sout_StreamNew( p_stream->p_sout, p_stream->psz_next );
401     if( !p_sys->p_out )
402     {
403         msg_Err( p_stream, "cannot create chain" );
404         vlc_object_destroy( p_sys );
405         return VLC_EGENERIC;
406     }
407
408     p_sys->i_master_drift = 0;
409
410     sout_CfgParse( p_stream, SOUT_CFG_PREFIX, ppsz_sout_options,
411                    p_stream->p_cfg );
412
413     /* Audio transcoding parameters */
414     var_Get( p_stream, SOUT_CFG_PREFIX "aenc", &val );
415     p_sys->psz_aenc = NULL;
416     p_sys->p_audio_cfg = NULL;
417     if( val.psz_string && *val.psz_string )
418     {
419         char *psz_next;
420         psz_next = sout_CfgCreate( &p_sys->psz_aenc, &p_sys->p_audio_cfg,
421                                    val.psz_string );
422         if( psz_next ) free( psz_next );
423     }
424     if( val.psz_string ) free( val.psz_string );
425
426     var_Get( p_stream, SOUT_CFG_PREFIX "acodec", &val );
427     p_sys->i_acodec = 0;
428     if( val.psz_string && *val.psz_string )
429     {
430         char fcc[4] = "    ";
431         memcpy( fcc, val.psz_string, __MIN( strlen( val.psz_string ), 4 ) );
432         p_sys->i_acodec = VLC_FOURCC( fcc[0], fcc[1], fcc[2], fcc[3] );
433     }
434     if( val.psz_string ) free( val.psz_string );
435
436     var_Get( p_stream, SOUT_CFG_PREFIX "ab", &val );
437     p_sys->i_abitrate = val.i_int;
438     if( p_sys->i_abitrate < 4000 ) p_sys->i_abitrate *= 1000;
439
440     var_Get( p_stream, SOUT_CFG_PREFIX "samplerate", &val );
441     p_sys->i_sample_rate = val.i_int;
442
443     var_Get( p_stream, SOUT_CFG_PREFIX "channels", &val );
444     p_sys->i_channels = val.i_int;
445
446     if( p_sys->i_acodec )
447     {
448         if( (strncmp( (char *)&p_sys->i_acodec, "mp3", 3) == 0) &&
449                             (p_sys->i_channels > 2) )
450         {
451             msg_Warn( p_stream, "%d channels invalid for mp3, forcing to 2",
452                       p_sys->i_channels );
453             p_sys->i_channels = 2;
454         }                    
455         msg_Dbg( p_stream, "codec audio=%4.4s %dHz %d channels %dKb/s",
456                  (char *)&p_sys->i_acodec, p_sys->i_sample_rate,
457                  p_sys->i_channels, p_sys->i_abitrate / 1000 );
458     }
459
460     /* Video transcoding parameters */
461     var_Get( p_stream, SOUT_CFG_PREFIX "venc", &val );
462     p_sys->psz_venc = NULL;
463     p_sys->p_video_cfg = NULL;
464     if( val.psz_string && *val.psz_string )
465     {
466         char *psz_next;
467         psz_next = sout_CfgCreate( &p_sys->psz_venc, &p_sys->p_video_cfg,
468                                    val.psz_string );
469         if( psz_next ) free( psz_next );
470     }
471     if( val.psz_string ) free( val.psz_string );
472
473     var_Get( p_stream, SOUT_CFG_PREFIX "vcodec", &val );
474     p_sys->i_vcodec = 0;
475     if( val.psz_string && *val.psz_string )
476     {
477         char fcc[4] = "    ";
478         memcpy( fcc, val.psz_string, __MIN( strlen( val.psz_string ), 4 ) );
479         p_sys->i_vcodec = VLC_FOURCC( fcc[0], fcc[1], fcc[2], fcc[3] );
480     }
481     if( val.psz_string ) free( val.psz_string );
482
483     var_Get( p_stream, SOUT_CFG_PREFIX "vb", &val );
484     p_sys->i_vbitrate = val.i_int;
485     if( p_sys->i_vbitrate < 16000 ) p_sys->i_vbitrate *= 1000;
486
487     var_Get( p_stream, SOUT_CFG_PREFIX "scale", &val );
488     p_sys->f_scale = val.f_float;
489
490     var_Get( p_stream, SOUT_CFG_PREFIX "fps", &val );
491     p_sys->f_fps = val.f_float;
492
493     var_Get( p_stream, SOUT_CFG_PREFIX "hurry-up", &val );
494     p_sys->b_hurry_up = val.b_bool;
495
496     var_Get( p_stream, SOUT_CFG_PREFIX "width", &val );
497     p_sys->i_width = val.i_int;
498
499     var_Get( p_stream, SOUT_CFG_PREFIX "height", &val );
500     p_sys->i_height = val.i_int;
501
502     var_Get( p_stream, SOUT_CFG_PREFIX "maxwidth", &val );
503     p_sys->i_maxwidth = val.i_int;
504
505     var_Get( p_stream, SOUT_CFG_PREFIX "maxheight", &val );
506     p_sys->i_maxheight = val.i_int;
507
508     var_Get( p_stream, SOUT_CFG_PREFIX "vfilter", &val );
509     p_sys->i_vfilters = 0;
510     if( val.psz_string && *val.psz_string )
511     {
512         char *psz_parser = val.psz_string;
513
514         while( psz_parser != NULL && *psz_parser != '\0' )
515         {
516             psz_parser = sout_CfgCreate(
517                                    &p_sys->psz_vfilters[p_sys->i_vfilters],
518                                    &p_sys->p_vfilters_cfg[p_sys->i_vfilters],
519                                    psz_parser );
520             p_sys->i_vfilters++;
521             if( psz_parser != NULL && *psz_parser != '\0' ) psz_parser++;
522         }
523     }
524     if( val.psz_string ) free( val.psz_string );
525     p_sys->psz_vfilters[p_sys->i_vfilters] = NULL;
526     p_sys->p_vfilters_cfg[p_sys->i_vfilters] = NULL;
527
528     var_Get( p_stream, SOUT_CFG_PREFIX "deinterlace", &val );
529     p_sys->b_deinterlace = val.b_bool;
530
531     var_Get( p_stream, SOUT_CFG_PREFIX "deinterlace-module", &val );
532     p_sys->psz_deinterlace = NULL;
533     p_sys->p_deinterlace_cfg = NULL;
534     if( val.psz_string && *val.psz_string )
535     {
536         char *psz_next;
537         psz_next = sout_CfgCreate( &p_sys->psz_deinterlace,
538                                    &p_sys->p_deinterlace_cfg,
539                                    val.psz_string );
540         if( psz_next ) free( psz_next );
541     }
542     if( val.psz_string ) free( val.psz_string );
543
544     var_Get( p_stream, SOUT_CFG_PREFIX "croptop", &val );
545     p_sys->i_crop_top = val.i_int;
546     var_Get( p_stream, SOUT_CFG_PREFIX "cropbottom", &val );
547     p_sys->i_crop_bottom = val.i_int;
548     var_Get( p_stream, SOUT_CFG_PREFIX "cropleft", &val );
549     p_sys->i_crop_left = val.i_int;
550     var_Get( p_stream, SOUT_CFG_PREFIX "cropright", &val );
551     p_sys->i_crop_right = val.i_int;
552
553     var_Get( p_stream, SOUT_CFG_PREFIX "threads", &val );
554     p_sys->i_threads = val.i_int;
555     var_Get( p_stream, SOUT_CFG_PREFIX "high-priority", &val );
556     p_sys->b_high_priority = val.b_bool;
557
558     if( p_sys->i_vcodec )
559     {
560         msg_Dbg( p_stream, "codec video=%4.4s %dx%d scaling: %f %dkb/s",
561                  (char *)&p_sys->i_vcodec, p_sys->i_width, p_sys->i_height,
562                  p_sys->f_scale, p_sys->i_vbitrate / 1000 );
563     }
564
565     /* Subpictures transcoding parameters */
566     p_sys->p_spu = NULL;
567     p_sys->psz_senc = NULL;
568     p_sys->p_spu_cfg = NULL;
569     p_sys->i_scodec = 0;
570
571     var_Get( p_stream, SOUT_CFG_PREFIX "senc", &val );
572     if( val.psz_string && *val.psz_string )
573     {
574         char *psz_next;
575         psz_next = sout_CfgCreate( &p_sys->psz_senc, &p_sys->p_spu_cfg,
576                                    val.psz_string );
577         if( psz_next ) free( psz_next );
578     }
579     if( val.psz_string ) free( val.psz_string );
580
581     var_Get( p_stream, SOUT_CFG_PREFIX "scodec", &val );
582     if( val.psz_string && *val.psz_string )
583     {
584         char fcc[4] = "    ";
585         memcpy( fcc, val.psz_string, __MIN( strlen( val.psz_string ), 4 ) );
586         p_sys->i_scodec = VLC_FOURCC( fcc[0], fcc[1], fcc[2], fcc[3] );
587     }
588     if( val.psz_string ) free( val.psz_string );
589
590     if( p_sys->i_scodec )
591     {
592         msg_Dbg( p_stream, "codec spu=%4.4s", (char *)&p_sys->i_scodec );
593     }
594
595     var_Get( p_stream, SOUT_CFG_PREFIX "soverlay", &val );
596     p_sys->b_soverlay = val.b_bool;
597
598     var_Get( p_stream, SOUT_CFG_PREFIX "sfilter", &val );
599     if( val.psz_string && *val.psz_string )
600     {
601         p_sys->p_spu = spu_Create( p_stream );
602         var_Create( p_sys->p_spu, "sub-filter", VLC_VAR_STRING );
603         var_Set( p_sys->p_spu, "sub-filter", val );
604         spu_Init( p_sys->p_spu );
605     }
606     if( val.psz_string ) free( val.psz_string );
607
608     /* OSD menu transcoding parameters */
609     p_sys->psz_osdenc = NULL;
610     p_sys->p_osd_cfg  = NULL;
611     p_sys->i_osdcodec = 0;
612     p_sys->b_es_osd   = VLC_FALSE;
613
614     var_Get( p_stream, SOUT_CFG_PREFIX "osd", &val );
615     p_sys->b_sout_osd = val.b_bool;
616     if( p_sys->b_sout_osd )
617     {
618         vlc_value_t osd_val;
619         char *psz_next;
620        
621         psz_next = sout_CfgCreate( &p_sys->psz_osdenc,
622                                    &p_sys->p_osd_cfg, strdup( "dvbsub") );
623         if( psz_next ) free( psz_next );
624         
625         p_sys->i_osdcodec = VLC_FOURCC('Y','U','V','P' );        
626
627         msg_Dbg( p_stream, "codec osd=%4.4s", (char *)&p_sys->i_osdcodec );
628
629         if( !p_sys->p_spu )
630         {
631             osd_val.psz_string = strdup("osdmenu");
632             p_sys->p_spu = spu_Create( p_stream );
633             var_Create( p_sys->p_spu, "sub-filter", VLC_VAR_STRING );
634             var_Set( p_sys->p_spu, "sub-filter", osd_val );
635             spu_Init( p_sys->p_spu );
636             if( osd_val.psz_string ) free( osd_val.psz_string );
637         }
638         else
639         {
640             osd_val.psz_string = strdup("osdmenu");
641             var_Set( p_sys->p_spu, "sub-filter", osd_val );
642             if( osd_val.psz_string ) free( osd_val.psz_string );
643         }         
644     }
645
646     /* Audio settings */
647     var_Get( p_stream, SOUT_CFG_PREFIX "audio-sync", &val );
648     p_sys->b_master_sync = val.b_bool;
649     if( p_sys->f_fps > 0 ) p_sys->b_master_sync = VLC_TRUE;
650
651     p_stream->pf_add    = Add;
652     p_stream->pf_del    = Del;
653     p_stream->pf_send   = Send;
654     p_stream->p_sys     = p_sys;
655
656     return VLC_SUCCESS;
657 }
658
659 /*****************************************************************************
660  * Close:
661  *****************************************************************************/
662 static void Close( vlc_object_t * p_this )
663 {
664     sout_stream_t       *p_stream = (sout_stream_t*)p_this;
665     sout_stream_sys_t   *p_sys = p_stream->p_sys;
666
667     sout_StreamDelete( p_sys->p_out );
668
669     while( p_sys->p_audio_cfg != NULL )
670     {
671         sout_cfg_t *p_next = p_sys->p_audio_cfg->p_next;
672
673         if( p_sys->p_audio_cfg->psz_name )
674             free( p_sys->p_audio_cfg->psz_name );
675         if( p_sys->p_audio_cfg->psz_value )
676             free( p_sys->p_audio_cfg->psz_value );
677         free( p_sys->p_audio_cfg );
678
679         p_sys->p_audio_cfg = p_next;
680     }
681     if( p_sys->psz_aenc ) free( p_sys->psz_aenc );
682
683     while( p_sys->p_video_cfg != NULL )
684     {
685         sout_cfg_t *p_next = p_sys->p_video_cfg->p_next;
686
687         if( p_sys->p_video_cfg->psz_name )
688             free( p_sys->p_video_cfg->psz_name );
689         if( p_sys->p_video_cfg->psz_value )
690             free( p_sys->p_video_cfg->psz_value );
691         free( p_sys->p_video_cfg );
692
693         p_sys->p_video_cfg = p_next;
694     }
695     if( p_sys->psz_venc ) free( p_sys->psz_venc );
696
697     while( p_sys->p_deinterlace_cfg != NULL )
698     {
699         sout_cfg_t *p_next = p_sys->p_deinterlace_cfg->p_next;
700
701         if( p_sys->p_deinterlace_cfg->psz_name )
702             free( p_sys->p_deinterlace_cfg->psz_name );
703         if( p_sys->p_deinterlace_cfg->psz_value )
704             free( p_sys->p_deinterlace_cfg->psz_value );
705         free( p_sys->p_deinterlace_cfg );
706
707         p_sys->p_deinterlace_cfg = p_next;
708     }
709     if( p_sys->psz_deinterlace ) free( p_sys->psz_deinterlace );
710
711     while( p_sys->p_spu_cfg != NULL )
712     {
713         sout_cfg_t *p_next = p_sys->p_spu_cfg->p_next;
714
715         if( p_sys->p_spu_cfg->psz_name )
716             free( p_sys->p_spu_cfg->psz_name );
717         if( p_sys->p_spu_cfg->psz_value )
718             free( p_sys->p_spu_cfg->psz_value );
719         free( p_sys->p_spu_cfg );
720
721         p_sys->p_spu_cfg = p_next;
722     }
723     if( p_sys->psz_senc ) free( p_sys->psz_senc );
724
725     if( p_sys->p_spu ) spu_Destroy( p_sys->p_spu );
726     
727     while( p_sys->p_osd_cfg != NULL )
728     {
729         sout_cfg_t *p_next = p_sys->p_osd_cfg->p_next;
730
731         if( p_sys->p_osd_cfg->psz_name )
732             free( p_sys->p_osd_cfg->psz_name );
733         if( p_sys->p_osd_cfg->psz_value )
734             free( p_sys->p_osd_cfg->psz_value );
735         free( p_sys->p_osd_cfg );
736
737         p_sys->p_osd_cfg = p_next;
738     }
739     if( p_sys->psz_osdenc ) free( p_sys->psz_osdenc );    
740
741     vlc_object_destroy( p_sys );
742 }
743
744 struct sout_stream_id_t
745 {
746     vlc_fourcc_t  b_transcode;
747
748     /* id of the out stream */
749     void *id;
750
751     /* Decoder */
752     decoder_t       *p_decoder;
753
754     /* Filters */
755     filter_t        *pp_filter[10];
756     int             i_filter;
757     filter_t        *pp_vfilter[10];
758     int             i_vfilter;
759
760     /* Encoder */
761     encoder_t       *p_encoder;
762
763     /* Sync */
764     date_t          interpolated_pts;
765 };
766
767 static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
768 {
769     sout_stream_sys_t *p_sys = p_stream->p_sys;
770     sout_stream_id_t *id;
771
772     id = malloc( sizeof( sout_stream_id_t ) );
773     memset( id, 0, sizeof(sout_stream_id_t) );
774
775     id->id = NULL;
776     id->p_decoder = NULL;
777     id->p_encoder = NULL;
778
779     /* Create decoder object */
780     id->p_decoder = vlc_object_create( p_stream, VLC_OBJECT_DECODER );
781     if( !id->p_decoder )
782     {
783         msg_Err( p_stream, "out of memory" );
784         goto error;
785     }
786     vlc_object_attach( id->p_decoder, p_stream );
787     id->p_decoder->p_module = NULL;
788     id->p_decoder->fmt_in = *p_fmt;
789     id->p_decoder->b_pace_control = VLC_TRUE;
790
791     /* Create encoder object */
792     id->p_encoder = vlc_object_create( p_stream, VLC_OBJECT_ENCODER );
793     if( !id->p_encoder )
794     {
795         msg_Err( p_stream, "out of memory" );
796         goto error;
797     }
798     vlc_object_attach( id->p_encoder, p_stream );
799     id->p_encoder->p_module = NULL;
800
801     /* Create destination format */
802     es_format_Init( &id->p_encoder->fmt_out, p_fmt->i_cat, 0 );
803     id->p_encoder->fmt_out.i_id    = p_fmt->i_id;
804     id->p_encoder->fmt_out.i_group = p_fmt->i_group;
805     if( p_fmt->psz_language )
806         id->p_encoder->fmt_out.psz_language = strdup( p_fmt->psz_language );
807
808     if( p_fmt->i_cat == AUDIO_ES && (p_sys->i_acodec || p_sys->psz_aenc) )
809     {
810         msg_Dbg( p_stream,
811                  "creating audio transcoding from fcc=`%4.4s' to fcc=`%4.4s'",
812                  (char*)&p_fmt->i_codec, (char*)&p_sys->i_acodec );
813
814         /* Complete destination format */
815         id->p_encoder->fmt_out.i_codec = p_sys->i_acodec;
816         id->p_encoder->fmt_out.audio.i_rate = p_sys->i_sample_rate > 0 ?
817             p_sys->i_sample_rate : (int)p_fmt->audio.i_rate;
818         id->p_encoder->fmt_out.i_bitrate = p_sys->i_abitrate;
819         id->p_encoder->fmt_out.audio.i_bitspersample =
820             p_fmt->audio.i_bitspersample;
821         id->p_encoder->fmt_out.audio.i_channels = p_sys->i_channels > 0 ?
822             p_sys->i_channels : p_fmt->audio.i_channels;
823         /* Sanity check for audio channels */
824         id->p_encoder->fmt_out.audio.i_channels =
825             __MIN( id->p_encoder->fmt_out.audio.i_channels,
826                    id->p_decoder->fmt_in.audio.i_channels );
827         id->p_encoder->fmt_out.audio.i_original_channels =
828             id->p_decoder->fmt_in.audio.i_physical_channels;
829         if( id->p_decoder->fmt_in.audio.i_channels ==
830             id->p_encoder->fmt_out.audio.i_channels )
831         {
832             id->p_encoder->fmt_out.audio.i_physical_channels =
833                 id->p_decoder->fmt_in.audio.i_physical_channels;
834         }
835         else
836         {
837             id->p_encoder->fmt_out.audio.i_physical_channels =
838                 pi_channels_maps[id->p_encoder->fmt_out.audio.i_channels];
839         }
840
841         /* Build decoder -> filter -> encoder chain */
842         if( transcode_audio_new( p_stream, id ) )
843         {
844             msg_Err( p_stream, "cannot create audio chain" );
845             goto error;
846         }
847
848         /* Open output stream */
849         id->id = p_sys->p_out->pf_add( p_sys->p_out, &id->p_encoder->fmt_out );
850         id->b_transcode = VLC_TRUE;
851
852         if( !id->id )
853         {
854             transcode_audio_close( p_stream, id );
855             goto error;
856         }
857
858         date_Init( &id->interpolated_pts, p_fmt->audio.i_rate, 1 );
859     }
860     else if( p_fmt->i_cat == VIDEO_ES &&
861              (p_sys->i_vcodec != 0 || p_sys->psz_venc) )
862     {
863         msg_Dbg( p_stream,
864                  "creating video transcoding from fcc=`%4.4s' to fcc=`%4.4s'",
865                  (char*)&p_fmt->i_codec, (char*)&p_sys->i_vcodec );
866
867         /* Complete destination format */
868         id->p_encoder->fmt_out.i_codec = p_sys->i_vcodec;
869         id->p_encoder->fmt_out.video.i_width  = p_sys->i_width;
870         id->p_encoder->fmt_out.video.i_height = p_sys->i_height;
871         id->p_encoder->fmt_out.i_bitrate = p_sys->i_vbitrate;
872
873         /* Build decoder -> filter -> encoder chain */
874         if( transcode_video_new( p_stream, id ) )
875         {
876             msg_Err( p_stream, "cannot create video chain" );
877             goto error;
878         }
879
880         /* Stream will be added later on because we don't know
881          * all the characteristics of the decoded stream yet */
882         id->b_transcode = VLC_TRUE;
883
884         if( p_sys->f_fps > 0 )
885         {
886             id->p_encoder->fmt_out.video.i_frame_rate =
887                 (p_sys->f_fps * 1001) + 0.5;
888             id->p_encoder->fmt_out.video.i_frame_rate_base = 1001;
889         }
890     }
891     else if( p_fmt->i_cat == SPU_ES && (p_sys->i_scodec || p_sys->psz_senc) )
892     {
893         msg_Dbg( p_stream, "creating subtitles transcoding from fcc=`%4.4s' "
894                  "to fcc=`%4.4s'", (char*)&p_fmt->i_codec,
895                  (char*)&p_sys->i_scodec );
896
897         /* Complete destination format */
898         id->p_encoder->fmt_out.i_codec = p_sys->i_scodec;
899
900         /* build decoder -> filter -> encoder */
901         if( transcode_spu_new( p_stream, id ) )
902         {
903             msg_Err( p_stream, "cannot create subtitles chain" );
904             goto error;
905         }
906
907         /* open output stream */
908         id->id = p_sys->p_out->pf_add( p_sys->p_out, &id->p_encoder->fmt_out );
909         id->b_transcode = VLC_TRUE;
910
911         if( !id->id )
912         {
913             transcode_spu_close( p_stream, id );
914             goto error;
915         }
916     }
917     else if( p_fmt->i_cat == SPU_ES && p_sys->b_soverlay )
918     {
919         msg_Dbg( p_stream, "subtitles (fcc=`%4.4s') overlaying",
920                  (char*)&p_fmt->i_codec );
921
922         id->b_transcode = VLC_TRUE;
923
924         /* Build decoder -> filter -> overlaying chain */
925         if( transcode_spu_new( p_stream, id ) )
926         {
927             msg_Err( p_stream, "cannot create subtitles chain" );
928             goto error;
929         }
930     }
931     else
932     {
933         msg_Dbg( p_stream, "not transcoding a stream (fcc=`%4.4s')",
934                  (char*)&p_fmt->i_codec );
935         id->id = p_sys->p_out->pf_add( p_sys->p_out, p_fmt );
936         id->b_transcode = VLC_FALSE;
937
938         if( !id->id ) goto error;
939     }
940
941     if( p_sys->b_sout_osd )
942     {
943         /* Create a fake OSD menu elementary stream */
944         if( !p_sys->b_es_osd && (p_sys->i_osdcodec != 0 || p_sys->psz_osdenc) )
945         {
946             if( transcode_osd_new( p_stream, p_sys->id_osd ) )
947             {
948                 msg_Err( p_stream, "cannot create osd chain" );
949                 goto error;
950             }
951             p_sys->b_es_osd = VLC_TRUE;
952         }
953     }
954     return id;
955
956  error:
957     if( id->p_decoder )
958     {
959         vlc_object_detach( id->p_decoder );
960         vlc_object_destroy( id->p_decoder );
961     }
962
963     if( id->p_encoder )
964     {
965         vlc_object_detach( id->p_encoder );
966         es_format_Clean( &id->p_encoder->fmt_out );
967         vlc_object_destroy( id->p_encoder );
968     }
969
970     free( id );
971     return NULL;
972 }
973
974 static int Del( sout_stream_t *p_stream, sout_stream_id_t *id )
975 {
976     sout_stream_sys_t *p_sys = p_stream->p_sys;
977
978     if( p_sys->b_es_osd )
979         transcode_osd_close( p_stream, p_sys->id_osd );
980
981     if( id->b_transcode )
982     {
983         switch( id->p_decoder->fmt_in.i_cat )
984         {
985         case AUDIO_ES:
986             transcode_audio_close( p_stream, id );
987             break;
988         case VIDEO_ES:
989             transcode_video_close( p_stream, id );
990             break;
991         case SPU_ES:
992             transcode_spu_close( p_stream, id );
993             break;
994         }
995     }
996
997     if( id->id ) p_sys->p_out->pf_del( p_sys->p_out, id->id );
998
999     if( id->p_decoder )
1000     {
1001         vlc_object_detach( id->p_decoder );
1002         vlc_object_destroy( id->p_decoder );
1003     }
1004
1005     if( id->p_encoder )
1006     {
1007         vlc_object_detach( id->p_encoder );
1008         es_format_Clean( &id->p_encoder->fmt_out );
1009         vlc_object_destroy( id->p_encoder );
1010     }
1011
1012     free( id );
1013
1014     return VLC_SUCCESS;
1015 }
1016
1017 static int Send( sout_stream_t *p_stream, sout_stream_id_t *id,
1018                  block_t *p_buffer )
1019 {
1020     sout_stream_sys_t *p_sys = p_stream->p_sys;
1021     block_t *p_out = NULL;
1022
1023     if( !id->b_transcode && id->id )
1024     {
1025         /* Transcode OSD menu pictures. */
1026         if( p_sys->b_es_osd )
1027         {
1028             transcode_osd_process( p_stream, id, p_buffer, &p_out );                
1029         }
1030         return p_sys->p_out->pf_send( p_sys->p_out, id->id, p_buffer );
1031     }
1032     else if( !id->b_transcode )
1033     {
1034         block_Release( p_buffer );
1035         return VLC_EGENERIC;
1036     }
1037
1038     switch( id->p_decoder->fmt_in.i_cat )
1039     {
1040     case AUDIO_ES:
1041         transcode_audio_process( p_stream, id, p_buffer, &p_out );
1042         break;
1043
1044     case VIDEO_ES:
1045         if( transcode_video_process( p_stream, id, p_buffer, &p_out )
1046             != VLC_SUCCESS )
1047         {
1048             return VLC_EGENERIC;
1049         }
1050         break;
1051
1052     case SPU_ES:
1053         if( transcode_spu_process( p_stream, id, p_buffer, &p_out ) !=
1054             VLC_SUCCESS )
1055         {
1056             return VLC_EGENERIC;
1057         }
1058         break;
1059
1060     default:
1061         p_out = NULL;
1062         block_Release( p_buffer );
1063         break;
1064     }
1065
1066     if( p_out ) return p_sys->p_out->pf_send( p_sys->p_out, id->id, p_out );
1067     return VLC_SUCCESS;
1068 }
1069
1070 /****************************************************************************
1071  * decoder reencoder part
1072  ****************************************************************************/
1073 int audio_BitsPerSample( vlc_fourcc_t i_format )
1074 {
1075     switch( i_format )
1076     {
1077     case VLC_FOURCC('u','8',' ',' '):
1078     case VLC_FOURCC('s','8',' ',' '):
1079         return 8;
1080
1081     case VLC_FOURCC('u','1','6','l'):
1082     case VLC_FOURCC('s','1','6','l'):
1083     case VLC_FOURCC('u','1','6','b'):
1084     case VLC_FOURCC('s','1','6','b'):
1085         return 16;
1086
1087     case VLC_FOURCC('u','2','4','l'):
1088     case VLC_FOURCC('s','2','4','l'):
1089     case VLC_FOURCC('u','2','4','b'):
1090     case VLC_FOURCC('s','2','4','b'):
1091         return 24;
1092
1093     case VLC_FOURCC('u','3','2','l'):
1094     case VLC_FOURCC('s','3','2','l'):
1095     case VLC_FOURCC('u','3','2','b'):
1096     case VLC_FOURCC('s','3','2','b'):
1097     case VLC_FOURCC('f','l','3','2'):
1098     case VLC_FOURCC('f','i','3','2'):
1099         return 32;
1100
1101     case VLC_FOURCC('f','l','6','4'):
1102         return 64;
1103     }
1104
1105     return 0;
1106 }
1107
1108 static filter_t *transcode_audio_filter_new( sout_stream_t *p_stream,
1109                                              sout_stream_id_t *id,
1110                                              es_format_t *p_fmt_in,
1111                                              es_format_t *p_fmt_out )
1112 {
1113     filter_t *p_filter = vlc_object_create( p_stream, VLC_OBJECT_FILTER );
1114
1115     vlc_object_attach( p_filter, p_stream );
1116     p_filter->pf_audio_buffer_new = (block_t* (*) (filter_t*, int))__block_New;
1117
1118     p_filter->fmt_in = *p_fmt_in;
1119     p_filter->fmt_out = *p_fmt_out;
1120
1121     p_filter->p_module = module_Need( p_filter, "audio filter2", 0, 0 );
1122     if( p_filter->p_module )
1123     {
1124         p_filter->fmt_out.audio.i_bitspersample = 
1125             audio_BitsPerSample( p_filter->fmt_out.i_codec );
1126         *p_fmt_in = p_filter->fmt_out;
1127     }
1128     else
1129     {
1130         vlc_object_detach( p_filter );
1131         vlc_object_destroy( p_filter );
1132         p_filter = 0;
1133     }
1134
1135     return p_filter;
1136 }
1137
1138 static int transcode_audio_new( sout_stream_t *p_stream,
1139                                 sout_stream_id_t *id )
1140 {
1141     sout_stream_sys_t *p_sys = p_stream->p_sys;
1142     es_format_t fmt_last;
1143     int i_pass = 6;
1144
1145     /*
1146      * Open decoder
1147      */
1148
1149     /* Initialization of decoder structures */
1150     id->p_decoder->fmt_out = id->p_decoder->fmt_in;
1151     id->p_decoder->fmt_out.i_extra = 0;
1152     id->p_decoder->fmt_out.p_extra = 0;
1153     id->p_decoder->pf_decode_audio = 0;
1154     id->p_decoder->pf_aout_buffer_new = audio_new_buffer;
1155     id->p_decoder->pf_aout_buffer_del = audio_del_buffer;
1156     /* id->p_decoder->p_cfg = p_sys->p_audio_cfg; */
1157
1158     id->p_decoder->p_module =
1159         module_Need( id->p_decoder, "decoder", "$codec", 0 );
1160
1161     if( !id->p_decoder->p_module )
1162     {
1163         msg_Err( p_stream, "cannot find decoder" );
1164         return VLC_EGENERIC;
1165     }
1166     id->p_decoder->fmt_out.audio.i_bitspersample = 
1167         audio_BitsPerSample( id->p_decoder->fmt_out.i_codec );
1168     fmt_last = id->p_decoder->fmt_out;
1169     /* FIX decoders so we don't have to do this */
1170     fmt_last.audio.i_rate = id->p_decoder->fmt_in.audio.i_rate;
1171
1172     /*
1173      * Open encoder
1174      */
1175
1176     /* Initialization of encoder format structures */
1177     es_format_Init( &id->p_encoder->fmt_in, id->p_decoder->fmt_in.i_cat,
1178                     id->p_decoder->fmt_out.i_codec );
1179     id->p_encoder->fmt_in.audio.i_format = id->p_decoder->fmt_out.i_codec;
1180
1181     /* Initialization of encoder format structures */
1182     es_format_Init( &id->p_encoder->fmt_in, AUDIO_ES, AOUT_FMT_S16_NE );
1183     id->p_encoder->fmt_in.audio.i_format = AOUT_FMT_S16_NE;
1184     id->p_encoder->fmt_in.audio.i_rate = id->p_encoder->fmt_out.audio.i_rate;
1185     id->p_encoder->fmt_in.audio.i_physical_channels =
1186         id->p_encoder->fmt_out.audio.i_physical_channels;
1187     id->p_encoder->fmt_in.audio.i_original_channels =
1188         id->p_encoder->fmt_out.audio.i_original_channels;
1189     id->p_encoder->fmt_in.audio.i_channels =
1190         id->p_encoder->fmt_out.audio.i_channels;
1191     id->p_encoder->fmt_in.audio.i_bitspersample =
1192         audio_BitsPerSample( id->p_encoder->fmt_in.i_codec );
1193
1194     id->p_encoder->p_cfg = p_stream->p_sys->p_audio_cfg;
1195
1196     id->p_encoder->p_module =
1197         module_Need( id->p_encoder, "encoder", p_sys->psz_aenc, VLC_TRUE );
1198     if( !id->p_encoder->p_module )
1199     {
1200         msg_Err( p_stream, "cannot find encoder" );
1201         module_Unneed( id->p_decoder, id->p_decoder->p_module );
1202         id->p_decoder->p_module = 0;
1203         return VLC_EGENERIC;
1204     }
1205     id->p_encoder->fmt_in.audio.i_format = id->p_encoder->fmt_in.i_codec;
1206     id->p_encoder->fmt_in.audio.i_bitspersample =
1207         audio_BitsPerSample( id->p_encoder->fmt_in.i_codec );
1208
1209     /* Load conversion filters */
1210     if( fmt_last.audio.i_channels != id->p_encoder->fmt_in.audio.i_channels ||
1211         fmt_last.audio.i_rate != id->p_encoder->fmt_in.audio.i_rate )
1212     {
1213         /* We'll have to go through fl32 first */
1214         es_format_t fmt_out = id->p_encoder->fmt_in;
1215         fmt_out.i_codec = fmt_out.audio.i_format = VLC_FOURCC('f','l','3','2');
1216
1217         id->pp_filter[id->i_filter] =
1218             transcode_audio_filter_new( p_stream, id, &fmt_last, &fmt_out );
1219
1220         if( id->pp_filter[id->i_filter] ) id->i_filter++;
1221     }
1222
1223     while( i_pass-- )
1224     {
1225         if( fmt_last.audio.i_channels !=
1226             id->p_encoder->fmt_in.audio.i_channels ||
1227             fmt_last.audio.i_rate != id->p_encoder->fmt_in.audio.i_rate ||
1228             fmt_last.i_codec != id->p_encoder->fmt_in.i_codec )
1229         {
1230             id->pp_filter[id->i_filter] =
1231                 transcode_audio_filter_new( p_stream, id, &fmt_last,
1232                                             &id->p_encoder->fmt_in );
1233
1234             if( id->pp_filter[id->i_filter] ) id->i_filter++;
1235             else break;
1236         }
1237     }
1238
1239     /* Final checks to see if conversions were successful */
1240     if( fmt_last.i_codec != id->p_encoder->fmt_in.i_codec )
1241     {
1242         msg_Err( p_stream, "no audio filter found (%4.4s->%4.4s)",
1243                  (char *)&fmt_last.i_codec,
1244                  (char *)&id->p_encoder->fmt_in.i_codec );
1245         transcode_audio_close( p_stream, id );
1246         return VLC_EGENERIC;
1247     }
1248
1249     if( fmt_last.audio.i_channels != id->p_encoder->fmt_in.audio.i_channels )
1250     {
1251         msg_Err( p_stream, "no audio filter found for mixing from"
1252                  " %i to %i channels", fmt_last.audio.i_channels,
1253                  id->p_encoder->fmt_in.audio.i_channels );
1254 #if 0
1255         /* FIXME : this might work, but only if the encoder is restarted */
1256         id->p_encoder->fmt_in.audio.i_channels = fmt_last.audio.i_channels;
1257         id->p_encoder->fmt_out.audio.i_channels = fmt_last.audio.i_channels;
1258
1259         id->p_encoder->fmt_in.audio.i_physical_channels =
1260             id->p_encoder->fmt_in.audio.i_original_channels =
1261                 fmt_last.audio.i_physical_channels;
1262         id->p_encoder->fmt_out.audio.i_physical_channels =
1263             id->p_encoder->fmt_out.audio.i_original_channels =
1264                 fmt_last.audio.i_physical_channels;
1265 #else
1266         transcode_audio_close( p_stream, id );
1267         return VLC_EGENERIC;
1268 #endif
1269     }
1270
1271     if( fmt_last.audio.i_rate != id->p_encoder->fmt_in.audio.i_rate )
1272     {
1273         msg_Err( p_stream, "no audio filter found for resampling from"
1274                  " %iHz to %iHz", fmt_last.audio.i_rate,
1275                  id->p_encoder->fmt_in.audio.i_rate );
1276 #if 0
1277         /* FIXME : this might work, but only if the encoder is restarted */
1278         id->p_encoder->fmt_in.audio.i_rate = fmt_last.audio.i_rate;
1279         id->p_encoder->fmt_out.audio.i_rate = fmt_last.audio.i_rate;
1280 #else
1281         transcode_audio_close( p_stream, id );
1282         return VLC_EGENERIC;
1283 #endif
1284     }
1285
1286     /* FIXME: Hack for mp3 transcoding support */
1287     if( id->p_encoder->fmt_out.i_codec == VLC_FOURCC( 'm','p','3',' ' ) )
1288         id->p_encoder->fmt_out.i_codec = VLC_FOURCC( 'm','p','g','a' );
1289
1290     return VLC_SUCCESS;
1291 }
1292
1293 static void transcode_audio_close( sout_stream_t *p_stream,
1294                                    sout_stream_id_t *id )
1295 {
1296     int i;
1297
1298     /* Close decoder */
1299     if( id->p_decoder->p_module )
1300         module_Unneed( id->p_decoder, id->p_decoder->p_module );
1301     id->p_decoder->p_module = 0;
1302
1303     /* Close encoder */
1304     if( id->p_encoder->p_module )
1305         module_Unneed( id->p_encoder, id->p_encoder->p_module );
1306     id->p_encoder->p_module = 0;
1307
1308     /* Close filters */
1309     for( i = 0; i < id->i_filter; i++ )
1310     {
1311         vlc_object_detach( id->pp_filter[i] );
1312         if( id->pp_filter[i]->p_module )
1313             module_Unneed( id->pp_filter[i], id->pp_filter[i]->p_module );
1314         vlc_object_destroy( id->pp_filter[i] );
1315     }
1316 }
1317
1318 static int transcode_audio_process( sout_stream_t *p_stream,
1319                                     sout_stream_id_t *id,
1320                                     block_t *in, block_t **out )
1321 {
1322     sout_stream_sys_t *p_sys = p_stream->p_sys;
1323     aout_buffer_t *p_audio_buf;
1324     block_t *p_block, *p_audio_block;
1325     int i;
1326     *out = NULL;
1327
1328     while( (p_audio_buf = id->p_decoder->pf_decode_audio( id->p_decoder,
1329                                                           &in )) )
1330     {
1331         if( p_sys->b_master_sync )
1332         {
1333             mtime_t i_dts = date_Get( &id->interpolated_pts ) + 1;
1334             if ( p_audio_buf->start_date - i_dts > MASTER_SYNC_MAX_DRIFT
1335                   || p_audio_buf->start_date - i_dts < -MASTER_SYNC_MAX_DRIFT )
1336             {
1337                 msg_Dbg( p_stream, "drift is too high, resetting master sync" );
1338                 date_Set( &id->interpolated_pts, p_audio_buf->start_date );
1339                 i_dts = p_audio_buf->start_date + 1;
1340             }
1341             p_sys->i_master_drift = p_audio_buf->start_date - i_dts;
1342             date_Increment( &id->interpolated_pts, p_audio_buf->i_nb_samples );
1343             p_audio_buf->start_date -= p_sys->i_master_drift;
1344             p_audio_buf->end_date -= p_sys->i_master_drift;
1345         }
1346
1347         p_audio_block = p_audio_buf->p_sys;
1348         p_audio_block->i_buffer = p_audio_buf->i_nb_bytes;
1349         p_audio_block->i_dts = p_audio_block->i_pts =
1350             p_audio_buf->start_date;
1351         p_audio_block->i_length = p_audio_buf->end_date -
1352             p_audio_buf->start_date;
1353         p_audio_block->i_samples = p_audio_buf->i_nb_samples;
1354
1355         /* Run filter chain */
1356         for( i = 0; i < id->i_filter; i++ )
1357         {
1358             p_audio_block =
1359                 id->pp_filter[i]->pf_audio_filter( id->pp_filter[i],
1360                                                    p_audio_block );
1361         }
1362
1363         p_audio_buf->p_buffer = p_audio_block->p_buffer;
1364         p_audio_buf->i_nb_bytes = p_audio_block->i_buffer;
1365         p_audio_buf->i_nb_samples = p_audio_block->i_samples;
1366         p_audio_buf->start_date = p_audio_block->i_dts;
1367         p_audio_buf->end_date = p_audio_block->i_dts + p_audio_block->i_length;
1368
1369         p_block = id->p_encoder->pf_encode_audio( id->p_encoder, p_audio_buf );
1370         block_ChainAppend( out, p_block );
1371         block_Release( p_audio_block );
1372         free( p_audio_buf );
1373     }
1374
1375     return VLC_SUCCESS;
1376 }
1377
1378 static void audio_release_buffer( aout_buffer_t *p_buffer )
1379 {
1380     if( p_buffer && p_buffer->p_sys ) block_Release( p_buffer->p_sys );
1381     if( p_buffer ) free( p_buffer );
1382 }
1383
1384 static aout_buffer_t *audio_new_buffer( decoder_t *p_dec, int i_samples )
1385 {
1386     aout_buffer_t *p_buffer;
1387     block_t *p_block;
1388     int i_size;
1389
1390     if( p_dec->fmt_out.audio.i_bitspersample )
1391     {
1392         i_size = i_samples * p_dec->fmt_out.audio.i_bitspersample / 8 *
1393             p_dec->fmt_out.audio.i_channels;
1394     }
1395     else if( p_dec->fmt_out.audio.i_bytes_per_frame &&
1396              p_dec->fmt_out.audio.i_frame_length )
1397     {
1398         i_size = i_samples * p_dec->fmt_out.audio.i_bytes_per_frame /
1399             p_dec->fmt_out.audio.i_frame_length;
1400     }
1401     else
1402     {
1403         i_size = i_samples * 4 * p_dec->fmt_out.audio.i_channels;
1404     }
1405
1406     p_buffer = malloc( sizeof(aout_buffer_t) );
1407     p_buffer->pf_release = audio_release_buffer;
1408     p_buffer->p_sys = p_block = block_New( p_dec, i_size );
1409
1410     p_buffer->p_buffer = p_block->p_buffer;
1411     p_buffer->i_size = p_buffer->i_nb_bytes = p_block->i_buffer;
1412     p_buffer->i_nb_samples = i_samples;
1413     p_block->i_samples = i_samples;
1414
1415     return p_buffer;
1416 }
1417
1418 static void audio_del_buffer( decoder_t *p_dec, aout_buffer_t *p_buffer )
1419 {
1420     if( p_buffer && p_buffer->p_sys ) block_Release( p_buffer->p_sys );
1421     if( p_buffer ) free( p_buffer );
1422 }
1423
1424 /*
1425  * video
1426  */
1427 static int transcode_video_new( sout_stream_t *p_stream, sout_stream_id_t *id )
1428 {
1429     sout_stream_sys_t *p_sys = p_stream->p_sys;
1430     int i;
1431
1432     /*
1433      * Open decoder
1434      */
1435
1436     /* Initialization of decoder structures */
1437     id->p_decoder->fmt_out = id->p_decoder->fmt_in;
1438     id->p_decoder->fmt_out.i_extra = 0;
1439     id->p_decoder->fmt_out.p_extra = 0;
1440     id->p_decoder->pf_decode_video = 0;
1441     id->p_decoder->pf_vout_buffer_new = video_new_buffer_decoder;
1442     id->p_decoder->pf_vout_buffer_del = video_del_buffer_decoder;
1443     id->p_decoder->pf_picture_link    = video_link_picture_decoder;
1444     id->p_decoder->pf_picture_unlink  = video_unlink_picture_decoder;
1445     id->p_decoder->p_owner = malloc( sizeof(decoder_owner_sys_t) );
1446     for( i = 0; i < PICTURE_RING_SIZE; i++ )
1447         id->p_decoder->p_owner->pp_pics[i] = 0;
1448     id->p_decoder->p_owner->p_sys = p_sys;
1449     /* id->p_decoder->p_cfg = p_sys->p_video_cfg; */
1450
1451     id->p_decoder->p_module =
1452         module_Need( id->p_decoder, "decoder", "$codec", 0 );
1453
1454     if( !id->p_decoder->p_module )
1455     {
1456         msg_Err( p_stream, "cannot find decoder" );
1457         return VLC_EGENERIC;
1458     }
1459
1460     /*
1461      * Open encoder.
1462      * Because some info about the decoded input will only be available
1463      * once the first frame is decoded, we actually only test the availability
1464      * of the encoder here.
1465      */
1466
1467     /* Initialization of encoder format structures */
1468     es_format_Init( &id->p_encoder->fmt_in, id->p_decoder->fmt_in.i_cat,
1469                     id->p_decoder->fmt_out.i_codec );
1470     id->p_encoder->fmt_in.video.i_chroma = id->p_decoder->fmt_out.i_codec;
1471
1472     /* The dimensions will be set properly later on.
1473      * Just put sensible values so we can test an encoder is available. */
1474     id->p_encoder->fmt_in.video.i_width =
1475         id->p_encoder->fmt_out.video.i_width ?
1476         id->p_encoder->fmt_out.video.i_width :
1477         id->p_decoder->fmt_in.video.i_width ?
1478         id->p_decoder->fmt_in.video.i_width : 16;
1479     id->p_encoder->fmt_in.video.i_height =
1480         id->p_encoder->fmt_out.video.i_height ?
1481         id->p_encoder->fmt_out.video.i_height :
1482         id->p_decoder->fmt_in.video.i_height ?
1483         id->p_decoder->fmt_in.video.i_height : 16;
1484     id->p_encoder->fmt_in.video.i_frame_rate = 25;
1485     id->p_encoder->fmt_in.video.i_frame_rate_base = 1;
1486
1487     id->p_encoder->i_threads = p_sys->i_threads;
1488     id->p_encoder->p_cfg = p_sys->p_video_cfg;
1489
1490     id->p_encoder->p_module =
1491         module_Need( id->p_encoder, "encoder", p_sys->psz_venc, VLC_TRUE );
1492     if( !id->p_encoder->p_module )
1493     {
1494         msg_Err( p_stream, "cannot find encoder" );
1495         module_Unneed( id->p_decoder, id->p_decoder->p_module );
1496         id->p_decoder->p_module = 0;
1497         return VLC_EGENERIC;
1498     }
1499
1500     /* Close the encoder.
1501      * We'll open it only when we have the first frame. */
1502     module_Unneed( id->p_encoder, id->p_encoder->p_module );
1503     id->p_encoder->p_module = NULL;
1504
1505     if( p_sys->i_threads >= 1 )
1506     {
1507         int i_priority = p_sys->b_high_priority ? VLC_THREAD_PRIORITY_OUTPUT :
1508                            VLC_THREAD_PRIORITY_VIDEO;
1509         p_sys->id_video = id;
1510         vlc_mutex_init( p_stream, &p_sys->lock_out );
1511         vlc_cond_init( p_stream, &p_sys->cond );
1512         memset( p_sys->pp_pics, 0, sizeof(p_sys->pp_pics) );
1513         p_sys->i_first_pic = 0;
1514         p_sys->i_last_pic = 0;
1515         p_sys->p_buffers = NULL;
1516         p_sys->b_die = p_sys->b_error = 0;
1517         if( vlc_thread_create( p_sys, "encoder", EncoderThread, i_priority,
1518                                VLC_FALSE ) )
1519         {
1520             msg_Err( p_stream, "cannot spawn encoder thread" );
1521             module_Unneed( id->p_decoder, id->p_decoder->p_module );
1522             id->p_decoder->p_module = 0;
1523             return VLC_EGENERIC;
1524         }
1525     }
1526
1527     return VLC_SUCCESS;
1528 }
1529
1530 static int transcode_video_encoder_open( sout_stream_t *p_stream,
1531                                          sout_stream_id_t *id )
1532 {
1533     sout_stream_sys_t *p_sys = p_stream->p_sys;
1534
1535     /* Hack because of the copy packetizer which can fail to detect the
1536      * proper size (which forces us to wait until the 1st frame
1537      * is decoded) */
1538     int i_width = id->p_decoder->fmt_out.video.i_width -
1539         p_sys->i_crop_left - p_sys->i_crop_right;
1540     int i_height = id->p_decoder->fmt_out.video.i_height -
1541         p_sys->i_crop_top - p_sys->i_crop_bottom;
1542
1543     if( id->p_encoder->fmt_out.video.i_width <= 0 &&
1544         id->p_encoder->fmt_out.video.i_height <= 0 && p_sys->f_scale )
1545     {
1546         /* Apply the scaling */
1547         id->p_encoder->fmt_out.video.i_width = i_width * p_sys->f_scale;
1548         id->p_encoder->fmt_out.video.i_height = i_height * p_sys->f_scale;
1549     }
1550     else if( id->p_encoder->fmt_out.video.i_width > 0 &&
1551              id->p_encoder->fmt_out.video.i_height <= 0 )
1552     {
1553         id->p_encoder->fmt_out.video.i_height =
1554             id->p_encoder->fmt_out.video.i_width / (double)i_width * i_height;
1555     }
1556     else if( id->p_encoder->fmt_out.video.i_width <= 0 &&
1557              id->p_encoder->fmt_out.video.i_height > 0 )
1558     {
1559         id->p_encoder->fmt_out.video.i_width =
1560             id->p_encoder->fmt_out.video.i_height / (double)i_height * i_width;
1561     }
1562
1563     if( p_sys->i_maxwidth
1564          && id->p_encoder->fmt_out.video.i_width > p_sys->i_maxwidth )
1565         id->p_encoder->fmt_out.video.i_width = p_sys->i_maxwidth;
1566     if( p_sys->i_maxheight
1567          && id->p_encoder->fmt_out.video.i_height > p_sys->i_maxheight )
1568         id->p_encoder->fmt_out.video.i_height = p_sys->i_maxheight;
1569
1570     /* Make sure the size is at least a multiple of 2 */
1571     id->p_encoder->fmt_out.video.i_width =
1572         (id->p_encoder->fmt_out.video.i_width + 1) >> 1 << 1;
1573     id->p_encoder->fmt_out.video.i_height =
1574         (id->p_encoder->fmt_out.video.i_height + 1) >> 1 << 1;
1575
1576     id->p_encoder->fmt_in.video.i_width =
1577         id->p_encoder->fmt_out.video.i_width;
1578     id->p_encoder->fmt_in.video.i_height =
1579         id->p_encoder->fmt_out.video.i_height;
1580
1581     if( !id->p_encoder->fmt_out.video.i_frame_rate ||
1582         !id->p_encoder->fmt_out.video.i_frame_rate_base )
1583     {
1584         if( id->p_decoder->fmt_out.video.i_frame_rate &&
1585             id->p_decoder->fmt_out.video.i_frame_rate_base )
1586         {
1587             id->p_encoder->fmt_out.video.i_frame_rate =
1588                 id->p_decoder->fmt_out.video.i_frame_rate;
1589             id->p_encoder->fmt_out.video.i_frame_rate_base =
1590                 id->p_decoder->fmt_out.video.i_frame_rate_base;
1591         }
1592         else
1593         {
1594             /* Pick a sensible default value */
1595             id->p_encoder->fmt_out.video.i_frame_rate = 25;
1596             id->p_encoder->fmt_out.video.i_frame_rate_base = 1;
1597         }
1598     }
1599
1600     id->p_encoder->fmt_in.video.i_frame_rate =
1601         id->p_encoder->fmt_out.video.i_frame_rate;
1602     id->p_encoder->fmt_in.video.i_frame_rate_base =
1603         id->p_encoder->fmt_out.video.i_frame_rate_base;
1604
1605     date_Init( &id->interpolated_pts,
1606                id->p_encoder->fmt_out.video.i_frame_rate,
1607                id->p_encoder->fmt_out.video.i_frame_rate_base );
1608
1609     /* Check whether a particular aspect ratio was requested */
1610     if( !id->p_encoder->fmt_out.video.i_aspect )
1611     {
1612         id->p_encoder->fmt_out.video.i_aspect =
1613             id->p_decoder->fmt_out.video.i_aspect;
1614     }
1615     id->p_encoder->fmt_in.video.i_aspect =
1616         id->p_encoder->fmt_out.video.i_aspect;
1617
1618     id->p_encoder->p_module =
1619         module_Need( id->p_encoder, "encoder", p_sys->psz_venc, VLC_TRUE );
1620     if( !id->p_encoder->p_module )
1621     {
1622         msg_Err( p_stream, "cannot find encoder" );
1623         return VLC_EGENERIC;
1624     }
1625
1626     id->p_encoder->fmt_in.video.i_chroma = id->p_encoder->fmt_in.i_codec;
1627
1628     /* Hack for mp2v/mp1v transcoding support */
1629     if( id->p_encoder->fmt_out.i_codec == VLC_FOURCC('m','p','1','v') ||
1630         id->p_encoder->fmt_out.i_codec == VLC_FOURCC('m','p','2','v') )
1631     {
1632         id->p_encoder->fmt_out.i_codec = VLC_FOURCC('m','p','g','v');
1633     }
1634
1635     id->id = p_stream->p_sys->p_out->pf_add( p_stream->p_sys->p_out,
1636                                              &id->p_encoder->fmt_out );
1637     if( !id->id )
1638     {
1639         msg_Err( p_stream, "cannot add this stream" );
1640         return VLC_EGENERIC;
1641     }
1642
1643     return VLC_SUCCESS;
1644 }
1645
1646 static void transcode_video_close( sout_stream_t *p_stream,
1647                                    sout_stream_id_t *id )
1648 {
1649     int i, j;
1650
1651     if( p_stream->p_sys->i_threads >= 1 )
1652     {
1653         vlc_mutex_lock( &p_stream->p_sys->lock_out );
1654         p_stream->p_sys->b_die = 1;
1655         vlc_cond_signal( &p_stream->p_sys->cond );
1656         vlc_mutex_unlock( &p_stream->p_sys->lock_out );
1657         vlc_thread_join( p_stream->p_sys );
1658         vlc_mutex_destroy( &p_stream->p_sys->lock_out );
1659         vlc_cond_destroy( &p_stream->p_sys->cond );
1660     }
1661
1662     /* Close decoder */
1663     if( id->p_decoder->p_module )
1664         module_Unneed( id->p_decoder, id->p_decoder->p_module );
1665
1666     if( id->p_decoder->p_owner )
1667     {
1668         /* Clean-up pictures ring buffer */
1669         for( i = 0; i < PICTURE_RING_SIZE; i++ )
1670         {
1671             if( id->p_decoder->p_owner->pp_pics[i] )
1672                 video_del_buffer( VLC_OBJECT(id->p_decoder),
1673                                   id->p_decoder->p_owner->pp_pics[i] );
1674         }
1675         free( id->p_decoder->p_owner );
1676     }
1677
1678     /* Close encoder */
1679     if( id->p_encoder->p_module )
1680         module_Unneed( id->p_encoder, id->p_encoder->p_module );
1681
1682     /* Close filters */
1683     for( i = 0; i < id->i_filter; i++ )
1684     {
1685         vlc_object_detach( id->pp_filter[i] );
1686         if( id->pp_filter[i]->p_module )
1687             module_Unneed( id->pp_filter[i], id->pp_filter[i]->p_module );
1688
1689         /* Clean-up pictures ring buffer */
1690         for( j = 0; j < PICTURE_RING_SIZE; j++ )
1691         {
1692             if( id->pp_filter[i]->p_owner->pp_pics[j] )
1693                 video_del_buffer( VLC_OBJECT(id->pp_filter[i]),
1694                                   id->pp_filter[i]->p_owner->pp_pics[j] );
1695         }
1696         free( id->pp_filter[i]->p_owner );
1697
1698         vlc_object_destroy( id->pp_filter[i] );
1699     }
1700     for( i = 0; i < id->i_vfilter; i++ )
1701     {
1702         vlc_object_detach( id->pp_vfilter[i] );
1703         if( id->pp_vfilter[i]->p_module )
1704             module_Unneed( id->pp_vfilter[i], id->pp_vfilter[i]->p_module );
1705
1706         /* Clean-up pictures ring buffer */
1707         for( j = 0; j < PICTURE_RING_SIZE; j++ )
1708         {
1709             if( id->pp_vfilter[i]->p_owner->pp_pics[j] )
1710                 video_del_buffer( VLC_OBJECT(id->pp_vfilter[i]),
1711                                   id->pp_vfilter[i]->p_owner->pp_pics[j] );
1712         }
1713         free( id->pp_vfilter[i]->p_owner );
1714
1715         vlc_object_destroy( id->pp_vfilter[i] );
1716     }
1717 }
1718
1719 static int transcode_video_process( sout_stream_t *p_stream,
1720                                     sout_stream_id_t *id,
1721                                     block_t *in, block_t **out )
1722 {
1723     sout_stream_sys_t *p_sys = p_stream->p_sys;
1724     int i_duplicate = 1, i;
1725     picture_t *p_pic, *p_pic2 = NULL;
1726     *out = NULL;
1727
1728     while( (p_pic = id->p_decoder->pf_decode_video( id->p_decoder, &in )) )
1729     {
1730         subpicture_t *p_subpic = 0;
1731
1732         if( p_stream->p_sout->i_out_pace_nocontrol && p_sys->b_hurry_up )
1733         {
1734             mtime_t current_date = mdate();
1735             if( current_date + 50000 > p_pic->date )
1736             {
1737                 msg_Dbg( p_stream, "late picture skipped ("I64Fd")",
1738                          current_date + 50000 - p_pic->date );
1739                 p_pic->pf_release( p_pic );
1740                 continue;
1741             }
1742         }
1743
1744         if( p_sys->b_master_sync )
1745         {
1746             mtime_t i_video_drift;
1747             mtime_t i_master_drift = p_sys->i_master_drift;
1748             mtime_t i_pts;
1749
1750             i_pts = date_Get( &id->interpolated_pts ) + 1;
1751             if ( p_pic->date - i_pts > MASTER_SYNC_MAX_DRIFT
1752                   || p_pic->date - i_pts < -MASTER_SYNC_MAX_DRIFT )
1753             {
1754                 msg_Dbg( p_stream, "drift is too high, resetting master sync" );
1755                 date_Set( &id->interpolated_pts, p_pic->date );
1756                 i_pts = p_pic->date + 1;
1757             }
1758             i_video_drift = p_pic->date - i_pts;
1759             i_duplicate = 1;
1760
1761             /* Set the pts of the frame being encoded */
1762             p_pic->date = i_pts;
1763
1764             if( i_video_drift < i_master_drift - 50000 )
1765             {
1766 #if 0
1767                 msg_Dbg( p_stream, "dropping frame (%i)",
1768                          (int)(i_video_drift - i_master_drift) );
1769 #endif
1770                 p_pic->pf_release( p_pic );
1771                 continue;
1772             }
1773             else if( i_video_drift > i_master_drift + 50000 )
1774             {
1775 #if 0
1776                 msg_Dbg( p_stream, "adding frame (%i)",
1777                          (int)(i_video_drift - i_master_drift) );
1778 #endif
1779                 i_duplicate = 2;
1780             }
1781         }
1782
1783         if( !id->p_encoder->p_module )
1784         {
1785             if( transcode_video_encoder_open( p_stream, id ) != VLC_SUCCESS )
1786             {
1787                 p_pic->pf_release( p_pic );
1788                 transcode_video_close( p_stream, id );
1789                 id->b_transcode = VLC_FALSE;
1790                 return VLC_EGENERIC;
1791             }
1792
1793             /* Deinterlace */
1794             if( p_stream->p_sys->b_deinterlace )
1795             {
1796                 id->pp_filter[id->i_filter] =
1797                     vlc_object_create( p_stream, VLC_OBJECT_FILTER );
1798                 vlc_object_attach( id->pp_filter[id->i_filter], p_stream );
1799
1800                 id->pp_filter[id->i_filter]->pf_vout_buffer_new =
1801                     video_new_buffer_filter;
1802                 id->pp_filter[id->i_filter]->pf_vout_buffer_del =
1803                     video_del_buffer_filter;
1804
1805                 id->pp_filter[id->i_filter]->fmt_in = id->p_decoder->fmt_out;
1806                 id->pp_filter[id->i_filter]->fmt_out = id->p_decoder->fmt_out;
1807                 id->pp_filter[id->i_filter]->p_cfg = p_sys->p_deinterlace_cfg;
1808                 id->pp_filter[id->i_filter]->p_module =
1809                     module_Need( id->pp_filter[id->i_filter],
1810                                  "video filter2", p_sys->psz_deinterlace, 0 );
1811                 if( id->pp_filter[id->i_filter]->p_module )
1812                 {
1813                     id->pp_filter[id->i_filter]->p_owner =
1814                         malloc( sizeof(filter_owner_sys_t) );
1815                     for( i = 0; i < PICTURE_RING_SIZE; i++ )
1816                         id->pp_filter[id->i_filter]->p_owner->pp_pics[i] = 0;
1817                     id->pp_filter[id->i_filter]->p_owner->p_sys = p_sys;
1818
1819                     id->i_filter++;
1820                 }
1821                 else
1822                 {
1823                     msg_Dbg( p_stream, "no video filter found" );
1824                     vlc_object_detach( id->pp_filter[id->i_filter] );
1825                     vlc_object_destroy( id->pp_filter[id->i_filter] );
1826                 }
1827             }
1828
1829             /* Check if we need a filter for chroma conversion or resizing */
1830             if( id->p_decoder->fmt_out.video.i_chroma !=
1831                 id->p_encoder->fmt_in.video.i_chroma ||
1832                 id->p_decoder->fmt_out.video.i_width !=
1833                 id->p_encoder->fmt_out.video.i_width ||
1834                 id->p_decoder->fmt_out.video.i_height !=
1835                 id->p_encoder->fmt_out.video.i_height ||
1836                 p_sys->i_crop_top > 0 || p_sys->i_crop_bottom > 0 ||
1837                 p_sys->i_crop_left > 0 || p_sys->i_crop_right > 0 )
1838             {
1839                 id->pp_filter[id->i_filter] =
1840                     vlc_object_create( p_stream, VLC_OBJECT_FILTER );
1841                 vlc_object_attach( id->pp_filter[id->i_filter], p_stream );
1842
1843                 id->pp_filter[id->i_filter]->pf_vout_buffer_new =
1844                     video_new_buffer_filter;
1845                 id->pp_filter[id->i_filter]->pf_vout_buffer_del =
1846                     video_del_buffer_filter;
1847
1848                 id->pp_filter[id->i_filter]->fmt_in = id->p_decoder->fmt_out;
1849                 id->pp_filter[id->i_filter]->fmt_out = id->p_encoder->fmt_in;
1850                 id->pp_filter[id->i_filter]->p_cfg = NULL;
1851                 id->pp_filter[id->i_filter]->p_module =
1852                     module_Need( id->pp_filter[id->i_filter],
1853                                  "video filter2", 0, 0 );
1854                 if( id->pp_filter[id->i_filter]->p_module )
1855                 {
1856                     id->pp_filter[id->i_filter]->p_owner =
1857                         malloc( sizeof(filter_owner_sys_t) );
1858                     for( i = 0; i < PICTURE_RING_SIZE; i++ )
1859                         id->pp_filter[id->i_filter]->p_owner->pp_pics[i] = 0;
1860                     id->pp_filter[id->i_filter]->p_owner->p_sys = p_sys;
1861
1862                     id->i_filter++;
1863                 }
1864                 else
1865                 {
1866                     msg_Dbg( p_stream, "no video filter found" );
1867                     vlc_object_detach( id->pp_filter[id->i_filter] );
1868                     vlc_object_destroy( id->pp_filter[id->i_filter] );
1869
1870                     p_pic->pf_release( p_pic );
1871                     transcode_video_close( p_stream, id );
1872                     id->b_transcode = VLC_FALSE;
1873                     return VLC_EGENERIC;
1874                 }
1875             }
1876
1877             for( i = 0; i < p_sys->i_vfilters; i++ )
1878             {
1879                 id->pp_vfilter[id->i_vfilter] =
1880                     vlc_object_create( p_stream, VLC_OBJECT_FILTER );
1881                 vlc_object_attach( id->pp_vfilter[id->i_vfilter], p_stream );
1882
1883                 id->pp_vfilter[id->i_vfilter]->pf_vout_buffer_new =
1884                     video_new_buffer_filter;
1885                 id->pp_vfilter[id->i_vfilter]->pf_vout_buffer_del =
1886                     video_del_buffer_filter;
1887
1888                 id->pp_vfilter[id->i_vfilter]->fmt_in = id->p_encoder->fmt_in;
1889                 id->pp_vfilter[id->i_vfilter]->fmt_out = id->p_encoder->fmt_in;
1890                 id->pp_vfilter[id->i_vfilter]->p_cfg = p_sys->p_vfilters_cfg[i];
1891                 id->pp_vfilter[id->i_vfilter]->p_module =
1892                     module_Need( id->pp_vfilter[id->i_vfilter],
1893                           "video filter2", p_sys->psz_vfilters[i], 0 );
1894                 if( id->pp_vfilter[id->i_vfilter]->p_module )
1895                 {
1896                     id->pp_vfilter[id->i_vfilter]->p_owner =
1897                         malloc( sizeof(filter_owner_sys_t) );
1898                     for( i = 0; i < PICTURE_RING_SIZE; i++ )
1899                         id->pp_vfilter[id->i_vfilter]->p_owner->pp_pics[i] = 0;
1900                     id->pp_vfilter[id->i_vfilter]->p_owner->p_sys = p_sys;
1901
1902                     id->i_vfilter++;
1903                 }
1904                 else
1905                 {
1906                     msg_Dbg( p_stream, "no video filter found" );
1907                     vlc_object_detach( id->pp_vfilter[id->i_vfilter] );
1908                     vlc_object_destroy( id->pp_vfilter[id->i_vfilter] );
1909                 }
1910             }
1911         }
1912
1913         /* Run filter chain */
1914         for( i = 0; i < id->i_filter; i++ )
1915         {
1916             p_pic = id->pp_filter[i]->pf_video_filter(id->pp_filter[i], p_pic);
1917         }
1918
1919         /*
1920          * Encoding
1921          */
1922
1923         /* Check if we have a subpicture to overlay */
1924         if( p_sys->p_spu )
1925         {
1926             p_subpic = spu_SortSubpictures( p_sys->p_spu, p_pic->date );
1927             /* TODO: get another pic */
1928         }
1929
1930         /* Overlay subpicture */
1931         if( p_subpic )
1932         {
1933             int i_scale_width, i_scale_height;
1934             video_format_t *p_fmt;
1935
1936             i_scale_width = id->p_encoder->fmt_in.video.i_width * 1000 /
1937                 id->p_decoder->fmt_out.video.i_width;
1938             i_scale_height = id->p_encoder->fmt_in.video.i_height * 1000 /
1939                 id->p_decoder->fmt_out.video.i_height;
1940
1941             if( p_pic->i_refcount && !id->i_filter )
1942             {
1943                 /* We can't modify the picture, we need to duplicate it */
1944                 picture_t *p_tmp = video_new_buffer_decoder( id->p_decoder );
1945                 if( p_tmp )
1946                 {
1947                     vout_CopyPicture( p_stream, p_tmp, p_pic );
1948                     p_pic->pf_release( p_pic );
1949                     p_pic = p_tmp;
1950                 }
1951             }
1952
1953             if( id->i_filter )
1954                 p_fmt = &id->pp_filter[id->i_filter -1]->fmt_out.video;
1955             else
1956                 p_fmt = &id->p_decoder->fmt_out.video;
1957
1958             /* FIXME (shouldn't have to be done here) */
1959             p_fmt->i_sar_num = p_fmt->i_aspect *
1960                 p_fmt->i_height / p_fmt->i_width;
1961             p_fmt->i_sar_den = VOUT_ASPECT_FACTOR;
1962
1963             spu_RenderSubpictures( p_sys->p_spu, p_fmt, p_pic, p_pic, p_subpic,
1964                                    i_scale_width, i_scale_height );
1965         }
1966
1967         /* Run vfilter chain */
1968         for( i = 0; i < id->i_vfilter; i++ )
1969         {
1970             p_pic = id->pp_vfilter[i]->pf_video_filter(id->pp_vfilter[i], p_pic);
1971         }
1972
1973         if( p_sys->i_threads == 0 )
1974         {
1975             block_t *p_block;
1976             p_block = id->p_encoder->pf_encode_video( id->p_encoder, p_pic );
1977             block_ChainAppend( out, p_block );
1978         }
1979
1980         if( p_sys->b_master_sync )
1981         {
1982             mtime_t i_pts = date_Get( &id->interpolated_pts ) + 1;
1983             if ( p_pic->date - i_pts > MASTER_SYNC_MAX_DRIFT
1984                   || p_pic->date - i_pts < -MASTER_SYNC_MAX_DRIFT )
1985             {
1986                 msg_Dbg( p_stream, "drift is too high, resetting master sync" );
1987                 date_Set( &id->interpolated_pts, p_pic->date );
1988                 i_pts = p_pic->date + 1;
1989             }
1990             date_Increment( &id->interpolated_pts, 1 );
1991         }
1992
1993         if( p_sys->b_master_sync && i_duplicate > 1 )
1994         {
1995             mtime_t i_pts = date_Get( &id->interpolated_pts ) + 1;
1996             if ( p_pic->date - i_pts > MASTER_SYNC_MAX_DRIFT
1997                   || p_pic->date - i_pts < -MASTER_SYNC_MAX_DRIFT )
1998             {
1999                 msg_Dbg( p_stream, "drift is too high, resetting master sync" );
2000                 date_Set( &id->interpolated_pts, p_pic->date );
2001                 i_pts = p_pic->date + 1;
2002             }
2003             date_Increment( &id->interpolated_pts, 1 );
2004
2005             if( p_sys->i_threads >= 1 )
2006             {
2007                 /* We can't modify the picture, we need to duplicate it */
2008                 p_pic2 = video_new_buffer_decoder( id->p_decoder );
2009                 if( p_pic2 != NULL )
2010                 {
2011                     vout_CopyPicture( p_stream, p_pic2, p_pic );
2012                     p_pic2->date = i_pts;
2013                 }
2014             }
2015             else
2016             {
2017                 block_t *p_block;
2018                 p_pic->date = i_pts;
2019                 p_block = id->p_encoder->pf_encode_video(id->p_encoder, p_pic);
2020                 block_ChainAppend( out, p_block );
2021             }
2022         }
2023
2024         if( p_sys->i_threads == 0 )
2025         {
2026             p_pic->pf_release( p_pic );
2027         }
2028         else
2029         {
2030             vlc_mutex_lock( &p_sys->lock_out );
2031             p_sys->pp_pics[p_sys->i_last_pic++] = p_pic;
2032             p_sys->i_last_pic %= PICTURE_RING_SIZE;
2033             *out = p_sys->p_buffers;
2034             p_sys->p_buffers = NULL;
2035             if( p_pic2 != NULL )
2036             {
2037                 p_sys->pp_pics[p_sys->i_last_pic++] = p_pic2;
2038                 p_sys->i_last_pic %= PICTURE_RING_SIZE;
2039             }
2040             vlc_cond_signal( &p_sys->cond );
2041             vlc_mutex_unlock( &p_sys->lock_out );
2042         }
2043
2044     }
2045
2046     return VLC_SUCCESS;
2047 }
2048
2049 static int EncoderThread( sout_stream_sys_t *p_sys )
2050 {
2051     sout_stream_id_t *id = p_sys->id_video;
2052     picture_t *p_pic;
2053
2054     while( !p_sys->b_die && !p_sys->b_error )
2055     {
2056         block_t *p_block;
2057
2058         vlc_mutex_lock( &p_sys->lock_out );
2059         while( p_sys->i_last_pic == p_sys->i_first_pic )
2060         {
2061             vlc_cond_wait( &p_sys->cond, &p_sys->lock_out );
2062             if( p_sys->b_die || p_sys->b_error ) break;
2063         }
2064         if( p_sys->b_die || p_sys->b_error )
2065         {
2066             vlc_mutex_unlock( &p_sys->lock_out );
2067             break;
2068         }
2069
2070         p_pic = p_sys->pp_pics[p_sys->i_first_pic++];
2071         p_sys->i_first_pic %= PICTURE_RING_SIZE;
2072         vlc_mutex_unlock( &p_sys->lock_out );
2073
2074         p_block = id->p_encoder->pf_encode_video( id->p_encoder, p_pic );
2075         vlc_mutex_lock( &p_sys->lock_out );
2076         block_ChainAppend( &p_sys->p_buffers, p_block );
2077
2078         vlc_mutex_unlock( &p_sys->lock_out );
2079
2080         p_pic->pf_release( p_pic );
2081     }
2082
2083     while( p_sys->i_last_pic != p_sys->i_first_pic )
2084     {
2085         p_pic = p_sys->pp_pics[p_sys->i_first_pic++];
2086         p_sys->i_first_pic %= PICTURE_RING_SIZE;
2087
2088         p_pic->pf_release( p_pic );
2089     }
2090
2091     block_ChainRelease( p_sys->p_buffers );
2092
2093     return 0;
2094 }
2095
2096 struct picture_sys_t
2097 {
2098     vlc_object_t *p_owner;
2099 };
2100
2101 static void video_release_buffer( picture_t *p_pic )
2102 {
2103     if( p_pic && !p_pic->i_refcount && p_pic->pf_release && p_pic->p_sys )
2104     {
2105         video_del_buffer_decoder( (decoder_t *)p_pic->p_sys->p_owner, p_pic );
2106     }
2107     else if( p_pic && p_pic->i_refcount > 0 ) p_pic->i_refcount--;
2108 }
2109
2110 static picture_t *video_new_buffer( vlc_object_t *p_this, picture_t **pp_ring,
2111                                     sout_stream_sys_t *p_sys )
2112 {
2113     decoder_t *p_dec = (decoder_t *)p_this;
2114     picture_t *p_pic;
2115     int i;
2116
2117     /* Find an empty space in the picture ring buffer */
2118     for( i = 0; i < PICTURE_RING_SIZE; i++ )
2119     {
2120         if( pp_ring[i] != 0 && pp_ring[i]->i_status == DESTROYED_PICTURE )
2121         {
2122             pp_ring[i]->i_status = RESERVED_PICTURE;
2123             return pp_ring[i];
2124         }
2125     }
2126     for( i = 0; i < PICTURE_RING_SIZE; i++ )
2127     {
2128         if( pp_ring[i] == 0 ) break;
2129     }
2130
2131     if( i == PICTURE_RING_SIZE && p_sys->i_threads >= 1 )
2132     {
2133         int i_first_pic = p_sys->i_first_pic;
2134
2135         if( p_sys->i_first_pic != p_sys->i_last_pic )
2136         {
2137             /* Encoder still has stuff to encode, wait to clear-up the list */
2138             while( p_sys->i_first_pic == i_first_pic )
2139                 msleep( 100000 );
2140         }
2141
2142         /* Find an empty space in the picture ring buffer */
2143         for( i = 0; i < PICTURE_RING_SIZE; i++ )
2144         {
2145             if( pp_ring[i] != 0 && pp_ring[i]->i_status == DESTROYED_PICTURE )
2146             {
2147                 pp_ring[i]->i_status = RESERVED_PICTURE;
2148                 return pp_ring[i];
2149             }
2150         }
2151         for( i = 0; i < PICTURE_RING_SIZE; i++ )
2152         {
2153             if( pp_ring[i] == 0 ) break;
2154         }
2155     }
2156
2157     if( i == PICTURE_RING_SIZE )
2158     {
2159         msg_Err( p_this, "decoder/filter is leaking pictures, "
2160                  "resetting its ring buffer" );
2161
2162         for( i = 0; i < PICTURE_RING_SIZE; i++ )
2163         {
2164             pp_ring[i]->pf_release( pp_ring[i] );
2165         }
2166
2167         i = 0;
2168     }
2169
2170     p_pic = malloc( sizeof(picture_t) );
2171     p_dec->fmt_out.video.i_chroma = p_dec->fmt_out.i_codec;
2172     vout_AllocatePicture( VLC_OBJECT(p_dec), p_pic,
2173                           p_dec->fmt_out.video.i_chroma,
2174                           p_dec->fmt_out.video.i_width,
2175                           p_dec->fmt_out.video.i_height,
2176                           p_dec->fmt_out.video.i_aspect );
2177
2178     if( !p_pic->i_planes )
2179     {
2180         free( p_pic );
2181         return 0;
2182     }
2183
2184     p_pic->pf_release = video_release_buffer;
2185     p_pic->p_sys = malloc( sizeof(picture_sys_t) );
2186     p_pic->p_sys->p_owner = p_this;
2187     p_pic->i_status = RESERVED_PICTURE;
2188
2189     pp_ring[i] = p_pic;
2190
2191     return p_pic;
2192 }
2193
2194 static picture_t *video_new_buffer_decoder( decoder_t *p_dec )
2195 {
2196     return video_new_buffer( VLC_OBJECT(p_dec),
2197                              p_dec->p_owner->pp_pics, p_dec->p_owner->p_sys );
2198 }
2199
2200 static picture_t *video_new_buffer_filter( filter_t *p_filter )
2201 {
2202     return video_new_buffer( VLC_OBJECT(p_filter),
2203                              p_filter->p_owner->pp_pics,
2204                              p_filter->p_owner->p_sys );
2205 }
2206
2207 static void video_del_buffer( vlc_object_t *p_this, picture_t *p_pic )
2208 {
2209     if( p_pic && p_pic->p_data_orig ) free( p_pic->p_data_orig );
2210     if( p_pic && p_pic->p_sys ) free( p_pic->p_sys );
2211     if( p_pic ) free( p_pic );
2212 }
2213
2214 static void video_del_buffer_decoder( decoder_t *p_decoder, picture_t *p_pic )
2215 {
2216     p_pic->i_refcount = 0;
2217     p_pic->i_status = DESTROYED_PICTURE;
2218 }
2219
2220 static void video_del_buffer_filter( filter_t *p_filter, picture_t *p_pic )
2221 {
2222     p_pic->i_refcount = 0;
2223     p_pic->i_status = DESTROYED_PICTURE;
2224 }
2225
2226 static void video_link_picture_decoder( decoder_t *p_dec, picture_t *p_pic )
2227 {
2228     p_pic->i_refcount++;
2229 }
2230
2231 static void video_unlink_picture_decoder( decoder_t *p_dec, picture_t *p_pic )
2232 {
2233     video_release_buffer( p_pic );
2234 }
2235
2236 /*
2237  * SPU
2238  */
2239 static subpicture_t *spu_new_buffer( decoder_t * );
2240 static void spu_del_buffer( decoder_t *, subpicture_t * );
2241
2242 static int transcode_spu_new( sout_stream_t *p_stream, sout_stream_id_t *id )
2243 {
2244     sout_stream_sys_t *p_sys = p_stream->p_sys;
2245
2246     /*
2247      * Open decoder
2248      */
2249
2250     /* Initialization of decoder structures */
2251     id->p_decoder->pf_spu_buffer_new = spu_new_buffer;
2252     id->p_decoder->pf_spu_buffer_del = spu_del_buffer;
2253     id->p_decoder->p_owner = (decoder_owner_sys_t *)p_stream;
2254     /* id->p_decoder->p_cfg = p_sys->p_spu_cfg; */
2255
2256     id->p_decoder->p_module =
2257         module_Need( id->p_decoder, "decoder", "$codec", 0 );
2258
2259     if( !id->p_decoder->p_module )
2260     {
2261         msg_Err( p_stream, "cannot find decoder" );
2262         return VLC_EGENERIC;
2263     }
2264
2265     if( !p_sys->b_soverlay )
2266     {
2267         /*
2268          * Open encoder
2269          */
2270
2271         /* Initialization of encoder format structures */
2272         es_format_Init( &id->p_encoder->fmt_in, id->p_decoder->fmt_in.i_cat,
2273                         id->p_decoder->fmt_in.i_codec );
2274
2275         id->p_encoder->p_cfg = p_sys->p_spu_cfg;
2276
2277         id->p_encoder->p_module =
2278             module_Need( id->p_encoder, "encoder", p_sys->psz_senc, VLC_TRUE );
2279
2280         if( !id->p_encoder->p_module )
2281         {
2282             module_Unneed( id->p_decoder, id->p_decoder->p_module );
2283             msg_Err( p_stream, "cannot find encoder" );
2284             return VLC_EGENERIC;
2285         }
2286     }
2287
2288     if( !p_sys->p_spu )
2289     {
2290         p_sys->p_spu = spu_Create( p_stream );
2291         spu_Init( p_sys->p_spu );
2292     }
2293
2294     return VLC_SUCCESS;
2295 }
2296
2297 static void transcode_spu_close( sout_stream_t *p_stream, sout_stream_id_t *id)
2298 {
2299     /* Close decoder */
2300     if( id->p_decoder->p_module )
2301         module_Unneed( id->p_decoder, id->p_decoder->p_module );
2302
2303     /* Close encoder */
2304     if( id->p_encoder->p_module )
2305         module_Unneed( id->p_encoder, id->p_encoder->p_module );
2306 }
2307
2308 static int transcode_spu_process( sout_stream_t *p_stream,
2309                                   sout_stream_id_t *id,
2310                                   block_t *in, block_t **out )
2311 {
2312     sout_stream_sys_t *p_sys = p_stream->p_sys;
2313     subpicture_t *p_subpic;
2314     *out = NULL;
2315
2316     p_subpic = id->p_decoder->pf_decode_sub( id->p_decoder, &in );
2317     if( !p_subpic ) return VLC_EGENERIC;
2318
2319     if( p_sys->b_master_sync && p_sys->i_master_drift )
2320     {
2321         p_subpic->i_start -= p_sys->i_master_drift;
2322         if( p_subpic->i_stop ) p_subpic->i_stop -= p_sys->i_master_drift;
2323     }
2324
2325     if( p_sys->b_soverlay )
2326     {
2327         spu_DisplaySubpicture( p_sys->p_spu, p_subpic );
2328     }
2329     else
2330     {
2331         block_t *p_block;
2332
2333         p_block = id->p_encoder->pf_encode_sub( id->p_encoder, p_subpic );
2334         spu_del_buffer( id->p_decoder, p_subpic );
2335
2336         if( p_block )
2337         {
2338             block_ChainAppend( out, p_block );
2339             return VLC_SUCCESS;
2340         }
2341     }
2342
2343     return VLC_EGENERIC;
2344 }
2345
2346 static subpicture_t *spu_new_buffer( decoder_t *p_dec )
2347 {
2348     sout_stream_t *p_stream = (sout_stream_t *)p_dec->p_owner;
2349     return spu_CreateSubpicture( p_stream->p_sys->p_spu );
2350 }
2351
2352 static void spu_del_buffer( decoder_t *p_dec, subpicture_t *p_subpic )
2353 {
2354     sout_stream_t *p_stream = (sout_stream_t *)p_dec->p_owner;
2355     spu_DestroySubpicture( p_stream->p_sys->p_spu, p_subpic );
2356 }
2357
2358 /*
2359  * OSD menu
2360  */
2361 static int transcode_osd_new( sout_stream_t *p_stream, sout_stream_id_t *id )
2362 {
2363     sout_stream_sys_t *p_sys = p_stream->p_sys;
2364     es_format_t fmt;
2365
2366     fmt.i_cat = SPU_ES;
2367     fmt.i_id = 0xbd1f; /* pid ?? */
2368     fmt.i_group = 3;   /* pmt entry ?? */
2369     fmt.i_codec = VLC_FOURCC( 'Y', 'U', 'V', 'A' );
2370     fmt.psz_language = strdup( "osd" );
2371
2372     id = malloc( sizeof( sout_stream_id_t ) );
2373     memset( id, 0, sizeof(sout_stream_id_t) );
2374
2375     id->id = NULL;
2376     id->p_decoder = NULL;
2377     id->p_encoder = NULL;
2378
2379     /* Create encoder object */
2380     id->p_encoder = vlc_object_create( p_stream, VLC_OBJECT_ENCODER );
2381     if( !id->p_encoder )
2382     {
2383         msg_Err( p_stream, "out of memory" );
2384         goto error;
2385     }
2386     vlc_object_attach( id->p_encoder, p_stream );
2387     id->p_encoder->p_module = NULL;
2388
2389     /* Create fake destination format */
2390     es_format_Init( &id->p_encoder->fmt_out, fmt.i_cat, 0 );
2391     id->p_encoder->fmt_out.i_id    = fmt.i_id;
2392     id->p_encoder->fmt_out.i_group = fmt.i_group;
2393     id->p_encoder->fmt_out.psz_language = strdup( fmt.psz_language );
2394     
2395     if( p_sys->i_osdcodec != 0 || p_sys->psz_osdenc )
2396     {
2397         msg_Dbg( p_stream, "creating osdmenu transcoding from fcc=`%4.4s' "
2398                  "to fcc=`%4.4s'", (char*)&fmt.i_codec,
2399                  (char*)&p_sys->i_osdcodec );
2400
2401         /* Complete destination format */
2402         id->p_encoder->fmt_out.i_codec = p_sys->i_osdcodec;
2403         
2404         /*
2405          * Open encoder
2406          */
2407         
2408         /* Initialization of encoder format structures */
2409         es_format_Init( &id->p_encoder->fmt_in, fmt.i_cat, fmt.i_codec );
2410         id->p_encoder->fmt_in.psz_language = strdup( fmt.psz_language );
2411
2412         id->p_encoder->p_cfg = p_sys->p_osd_cfg;
2413
2414         id->p_encoder->p_module =
2415             module_Need( id->p_encoder, "encoder", p_sys->psz_osdenc, VLC_TRUE );
2416
2417         if( !id->p_encoder->p_module )
2418         {
2419             msg_Err( p_stream, "cannot find encoder" );
2420             goto error;
2421         }
2422         
2423         /* open output stream */
2424         id->id = p_sys->p_out->pf_add( p_sys->p_out, &id->p_encoder->fmt_out );
2425         id->b_transcode = VLC_TRUE;
2426
2427         if( !id->id ) goto error;
2428     }
2429     else
2430     {
2431         msg_Dbg( p_stream, "not transcoding a stream (fcc=`%4.4s')",
2432                  (char*)&fmt.i_codec );
2433         id->id = p_sys->p_out->pf_add( p_sys->p_out, &fmt );
2434         id->b_transcode = VLC_FALSE;
2435
2436         if( !id->id ) goto error;
2437     }
2438
2439     p_sys->id_osd = id;
2440     p_sys->b_es_osd = VLC_TRUE;
2441
2442     if( !p_sys->p_spu )
2443     {
2444         p_sys->p_spu = spu_Create( p_stream );
2445         if( spu_Init( p_sys->p_spu ) != VLC_SUCCESS )
2446             msg_Err( p_sys, "spu initialisation failed" );
2447     }
2448     
2449     if( fmt.psz_language )
2450         free( fmt.psz_language );
2451         
2452     return VLC_SUCCESS;
2453     
2454  error:
2455     msg_Err( p_stream, "starting osd encoding thread failed" );
2456     if( id->p_encoder->p_module )
2457             module_Unneed( id->p_encoder, id->p_encoder->p_module );
2458     if( id->p_encoder )
2459     {
2460         vlc_object_detach( id->p_encoder );
2461         vlc_object_destroy( id->p_encoder );
2462     }
2463     if( fmt.psz_language ) free( fmt.psz_language );
2464     if( id ) free( id );
2465     p_sys->id_osd = NULL;
2466     p_sys->b_es_osd = VLC_FALSE;
2467     return VLC_EGENERIC;
2468 }
2469     
2470 static void transcode_osd_close( sout_stream_t *p_stream, sout_stream_id_t *id)
2471 {    
2472     sout_stream_sys_t *p_sys = p_stream->p_sys;
2473     
2474     /* Close encoder */
2475     if( p_sys->b_es_osd && id )
2476     {
2477         if( id->p_encoder->p_module )
2478             module_Unneed( id->p_encoder, id->p_encoder->p_module );
2479         
2480         if( id->id ) p_sys->p_out->pf_del( p_sys->p_out, id->id );
2481     
2482         if( id->p_encoder )
2483         {
2484             vlc_object_detach( id->p_encoder );
2485             vlc_object_destroy( id->p_encoder );
2486         }
2487     }
2488     p_sys->b_es_osd = VLC_FALSE;
2489     if( id ) free( id );
2490 }
2491
2492 static int transcode_osd_process( sout_stream_t *p_stream,
2493                                   sout_stream_id_t *id,
2494                                   block_t *in, block_t **out )
2495 {
2496     sout_stream_sys_t *p_sys = p_stream->p_sys;
2497     subpicture_t *p_subpic = NULL;
2498     
2499     /* Check if we have a subpicture to send */
2500     if( p_sys->p_spu && in->i_dts > 0)
2501     {
2502         p_subpic = spu_SortSubpictures( p_sys->p_spu, in->i_dts );
2503     }
2504     else
2505     {
2506         msg_Warn( p_stream, "spu channel not initialized, doing it now" );
2507         if( !p_sys->p_spu )
2508         {
2509             p_sys->p_spu = spu_Create( p_stream );
2510             if( spu_Init( p_sys->p_spu ) != VLC_SUCCESS )
2511                 msg_Err( p_stream, "spu initialisation failed" );
2512         }
2513     }
2514         
2515     if( p_subpic )
2516     {
2517         block_t *p_block = NULL;
2518         
2519         if( p_sys->b_master_sync && p_sys->i_master_drift )
2520         {
2521             p_subpic->i_start -= p_sys->i_master_drift;
2522             if( p_subpic->i_stop ) p_subpic->i_stop -= p_sys->i_master_drift;
2523         }
2524     
2525         p_block = p_sys->id_osd->p_encoder->pf_encode_sub( p_sys->id_osd->p_encoder, p_subpic );
2526         if( p_block )
2527         {
2528             p_block->i_dts = p_block->i_pts = in->i_dts;
2529             block_ChainAppend( out, p_block );
2530             if( *out )
2531             {
2532                 if( p_sys->p_out->pf_send( p_sys->p_out, p_sys->id_osd->id, *out ) == VLC_SUCCESS )
2533                     spu_DestroySubpicture( p_sys->p_spu, p_subpic );
2534             }            
2535             return VLC_SUCCESS;            
2536         }
2537     }
2538     return VLC_EGENERIC;
2539 }