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

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

midi

开发平台：

Unix_Linux

encoder.c：源码内容
  1. /*****************************************************************************
  2.  * encoder.c: video and audio encoder using the ffmpeg library
  3.  *****************************************************************************
  4.  * Copyright (C) 1999-2004 the VideoLAN team
  5.  * $Id: 40dcb2bc9cc4a2344268a369a1f6b60d12cd6861 $
  6.  *
  7.  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
  8.  *          Gildas Bazin <gbazin@videolan.org>
  9.  *          Christophe Massiot <massiot@via.ecp.fr>
  10.  * Part of the file Copyright (C) FFMPEG Project Developers
  11.  * (mpeg4_default matrixes)
  12.  *
  13.  * This program is free software; you can redistribute it and/or modify
  14.  * it under the terms of the GNU General Public License as published by
  15.  * the Free Software Foundation; either version 2 of the License, or
  16.  * (at your option) any later version.
  17.  *
  18.  * This program is distributed in the hope that it will be useful,
  19.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21.  * GNU General Public License for more details.
  22.  *
  23.  * You should have received a copy of the GNU General Public License
  24.  * along with this program; if not, write to the Free Software
  25.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  26.  *****************************************************************************/
  28. /*****************************************************************************
  29.  * Preamble
  30.  *****************************************************************************/
  31. #ifdef HAVE_CONFIG_H
  32. # include "config.h"
  33. #endif
  35. #include <vlc_common.h>
  36. #include <vlc_vout.h>
  37. #include <vlc_aout.h>
  38. #include <vlc_sout.h>
  39. #include <vlc_codec.h>
  40. #include <vlc_dialog.h>
  41. #include <vlc_avcodec.h>
  43. /* ffmpeg header */
  44. #define HAVE_MMX 1
  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. #include "avcodec.h"
  55. #define HURRY_UP_GUARD1 (450000)
  56. #define HURRY_UP_GUARD2 (300000)
  57. #define HURRY_UP_GUARD3 (100000)
  59. #define MAX_FRAME_DELAY (FF_MAX_B_FRAMES + 2)
  61. /*****************************************************************************
  62.  * Local prototypes
  63.  *****************************************************************************/
  64. int  OpenEncoder ( vlc_object_t * );
  65. void CloseEncoder( vlc_object_t * );
  67. static block_t *EncodeVideo( encoder_t *, picture_t * );
  68. static block_t *EncodeAudio( encoder_t *, aout_buffer_t * );
  70. struct thread_context_t;
  72. /*****************************************************************************
  73.  * thread_context_t : for multithreaded encoding
  74.  *****************************************************************************/
  75. struct thread_context_t
  76. {
  77.     VLC_COMMON_MEMBERS
  79.     AVCodecContext  *p_context;
  80.     int             (* pf_func)(AVCodecContext *c, void *arg);
  81.     void            *arg;
  82.     int             i_ret;
  84.     vlc_mutex_t     lock;
  85.     vlc_cond_t      cond;
  86.     bool            b_work, b_done;
  87. };
  89. /*****************************************************************************
  90.  * encoder_sys_t : ffmpeg encoder descriptor
  91.  *****************************************************************************/
  92. struct encoder_sys_t
  93. {
  94.     /*
  95.      * Ffmpeg properties
  96.      */
  97.     AVCodec         *p_codec;
  98.     AVCodecContext  *p_context;
  100.     /*
  101.      * Common properties
  102.      */
  103.     char *p_buffer;
  104.     uint8_t *p_buffer_out;
  105.     size_t i_buffer_out;
  107.     /*
  108.      * Video properties
  109.      */
  110.     mtime_t i_last_ref_pts;
  111.     mtime_t i_buggy_pts_detect;
  112.     mtime_t i_last_pts;
  113.     bool    b_inited;
  115.     /*
  116.      * Audio properties
  117.      */
  118.     int i_frame_size;
  119.     int i_samples_delay;
  120.     mtime_t i_pts;
  122.     /* Encoding settings */
  123.     int        i_key_int;
  124.     int        i_b_frames;
  125.     int        i_vtolerance;
  126.     int        i_qmin;
  127.     int        i_qmax;
  128.     int        i_hq;
  129.     int        i_rc_buffer_size;
  130.     float      f_rc_buffer_aggressivity;
  131.     bool       b_pre_me;
  132.     bool       b_hurry_up;
  133.     bool       b_interlace, b_interlace_me;
  134.     float      f_i_quant_factor;
  135.     int        i_noise_reduction;
  136.     bool       b_mpeg4_matrix;
  137.     bool       b_trellis;
  138.     int        i_quality; /* for VBR */
  139.     float      f_lumi_masking, f_dark_masking, f_p_masking, f_border_masking;
  140.     int        i_luma_elim, i_chroma_elim;
  141. #if LIBAVCODEC_VERSION_INT >= ((51<<16)+(40<<8)+4)
  142.     int        i_aac_profile; /* AAC profile to use.*/
  143. #endif
  144.     /* Used to work around stupid timestamping behaviour in libavcodec */
  145.     uint64_t i_framenum;
  146.     mtime_t  pi_delay_pts[MAX_FRAME_DELAY];
  147. };
  149. static const char *const ppsz_enc_options[] = {
  150.     "keyint", "bframes", "vt", "qmin", "qmax", "hq",
  151.     "rc-buffer-size", "rc-buffer-aggressivity", "pre-me", "hurry-up",
  152.     "interlace", "i-quant-factor", "noise-reduction", "mpeg4-matrix",
  153.     "trellis", "qscale", "strict", "lumi-masking", "dark-masking",
  154.     "p-masking", "border-masking", "luma-elim-threshold",
  155.     "chroma-elim-threshold",
  156. #if LIBAVCODEC_VERSION_INT >= ((51<<16)+(40<<8)+4)
  157.      "aac-profile",
  158. #endif
  159.      NULL
  160. };
  162. static const uint16_t mpa_bitrate_tab[2][15] =
  163. {
  164.     {0, 32, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320, 384},
  165.     {0, 8, 16, 24, 32, 40, 48, 56, 64, 80, 96, 112, 128, 144, 160}
  166. };
  168. static const uint16_t mpa_freq_tab[6] =
  169. { 44100, 48000, 32000, 22050, 24000, 16000 };
  171. static const uint16_t mpeg4_default_intra_matrix[64] = {
  172.   8, 17, 18, 19, 21, 23, 25, 27,
  173.  17, 18, 19, 21, 23, 25, 27, 28,
  174.  20, 21, 22, 23, 24, 26, 28, 30,
  175.  21, 22, 23, 24, 26, 28, 30, 32,
  176.  22, 23, 24, 26, 28, 30, 32, 35,
  177.  23, 24, 26, 28, 30, 32, 35, 38,
  178.  25, 26, 28, 30, 32, 35, 38, 41,
  179.  27, 28, 30, 32, 35, 38, 41, 45,
  180. };
  182. static const uint16_t mpeg4_default_non_intra_matrix[64] = {
  183.  16, 17, 18, 19, 20, 21, 22, 23,
  184.  17, 18, 19, 20, 21, 22, 23, 24,
  185.  18, 19, 20, 21, 22, 23, 24, 25,
  186.  19, 20, 21, 22, 23, 24, 26, 27,
  187.  20, 21, 22, 23, 25, 26, 27, 28,
  188.  21, 22, 23, 24, 26, 27, 28, 30,
  189.  22, 23, 24, 26, 27, 28, 30, 31,
  190.  23, 24, 25, 27, 28, 30, 31, 33,
  191. };
  193. /*****************************************************************************
  194.  * OpenEncoder: probe the encoder
  195.  *****************************************************************************/
  197. int OpenEncoder( vlc_object_t *p_this )
  198. {
  199.     encoder_t *p_enc = (encoder_t *)p_this;
  200.     encoder_sys_t *p_sys = p_enc->p_sys;
  201.     AVCodecContext *p_context;
  202.     AVCodec *p_codec;
  203.     int i_codec_id, i_cat;
  204.     const char *psz_namecodec;
  205.     vlc_value_t val;
  207.     if( !GetFfmpegCodec( p_enc->fmt_out.i_codec, &i_cat, &i_codec_id,
  208.                              &psz_namecodec ) )
  209.     {
  210.         if( TestFfmpegChroma( -1, p_enc->fmt_out.i_codec ) != VLC_SUCCESS )
  211.         {
  212.             /* handed chroma output */
  213.             return VLC_EGENERIC;
  214.         }
  215.         i_cat      = VIDEO_ES;
  216.         i_codec_id = CODEC_ID_RAWVIDEO;
  217.         psz_namecodec = "Raw video";
  218.     }
  220.     if( p_enc->fmt_out.i_cat == VIDEO_ES && i_cat != VIDEO_ES )
  221.     {
  222.         msg_Err( p_enc, ""%s" is not a video encoder", psz_namecodec );
  223.         dialog_Fatal( p_enc, _("Streaming / Transcoding failed"),
  224.                         _(""%s" is no video encoder."), psz_namecodec );
  225.         return VLC_EGENERIC;
  226.     }
  228.     if( p_enc->fmt_out.i_cat == AUDIO_ES && i_cat != AUDIO_ES )
  229.     {
  230.         msg_Err( p_enc, ""%s" is not an audio encoder", psz_namecodec );
  231.         dialog_Fatal( p_enc, _("Streaming / Transcoding failed"),
  232.                         _(""%s" is no audio encoder."), psz_namecodec );
  233.         return VLC_EGENERIC;
  234.     }
  236.     if( p_enc->fmt_out.i_cat == SPU_ES )
  237.     {
  238.         /* We don't support subtitle encoding */
  239.         return VLC_EGENERIC;
  240.     }
  242.     /* Initialization must be done before avcodec_find_encoder() */
  243.     InitLibavcodec( p_this );
  245.     p_codec = avcodec_find_encoder( i_codec_id );
  246.     if( !p_codec )
  247.     {
  248.         msg_Err( p_enc, "cannot find encoder %sn"
  249. "*** Your FFMPEG installation is crippled.   ***n"
  250. "*** Please check with your FFMPEG packager. ***n"
  251. "*** This is NOT a VLC media player issue.   ***", psz_namecodec );
  253.         dialog_Fatal( p_enc, _("Streaming / Transcoding failed"), _(
  254. /* I have had enough of all these MPEG-3 transcoding bug reports.
  255.  * Downstream packager, you had better not patch this out, or I will be really
  256.  * annoyed. Think about it - you don't want to fork the VLC translation files,
  257.  * do you? -- Courmisch, 2008-10-22 */
  258. "It seems your FFMPEG (libavcodec) installation lacks the following encoder:n"
  259. "%s.n"
  260. "If you don't know how to fix this, ask for support from your distribution.n"
  261. "n"
  262. "This is not an error inside VLC media player.n"
  263. "Do not contact the VideoLAN project about this issue.n"),
  264.             psz_namecodec );
  265.         return VLC_EGENERIC;
  266.     }
  268.     /* Allocate the memory needed to store the encoder's structure */
  269.     if( ( p_sys = calloc( 1, sizeof(encoder_sys_t) ) ) == NULL )
  270.         return VLC_ENOMEM;
  271.     p_enc->p_sys = p_sys;
  272.     p_sys->p_codec = p_codec;
  274.     p_enc->pf_encode_video = EncodeVideo;
  275.     p_enc->pf_encode_audio = EncodeAudio;
  277.     p_sys->p_buffer = NULL;
  278.     p_sys->p_buffer_out = NULL;
  279.     p_sys->i_buffer_out = 0;
  281.     p_sys->p_context = p_context = avcodec_alloc_context();
  282.     p_context->debug = config_GetInt( p_enc, "ffmpeg-debug" );
  283.     p_context->opaque = (void *)p_this;
  285.     /* Set CPU capabilities */
  286.     unsigned i_cpu = vlc_CPU();
  287.     p_context->dsp_mask = 0;
  288.     if( !(i_cpu & CPU_CAPABILITY_MMX) )
  289.     {
  290.         p_context->dsp_mask |= FF_MM_MMX;
  291.     }
  292.     if( !(i_cpu & CPU_CAPABILITY_MMXEXT) )
  293.     {
  294.         p_context->dsp_mask |= FF_MM_MMXEXT;
  295.     }
  296.     if( !(i_cpu & CPU_CAPABILITY_3DNOW) )
  297.     {
  298.         p_context->dsp_mask |= FF_MM_3DNOW;
  299.     }
  300.     if( !(i_cpu & CPU_CAPABILITY_SSE) )
  301.     {
  302.         p_context->dsp_mask |= FF_MM_SSE;
  303.         p_context->dsp_mask |= FF_MM_SSE2;
  304.     }
  306.     config_ChainParse( p_enc, ENC_CFG_PREFIX, ppsz_enc_options, p_enc->p_cfg );
  308.     var_Get( p_enc, ENC_CFG_PREFIX "keyint", &val );
  309.     p_sys->i_key_int = val.i_int;
  311.     var_Get( p_enc, ENC_CFG_PREFIX "bframes", &val );
  312.     p_sys->i_b_frames = val.i_int;
  314.     var_Get( p_enc, ENC_CFG_PREFIX "vt", &val );
  315.     p_sys->i_vtolerance = val.i_int * 1000;
  317.     var_Get( p_enc, ENC_CFG_PREFIX "interlace", &val );
  318.     p_sys->b_interlace = val.b_bool;
  320.     var_Get( p_enc, ENC_CFG_PREFIX "interlace-me", &val );
  321.     p_sys->b_interlace_me = val.b_bool;
  323.     var_Get( p_enc, ENC_CFG_PREFIX "pre-me", &val );
  324.     p_sys->b_pre_me = val.b_bool;
  326.     var_Get( p_enc, ENC_CFG_PREFIX "hurry-up", &val );
  327.     p_sys->b_hurry_up = val.b_bool;
  328.     if( p_sys->b_hurry_up )
  329.     {
  330.         /* hurry up mode needs noise reduction, even small */
  331.         p_sys->i_noise_reduction = 1;
  332.     }
  334.     var_Get( p_enc, ENC_CFG_PREFIX "rc-buffer-size", &val );
  335.     p_sys->i_rc_buffer_size = val.i_int;
  336.     var_Get( p_enc, ENC_CFG_PREFIX "rc-buffer-aggressivity", &val );
  337.     p_sys->f_rc_buffer_aggressivity = val.f_float;
  339.     var_Get( p_enc, ENC_CFG_PREFIX "i-quant-factor", &val );
  340.     p_sys->f_i_quant_factor = val.f_float;
  342.     var_Get( p_enc, ENC_CFG_PREFIX "noise-reduction", &val );
  343.     p_sys->i_noise_reduction = val.i_int;
  345.     var_Get( p_enc, ENC_CFG_PREFIX "mpeg4-matrix", &val );
  346.     p_sys->b_mpeg4_matrix = val.b_bool;
  348.     var_Get( p_enc, ENC_CFG_PREFIX "qscale", &val );
  349.     if( val.f_float < 0.01 || val.f_float > 255.0 ) val.f_float = 0;
  350.     p_sys->i_quality = (int)(FF_QP2LAMBDA * val.f_float + 0.5);
  352.     var_Get( p_enc, ENC_CFG_PREFIX "hq", &val );
  353.     p_sys->i_hq = FF_MB_DECISION_RD;
  354.     if( val.psz_string && *val.psz_string )
  355.     {
  356.         if( !strcmp( val.psz_string, "rd" ) )
  357.             p_sys->i_hq = FF_MB_DECISION_RD;
  358.         else if( !strcmp( val.psz_string, "bits" ) )
  359.             p_sys->i_hq = FF_MB_DECISION_BITS;
  360.         else if( !strcmp( val.psz_string, "simple" ) )
  361.             p_sys->i_hq = FF_MB_DECISION_SIMPLE;
  362.         else
  363.             p_sys->i_hq = FF_MB_DECISION_RD;
  364.     }
  365.     else
  366.         p_sys->i_hq = FF_MB_DECISION_RD;
  367.     free( val.psz_string );
  369.     var_Get( p_enc, ENC_CFG_PREFIX "qmin", &val );
  370.     p_sys->i_qmin = val.i_int;
  371.     var_Get( p_enc, ENC_CFG_PREFIX "qmax", &val );
  372.     p_sys->i_qmax = val.i_int;
  373.     var_Get( p_enc, ENC_CFG_PREFIX "trellis", &val );
  374.     p_sys->b_trellis = val.b_bool;
  376.     var_Get( p_enc, ENC_CFG_PREFIX "strict", &val );
  377.     if( val.i_int < - 1 || val.i_int > 1 ) val.i_int = 0;
  378.     p_context->strict_std_compliance = val.i_int;
  380.     var_Get( p_enc, ENC_CFG_PREFIX "lumi-masking", &val );
  381.     p_sys->f_lumi_masking = val.f_float;
  382.     var_Get( p_enc, ENC_CFG_PREFIX "dark-masking", &val );
  383.     p_sys->f_dark_masking = val.f_float;
  384.     var_Get( p_enc, ENC_CFG_PREFIX "p-masking", &val );
  385.     p_sys->f_p_masking = val.f_float;
  386.     var_Get( p_enc, ENC_CFG_PREFIX "border-masking", &val );
  387.     p_sys->f_border_masking = val.f_float;
  388.     var_Get( p_enc, ENC_CFG_PREFIX "luma-elim-threshold", &val );
  389.     p_sys->i_luma_elim = val.i_int;
  390.     var_Get( p_enc, ENC_CFG_PREFIX "chroma-elim-threshold", &val );
  391.     p_sys->i_chroma_elim = val.i_int;
  393. #if LIBAVCODEC_VERSION_INT >= ((51<<16)+(40<<8)+4)
  394.     var_Get( p_enc, ENC_CFG_PREFIX "aac-profile", &val );
  395.     /* ffmpeg uses faac encoder atm, and it has issues with
  396.      * other than low-complexity profile, so default to that */
  397.     p_sys->i_aac_profile = FF_PROFILE_AAC_LOW;
  398.     if( val.psz_string && *val.psz_string )
  399.     {
  400.         if( !strncmp( val.psz_string, "main", 4 ) )
  401.             p_sys->i_aac_profile = FF_PROFILE_AAC_MAIN;
  402.         else if( !strncmp( val.psz_string, "low", 3 ) )
  403.             p_sys->i_aac_profile = FF_PROFILE_AAC_LOW;
  404. #if 0    /* Not supported by FAAC encoder */
  405.         else if( !strncmp( val.psz_string, "ssr", 3 ) )
  406.             p_sys->i_aac_profile = FF_PROFILE_AAC_SSR;
  407. #endif
  408.         else  if( !strncmp( val.psz_string, "ltp", 3 ) )
  409.             p_sys->i_aac_profile = FF_PROFILE_AAC_LTP;
  410.         else
  411.         {
  412.             msg_Warn( p_enc, "unknown AAC profile requested, setting it to low" );
  413.             p_sys->i_aac_profile = FF_PROFILE_AAC_LOW;
  414.         }
  415.     }
  416.     free( val.psz_string );
  417. #endif
  419.     if( p_enc->fmt_in.i_cat == VIDEO_ES )
  420.     {
  421.         int i_aspect_num, i_aspect_den;
  423.         if( !p_enc->fmt_in.video.i_width || !p_enc->fmt_in.video.i_height )
  424.         {
  425.             msg_Warn( p_enc, "invalid size %ix%i", p_enc->fmt_in.video.i_width,
  426.                       p_enc->fmt_in.video.i_height );
  427.             free( p_sys );
  428.             return VLC_EGENERIC;
  429.         }
  431.         p_context->width = p_enc->fmt_in.video.i_width;
  432.         p_context->height = p_enc->fmt_in.video.i_height;
  434.         p_context->time_base.num = p_enc->fmt_in.video.i_frame_rate_base;
  435.         p_context->time_base.den = p_enc->fmt_in.video.i_frame_rate;
  437.         /* Defaults from ffmpeg.c */
  438.         p_context->qblur = 0.5;
  439.         p_context->qcompress = 0.5;
  440.         p_context->b_quant_offset = 1.25;
  441.         p_context->b_quant_factor = 1.25;
  442.         p_context->i_quant_offset = 0.0;
  443.         p_context->i_quant_factor = -0.8;
  445.         p_context->lumi_masking = p_sys->f_lumi_masking;
  446.         p_context->dark_masking = p_sys->f_dark_masking;
  447.         p_context->p_masking = p_sys->f_p_masking;
  448.         p_context->border_masking = p_sys->f_border_masking;
  449.         p_context->luma_elim_threshold = p_sys->i_luma_elim;
  450.         p_context->chroma_elim_threshold = p_sys->i_chroma_elim;
  452.         if( p_sys->i_key_int > 0 )
  453.             p_context->gop_size = p_sys->i_key_int;
  454.         p_context->max_b_frames =
  455.             __MAX( __MIN( p_sys->i_b_frames, FF_MAX_B_FRAMES ), 0 );
  456.         p_context->b_frame_strategy = 0;
  457.         if( !p_context->max_b_frames  &&
  458.             (  p_enc->fmt_out.i_codec == VLC_FOURCC('m', 'p', '2', 'v') ||
  459.                p_enc->fmt_out.i_codec == VLC_FOURCC('m', 'p', '1', 'v') ) )
  460.             p_context->flags |= CODEC_FLAG_LOW_DELAY;
  462.         av_reduce( &i_aspect_num, &i_aspect_den,
  463.                    p_enc->fmt_in.video.i_aspect,
  464.                    VOUT_ASPECT_FACTOR, 1 << 30 /* something big */ );
  465.         av_reduce( &p_context->sample_aspect_ratio.num,
  466.                    &p_context->sample_aspect_ratio.den,
  467.                    i_aspect_num * (int64_t)p_context->height,
  468.                    i_aspect_den * (int64_t)p_context->width, 1 << 30 );
  470.         p_sys->i_buffer_out = p_context->height * p_context->width * 3;
  471.         if( p_sys->i_buffer_out < FF_MIN_BUFFER_SIZE )
  472.             p_sys->i_buffer_out = FF_MIN_BUFFER_SIZE;
  473.         p_sys->p_buffer_out = malloc( p_sys->i_buffer_out );
  475.         p_enc->fmt_in.i_codec = VLC_FOURCC('I','4','2','0');
  476.         p_enc->fmt_in.video.i_chroma = p_enc->fmt_in.i_codec;
  477.         GetFfmpegChroma( &p_context->pix_fmt, p_enc->fmt_in.video );
  479.         if( p_codec->pix_fmts )
  480.         {
  481.             const enum PixelFormat *p = p_codec->pix_fmts;
  482.             for( ; *p != -1; p++ )
  483.             {
  484.                 if( *p == p_context->pix_fmt ) break;
  485.             }
  486.             if( *p == -1 ) p_context->pix_fmt = p_codec->pix_fmts[0];
  487.             GetVlcChroma( &p_enc->fmt_in.video, p_context->pix_fmt );
  488.             p_enc->fmt_in.i_codec = p_enc->fmt_in.video.i_chroma;
  489.         }
  492.         if ( p_sys->f_i_quant_factor != 0.0 )
  493.             p_context->i_quant_factor = p_sys->f_i_quant_factor;
  495.         p_context->noise_reduction = p_sys->i_noise_reduction;
  497.         if ( p_sys->b_mpeg4_matrix )
  498.         {
  499.             p_context->intra_matrix = mpeg4_default_intra_matrix;
  500.             p_context->inter_matrix = mpeg4_default_non_intra_matrix;
  501.         }
  503.         if ( p_sys->b_pre_me )
  504.         {
  505.             p_context->pre_me = 1;
  506.             p_context->me_pre_cmp = FF_CMP_CHROMA;
  507.         }
  509.         if ( p_sys->b_interlace )
  510.         {
  511.             if ( p_context->height <= 280 )
  512.             {
  513.                 if ( p_context->height != 16 || p_context->width != 16 )
  514.                     msg_Warn( p_enc,
  515.                         "disabling interlaced video because height=%d <= 280",
  516.                         p_context->height );
  517.             }
  518.             else
  519.             {
  520.                 p_context->flags |= CODEC_FLAG_INTERLACED_DCT;
  521.                 if ( p_sys->b_interlace_me )
  522.                     p_context->flags |= CODEC_FLAG_INTERLACED_ME;
  523.             }
  524.         }
  526. #if LIBAVCODEC_VERSION_INT < ((52<<16)+(0<<8)+0)
  527.         if ( p_sys->b_trellis )
  528.             p_context->flags |= CODEC_FLAG_TRELLIS_QUANT;
  529. #else
  530.         p_context->trellis = p_sys->b_trellis;
  531. #endif
  533.         if ( p_sys->i_qmin > 0 && p_sys->i_qmin == p_sys->i_qmax )
  534.             p_context->flags |= CODEC_FLAG_QSCALE;
  536.         if ( p_enc->i_threads >= 1 )
  537.             avcodec_thread_init( p_context, p_enc->i_threads );
  539.         if( p_sys->i_vtolerance > 0 )
  540.             p_context->bit_rate_tolerance = p_sys->i_vtolerance;
  542.         /* usually if someone sets bitrate, he likes more to get that bitrate
  543.          * over quality should help 'normal' user to get asked bitrate
  544.          */
  545.         if( p_enc->fmt_out.i_bitrate > 0 && p_sys->i_qmax == 0 && p_sys->i_qmin == 0 )
  546.         {
  547.             p_sys->i_qmax = 51;
  548.             p_sys->i_qmin = 3;
  549.         }
  551.         if( p_sys->i_qmin > 0 )
  552.         {
  553.             p_context->mb_qmin = p_context->qmin = p_sys->i_qmin;
  554.             p_context->mb_lmin = p_context->lmin = p_sys->i_qmin * FF_QP2LAMBDA;
  555.         }
  556.         if( p_sys->i_qmax > 0 )
  557.         {
  558.             p_context->mb_qmax = p_context->qmax = p_sys->i_qmax;
  559.             p_context->mb_lmax = p_context->lmax = p_sys->i_qmax * FF_QP2LAMBDA;
  560.         }
  561.         p_context->max_qdiff = 3;
  563.         p_context->mb_decision = p_sys->i_hq;
  565.         if( p_sys->i_quality )
  566.         {
  567.             p_context->flags |= CODEC_FLAG_QSCALE;
  568.             p_context->global_quality = p_sys->i_quality;
  569.         }
  570.         else
  571.         {
  572.             p_context->rc_qsquish = 1.0;
  573.             p_context->rc_max_rate = p_enc->fmt_out.i_bitrate;
  574.             p_context->rc_min_rate = p_enc->fmt_out.i_bitrate;
  575.             p_context->rc_buffer_size = p_sys->i_rc_buffer_size;
  576.             /* This is from ffmpeg's ffmpeg.c : */
  577.             p_context->rc_initial_buffer_occupancy
  578.                 = p_sys->i_rc_buffer_size * 3/4;
  579.             p_context->rc_buffer_aggressivity = p_sys->f_rc_buffer_aggressivity;
  580.         }
  581.     }
  582.     else if( p_enc->fmt_in.i_cat == AUDIO_ES )
  583.     {
  584.         /* work around bug in libmp3lame encoding */
  585.         if( i_codec_id == CODEC_ID_MP3 && p_enc->fmt_in.audio.i_channels > 2 )
  586.             p_enc->fmt_in.audio.i_channels = 2;
  588.         p_enc->fmt_in.i_codec  = AOUT_FMT_S16_NE;
  589.         p_context->sample_rate = p_enc->fmt_out.audio.i_rate;
  590.         p_context->channels    = p_enc->fmt_out.audio.i_channels;
  592.         if ( p_enc->fmt_out.i_codec == VLC_FOURCC('m','p','4','a') )
  593.         {
  594.             /* XXX: FAAC does resample only when setting the INPUT samplerate
  595.              * to the desired value (-R option of the faac frontend)
  596.             p_enc->fmt_in.audio.i_rate = p_context->sample_rate;*/
  597. #if LIBAVCODEC_VERSION_INT >= ((51<<16)+(40<<8)+4)
  598.             /* vlc should default to low-complexity profile, faac encoder
  599.              * has bug and aac audio has issues otherwise atm */
  600.             p_context->profile = p_sys->i_aac_profile;
  601. #endif
  602.         }
  603.     }
  605.     /* Misc parameters */
  606.     p_context->bit_rate = p_enc->fmt_out.i_bitrate;
  608.     if( i_codec_id == CODEC_ID_RAWVIDEO )
  609.     {
  610.         /* XXX: hack: Force same codec (will be handled by transcode) */
  611.         p_enc->fmt_in.video.i_chroma = p_enc->fmt_in.i_codec = p_enc->fmt_out.i_codec;
  612.         GetFfmpegChroma( &p_context->pix_fmt, p_enc->fmt_in.video );
  613.     }
  615.     /* Make sure we get extradata filled by the encoder */
  616.     p_context->extradata_size = 0;
  617.     p_context->extradata = NULL;
  618.     p_context->flags |= CODEC_FLAG_GLOBAL_HEADER;
  620.     int ret;
  621.     vlc_avcodec_lock();
  622.     ret = avcodec_open( p_context, p_codec );
  623.     vlc_avcodec_unlock();
  624.     if( ret )
  625.     {
  626.         if( p_enc->fmt_in.i_cat == AUDIO_ES &&
  627.              (p_context->channels > 2 || i_codec_id == CODEC_ID_MP2
  628.                || i_codec_id == CODEC_ID_MP3) )
  629.         {
  630.             if( p_context->channels > 2 )
  631.             {
  632.                 p_context->channels = 2;
  633.                 p_enc->fmt_in.audio.i_channels = 2; // FIXME
  634.                 msg_Warn( p_enc, "stereo mode selected (codec limitation)" );
  635.             }
  637.             if( i_codec_id == CODEC_ID_MP2 || i_codec_id == CODEC_ID_MP3 )
  638.             {
  639.                 int i_frequency, i;
  641.                 for ( i_frequency = 0; i_frequency < 6; i_frequency++ )
  642.                 {
  643.                     if ( p_enc->fmt_out.audio.i_rate
  644.                             == mpa_freq_tab[i_frequency] )
  645.                         break;
  646.                 }
  647.                 if ( i_frequency == 6 )
  648.                 {
  649.                     msg_Err( p_enc, "MPEG audio doesn't support frequency=%d",
  650.                              p_enc->fmt_out.audio.i_rate );
  651.                     free( p_sys );
  652.                     return VLC_EGENERIC;
  653.                 }
  655.                 for ( i = 1; i < 14; i++ )
  656.                 {
  657.                     if ( p_enc->fmt_out.i_bitrate / 1000
  658.                           <= mpa_bitrate_tab[i_frequency / 3][i] )
  659.                         break;
  660.                 }
  661.                 if ( p_enc->fmt_out.i_bitrate / 1000
  662.                       != mpa_bitrate_tab[i_frequency / 3][i] )
  663.                 {
  664.                     msg_Warn( p_enc,
  665.                               "MPEG audio doesn't support bitrate=%d, using %d",
  666.                               p_enc->fmt_out.i_bitrate,
  667.                               mpa_bitrate_tab[i_frequency / 3][i] * 1000 );
  668.                     p_enc->fmt_out.i_bitrate =
  669.                         mpa_bitrate_tab[i_frequency / 3][i] * 1000;
  670.                     p_context->bit_rate = p_enc->fmt_out.i_bitrate;
  671.                 }
  672.             }
  674.             p_context->codec = NULL;
  675.             vlc_avcodec_lock();
  676.             ret = avcodec_open( p_context, p_codec );
  677.             vlc_avcodec_unlock();
  678.             if( ret )
  679.             {
  680.                 msg_Err( p_enc, "cannot open encoder" );
  681.                 dialog_Fatal( p_enc,
  682.                                 _("Streaming / Transcoding failed"),
  683.                                 "%s", _("VLC could not open the encoder.") );
  684.                 free( p_sys );
  685.                 return VLC_EGENERIC;
  686.             }
  687.         }
  688.         else
  689.         {
  690.             msg_Err( p_enc, "cannot open encoder" );
  691.             dialog_Fatal( p_enc, _("Streaming / Transcoding failed"),
  692.                             "%s", _("VLC could not open the encoder.") );
  693.             free( p_sys );
  694.             return VLC_EGENERIC;
  695.         }
  696.     }
  698.     if( i_codec_id == CODEC_ID_FLAC )
  699.     {
  700.         p_enc->fmt_out.i_extra = 4 + 1 + 3 + p_context->extradata_size;
  701.         p_enc->fmt_out.p_extra = malloc( p_enc->fmt_out.i_extra );
  702.         if( p_enc->fmt_out.p_extra )
  703.         {
  704.             uint8_t *p = p_enc->fmt_out.p_extra;
  705.             p[0] = 0x66;    /* f */
  706.             p[1] = 0x4C;    /* L */
  707.             p[2] = 0x61;    /* a */
  708.             p[3] = 0x43;    /* C */
  709.             p[4] = 0x80;    /* streaminfo block, last block before audio */
  710.             p[5] = ( p_context->extradata_size >> 16 ) & 0xff;
  711.             p[6] = ( p_context->extradata_size >>  8 ) & 0xff;
  712.             p[7] = ( p_context->extradata_size       ) & 0xff;
  713.             memcpy( &p[8], p_context->extradata, p_context->extradata_size );
  714.         }
  715.         else
  716.         {
  717.             p_enc->fmt_out.i_extra = 0;
  718.         }
  719.     }
  720.     else
  721.     {
  722.         p_enc->fmt_out.i_extra = p_context->extradata_size;
  723.         if( p_enc->fmt_out.i_extra )
  724.         {
  725.             p_enc->fmt_out.p_extra = malloc( p_enc->fmt_out.i_extra );
  726.             memcpy( p_enc->fmt_out.p_extra, p_context->extradata,
  727.                     p_enc->fmt_out.i_extra );
  728.         }
  729.     }
  731.     p_context->flags &= ~CODEC_FLAG_GLOBAL_HEADER;
  733.     if( p_enc->fmt_in.i_cat == AUDIO_ES )
  734.     {
  735.         p_sys->i_buffer_out = 2 * AVCODEC_MAX_AUDIO_FRAME_SIZE;
  736.         p_sys->p_buffer_out = malloc( p_sys->i_buffer_out );
  737.         p_sys->i_frame_size = p_context->frame_size * 2 * p_context->channels;
  738.         p_sys->p_buffer = malloc( p_sys->i_frame_size );
  739.         p_enc->fmt_out.audio.i_blockalign = p_context->block_align;
  740.     }
  742.     msg_Dbg( p_enc, "found encoder %s", psz_namecodec );
  744.     return VLC_SUCCESS;
  745. }
  747. /****************************************************************************
  748.  * EncodeVideo: the whole thing
  749.  ****************************************************************************/
  750. static block_t *EncodeVideo( encoder_t *p_enc, picture_t *p_pict )
  751. {
  752.     encoder_sys_t *p_sys = p_enc->p_sys;
  753.     AVFrame frame;
  754.     int i_out, i_plane;
  757.     memset( &frame, 0, sizeof( AVFrame ) );
  758.     for( i_plane = 0; i_plane < p_pict->i_planes; i_plane++ )
  759.     {
  760.         frame.data[i_plane] = p_pict->p[i_plane].p_pixels;
  761.         frame.linesize[i_plane] = p_pict->p[i_plane].i_pitch;
  762.     }
  764.     /* Let ffmpeg select the frame type */
  765.     frame.pict_type = 0;
  767.     frame.repeat_pict = p_pict->i_nb_fields - 2;
  768.     frame.interlaced_frame = !p_pict->b_progressive;
  769.     frame.top_field_first = !!p_pict->b_top_field_first;
  771.     /* Set the pts of the frame being encoded (segfaults with mpeg4!)*/
  772.     if( p_enc->fmt_out.i_codec != VLC_FOURCC( 'm', 'p', '4', 'v' ) )
  773.     {
  774.         frame.pts = p_pict->date ? p_pict->date : (int64_t)AV_NOPTS_VALUE;
  776.         if ( p_sys->b_hurry_up && frame.pts != (int64_t)AV_NOPTS_VALUE )
  777.         {
  778.             mtime_t current_date = mdate();
  780.             if ( current_date + HURRY_UP_GUARD3 > frame.pts )
  781.             {
  782.                 p_sys->p_context->mb_decision = FF_MB_DECISION_SIMPLE;
  783. #if LIBAVCODEC_VERSION_INT < ((52<<16)+(0<<8)+0)
  784.                 p_sys->p_context->flags &= ~CODEC_FLAG_TRELLIS_QUANT;
  785. #else
  786.                 p_sys->p_context->trellis = 0;
  787. #endif
  788.                 msg_Dbg( p_enc, "hurry up mode 3" );
  789.             }
  790.             else
  791.             {
  792.                 p_sys->p_context->mb_decision = p_sys->i_hq;
  794.                 if ( current_date + HURRY_UP_GUARD2 > frame.pts )
  795.                 {
  796. #if LIBAVCODEC_VERSION_INT < ((52<<16)+(0<<8)+0)
  797.                     p_sys->p_context->flags &= ~CODEC_FLAG_TRELLIS_QUANT;
  798. #else
  799.                     p_sys->p_context->trellis = 0;
  800. #endif
  801.                     p_sys->p_context->noise_reduction = p_sys->i_noise_reduction
  802.                          + (HURRY_UP_GUARD2 + current_date - frame.pts) / 500;
  803.                     msg_Dbg( p_enc, "hurry up mode 2" );
  804.                 }
  805.                 else
  806.                 {
  807. #if LIBAVCODEC_VERSION_INT < ((52<<16)+(0<<8)+0)
  808.                     if ( p_sys->b_trellis )
  809.                         p_sys->p_context->flags |= CODEC_FLAG_TRELLIS_QUANT;
  810. #else
  811.                     p_sys->p_context->trellis = p_sys->b_trellis;
  812. #endif
  814.                     p_sys->p_context->noise_reduction =
  815.                         p_sys->i_noise_reduction;
  816.                 }
  817.             }
  819.             if ( current_date + HURRY_UP_GUARD1 > frame.pts )
  820.             {
  821.                 frame.pict_type = FF_P_TYPE;
  822.                 /* msg_Dbg( p_enc, "hurry up mode 1 %lld", current_date + HURRY_UP_GUARD1 - frame.pts ); */
  823.             }
  824.         }
  825.     }
  826.     else
  827.     {
  828.         frame.pts = (int64_t)AV_NOPTS_VALUE;
  829.     }
  831.     if ( frame.pts != (int64_t)AV_NOPTS_VALUE && frame.pts != 0 )
  832.     {
  833.         if ( p_sys->i_last_pts == frame.pts )
  834.         {
  835.             msg_Warn( p_enc, "almost fed libavcodec with two frames with the "
  836.                       "same PTS (%"PRId64 ")", frame.pts );
  837.             return NULL;
  838.         }
  839.         else if ( p_sys->i_last_pts > frame.pts )
  840.         {
  841.             msg_Warn( p_enc, "almost fed libavcodec with a frame in the "
  842.                       "past (current: %"PRId64 ", last: %"PRId64")",
  843.                       frame.pts, p_sys->i_last_pts );
  844.             return NULL;
  845.         }
  846.         else
  847.         {
  848.             p_sys->i_last_pts = frame.pts;
  849.         }
  850.     }
  852.     frame.quality = p_sys->i_quality;
  854.     /* Ugly work-around for stupid libavcodec behaviour */
  855.     p_sys->i_framenum++;
  856.     p_sys->pi_delay_pts[p_sys->i_framenum % MAX_FRAME_DELAY] = frame.pts;
  857.     frame.pts = p_sys->i_framenum * AV_TIME_BASE *
  858.         p_enc->fmt_in.video.i_frame_rate_base;
  859.     frame.pts += p_enc->fmt_in.video.i_frame_rate - 1;
  860.     frame.pts /= p_enc->fmt_in.video.i_frame_rate;
  861.     /* End work-around */
  863.     i_out = avcodec_encode_video( p_sys->p_context, p_sys->p_buffer_out,
  864.                                   p_sys->i_buffer_out, &frame );
  866.     if( i_out > 0 )
  867.     {
  868.         block_t *p_block = block_New( p_enc, i_out );
  869.         memcpy( p_block->p_buffer, p_sys->p_buffer_out, i_out );
  871.         /* FIXME, 3-2 pulldown is not handled correctly */
  872.         p_block->i_length = INT64_C(1000000) *
  873.             p_enc->fmt_in.video.i_frame_rate_base /
  874.                 p_enc->fmt_in.video.i_frame_rate;
  876.         if( !p_sys->p_context->max_b_frames || !p_sys->p_context->delay )
  877.         {
  878.             /* No delay -> output pts == input pts */
  879.             p_block->i_pts = p_block->i_dts = p_pict->date;
  880.         }
  881.         else if( p_sys->p_context->coded_frame->pts != (int64_t)AV_NOPTS_VALUE &&
  882.             p_sys->p_context->coded_frame->pts != 0 &&
  883.             p_sys->i_buggy_pts_detect != p_sys->p_context->coded_frame->pts )
  884.         {
  885.             p_sys->i_buggy_pts_detect = p_sys->p_context->coded_frame->pts;
  886.             p_block->i_pts = p_sys->p_context->coded_frame->pts;
  888.             /* Ugly work-around for stupid libavcodec behaviour */
  889.             {
  890.                 int64_t i_framenum = p_block->i_pts *
  891.                     p_enc->fmt_in.video.i_frame_rate /
  892.                     p_enc->fmt_in.video.i_frame_rate_base / AV_TIME_BASE;
  894.                 p_block->i_pts = p_sys->pi_delay_pts[i_framenum % MAX_FRAME_DELAY];
  895.             }
  896.             /* End work-around */
  898.             if( p_sys->p_context->coded_frame->pict_type != FF_I_TYPE &&
  899.                 p_sys->p_context->coded_frame->pict_type != FF_P_TYPE )
  900.             {
  901.                 p_block->i_dts = p_block->i_pts;
  902.             }
  903.             else
  904.             {
  905.                 if( p_sys->i_last_ref_pts )
  906.                 {
  907.                     p_block->i_dts = p_sys->i_last_ref_pts;
  908.                 }
  909.                 else
  910.                 {
  911.                     /* Let's put something sensible */
  912.                     p_block->i_dts = p_block->i_pts;
  913.                 }
  915.                 p_sys->i_last_ref_pts = p_block->i_pts;
  916.             }
  917.         }
  918.         else
  919.         {
  920.             /* Buggy libavcodec which doesn't update coded_frame->pts
  921.              * correctly */
  922.             p_block->i_dts = p_block->i_pts = p_pict->date;
  923.         }
  925.         switch ( p_sys->p_context->coded_frame->pict_type )
  926.         {
  927.         case FF_I_TYPE:
  928.             p_block->i_flags |= BLOCK_FLAG_TYPE_I;
  929.             break;
  930.         case FF_P_TYPE:
  931.             p_block->i_flags |= BLOCK_FLAG_TYPE_P;
  932.             break;
  933.         case FF_B_TYPE:
  934.             p_block->i_flags |= BLOCK_FLAG_TYPE_B;
  935.             break;
  936.         }
  938.         return p_block;
  939.     }
  941.     return NULL;
  942. }
  944. /****************************************************************************
  945.  * EncodeAudio: the whole thing
  946.  ****************************************************************************/
  947. static block_t *EncodeAudio( encoder_t *p_enc, aout_buffer_t *p_aout_buf )
  948. {
  949.     encoder_sys_t *p_sys = p_enc->p_sys;
  950.     block_t *p_block, *p_chain = NULL;
  952.     uint8_t *p_buffer = p_aout_buf->p_buffer;
  953.     int i_samples = p_aout_buf->i_nb_samples;
  954.     int i_samples_delay = p_sys->i_samples_delay;
  956.     p_sys->i_pts = p_aout_buf->start_date -
  957.                 (mtime_t)1000000 * (mtime_t)p_sys->i_samples_delay /
  958.                 (mtime_t)p_enc->fmt_in.audio.i_rate;
  960.     p_sys->i_samples_delay += i_samples;
  962.     while( p_sys->i_samples_delay >= p_sys->p_context->frame_size )
  963.     {
  964.         int16_t *p_samples;
  965.         int i_out;
  967.         if( i_samples_delay )
  968.         {
  969.             /* Take care of the left-over from last time */
  970.             int i_delay_size = i_samples_delay * 2 *
  971.                                  p_sys->p_context->channels;
  972.             int i_size = p_sys->i_frame_size - i_delay_size;
  974.             p_samples = (int16_t *)p_sys->p_buffer;
  975.             memcpy( p_sys->p_buffer + i_delay_size, p_buffer, i_size );
  976.             p_buffer -= i_delay_size;
  977.             i_samples += i_samples_delay;
  978.             i_samples_delay = 0;
  979.         }
  980.         else
  981.         {
  982.             p_samples = (int16_t *)p_buffer;
  983.         }
  985.         i_out = avcodec_encode_audio( p_sys->p_context, p_sys->p_buffer_out,
  986.                                       p_sys->i_buffer_out, p_samples );
  988. #if 0
  989.         msg_Warn( p_enc, "avcodec_encode_audio: %d", i_out );
  990. #endif
  991.         if( i_out < 0 ) break;
  993.         p_buffer += p_sys->i_frame_size;
  994.         p_sys->i_samples_delay -= p_sys->p_context->frame_size;
  995.         i_samples -= p_sys->p_context->frame_size;
  997.         if( i_out == 0 ) continue;
  999.         p_block = block_New( p_enc, i_out );
  1000.         memcpy( p_block->p_buffer, p_sys->p_buffer_out, i_out );
  1002.         p_block->i_length = (mtime_t)1000000 *
  1003.             (mtime_t)p_sys->p_context->frame_size /
  1004.             (mtime_t)p_sys->p_context->sample_rate;
  1006.         p_block->i_dts = p_block->i_pts = p_sys->i_pts;
  1008.         /* Update pts */
  1009.         p_sys->i_pts += p_block->i_length;
  1010.         block_ChainAppend( &p_chain, p_block );
  1011.     }
  1013.     /* Backup the remaining raw samples */
  1014.     if( i_samples )
  1015.     {
  1016.         memcpy( p_sys->p_buffer + i_samples_delay * 2 *
  1017.                 p_sys->p_context->channels, p_buffer,
  1018.                 i_samples * 2 * p_sys->p_context->channels );
  1019.     }
  1021.     return p_chain;
  1022. }
  1024. /*****************************************************************************
  1025.  * CloseEncoder: ffmpeg encoder destruction
  1026.  *****************************************************************************/
  1027. void CloseEncoder( vlc_object_t *p_this )
  1028. {
  1029.     encoder_t *p_enc = (encoder_t *)p_this;
  1030.     encoder_sys_t *p_sys = p_enc->p_sys;
  1032.     vlc_avcodec_lock();
  1033.     avcodec_close( p_sys->p_context );
  1034.     vlc_avcodec_unlock();
  1035.     av_free( p_sys->p_context );
  1037.     free( p_sys->p_buffer );
  1038.     free( p_sys->p_buffer_out );
  1040.     free( p_sys );
  1041. }