开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > midi > 查看源码
switcher.c
资源名称：vlc-1.0.5.zip　[点击查看]
上传用户：kjfoods
上传日期：2020-07-06
资源大小：29949k
文件大小：30k
源码类别：

midi

开发平台：

Unix_Linux

switcher.c：源码内容
  1. /*****************************************************************************
  2.  * switcher.c: MPEG2 video switcher module
  3.  *****************************************************************************
  4.  * Copyright (C) 2004 the VideoLAN team
  5.  * $Id: 9faf03eff2a2f97518feb9d561f1148c43da6558 $
  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., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22.  *****************************************************************************/
  24. /*****************************************************************************
  25.  * Preamble
  26.  *****************************************************************************/
  27. #include <math.h>
  29. #ifdef HAVE_CONFIG_H
  30. # include "config.h"
  31. #endif
  33. #include <vlc_common.h>
  34. #include <vlc_plugin.h>
  35. #include <vlc_sout.h>
  36. #include <vlc_vout.h>
  37. #include <vlc_avcodec.h>
  39. #include <vlc_block.h>
  41. #include <vlc_charset.h>
  42. #include <vlc_network.h>
  44. #define HAVE_MMX
  45. #ifdef HAVE_LIBAVCODEC_AVCODEC_H
  46. #   include <libavcodec/avcodec.h>
  47. #elif defined(HAVE_FFMPEG_AVCODEC_H)
  48. #   include <ffmpeg/avcodec.h>
  49. #else
  50. #   include <avcodec.h>
  51. #endif
  53. #ifdef HAVE_POSTPROC_POSTPROCESS_H
  54. #   include <postproc/postprocess.h>
  55. #else
  56. #   include <libpostproc/postprocess.h>
  57. #endif
  59. #define SOUT_CFG_PREFIX "sout-switcher-"
  60. #define MAX_PICTURES 10
  61. #define MAX_AUDIO 30
  62. #define MAX_THRESHOLD 99999999
  64. /*****************************************************************************
  65.  * Local prototypes
  66.  *****************************************************************************/
  67. static int      Open    ( vlc_object_t * );
  68. static void     Close   ( vlc_object_t * );
  70. static sout_stream_id_t *Add( sout_stream_t *, es_format_t * );
  71. static int Del( sout_stream_t *, sout_stream_id_t * );
  72. static int Send( sout_stream_t *, sout_stream_id_t *, block_t * );
  74. static mtime_t Process( sout_stream_t *p_stream, sout_stream_id_t *id,
  75.                         mtime_t i_max_dts );
  76. static int UnpackFromFile( sout_stream_t *p_stream, const char *psz_file,
  77.                            int i_width, int i_height,
  78.                            picture_t *p_pic );
  79. static void NetCommand( sout_stream_t *p_stream );
  80. static mtime_t VideoCommand( sout_stream_t *p_stream, sout_stream_id_t *id );
  81. static block_t *VideoGetBuffer( sout_stream_t *p_stream, sout_stream_id_t *id,
  82.                                 block_t *p_buffer );
  83. static block_t *AudioGetBuffer( sout_stream_t *p_stream, sout_stream_id_t *id,
  84.                                 block_t *p_buffer );
  86. /*****************************************************************************
  87.  * Module descriptor
  88.  *****************************************************************************/
  89. #define FILES_TEXT N_("Files")
  90. #define FILES_LONGTEXT N_( 
  91.     "Full paths of the files separated by colons." )
  92. #define SIZES_TEXT N_("Sizes")
  93. #define SIZES_LONGTEXT N_( 
  94.     "List of sizes separated by colons (720x576:480x576)." )
  95. #define RATIO_TEXT N_("Aspect ratio")
  96. #define RATIO_LONGTEXT N_( 
  97.     "Aspect ratio (4:3, 16:9)." )
  98. #define PORT_TEXT N_("Command UDP port")
  99. #define PORT_LONGTEXT N_( 
  100.     "UDP port to listen to for commands." )
  101. #define COMMAND_TEXT N_("Command")
  102. #define COMMAND_LONGTEXT N_( 
  103.     "Initial command to execute." )
  104. #define GOP_TEXT N_("GOP size")
  105. #define GOP_LONGTEXT N_( 
  106.     "Number of P frames between two I frames." )
  107. #define QSCALE_TEXT N_("Quantizer scale")
  108. #define QSCALE_LONGTEXT N_( 
  109.     "Fixed quantizer scale to use." )
  110. #define AUDIO_TEXT N_("Mute audio")
  111. #define AUDIO_LONGTEXT N_( 
  112.     "Mute audio when command is not 0." )
  114. vlc_module_begin ()
  115.     set_description( N_("MPEG2 video switcher stream output") )
  116.     set_capability( "sout stream", 50 )
  117.     add_shortcut( "switcher" )
  118.     set_callbacks( Open, Close )
  120.     add_string( SOUT_CFG_PREFIX "files", "", NULL, FILES_TEXT,
  121.                 FILES_LONGTEXT, false )
  122.     add_string( SOUT_CFG_PREFIX "sizes", "", NULL, SIZES_TEXT,
  123.                 SIZES_LONGTEXT, false )
  124.     add_string( SOUT_CFG_PREFIX "aspect-ratio", "4:3", NULL, RATIO_TEXT,
  125.                 RATIO_LONGTEXT, false )
  126.     add_integer( SOUT_CFG_PREFIX "port", 5001, NULL,
  127.                  PORT_TEXT, PORT_LONGTEXT, true )
  128.     add_integer( SOUT_CFG_PREFIX "command", 0, NULL,
  129.                  COMMAND_TEXT, COMMAND_LONGTEXT, true )
  130.     add_integer( SOUT_CFG_PREFIX "gop", 8, NULL,
  131.                  GOP_TEXT, GOP_LONGTEXT, true )
  132.     add_integer( SOUT_CFG_PREFIX "qscale", 5, NULL,
  133.                  QSCALE_TEXT, QSCALE_LONGTEXT, true )
  134.     add_bool( SOUT_CFG_PREFIX "mute-audio", 1, NULL,
  135.               AUDIO_TEXT, AUDIO_LONGTEXT, true )
  136. vlc_module_end ()
  138. static const char *const ppsz_sout_options[] = {
  139.     "files", "sizes", "aspect-ratio", "port", "command", "gop", "qscale",
  140.     "mute-audio", NULL
  141. };
  143. struct sout_stream_sys_t
  144. {
  145.     sout_stream_t   *p_out;
  146.     int             i_gop;
  147.     int             i_qscale;
  148.     int             i_aspect;
  149.     sout_stream_id_t *pp_audio_ids[MAX_AUDIO];
  150.     bool      b_audio;
  152.     /* Pictures */
  153.     picture_t       p_pictures[MAX_PICTURES];
  154.     int             i_nb_pictures;
  156.     /* Command */
  157.     int             i_fd;
  158.     int             i_cmd, i_old_cmd;
  159. };
  161. struct sout_stream_id_t
  162. {
  163.     void            *id;
  164.     bool      b_switcher_video;
  165.     bool      b_switcher_audio;
  166.     es_format_t     f_src;
  167.     block_t         *p_queued;
  169.     /* ffmpeg part */
  170.     AVCodec         *ff_enc;
  171.     AVCodecContext  *ff_enc_c;
  172.     AVFrame         *p_frame;
  173.     uint8_t         *p_buffer_out;
  174.     int             i_nb_pred;
  175.     int16_t         *p_samples;
  176. };
  178. /*****************************************************************************
  179.  * Open:
  180.  *****************************************************************************/
  181. static int Open( vlc_object_t *p_this )
  182. {
  183.     sout_stream_t     *p_stream = (sout_stream_t*)p_this;
  184.     sout_stream_sys_t *p_sys;
  185.     vlc_value_t       val;
  186.     char              *psz_files, *psz_sizes;
  187.     int               i_height = 0, i_width = 0;
  189.     p_sys = calloc( 1, sizeof(sout_stream_sys_t) );
  190.     if( !p_sys )
  191.         return VLC_ENOMEM;
  193.     p_sys->p_out = sout_StreamNew( p_stream->p_sout, p_stream->psz_next );
  194.     if( !p_sys->p_out )
  195.     {
  196.         msg_Err( p_stream, "cannot create chain" );
  197.         free( p_sys );
  198.         return VLC_EGENERIC;
  199.     }
  201.     config_ChainParse( p_stream, SOUT_CFG_PREFIX, ppsz_sout_options,
  202.                    p_stream->p_cfg );
  204.     var_Get( p_stream, SOUT_CFG_PREFIX "files", &val );
  205.     psz_files = val.psz_string;
  206.     var_Get( p_stream, SOUT_CFG_PREFIX "sizes", &val );
  207.     psz_sizes = val.psz_string;
  209.     p_sys->i_nb_pictures = 0;
  210.     while ( psz_files && *psz_files )
  211.     {
  212.         char * psz_file = psz_files;
  213.         char * psz_size = psz_sizes;
  215.         while ( *psz_files && *psz_files != ':' )
  216.             psz_files++;
  217.         if ( *psz_files == ':' )
  218.            *psz_files++ = '';
  220.         if ( *psz_sizes )
  221.         {
  222.             while ( *psz_sizes && *psz_sizes != ':' )
  223.                 psz_sizes++;
  224.             if ( *psz_sizes == ':' )
  225.                 *psz_sizes++ = '';
  226.             if ( sscanf( psz_size, "%dx%d", &i_width, &i_height ) != 2 )
  227.             {
  228.                 msg_Err( p_stream, "bad size %s for file %s", psz_size,
  229.                          psz_file );
  230.                 free( p_sys );
  231.                 return VLC_EGENERIC;
  232.             }
  233.         }
  235.         if ( UnpackFromFile( p_stream, psz_file, i_width, i_height,
  236.                              &p_sys->p_pictures[p_sys->i_nb_pictures] ) < 0 )
  237.         {
  238.             free( p_sys );
  239.             return VLC_EGENERIC;
  240.         }
  241.         p_sys->i_nb_pictures++;
  242.     }
  244.     var_Get( p_stream, SOUT_CFG_PREFIX "aspect-ratio", &val );
  245.     if ( val.psz_string )
  246.     {
  247.         char *psz_parser = strchr( val.psz_string, ':' );
  249.         if( psz_parser )
  250.         {
  251.             *psz_parser++ = '';
  252.             p_sys->i_aspect = atoi( val.psz_string ) * VOUT_ASPECT_FACTOR
  253.                 / atoi( psz_parser );
  254.         }
  255.         else
  256.         {
  257.             msg_Warn( p_stream, "bad aspect ratio %s", val.psz_string );
  258.             p_sys->i_aspect = 4 * VOUT_ASPECT_FACTOR / 3;
  259.         }
  261.         free( val.psz_string );
  262.     }
  263.     else
  264.     {
  265.         p_sys->i_aspect = 4 * VOUT_ASPECT_FACTOR / 3;
  266.     }
  268.     var_Get( p_stream, SOUT_CFG_PREFIX "port", &val );
  269.     p_sys->i_fd = net_ListenUDP1( p_stream, NULL, val.i_int );
  270.     if ( p_sys->i_fd < 0 )
  271.     {
  272.         free( p_sys );
  273.         return VLC_EGENERIC;
  274.     }
  276.     var_Get( p_stream, SOUT_CFG_PREFIX "command", &val );
  277.     p_sys->i_cmd = val.i_int;
  278.     p_sys->i_old_cmd = 0;
  280.     var_Get( p_stream, SOUT_CFG_PREFIX "gop", &val );
  281.     p_sys->i_gop = val.i_int;
  283.     var_Get( p_stream, SOUT_CFG_PREFIX "qscale", &val );
  284.     p_sys->i_qscale = val.i_int;
  286.     var_Get( p_stream, SOUT_CFG_PREFIX "mute-audio", &val );
  287.     p_sys->b_audio = val.b_bool;
  289.     p_stream->pf_add    = Add;
  290.     p_stream->pf_del    = Del;
  291.     p_stream->pf_send   = Send;
  292.     p_stream->p_sys     = p_sys;
  294.     avcodec_init();
  295.     avcodec_register_all();
  297.     return VLC_SUCCESS;
  298. }
  300. /*****************************************************************************
  301.  * Close:
  302.  *****************************************************************************/
  303. static void Close( vlc_object_t * p_this )
  304. {
  305.     sout_stream_t       *p_stream = (sout_stream_t *)p_this;
  306.     sout_stream_sys_t   *p_sys = p_stream->p_sys;
  308.     sout_StreamDelete( p_sys->p_out );
  310.     free( p_sys );
  311. }
  313. /*****************************************************************************
  314.  * Add: Add an input elementary stream
  315.  *****************************************************************************/
  316. static sout_stream_id_t *Add( sout_stream_t *p_stream, es_format_t *p_fmt )
  317. {
  318.     sout_stream_sys_t   *p_sys = p_stream->p_sys;
  319.     sout_stream_id_t    *id;
  321.     id = calloc( 1, sizeof( sout_stream_id_t ) );
  322.     if( !id )
  323.         return NULL;
  325.     if( p_fmt->i_cat == VIDEO_ES &&
  326.         ( p_fmt->i_codec == VLC_FOURCC('m', 'p', 'g', 'v') ||
  327.           p_fmt->i_codec == VLC_FOURCC('f', 'a', 'k', 'e') ) )
  328.     {
  329.         id->b_switcher_video = true;
  330.         p_fmt->i_codec = VLC_FOURCC('m', 'p', 'g', 'v');
  331.         msg_Dbg( p_stream, "creating video switcher for fcc=`%4.4s' cmd:%d",
  332.                  (char*)&p_fmt->i_codec, p_sys->i_cmd );
  333.     }
  334.     else if ( p_fmt->i_cat == AUDIO_ES &&
  335.               p_fmt->i_codec == VLC_FOURCC('m', 'p', 'g', 'a') &&
  336.               p_sys->b_audio )
  337.     {
  338.         int i_ff_codec = CODEC_ID_MP2;
  339.         int i;
  341.         id->b_switcher_audio = true;
  342.         msg_Dbg( p_stream, "creating audio switcher for fcc=`%4.4s' cmd:%d",
  343.                  (char*)&p_fmt->i_codec, p_sys->i_cmd );
  345.         /* Allocate the encoder right now. */
  346.         if( i_ff_codec == 0 )
  347.         {
  348.             msg_Err( p_stream, "cannot find encoder" );
  349.             free( id );
  350.             return NULL;
  351.         }
  353.         id->ff_enc = avcodec_find_encoder( i_ff_codec );
  354.         if( !id->ff_enc )
  355.         {
  356.             msg_Err( p_stream, "cannot find encoder (avcodec)" );
  357.             free( id );
  358.             return NULL;
  359.         }
  361.         id->ff_enc_c = avcodec_alloc_context();
  363.         /* Set CPU capabilities */
  364.         unsigned i_cpu = vlc_CPU();
  365.         id->ff_enc_c->dsp_mask = 0;
  366.         if( !(i_cpu & CPU_CAPABILITY_MMX) )
  367.         {
  368.             id->ff_enc_c->dsp_mask |= FF_MM_MMX;
  369.         }
  370.         if( !(i_cpu & CPU_CAPABILITY_MMXEXT) )
  371.         {
  372.             id->ff_enc_c->dsp_mask |= FF_MM_MMXEXT;
  373.         }
  374.         if( !(i_cpu & CPU_CAPABILITY_3DNOW) )
  375.         {
  376.             id->ff_enc_c->dsp_mask |= FF_MM_3DNOW;
  377.         }
  378.         if( !(i_cpu & CPU_CAPABILITY_SSE) )
  379.         {
  380.             id->ff_enc_c->dsp_mask |= FF_MM_SSE;
  381.             id->ff_enc_c->dsp_mask |= FF_MM_SSE2;
  382.         }
  384.         id->ff_enc_c->sample_rate = p_fmt->audio.i_rate;
  385.         id->ff_enc_c->channels    = p_fmt->audio.i_channels;
  386.         id->ff_enc_c->bit_rate    = p_fmt->i_bitrate;
  388.         vlc_avcodec_lock();
  389.         if( avcodec_open( id->ff_enc_c, id->ff_enc ) )
  390.         {
  391.             avcodec_unlock();
  392.             msg_Err( p_stream, "cannot open encoder" );
  393.             av_free( id->ff_enc_c );
  394.             free( id );
  395.             return NULL;
  396.         }
  397.         avcodec_unlock();
  399.         id->p_buffer_out = malloc( AVCODEC_MAX_AUDIO_FRAME_SIZE * 2 );
  400.         id->p_samples = calloc( id->ff_enc_c->frame_size * p_fmt->audio.i_channels,
  401.                                 sizeof(int16_t) );
  402.         if( !id->p_buffer_out || !id->p_samples )
  403.             goto error;
  405.         for( i = 0; i < MAX_AUDIO; i++ )
  406.         {
  407.             if( p_sys->pp_audio_ids[i] == NULL )
  408.             {
  409.                 p_sys->pp_audio_ids[i] = id;
  410.                 break;
  411.             }
  412.         }
  413.         if( i == MAX_AUDIO )
  414.         {
  415.             msg_Err( p_stream, "too many audio streams!" );
  416.             goto error;
  417.         }
  418.     }
  419.     else
  420.     {
  421.         msg_Dbg( p_stream, "do not know what to do when switching (fcc=`%4.4s')",
  422.                  (char*)&p_fmt->i_codec );
  423.     }
  425.     /* src format */
  426.     memcpy( &id->f_src, p_fmt, sizeof( es_format_t ) );
  428.     /* open output stream */
  429.     id->id = p_sys->p_out->pf_add( p_sys->p_out, p_fmt );
  431.     if( id->id != NULL )
  432.         return id;
  434. error:
  435.     vlc_avcodec_lock();
  436.     avcodec_close( id->ff_enc_c );
  437.     vlc_avcodec_unlock();
  438.     free( id->p_samples );
  439.     free( id->p_buffer_out );
  440.     av_free( id->ff_enc_c );
  441.     free( id );
  442.     return NULL;
  443. }
  445. /*****************************************************************************
  446.  * Del: Del an elementary stream
  447.  *****************************************************************************/
  448. static int Del( sout_stream_t *p_stream, sout_stream_id_t *id )
  449. {
  450.     sout_stream_sys_t   *p_sys = p_stream->p_sys;
  452.     if ( id->b_switcher_audio )
  453.     {
  454.         int i;
  455.         for ( i = 0; i < MAX_AUDIO; i++ )
  456.         {
  457.             if ( p_sys->pp_audio_ids[i] == id )
  458.             {
  459.                 p_sys->pp_audio_ids[i] = NULL;
  460.                 break;
  461.             }
  462.         }
  463.     }
  465.     if ( id->ff_enc )
  466.     {
  467.         vlc_avcodec_lock();
  468.         avcodec_close( id->ff_enc_c );
  469.         vlc_avcodec_unlock();
  470.         av_free( id->ff_enc_c );
  471.         av_free( id->p_frame );
  472.         free( id->p_buffer_out );
  473.     }
  475.     if ( id->id )
  476.     {
  477.         p_sys->p_out->pf_del( p_sys->p_out, id->id );
  478.     }
  479.     free( id );
  481.     return VLC_SUCCESS;
  482. }
  484. /*****************************************************************************
  485.  * Send: Process an input packet
  486.  *****************************************************************************/
  487. static int Send( sout_stream_t *p_stream, sout_stream_id_t *id,
  488.                  block_t *p_buffer )
  489. {
  490.     sout_stream_sys_t *p_sys = p_stream->p_sys;
  492.     if ( !id->id )
  493.     {
  494.         block_Release( p_buffer );
  495.         return VLC_EGENERIC;
  496.     }
  498.     if ( !id->b_switcher_video && !id->b_switcher_audio )
  499.     {
  500.         return p_sys->p_out->pf_send( p_sys->p_out, id->id, p_buffer );
  501.     }
  503.     block_ChainAppend( &id->p_queued, p_buffer );
  505.     if ( id->b_switcher_video )
  506.     {
  507.         /* Check for commands for every video frame. */
  508.         NetCommand( p_stream );
  510.         while ( id->p_queued != NULL )
  511.         {
  512.             mtime_t i_dts = 0;
  513.             int i;
  515.             if ( p_sys->i_old_cmd != p_sys->i_cmd )
  516.             {
  517.                 i_dts = VideoCommand( p_stream, id );
  518.             }
  520.             i_dts = Process( p_stream, id, i_dts );
  522.             for ( i = 0; i < MAX_AUDIO; i++ )
  523.             {
  524.                 if ( p_sys->pp_audio_ids[i] != NULL )
  525.                     Process( p_stream, p_sys->pp_audio_ids[i], i_dts );
  526.             }
  527.         }
  528.     }
  530.     return VLC_SUCCESS;
  531. }
  533. /*****************************************************************************
  534.  * Process: Process and dispatch buffers
  535.  *****************************************************************************/
  536. static mtime_t Process( sout_stream_t *p_stream, sout_stream_id_t *id,
  537.                         mtime_t i_max_dts )
  538. {
  539.     sout_stream_sys_t *p_sys = p_stream->p_sys;
  540.     mtime_t i_dts = 0;
  541.     block_t *p_blocks = NULL;
  542.     block_t *p_blocks_out = NULL;
  544.     /* Find out the blocks we need. */
  545.     while ( id->p_queued != NULL
  546.              && (!i_max_dts || id->p_queued->i_dts <= i_max_dts) )
  547.     {
  548.         block_t *p_next = id->p_queued->p_next;
  549.         id->p_queued->p_next = NULL;
  550.         i_dts = id->p_queued->i_dts;
  551.         block_ChainAppend( &p_blocks, id->p_queued );
  552.         id->p_queued = p_next;
  553.     }
  555.     if ( p_sys->i_old_cmd == 0 )
  556.     {
  557.         /* Full forward */
  558.         if ( p_blocks != NULL )
  559.             p_sys->p_out->pf_send( p_sys->p_out, id->id, p_blocks );
  560.         return i_dts;
  561.     }
  563.     if ( p_sys->i_old_cmd == -1 )
  564.     {
  565.         /* No output at all */
  566.         while ( p_blocks != NULL )
  567.         {
  568.             block_t * p_next = p_blocks->p_next;
  569.             block_Release( p_blocks );
  570.             p_blocks = p_next;
  571.         }
  572.         return i_dts;
  573.     }
  575.     while ( p_blocks != NULL )
  576.     {
  577.         block_t * p_next = p_blocks->p_next;
  578.         block_t * p_out;
  580.         if ( id->b_switcher_video )
  581.         {
  582.             p_out = VideoGetBuffer( p_stream, id, p_blocks );
  583.         }
  584.         else
  585.         {
  586.             p_out = AudioGetBuffer( p_stream, id, p_blocks );
  587.         }
  588.         p_blocks = p_next;
  589.         if ( p_out != NULL )
  590.         {
  591.             block_ChainAppend( &p_blocks_out, p_out );
  592.         }
  593.     }
  595.     if ( p_blocks_out != NULL )
  596.         p_sys->p_out->pf_send( p_sys->p_out, id->id, p_blocks_out );
  597.     return i_dts;
  598. }
  600. /*****************************************************************************
  601.  * UnpackFromFile: Read a YUV picture and store it in our format
  602.  *****************************************************************************/
  603. static int UnpackFromFile( sout_stream_t *p_stream, const char *psz_file,
  604.                            int i_width, int i_height,
  605.                            picture_t *p_pic )
  606. {
  607.     int i, j;
  608.     FILE *p_file = utf8_fopen( psz_file, "r" );
  610.     if ( p_file == NULL )
  611.     {
  612.         msg_Err( p_stream, "file %s not found", psz_file );
  613.         return -1;
  614.     }
  616.     vout_InitPicture( VLC_OBJECT(p_stream), p_pic, VLC_FOURCC('I','4','2','0'),
  617.                       i_width, i_height,
  618.                       i_width * VOUT_ASPECT_FACTOR / i_height );
  619.     for ( i = 0; i < p_pic->i_planes; i++ )
  620.     {
  621.         p_pic->p[i].p_pixels = malloc( p_pic->p[i].i_lines *
  622.                                            p_pic->p[i].i_pitch );
  623.         memset( p_pic->p[i].p_pixels, 0, p_pic->p[i].i_lines *
  624.                     p_pic->p[i].i_pitch );
  625.     }
  627.     for ( i = 0; i < i_height; i++ )
  628.     {
  629.         int i_chroma;
  630.         uint8_t p_buffer[i_width * 2];
  631.         uint8_t *p_char = p_buffer;
  632.         uint8_t *p_y = &p_pic->p[0].p_pixels[i * p_pic->p[0].i_pitch];
  633.         uint8_t *p_u = &p_pic->p[1].p_pixels[i/2 * p_pic->p[1].i_pitch];
  634.         uint8_t *p_v = &p_pic->p[2].p_pixels[i/2 * p_pic->p[2].i_pitch];
  636.         if ( fread( p_buffer, 2, i_width, p_file ) != (size_t)i_width )
  637.         {
  638.             msg_Err( p_stream, "premature end of file %s", psz_file );
  639.             fclose( p_file );
  640.             for ( i = 0; i < p_pic->i_planes; i++ )
  641.             {
  642.                 free( p_pic->p[i].p_pixels );
  643.             }
  644.             return -1;
  645.         }
  647.         i_chroma = 0;
  648.         for ( j = 0; j < i_width; j++ )
  649.         {
  650.             uint8_t **pp_chroma = i_chroma ? &p_v : &p_u;
  651.             i_chroma = !i_chroma;
  652.             if ( i & 1 )
  653.                 **pp_chroma = (**pp_chroma + *p_char + 1) / 2;
  654.             else
  655.                 **pp_chroma = *p_char;
  656.             (*pp_chroma)++;
  657.             p_char++;
  658.             *p_y++ = *p_char++;
  659.         }
  660.     }
  662.     fclose( p_file );
  663.     return 0;
  664. }
  666. /*****************************************************************************
  667.  * NetCommand: Get a command from the network
  668.  *****************************************************************************/
  669. static void NetCommand( sout_stream_t *p_stream )
  670. {
  671.     sout_stream_sys_t *p_sys = p_stream->p_sys;
  672.     char psz_buffer[11];
  673.     int i_len = recv( p_sys->i_fd, psz_buffer, sizeof( psz_buffer ) - 1, 0 );
  675.     if ( i_len > 0 )
  676.     {
  677.         psz_buffer[i_len] = '';
  678.         int i_cmd = strtol( psz_buffer, NULL, 0 );
  679.         if ( i_cmd < -1 || i_cmd > p_sys->i_nb_pictures )
  680.         {
  681.             msg_Err( p_stream, "got a wrong command (%d)", i_cmd );
  682.             return;
  683.         }
  685.         p_sys->i_cmd = i_cmd;
  687.         msg_Dbg( p_stream, "new command: %d old:%d", p_sys->i_cmd,
  688.                  p_sys->i_old_cmd );
  689.     }
  690. }
  692. /*****************************************************************************
  693.  * VideoCommand: Create/Delete a video encoder
  694.  *****************************************************************************/
  695. static mtime_t VideoCommand( sout_stream_t *p_stream, sout_stream_id_t *id )
  696. {
  697.     sout_stream_sys_t *p_sys = p_stream->p_sys;
  699.     if ( p_sys->i_cmd == 0 && !(id->p_queued->i_flags & BLOCK_FLAG_TYPE_I) )
  700.     {
  701.         mtime_t i_dts = id->p_queued->i_dts;
  702.         block_t *p_block = id->p_queued->p_next;
  704.         while ( p_block != NULL )
  705.         {
  706.             if ( p_block->i_flags & BLOCK_FLAG_TYPE_I )
  707.                 return i_dts;
  708.             i_dts = p_block->i_dts;
  709.             p_block = p_block->p_next;
  710.         }
  712.         return 0;
  713.     }
  715.     p_sys->i_old_cmd = p_sys->i_cmd;
  717.     if ( id->ff_enc )
  718.     {
  719.         vlc_avcodec_lock();
  720.         avcodec_close( id->ff_enc_c );
  721.         vlc_avcodec_unlock();
  722.         av_free( id->ff_enc_c );
  723.         av_free( id->p_frame );
  724.         free( id->p_buffer_out );
  725.         id->ff_enc = NULL;
  726.     }
  728.     if ( p_sys->i_cmd > 0 )
  729.     {
  730.         /* Create a new encoder. */
  731.         int i_ff_codec = CODEC_ID_MPEG2VIDEO;
  732.         int i_aspect_num, i_aspect_den;
  734.         if( i_ff_codec == 0 )
  735.         {
  736.             msg_Err( p_stream, "cannot find encoder" );
  737.             return 0;
  738.         }
  740.         id->ff_enc = avcodec_find_encoder( i_ff_codec );
  741.         if( !id->ff_enc )
  742.         {
  743.             msg_Err( p_stream, "cannot find encoder (avcodec)" );
  744.             return 0;
  745.         }
  747.         id->ff_enc_c = avcodec_alloc_context();
  749.         /* Set CPU capabilities */
  750.         unsigned i_cpu = vlc_CPU();
  751.         id->ff_enc_c->dsp_mask = 0;
  752.         if( !(i_cpu & CPU_CAPABILITY_MMX) )
  753.         {
  754.             id->ff_enc_c->dsp_mask |= FF_MM_MMX;
  755.         }
  756.         if( !(i_cpu & CPU_CAPABILITY_MMXEXT) )
  757.         {
  758.             id->ff_enc_c->dsp_mask |= FF_MM_MMXEXT;
  759.         }
  760.         if( !(i_cpu & CPU_CAPABILITY_3DNOW) )
  761.         {
  762.             id->ff_enc_c->dsp_mask |= FF_MM_3DNOW;
  763.         }
  764.         if( !(i_cpu & CPU_CAPABILITY_SSE) )
  765.         {
  766.             id->ff_enc_c->dsp_mask |= FF_MM_SSE;
  767.             id->ff_enc_c->dsp_mask |= FF_MM_SSE2;
  768.         }
  770.         id->ff_enc_c->width = p_sys->p_pictures[p_sys->i_cmd-1].format.i_width;
  771.         id->ff_enc_c->height = p_sys->p_pictures[p_sys->i_cmd-1].format.i_height;
  772.         av_reduce( &i_aspect_num, &i_aspect_den,
  773.                    p_sys->i_aspect,
  774.                    VOUT_ASPECT_FACTOR, 1 << 30 /* something big */ );
  775.         av_reduce( &id->ff_enc_c->sample_aspect_ratio.num,
  776.                    &id->ff_enc_c->sample_aspect_ratio.den,
  777.                    i_aspect_num * (int64_t)id->ff_enc_c->height,
  778.                    i_aspect_den * (int64_t)id->ff_enc_c->width, 1 << 30 );
  780. #if LIBAVCODEC_BUILD >= 4754
  781.         id->ff_enc_c->time_base.num = 1;
  782.         id->ff_enc_c->time_base.den = 25; /* FIXME */
  783. #else
  784.         id->ff_enc_c->frame_rate    = 25; /* FIXME */
  785.         id->ff_enc_c->frame_rate_base = 1;
  786. #endif
  788.         id->ff_enc_c->gop_size = 200;
  789.         id->ff_enc_c->max_b_frames = 0;
  791.         id->ff_enc_c->flags |= CODEC_FLAG_QSCALE
  792.                             | CODEC_FLAG_INPUT_PRESERVED
  793.                             | CODEC_FLAG_LOW_DELAY;
  795.         id->ff_enc_c->mb_decision = FF_MB_DECISION_SIMPLE;
  796.         id->ff_enc_c->pix_fmt = PIX_FMT_YUV420P;
  798.         avcodec_lock();
  799.         if( avcodec_open( id->ff_enc_c, id->ff_enc ) )
  800.         {
  801.             avcodec_unlock();
  802.             msg_Err( p_stream, "cannot open encoder" );
  803.             return 0;
  804.         }
  805.         avcodec_unlock();
  807.         id->p_buffer_out = malloc( id->ff_enc_c->width * id->ff_enc_c->height * 3 );
  808.         id->p_frame = avcodec_alloc_frame();
  809.         id->p_frame->linesize[0] = p_sys->p_pictures[p_sys->i_cmd-1].p[0].i_pitch;
  810.         id->p_frame->linesize[1] = p_sys->p_pictures[p_sys->i_cmd-1].p[1].i_pitch;
  811.         id->p_frame->linesize[2] = p_sys->p_pictures[p_sys->i_cmd-1].p[2].i_pitch;
  812.         id->p_frame->data[0] = p_sys->p_pictures[p_sys->i_cmd-1].p[0].p_pixels;
  813.         id->p_frame->data[1] = p_sys->p_pictures[p_sys->i_cmd-1].p[1].p_pixels;
  814.         id->p_frame->data[2] = p_sys->p_pictures[p_sys->i_cmd-1].p[2].p_pixels;
  816.         id->i_nb_pred = p_sys->i_gop;
  817.     }
  819.     return 0;
  820. }
  822. /*****************************************************************************
  823.  * VideoGetBuffer: Build an alternate video buffer
  824.  *****************************************************************************/
  825. static block_t *VideoGetBuffer( sout_stream_t *p_stream, sout_stream_id_t *id,
  826.                                 block_t *p_buffer )
  827. {
  828.     sout_stream_sys_t *p_sys = p_stream->p_sys;
  829.     int i_out;
  830.     block_t *p_out;
  832.     id->p_frame->quality = p_sys->i_qscale * powf(2.0, FF_LAMBDA_SHIFT + 7.0)
  833.                             / 139.0;
  834.     id->p_frame->interlaced_frame = 0;
  835.     id->p_frame->top_field_first = 1;
  836.     id->p_frame->pts = p_buffer->i_dts;
  838.     if ( id->i_nb_pred >= p_sys->i_gop )
  839.     {
  840.         id->p_frame->pict_type = FF_I_TYPE;
  841. #if 0
  842.         id->p_frame->me_threshold = 0;
  843.         id->p_frame->mb_threshold = 0;
  844. #endif
  845.         id->i_nb_pred = 0;
  846.     }
  847.     else
  848.     {
  849.         id->p_frame->pict_type = FF_P_TYPE;
  850. #if 0
  851.         if ( id->p_frame->mb_type != NULL )
  852.         {
  853.             id->p_frame->me_threshold = MAX_THRESHOLD;
  854.             id->p_frame->mb_threshold = MAX_THRESHOLD;
  855.         }
  856. #endif
  857.         id->i_nb_pred++;
  858.     }
  860.     i_out = avcodec_encode_video( id->ff_enc_c, id->p_buffer_out,
  861.                                   id->ff_enc_c->width * id->ff_enc_c->height * 3,
  862.                                   id->p_frame );
  864.     if ( i_out <= 0 )
  865.         return NULL;
  867. #if 0
  868.     if ( id->p_frame->mb_type == NULL
  869.           && id->ff_enc_c->coded_frame->pict_type != FF_I_TYPE )
  870.     {
  871.         int mb_width = (id->ff_enc_c->width + 15) / 16;
  872.         int mb_height = (id->ff_enc_c->height + 15) / 16;
  873.         int h_chroma_shift, v_chroma_shift;
  874.         int i;
  876.         avcodec_get_chroma_sub_sample( id->ff_enc_c->pix_fmt, &h_chroma_shift,
  877.                                        &v_chroma_shift );
  879.         id->p_frame->motion_subsample_log2
  880.             = id->ff_enc_c->coded_frame->motion_subsample_log2;
  881.         id->p_frame->mb_type = malloc( ((mb_width + 1) * (mb_height + 1) + 1)
  882.                                     * sizeof(uint32_t) );
  883.         vlc_memcpy( id->p_frame->mb_type, id->ff_enc_c->coded_frame->mb_type,
  884.                     (mb_width + 1) * mb_height * sizeof(id->p_frame->mb_type[0]));
  886.         for ( i = 0; i < 2; i++ )
  887.         {
  888.             int stride = ((16 * mb_width )
  889.                     >> id->ff_enc_c->coded_frame->motion_subsample_log2) + 1;
  890.             int height = ((16 * mb_height)
  891.                     >> id->ff_enc_c->coded_frame->motion_subsample_log2);
  892.             int b8_stride = mb_width * 2 + 1;
  894.             if ( id->ff_enc_c->coded_frame->motion_val[i] )
  895.             {
  896.                 id->p_frame->motion_val[i] = malloc( 2 * stride * height
  897.                                                 * sizeof(int16_t) );
  898.                 vlc_memcpy( id->p_frame->motion_val[i],
  899.                             id->ff_enc_c->coded_frame->motion_val[i],
  900.                             2 * stride * height * sizeof(int16_t) );
  901.             }
  902.             if ( id->ff_enc_c->coded_frame->ref_index[i] )
  903.             {
  904.                 id->p_frame->ref_index[i] = malloc( b8_stride * 2 * mb_height
  905.                                                * sizeof(int8_t) );
  906.                 vlc_memcpy( id->p_frame->ref_index[i],
  907.                             id->ff_enc_c->coded_frame->ref_index[i],
  908.                             b8_stride * 2 * mb_height * sizeof(int8_t));
  909.             }
  910.         }
  911.     }
  912. #endif
  914.     p_out = block_New( p_stream, i_out );
  915.     vlc_memcpy( p_out->p_buffer, id->p_buffer_out, i_out );
  916.     p_out->i_length = p_buffer->i_length;
  917.     p_out->i_pts = p_buffer->i_dts;
  918.     p_out->i_dts = p_buffer->i_dts;
  919.     p_out->i_rate = p_buffer->i_rate;
  921.     switch ( id->ff_enc_c->coded_frame->pict_type )
  922.     {
  923.     case FF_I_TYPE:
  924.         p_out->i_flags |= BLOCK_FLAG_TYPE_I;
  925.         break;
  926.     case FF_P_TYPE:
  927.         p_out->i_flags |= BLOCK_FLAG_TYPE_P;
  928.         break;
  929.     case FF_B_TYPE:
  930.         p_out->i_flags |= BLOCK_FLAG_TYPE_B;
  931.         break;
  932.     default:
  933.         break;
  934.     }
  936.     block_Release( p_buffer );
  938.     return p_out;
  939. }
  941. /*****************************************************************************
  942.  * AudioGetBuffer: Build an alternate audio buffer
  943.  *****************************************************************************/
  944. static block_t *AudioGetBuffer( sout_stream_t *p_stream, sout_stream_id_t *id,
  945.                                 block_t *p_buffer )
  946. {
  947.     int i_out;
  948.     block_t *p_out;
  950.     i_out = avcodec_encode_audio( id->ff_enc_c, id->p_buffer_out,
  951.                                   2 * AVCODEC_MAX_AUDIO_FRAME_SIZE,
  952.                                   id->p_samples );
  954.     if ( i_out <= 0 )
  955.         return NULL;
  957.     p_out = block_New( p_stream, i_out );
  958.     vlc_memcpy( p_out->p_buffer, id->p_buffer_out, i_out );
  959.     p_out->i_length = p_buffer->i_length;
  960.     p_out->i_pts = p_buffer->i_dts;
  961.     p_out->i_dts = p_buffer->i_dts;
  962.     p_out->i_rate = p_buffer->i_rate;
  964.     block_Release( p_buffer );
  966.     return p_out;
  967. }