* modules/stream_out/switcher.c: fixed forbidden empty string for gettext.
[vlc.git] / modules / stream_out / switcher.c
1 /*****************************************************************************
2  * switcher.c: MPEG2 video switcher module
3  *****************************************************************************
4  * Copyright (C) 2004 VideoLAN
5  * $Id$
6  *
7  * Authors: Christophe Massiot <massiot@via.ecp.fr>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
22  *****************************************************************************/
23
24 /*****************************************************************************
25  * Preamble
26  *****************************************************************************/
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <math.h>
30
31 #include <vlc/vlc.h>
32 #include <vlc/sout.h>
33 #include <vlc/vout.h>
34
35 #include "network.h"
36
37 #define HAVE_MMX
38 #ifdef HAVE_FFMPEG_AVCODEC_H
39 #   include <ffmpeg/avcodec.h>
40 #else
41 #   include <avcodec.h>
42 #endif
43
44 #ifdef HAVE_POSTPROC_POSTPROCESS_H
45 #   include <postproc/postprocess.h>
46 #else
47 #   include <libpostproc/postprocess.h>
48 #endif
49
50 #define SOUT_CFG_PREFIX "sout-switcher-"
51 #define MAX_PICTURES 12
52 #define MAX_AUDIO 12
53 #define AVCODEC_MAX_VIDEO_FRAME_SIZE (3*1024*1024)
54 #define MAX_THRESHOLD 99999999
55
56 /*****************************************************************************
57  * Local prototypes
58  *****************************************************************************/
59 static int      Open    ( vlc_object_t * );
60 static void     Close   ( vlc_object_t * );
61
62 static sout_stream_id_t *Add( sout_stream_t *, es_format_t * );
63 static int Del( sout_stream_t *, sout_stream_id_t * );
64 static int Send( sout_stream_t *, sout_stream_id_t *, block_t * );
65
66 static mtime_t Process( sout_stream_t *p_stream, sout_stream_id_t *id,
67                         mtime_t i_max_dts );
68 static int UnpackFromFile( sout_stream_t *p_stream, const char *psz_file,
69                            int i_width, int i_height,
70                            picture_t *p_pic );
71 static void NetCommand( sout_stream_t *p_stream );
72 static mtime_t VideoCommand( sout_stream_t *p_stream, sout_stream_id_t *id );
73 static block_t *VideoGetBuffer( sout_stream_t *p_stream, sout_stream_id_t *id,
74                                 block_t *p_buffer );
75 static block_t *AudioGetBuffer( sout_stream_t *p_stream, sout_stream_id_t *id,
76                                 block_t *p_buffer );
77
78 /*****************************************************************************
79  * Module descriptor
80  *****************************************************************************/
81 #define FILES_TEXT N_("Files")
82 #define FILES_LONGTEXT N_( \
83     "Full paths of the files separated by colons." )
84 #define SIZES_TEXT N_("Sizes")
85 #define SIZES_LONGTEXT N_( \
86     "List of sizes separated by colons (720x576:480x576)." )
87 #define PORT_TEXT N_("Command UDP port")
88 #define PORT_LONGTEXT N_( \
89     "UDP port to listen to for commands." )
90 #define COMMAND_TEXT N_("Command")
91 #define COMMAND_LONGTEXT N_( \
92     "Initial command to execute." )
93 #define GOP_TEXT N_("GOP size")
94 #define GOP_LONGTEXT N_( \
95     "Number of P frames between two I frames." )
96 #define QSCALE_TEXT N_("Quantizer scale")
97 #define QSCALE_LONGTEXT N_( \
98     "Quantizer scale." )
99
100 vlc_module_begin();
101     set_description( _("MPEG2 video switcher stream output") );
102     set_capability( "sout stream", 50 );
103     add_shortcut( "switcher" );
104     set_callbacks( Open, Close );
105
106     add_string( SOUT_CFG_PREFIX "files", "", NULL, FILES_TEXT,
107                 FILES_LONGTEXT, VLC_FALSE );
108     add_string( SOUT_CFG_PREFIX "sizes", "", NULL, SIZES_TEXT,
109                 SIZES_LONGTEXT, VLC_FALSE );
110     add_integer( SOUT_CFG_PREFIX "port", 5001, NULL,
111                  PORT_TEXT, PORT_LONGTEXT, VLC_TRUE );
112     add_integer( SOUT_CFG_PREFIX "command", 0, NULL,
113                  COMMAND_TEXT, COMMAND_LONGTEXT, VLC_TRUE );
114     add_integer( SOUT_CFG_PREFIX "gop", 8, NULL,
115                  GOP_TEXT, GOP_LONGTEXT, VLC_TRUE );
116     add_integer( SOUT_CFG_PREFIX "qscale", 5, NULL,
117                  QSCALE_TEXT, QSCALE_LONGTEXT, VLC_TRUE );
118 vlc_module_end();
119
120 static const char *ppsz_sout_options[] = {
121     "files", "sizes", "port", "command", "gop", "qscale", NULL
122 };
123
124 struct sout_stream_sys_t
125 {
126     sout_stream_t   *p_out;
127     int             i_gop;
128     int             i_qscale;
129     sout_stream_id_t *pp_audio_ids[MAX_AUDIO];
130
131     /* Pictures */
132     picture_t       p_pictures[MAX_PICTURES];
133     int             i_nb_pictures;
134
135     /* Command */
136     int             i_fd;
137     int             i_cmd, i_old_cmd;
138 };
139
140 struct sout_stream_id_t
141 {
142     void            *id;
143     vlc_bool_t      b_switcher_video;
144     vlc_bool_t      b_switcher_audio;
145     es_format_t     f_src;
146     block_t         *p_queued;
147
148     /* ffmpeg part */
149     AVCodec         *ff_enc;
150     AVCodecContext  *ff_enc_c;
151     AVFrame         *p_frame;
152     char            *p_buffer_out;
153     int             i_nb_pred;
154     int16_t         *p_samples;
155 };
156
157 /*****************************************************************************
158  * Open:
159  *****************************************************************************/
160 static int Open( vlc_object_t *p_this )
161 {
162     sout_stream_t     *p_stream = (sout_stream_t*)p_this;
163     sout_stream_sys_t *p_sys;
164     vlc_value_t       val;
165     char              *psz_files, *psz_sizes;
166     int               i_height = 0, i_width = 0;
167
168     p_sys = malloc( sizeof(sout_stream_sys_t) );
169     memset( p_sys, 0, sizeof(sout_stream_sys_t) );
170
171     p_sys->p_out = sout_StreamNew( p_stream->p_sout, p_stream->psz_next );
172     if( !p_sys->p_out )
173     {
174         msg_Err( p_stream, "cannot create chain" );
175         free( p_sys );
176         return VLC_EGENERIC;
177     }
178
179     sout_CfgParse( p_stream, SOUT_CFG_PREFIX, ppsz_sout_options,
180                    p_stream->p_cfg );
181
182     var_Get( p_stream, SOUT_CFG_PREFIX "files", &val );
183     psz_files = val.psz_string;
184     var_Get( p_stream, SOUT_CFG_PREFIX "sizes", &val );
185     psz_sizes = val.psz_string;
186
187     p_sys->i_nb_pictures = 0;
188     while ( psz_files && *psz_files )
189     {
190         char * psz_file = psz_files;
191         char * psz_size = psz_sizes;
192
193         while ( *psz_files && *psz_files != ':' )
194             psz_files++;
195         if ( *psz_files == ':' )
196            *psz_files++ = '\0';
197
198         if ( *psz_sizes )
199         {
200             while ( *psz_sizes && *psz_sizes != ':' )
201                 psz_sizes++;
202             if ( *psz_sizes == ':' )
203                 *psz_sizes++ = '\0';
204             if ( sscanf( psz_size, "%dx%d", &i_width, &i_height ) != 2 )
205             {
206                 msg_Err( p_stream, "bad size %s for file %s", psz_size,
207                          psz_file );
208                 free( p_sys );
209                 return VLC_EGENERIC;
210             }
211         }
212
213         if ( UnpackFromFile( p_stream, psz_file, i_width, i_height,
214                              &p_sys->p_pictures[p_sys->i_nb_pictures] ) < 0 )
215         {
216             free( p_sys );
217             return VLC_EGENERIC;
218         }
219         p_sys->i_nb_pictures++;
220     }
221
222     var_Get( p_stream, SOUT_CFG_PREFIX "port", &val );
223     p_sys->i_fd = net_OpenUDP( p_stream, NULL, val.i_int, NULL, 0 );
224     if ( p_sys->i_fd < 0 )
225     {
226         free( p_sys );
227         return VLC_EGENERIC;
228     }
229
230     var_Get( p_stream, SOUT_CFG_PREFIX "command", &val );
231     p_sys->i_cmd = val.i_int;
232     p_sys->i_old_cmd = 0;
233
234     var_Get( p_stream, SOUT_CFG_PREFIX "gop", &val );
235     p_sys->i_gop = val.i_int;
236
237     var_Get( p_stream, SOUT_CFG_PREFIX "qscale", &val );
238     p_sys->i_qscale = val.i_int;
239
240     p_stream->pf_add    = Add;
241     p_stream->pf_del    = Del;
242     p_stream->pf_send   = Send;
243     p_stream->p_sys     = p_sys;
244
245     avcodec_init();
246     avcodec_register_all();
247
248     return VLC_SUCCESS;
249 }
250
251 /*****************************************************************************
252  * Close:
253  *****************************************************************************/
254 static void Close( vlc_object_t * p_this )
255 {
256     sout_stream_t       *p_stream = (sout_stream_t *)p_this;
257     sout_stream_sys_t   *p_sys = p_stream->p_sys;
258
259     sout_StreamDelete( p_sys->p_out );
260
261     free( p_sys );
262 }
263
264 /*****************************************************************************
265  * Add: Add an input elementary stream
266  *****************************************************************************/
267 static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
268 {
269     sout_stream_sys_t   *p_sys = p_stream->p_sys;
270     sout_stream_id_t    *id;
271
272     id = malloc( sizeof( sout_stream_id_t ) );
273     memset( id, 0, sizeof( sout_stream_id_t ) );
274     id->id = NULL;
275
276     if ( p_fmt->i_cat == VIDEO_ES
277             && (p_fmt->i_codec == VLC_FOURCC('m', 'p', 'g', 'v')
278                  || p_fmt->i_codec == VLC_FOURCC('f', 'a', 'k', 'e')) )
279     {
280         id->b_switcher_video = VLC_TRUE;
281         p_fmt->i_codec = VLC_FOURCC('m', 'p', 'g', 'v');
282         msg_Dbg( p_stream,
283                  "creating video switcher for fcc=`%4.4s'",
284                  (char*)&p_fmt->i_codec );
285     }
286     else if ( p_fmt->i_cat == AUDIO_ES
287                && p_fmt->i_codec == VLC_FOURCC('m', 'p', 'g', 'a') )
288     {
289         int i_ff_codec = CODEC_ID_MP2;
290         int i;
291
292         id->b_switcher_audio = VLC_TRUE;
293         msg_Dbg( p_stream,
294                  "creating audio switcher for fcc=`%4.4s'",
295                  (char*)&p_fmt->i_codec );
296
297         /* Allocate the encoder right now. */
298         if( i_ff_codec == 0 )
299         {
300             msg_Err( p_stream, "cannot find encoder" );
301             return NULL;
302         }
303
304         id->ff_enc = avcodec_find_encoder( i_ff_codec );
305         if( !id->ff_enc )
306         {
307             msg_Err( p_stream, "cannot find encoder (avcodec)" );
308             return NULL;
309         }
310
311         id->ff_enc_c = avcodec_alloc_context();
312
313         /* Set CPU capabilities */
314         id->ff_enc_c->dsp_mask = 0;
315         if( !(p_stream->p_libvlc->i_cpu & CPU_CAPABILITY_MMX) )
316         {
317             id->ff_enc_c->dsp_mask |= FF_MM_MMX;
318         }
319         if( !(p_stream->p_libvlc->i_cpu & CPU_CAPABILITY_MMXEXT) )
320         {
321             id->ff_enc_c->dsp_mask |= FF_MM_MMXEXT;
322         }
323         if( !(p_stream->p_libvlc->i_cpu & CPU_CAPABILITY_3DNOW) )
324         {
325             id->ff_enc_c->dsp_mask |= FF_MM_3DNOW;
326         }
327         if( !(p_stream->p_libvlc->i_cpu & CPU_CAPABILITY_SSE) )
328         {
329             id->ff_enc_c->dsp_mask |= FF_MM_SSE;
330             id->ff_enc_c->dsp_mask |= FF_MM_SSE2;
331         }
332
333         id->ff_enc_c->sample_rate = p_fmt->audio.i_rate;
334         id->ff_enc_c->channels    = p_fmt->audio.i_channels;
335         id->ff_enc_c->bit_rate    = p_fmt->i_bitrate;
336
337         if( avcodec_open( id->ff_enc_c, id->ff_enc ) )
338         {
339             msg_Err( p_stream, "cannot open encoder" );
340             return NULL;
341         }
342
343         id->p_buffer_out = malloc( AVCODEC_MAX_AUDIO_FRAME_SIZE * 2 );
344         id->p_samples = malloc( id->ff_enc_c->frame_size
345                                  * p_fmt->audio.i_channels * sizeof(int16_t) );
346         memset( id->p_samples, 0,
347                 id->ff_enc_c->frame_size * p_fmt->audio.i_channels
348                  * sizeof(int16_t) );
349
350         for ( i = 0; i < MAX_AUDIO; i++ )
351         {
352             if ( p_sys->pp_audio_ids[i] == NULL )
353             {
354                 p_sys->pp_audio_ids[i] = id;
355                 break;
356             }
357         }
358         if ( i == MAX_AUDIO )
359         {
360             msg_Err( p_stream, "too many audio streams !" );
361             free( id );
362             return NULL;
363         }
364     }
365     else
366     {
367         msg_Dbg( p_stream, "do not know what to do when switching (fcc=`%4.4s')",
368                  (char*)&p_fmt->i_codec );
369     }
370
371     /* src format */
372     memcpy( &id->f_src, p_fmt, sizeof( es_format_t ) );
373
374     /* open output stream */
375     id->id = p_sys->p_out->pf_add( p_sys->p_out, p_fmt );
376
377     if ( id->id == NULL )
378     {
379         free( id );
380         return NULL;
381     }
382
383     return id;
384 }
385
386 /*****************************************************************************
387  * Del: Del an elementary stream
388  *****************************************************************************/
389 static int Del( sout_stream_t *p_stream, sout_stream_id_t *id )
390 {
391     sout_stream_sys_t   *p_sys = p_stream->p_sys;
392
393     if ( id->b_switcher_audio )
394     {
395         int i;
396         for ( i = 0; i < MAX_AUDIO; i++ )
397         {
398             if ( p_sys->pp_audio_ids[i] == id )
399             {
400                 p_sys->pp_audio_ids[i] = NULL;
401                 break;
402             }
403         }
404     }
405
406     if ( id->ff_enc )
407     {
408         avcodec_close( id->ff_enc_c );
409         av_free( id->ff_enc_c );
410         av_free( id->p_frame );
411         free( id->p_buffer_out );
412     }
413
414     if ( id->id )
415     {
416         p_sys->p_out->pf_del( p_sys->p_out, id->id );
417     }
418     free( id );
419
420     return VLC_SUCCESS;
421 }
422
423 /*****************************************************************************
424  * Send: Process an input packet
425  *****************************************************************************/
426 static int Send( sout_stream_t *p_stream, sout_stream_id_t *id,
427                  block_t *p_buffer )
428 {
429     sout_stream_sys_t *p_sys = p_stream->p_sys;
430
431     if ( !id->id )
432     {
433         block_Release( p_buffer );
434         return VLC_EGENERIC;
435     }
436
437     if ( !id->b_switcher_video && !id->b_switcher_audio )
438     {
439         return p_sys->p_out->pf_send( p_sys->p_out, id->id, p_buffer );
440     }
441
442     block_ChainAppend( &id->p_queued, p_buffer );
443
444     if ( id->b_switcher_video )
445     {
446         /* Check for commands for every video frame. */
447         NetCommand( p_stream );
448
449         while ( id->p_queued != NULL )
450         {
451             mtime_t i_dts = 0;
452             int i;
453
454             if ( p_sys->i_old_cmd != p_sys->i_cmd )
455             {
456                 i_dts = VideoCommand( p_stream, id );
457             }
458
459             i_dts = Process( p_stream, id, i_dts );
460
461             for ( i = 0; i < MAX_AUDIO; i++ )
462             {
463                 if ( p_sys->pp_audio_ids[i] != NULL )
464                     Process( p_stream, p_sys->pp_audio_ids[i], i_dts );
465             }
466         }
467     }
468
469     return VLC_SUCCESS;
470 }
471
472 /*****************************************************************************
473  * Process: Process and dispatch buffers
474  *****************************************************************************/
475 static mtime_t Process( sout_stream_t *p_stream, sout_stream_id_t *id,
476                         mtime_t i_max_dts )
477 {
478     sout_stream_sys_t *p_sys = p_stream->p_sys;
479     mtime_t i_dts = 0;
480     block_t *p_blocks = NULL;
481     block_t *p_blocks_out = NULL;
482
483     /* Find out the blocks we need. */
484     while ( id->p_queued != NULL
485              && (!i_max_dts || id->p_queued->i_dts <= i_max_dts) )
486     {
487         block_t *p_next = id->p_queued->p_next;
488         id->p_queued->p_next = NULL;
489         i_dts = id->p_queued->i_dts;
490         block_ChainAppend( &p_blocks, id->p_queued );
491         id->p_queued = p_next;
492     }
493
494     if ( p_sys->i_old_cmd == 0 )
495     {
496         /* Full forward */
497         if ( p_blocks != NULL )
498             p_sys->p_out->pf_send( p_sys->p_out, id->id, p_blocks );
499         return i_dts;
500     }
501
502     if ( p_sys->i_old_cmd == -1 )
503     {
504         /* No output at all */
505         while ( p_blocks != NULL )
506         {
507             block_t * p_next = p_blocks->p_next;
508             block_Release( p_blocks );
509             p_blocks = p_next;
510         }
511         return i_dts;
512     }
513
514     while ( p_blocks != NULL )
515     {
516         block_t * p_next = p_blocks->p_next;
517         block_t * p_out;
518
519         if ( id->b_switcher_video )
520         {
521             p_out = VideoGetBuffer( p_stream, id, p_blocks );
522         }
523         else
524         {
525             p_out = AudioGetBuffer( p_stream, id, p_blocks );
526         }
527         p_blocks = p_next;
528         if ( p_out != NULL )
529         {
530             block_ChainAppend( &p_blocks_out, p_out );
531         }
532     }
533
534     if ( p_blocks_out != NULL )
535         p_sys->p_out->pf_send( p_sys->p_out, id->id, p_blocks_out );
536     return i_dts;
537 }
538
539 /*****************************************************************************
540  * UnpackFromFile: Read a YUV picture and store it in our format
541  *****************************************************************************/
542 static int UnpackFromFile( sout_stream_t *p_stream, const char *psz_file,
543                            int i_width, int i_height,
544                            picture_t *p_pic )
545 {
546     int i, j;
547     FILE *p_file = fopen( psz_file, "r" );
548
549     if ( p_file == NULL )
550     {
551         msg_Err( p_stream, "file %s not found", psz_file );
552         return -1;
553     }
554
555     vout_InitPicture( VLC_OBJECT(p_stream), p_pic, VLC_FOURCC('I','4','2','0'),
556                       i_width, i_height,
557                       i_width * VOUT_ASPECT_FACTOR / i_height );
558     for ( i = 0; i < p_pic->i_planes; i++ )
559     {
560         p_pic->p[i].p_pixels = malloc( p_pic->p[i].i_lines *
561                                            p_pic->p[i].i_pitch );
562         memset( p_pic->p[i].p_pixels, 0, p_pic->p[i].i_lines *
563                     p_pic->p[i].i_pitch );
564     }
565
566     for ( i = 0; i < i_height; i++ )
567     {
568         int i_chroma;
569         uint8_t p_buffer[i_width * 2];
570         uint8_t *p_char = p_buffer;
571         uint8_t *p_y = &p_pic->p[0].p_pixels[i * p_pic->p[0].i_pitch];
572         uint8_t *p_u = &p_pic->p[1].p_pixels[i/2 * p_pic->p[1].i_pitch];
573         uint8_t *p_v = &p_pic->p[2].p_pixels[i/2 * p_pic->p[2].i_pitch];
574
575         if ( fread( p_buffer, 2, i_width, p_file ) != i_width )
576         {
577             msg_Err( p_stream, "premature end of file %s", psz_file );
578             fclose( p_file );
579             for ( i = 0; i < p_pic->i_planes; i++ )
580             {
581                 free( p_pic->p[i].p_pixels );
582             }
583             return -1;
584         }
585
586         i_chroma = 0;
587         for ( j = 0; j < i_width; j++ )
588         {
589             uint8_t **pp_chroma = i_chroma ? &p_v : &p_u;
590             i_chroma = !i_chroma;
591             if ( i & 1 )
592                 **pp_chroma = (**pp_chroma + *p_char + 1) / 2;
593             else
594                 **pp_chroma = *p_char;
595             (*pp_chroma)++;
596             p_char++;
597             *p_y++ = *p_char++;
598         }
599     }
600
601     fclose( p_file );
602     return 0;
603 }
604
605 /*****************************************************************************
606  * NetCommand: Get a command from the network
607  *****************************************************************************/
608 static void NetCommand( sout_stream_t *p_stream )
609 {
610     sout_stream_sys_t *p_sys = p_stream->p_sys;
611     char psz_buffer[10];
612     int i_len = net_ReadNonBlock( p_stream, p_sys->i_fd, NULL, psz_buffer,
613                                   sizeof( psz_buffer ), 0 );
614
615     if ( i_len > 0 )
616     {
617         int i_cmd = strtol( psz_buffer, NULL, 0 );
618         if ( i_cmd < -1 || i_cmd > p_sys->i_nb_pictures )
619         {
620             msg_Err( p_stream, "got a wrong command (%d)", i_cmd );
621             return;
622         }
623
624         p_sys->i_cmd = i_cmd;
625     }
626 }
627
628 /*****************************************************************************
629  * VideoCommand: Create/Delete a video encoder
630  *****************************************************************************/
631 static mtime_t VideoCommand( sout_stream_t *p_stream, sout_stream_id_t *id )
632 {
633     sout_stream_sys_t *p_sys = p_stream->p_sys;
634
635     if ( p_sys->i_cmd == 0 && !(id->p_queued->i_flags & BLOCK_FLAG_TYPE_I) )
636     {
637         mtime_t i_dts = id->p_queued->i_dts;
638         block_t *p_block = id->p_queued->p_next;
639
640         while ( p_block != NULL )
641         {
642             if ( p_block->i_flags & BLOCK_FLAG_TYPE_I )
643                 return i_dts;
644             i_dts = p_block->i_dts;
645             p_block = p_block->p_next;
646         }
647
648         return 0;
649     }
650
651     p_sys->i_old_cmd = p_sys->i_cmd;
652
653     if ( id->ff_enc )
654     {
655         avcodec_close( id->ff_enc_c );
656         av_free( id->ff_enc_c );
657         av_free( id->p_frame );
658         free( id->p_buffer_out );
659         id->ff_enc = NULL;
660     }
661
662     if ( p_sys->i_cmd > 0 )
663     {
664         /* Create a new encoder. */
665         int i_ff_codec = CODEC_ID_MPEG2VIDEO;
666         if( i_ff_codec == 0 )
667         {
668             msg_Err( p_stream, "cannot find encoder" );
669             return 0;
670         }
671
672         id->ff_enc = avcodec_find_encoder( i_ff_codec );
673         if( !id->ff_enc )
674         {
675             msg_Err( p_stream, "cannot find encoder (avcodec)" );
676             return 0;
677         }
678
679         id->ff_enc_c = avcodec_alloc_context();
680
681         /* Set CPU capabilities */
682         id->ff_enc_c->dsp_mask = 0;
683         if( !(p_stream->p_libvlc->i_cpu & CPU_CAPABILITY_MMX) )
684         {
685             id->ff_enc_c->dsp_mask |= FF_MM_MMX;
686         }
687         if( !(p_stream->p_libvlc->i_cpu & CPU_CAPABILITY_MMXEXT) )
688         {
689             id->ff_enc_c->dsp_mask |= FF_MM_MMXEXT;
690         }
691         if( !(p_stream->p_libvlc->i_cpu & CPU_CAPABILITY_3DNOW) )
692         {
693             id->ff_enc_c->dsp_mask |= FF_MM_3DNOW;
694         }
695         if( !(p_stream->p_libvlc->i_cpu & CPU_CAPABILITY_SSE) )
696         {
697             id->ff_enc_c->dsp_mask |= FF_MM_SSE;
698             id->ff_enc_c->dsp_mask |= FF_MM_SSE2;
699         }
700
701         id->ff_enc_c->width = p_sys->p_pictures[p_sys->i_cmd-1].format.i_width;
702         id->ff_enc_c->height = p_sys->p_pictures[p_sys->i_cmd-1].format.i_height;
703
704         id->ff_enc_c->frame_rate    = 25; /* FIXME */
705         id->ff_enc_c->frame_rate_base = 1;
706
707         id->ff_enc_c->gop_size = 200;
708         id->ff_enc_c->max_b_frames = 0;
709
710         id->ff_enc_c->flags |= CODEC_FLAG_QSCALE
711                             | CODEC_FLAG_INPUT_PRESERVED
712                             | CODEC_FLAG_LOW_DELAY;
713
714         id->ff_enc_c->mb_decision = FF_MB_DECISION_SIMPLE;
715
716         if( avcodec_open( id->ff_enc_c, id->ff_enc ) )
717         {
718             msg_Err( p_stream, "cannot open encoder" );
719             return 0;
720         }
721
722         id->p_buffer_out = malloc( AVCODEC_MAX_VIDEO_FRAME_SIZE );
723         id->p_frame = avcodec_alloc_frame();
724         id->p_frame->linesize[0] = p_sys->p_pictures[p_sys->i_cmd-1].p[0].i_pitch;
725         id->p_frame->linesize[1] = p_sys->p_pictures[p_sys->i_cmd-1].p[1].i_pitch;
726         id->p_frame->linesize[2] = p_sys->p_pictures[p_sys->i_cmd-1].p[2].i_pitch;
727         id->p_frame->data[0] = p_sys->p_pictures[p_sys->i_cmd-1].p[0].p_pixels;
728         id->p_frame->data[1] = p_sys->p_pictures[p_sys->i_cmd-1].p[1].p_pixels;
729         id->p_frame->data[2] = p_sys->p_pictures[p_sys->i_cmd-1].p[2].p_pixels;
730
731         id->i_nb_pred = p_sys->i_gop;
732     }
733
734     return 0;
735 }
736
737 /*****************************************************************************
738  * VideoGetBuffer: Build an alternate video buffer
739  *****************************************************************************/
740 static block_t *VideoGetBuffer( sout_stream_t *p_stream, sout_stream_id_t *id,
741                                 block_t *p_buffer )
742 {
743     sout_stream_sys_t *p_sys = p_stream->p_sys;
744     int i_out;
745     block_t *p_out;
746
747     id->p_frame->quality = p_sys->i_qscale * powf(2.0, FF_LAMBDA_SHIFT + 7.0)
748                             / 139.0;
749     id->p_frame->interlaced_frame = 0;
750     id->p_frame->top_field_first = 1;
751     id->p_frame->pts = p_buffer->i_dts;
752
753     if ( id->i_nb_pred >= p_sys->i_gop )
754     {
755         id->p_frame->pict_type = FF_I_TYPE;
756 #if 0
757         id->p_frame->me_threshold = 0;
758         id->p_frame->mb_threshold = 0;
759 #endif
760         id->i_nb_pred = 0;
761     }
762     else
763     {
764         id->p_frame->pict_type = FF_P_TYPE;
765 #if 0
766         if ( id->p_frame->mb_type != NULL )
767         {
768             id->p_frame->me_threshold = MAX_THRESHOLD;
769             id->p_frame->mb_threshold = MAX_THRESHOLD;
770         }
771 #endif
772         id->i_nb_pred++;
773     }
774
775     i_out = avcodec_encode_video( id->ff_enc_c, id->p_buffer_out,
776                                   AVCODEC_MAX_VIDEO_FRAME_SIZE,
777                                   id->p_frame );
778
779     if ( i_out <= 0 )
780         return NULL;
781
782 #if 0
783     if ( id->p_frame->mb_type == NULL
784           && id->ff_enc_c->coded_frame->pict_type != FF_I_TYPE )
785     {
786         int mb_width = (id->ff_enc_c->width + 15) / 16;
787         int mb_height = (id->ff_enc_c->height + 15) / 16;
788         int h_chroma_shift, v_chroma_shift;
789         int i;
790         
791         avcodec_get_chroma_sub_sample( id->ff_enc_c->pix_fmt, &h_chroma_shift,
792                                        &v_chroma_shift );
793
794         id->p_frame->motion_subsample_log2
795             = id->ff_enc_c->coded_frame->motion_subsample_log2;
796         id->p_frame->mb_type = malloc( ((mb_width + 1) * (mb_height + 1) + 1)
797                                     * sizeof(uint32_t) );
798         p_stream->p_vlc->pf_memcpy( id->p_frame->mb_type,
799                                     id->ff_enc_c->coded_frame->mb_type,
800                                     (mb_width + 1) * mb_height
801                                       * sizeof(id->p_frame->mb_type[0]));
802         
803         for ( i = 0; i < 2; i++ )
804         {
805             int stride = ((16 * mb_width )
806                     >> id->ff_enc_c->coded_frame->motion_subsample_log2) + 1;
807             int height = ((16 * mb_height)
808                     >> id->ff_enc_c->coded_frame->motion_subsample_log2);
809             int b8_stride = mb_width * 2 + 1;
810
811             if ( id->ff_enc_c->coded_frame->motion_val[i] )
812             {
813                 id->p_frame->motion_val[i] = malloc( 2 * stride * height
814                                                 * sizeof(int16_t) );
815                 p_stream->p_vlc->pf_memcpy( id->p_frame->motion_val[i],
816                                      id->ff_enc_c->coded_frame->motion_val[i],
817                                      2 * stride * height * sizeof(int16_t) );
818             }
819             if ( id->ff_enc_c->coded_frame->ref_index[i] )
820             {
821                 id->p_frame->ref_index[i] = malloc( b8_stride * 2 * mb_height
822                                                * sizeof(int8_t) );
823                 p_stream->p_vlc->pf_memcpy( id->p_frame->ref_index[i],
824                                  id->ff_enc_c->coded_frame->ref_index[i],
825                                  b8_stride * 2 * mb_height * sizeof(int8_t));
826             }
827         }
828     }
829 #endif
830
831     p_out = block_New( p_stream, i_out );
832     p_stream->p_vlc->pf_memcpy( p_out->p_buffer, id->p_buffer_out, i_out );
833     p_out->i_length = p_buffer->i_length;
834     p_out->i_pts = p_buffer->i_dts;
835     p_out->i_dts = p_buffer->i_dts;
836
837     switch ( id->ff_enc_c->coded_frame->pict_type )
838     {
839     case FF_I_TYPE:
840         p_out->i_flags |= BLOCK_FLAG_TYPE_I;
841         break;
842     case FF_P_TYPE:
843         p_out->i_flags |= BLOCK_FLAG_TYPE_P;
844         break;
845     case FF_B_TYPE:
846         p_out->i_flags |= BLOCK_FLAG_TYPE_B;
847         break;
848     default:
849         break;
850     }
851
852     block_Release( p_buffer );
853
854     return p_out;
855 }
856
857 /*****************************************************************************
858  * AudioGetBuffer: Build an alternate audio buffer
859  *****************************************************************************/
860 static block_t *AudioGetBuffer( sout_stream_t *p_stream, sout_stream_id_t *id,
861                                 block_t *p_buffer )
862 {
863     int i_out;
864     block_t *p_out;
865
866     i_out = avcodec_encode_audio( id->ff_enc_c, id->p_buffer_out,
867                                   2 * AVCODEC_MAX_AUDIO_FRAME_SIZE,
868                                   id->p_samples );
869
870     if ( i_out <= 0 )
871         return NULL;
872
873     p_out = block_New( p_stream, i_out );
874     p_stream->p_vlc->pf_memcpy( p_out->p_buffer, id->p_buffer_out, i_out );
875     p_out->i_length = p_buffer->i_length;
876     p_out->i_pts = p_buffer->i_dts;
877     p_out->i_dts = p_buffer->i_dts;
878
879     block_Release( p_buffer );
880
881     return p_out;
882 }
883