encoder.c 34.6 KB
Newer Older
gbazin's avatar
 
gbazin committed
1 2 3
/*****************************************************************************
 * encoder.c: video and audio encoder using the ffmpeg library
 *****************************************************************************
4
 * Copyright (C) 1999-2004 the VideoLAN team
gbazin's avatar
gbazin committed
5
 * $Id$
gbazin's avatar
 
gbazin committed
6 7
 *
 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8
 *          Gildas Bazin <gbazin@videolan.org>
9
 *          Christophe Massiot <massiot@via.ecp.fr>
gbazin's avatar
 
gbazin committed
10 11 12 13 14 15 16 17 18 19 20 21 22
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
dionoea's avatar
dionoea committed
23
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
gbazin's avatar
 
gbazin committed
24 25 26 27 28 29 30 31
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
#include <vlc/vlc.h>
#include <vlc/vout.h>
#include <vlc/aout.h>
32
#include <vlc/sout.h>
gbazin's avatar
 
gbazin committed
33 34 35
#include <vlc/decoder.h>

/* ffmpeg header */
36
#define HAVE_MMX 1
gbazin's avatar
 
gbazin committed
37 38 39 40 41 42
#ifdef HAVE_FFMPEG_AVCODEC_H
#   include <ffmpeg/avcodec.h>
#else
#   include <avcodec.h>
#endif

43 44 45
#if LIBAVCODEC_BUILD < 4704
#   define AV_NOPTS_VALUE 0
#endif
46 47 48
#if LIBAVCODEC_BUILD < 4684
#    define FF_QP2LAMBDA 118
#endif
49

gbazin's avatar
 
gbazin committed
50 51 52
#include "ffmpeg.h"

#define AVCODEC_MAX_VIDEO_FRAME_SIZE (3*1024*1024)
53 54 55
#define HURRY_UP_GUARD1 (450000)
#define HURRY_UP_GUARD2 (300000)
#define HURRY_UP_GUARD3 (100000)
gbazin's avatar
 
gbazin committed
56

57 58
#define MAX_FRAME_DELAY (FF_MAX_B_FRAMES + 2)

gbazin's avatar
 
gbazin committed
59 60 61
/*****************************************************************************
 * Local prototypes
 *****************************************************************************/
gbazin's avatar
 
gbazin committed
62 63
int  E_(OpenEncoder) ( vlc_object_t * );
void E_(CloseEncoder)( vlc_object_t * );
gbazin's avatar
 
gbazin committed
64 65 66 67

static block_t *EncodeVideo( encoder_t *, picture_t * );
static block_t *EncodeAudio( encoder_t *, aout_buffer_t * );

68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92
struct thread_context_t;
static int FfmpegThread( struct thread_context_t *p_context );
static int FfmpegExecute( AVCodecContext *s,
                          int (*pf_func)(AVCodecContext *c2, void *arg2),
                          void **arg, int *ret, int count );

/*****************************************************************************
 * thread_context_t : for multithreaded encoding
 *****************************************************************************/
#if LIBAVCODEC_BUILD >= 4702
struct thread_context_t
{
    VLC_COMMON_MEMBERS

    AVCodecContext  *p_context;
    int             (* pf_func)(AVCodecContext *c, void *arg);
    void            *arg;
    int             i_ret;

    vlc_mutex_t     lock;
    vlc_cond_t      cond;
    vlc_bool_t      b_work, b_done;
};
#endif

gbazin's avatar
 
gbazin committed
93 94 95 96 97 98 99 100 101 102 103 104 105 106
/*****************************************************************************
 * encoder_sys_t : ffmpeg encoder descriptor
 *****************************************************************************/
struct encoder_sys_t
{
    /*
     * Ffmpeg properties
     */
    AVCodec         *p_codec;
    AVCodecContext  *p_context;

    /*
     * Common properties
     */
gbazin's avatar
 
gbazin committed
107 108
    char *p_buffer;
    char *p_buffer_out;
gbazin's avatar
 
gbazin committed
109

gbazin's avatar
 
gbazin committed
110
    /*
111
     * Video properties
gbazin's avatar
 
gbazin committed
112
     */
gbazin's avatar
 
gbazin committed
113 114
    mtime_t i_last_ref_pts;
    mtime_t i_buggy_pts_detect;
115
    mtime_t i_last_pts;
116
    vlc_bool_t b_inited;
gbazin's avatar
 
gbazin committed
117

gbazin's avatar
 
gbazin committed
118 119 120
    /*
     * Audio properties
     */
gbazin's avatar
 
gbazin committed
121
    int i_frame_size;
gbazin's avatar
 
gbazin committed
122 123
    int i_samples_delay;
    mtime_t i_pts;
124 125 126 127 128 129 130 131 132 133 134 135 136

    /* Encoding settings */
    int        i_key_int;
    int        i_b_frames;
    int        i_vtolerance;
    int        i_qmin;
    int        i_qmax;
    int        i_hq;
    vlc_bool_t b_strict_rc;
    int        i_rc_buffer_size;
    float      f_rc_buffer_aggressivity;
    vlc_bool_t b_pre_me;
    vlc_bool_t b_hurry_up;
137
    vlc_bool_t b_interlace, b_interlace_me;
138 139 140 141
    float      f_i_quant_factor;
    int        i_noise_reduction;
    vlc_bool_t b_mpeg4_matrix;
    vlc_bool_t b_trellis;
gbazin's avatar
gbazin committed
142
    int        i_quality; /* for VBR */
143 144
    float      f_lumi_masking, f_dark_masking, f_p_masking, f_border_masking;
    int        i_luma_elim, i_chroma_elim;
145 146 147 148

    /* Used to work around stupid timestamping behaviour in libavcodec */
    uint64_t i_framenum;
    mtime_t  pi_delay_pts[MAX_FRAME_DELAY];
149 150 151
};

static const char *ppsz_enc_options[] = {
152
    "keyint", "bframes", "vt", "qmin", "qmax", "hq", "strict-rc",
153 154
    "rc-buffer-size", "rc-buffer-aggressivity", "pre-me", "hurry-up",
    "interlace", "i-quant-factor", "noise-reduction", "mpeg4-matrix",
155 156 157
    "trellis", "qscale", "strict", "lumi-masking", "dark-masking",
    "p-masking", "border-masking", "luma-elim-threshold",
    "chroma-elim-threshold", NULL
gbazin's avatar
 
gbazin committed
158 159
};

160 161 162 163 164 165 166 167 168
static const uint16_t mpa_bitrate_tab[2][15] =
{
    {0, 32, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320, 384},
    {0, 8, 16, 24, 32, 40, 48, 56, 64, 80, 96, 112, 128, 144, 160}
};

static const uint16_t mpa_freq_tab[6] =
{ 44100, 48000, 32000, 22050, 24000, 16000 };

gbazin's avatar
 
gbazin committed
169
/*****************************************************************************
gbazin's avatar
 
gbazin committed
170
 * OpenEncoder: probe the encoder
gbazin's avatar
 
gbazin committed
171
 *****************************************************************************/
172 173
extern int16_t IMPORT_SYMBOL ff_mpeg4_default_intra_matrix[];
extern int16_t IMPORT_SYMBOL ff_mpeg4_default_non_intra_matrix[];
174

gbazin's avatar
 
gbazin committed
175
int E_(OpenEncoder)( vlc_object_t *p_this )
gbazin's avatar
 
gbazin committed
176 177 178
{
    encoder_t *p_enc = (encoder_t *)p_this;
    encoder_sys_t *p_sys = p_enc->p_sys;
gbazin's avatar
 
gbazin committed
179
    AVCodecContext *p_context;
gbazin's avatar
 
gbazin committed
180
    AVCodec *p_codec;
gbazin's avatar
 
gbazin committed
181 182
    int i_codec_id, i_cat;
    char *psz_namecodec;
183
    vlc_value_t val;
184 185 186
    vlc_value_t lockval;

    var_Get( p_enc->p_libvlc, "avcodec", &lockval );
gbazin's avatar
 
gbazin committed
187

gbazin's avatar
 
gbazin committed
188
    if( !E_(GetFfmpegCodec)( p_enc->fmt_out.i_codec, &i_cat, &i_codec_id,
gbazin's avatar
 
gbazin committed
189
                             &psz_namecodec ) )
gbazin's avatar
 
gbazin committed
190
    {
191 192 193 194 195 196 197 198
        if( E_(GetFfmpegChroma)( p_enc->fmt_out.i_codec ) < 0 )
        {
            /* handed chroma output */
            return VLC_EGENERIC;
        }
        i_cat      = VIDEO_ES;
        i_codec_id = CODEC_ID_RAWVIDEO;
        psz_namecodec = "Raw video";
gbazin's avatar
 
gbazin committed
199 200
    }

201

gbazin's avatar
 
gbazin committed
202
    if( p_enc->fmt_out.i_cat == VIDEO_ES && i_cat != VIDEO_ES )
gbazin's avatar
 
gbazin committed
203
    {
gbazin's avatar
 
gbazin committed
204
        msg_Err( p_enc, "\"%s\" is not a video encoder", psz_namecodec );
gbazin's avatar
 
gbazin committed
205 206 207
        return VLC_EGENERIC;
    }

gbazin's avatar
 
gbazin committed
208 209 210 211 212 213
    if( p_enc->fmt_out.i_cat == AUDIO_ES && i_cat != AUDIO_ES )
    {
        msg_Err( p_enc, "\"%s\" is not an audio encoder", psz_namecodec );
        return VLC_EGENERIC;
    }

gbazin's avatar
 
gbazin committed
214
    /* Initialization must be done before avcodec_find_decoder() */
gbazin's avatar
 
gbazin committed
215 216
    E_(InitLibavcodec)(p_this);

gbazin's avatar
 
gbazin committed
217 218 219 220 221 222 223
    p_codec = avcodec_find_encoder( i_codec_id );
    if( !p_codec )
    {
        msg_Err( p_enc, "cannot find encoder %s", psz_namecodec );
        return VLC_EGENERIC;
    }

gbazin's avatar
 
gbazin committed
224 225 226 227 228 229
    /* Allocate the memory needed to store the decoder's structure */
    if( ( p_sys = (encoder_sys_t *)malloc(sizeof(encoder_sys_t)) ) == NULL )
    {
        msg_Err( p_enc, "out of memory" );
        return VLC_EGENERIC;
    }
230
    memset( p_sys, 0, sizeof(encoder_sys_t) );
gbazin's avatar
 
gbazin committed
231 232 233 234
    p_enc->p_sys = p_sys;
    p_sys->p_codec = p_codec;

    p_enc->pf_encode_video = EncodeVideo;
gbazin's avatar
 
gbazin committed
235
    p_enc->pf_encode_audio = EncodeAudio;
gbazin's avatar
 
gbazin committed
236

gbazin's avatar
 
gbazin committed
237 238 239 240
    p_sys->p_buffer_out = NULL;
    p_sys->p_buffer = NULL;

    p_sys->p_context = p_context = avcodec_alloc_context();
241 242
    p_context->debug = config_GetInt( p_enc, "ffmpeg-debug" );
    p_context->opaque = (void *)p_this;
gbazin's avatar
 
gbazin committed
243 244 245

    /* Set CPU capabilities */
    p_context->dsp_mask = 0;
gbazin's avatar
 
gbazin committed
246
    if( !(p_enc->p_libvlc->i_cpu & CPU_CAPABILITY_MMX) )
gbazin's avatar
 
gbazin committed
247
    {
gbazin's avatar
 
gbazin committed
248
        p_context->dsp_mask |= FF_MM_MMX;
gbazin's avatar
 
gbazin committed
249
    }
gbazin's avatar
 
gbazin committed
250
    if( !(p_enc->p_libvlc->i_cpu & CPU_CAPABILITY_MMXEXT) )
gbazin's avatar
 
gbazin committed
251
    {
gbazin's avatar
 
gbazin committed
252
        p_context->dsp_mask |= FF_MM_MMXEXT;
gbazin's avatar
 
gbazin committed
253
    }
gbazin's avatar
 
gbazin committed
254
    if( !(p_enc->p_libvlc->i_cpu & CPU_CAPABILITY_3DNOW) )
gbazin's avatar
 
gbazin committed
255
    {
gbazin's avatar
 
gbazin committed
256
        p_context->dsp_mask |= FF_MM_3DNOW;
gbazin's avatar
 
gbazin committed
257
    }
gbazin's avatar
 
gbazin committed
258
    if( !(p_enc->p_libvlc->i_cpu & CPU_CAPABILITY_SSE) )
gbazin's avatar
 
gbazin committed
259
    {
gbazin's avatar
 
gbazin committed
260 261
        p_context->dsp_mask |= FF_MM_SSE;
        p_context->dsp_mask |= FF_MM_SSE2;
gbazin's avatar
 
gbazin committed
262 263
    }

264
    sout_CfgParse( p_enc, ENC_CFG_PREFIX, ppsz_enc_options, p_enc->p_cfg );
265 266 267 268 269 270 271 272 273 274 275 276 277

    var_Get( p_enc, ENC_CFG_PREFIX "keyint", &val );
    p_sys->i_key_int = val.i_int;

    var_Get( p_enc, ENC_CFG_PREFIX "bframes", &val );
    p_sys->i_b_frames = val.i_int;

    var_Get( p_enc, ENC_CFG_PREFIX "vt", &val );
    p_sys->i_vtolerance = val.i_int;

    var_Get( p_enc, ENC_CFG_PREFIX "interlace", &val );
    p_sys->b_interlace = val.b_bool;

278 279 280
    var_Get( p_enc, ENC_CFG_PREFIX "interlace-me", &val );
    p_sys->b_interlace_me = val.b_bool;

281
    var_Get( p_enc, ENC_CFG_PREFIX "pre-me", &val );
282 283
    p_sys->b_pre_me = val.b_bool;

284
    var_Get( p_enc, ENC_CFG_PREFIX "hurry-up", &val );
285 286 287 288 289 290 291
    p_sys->b_hurry_up = val.b_bool;
    if( p_sys->b_hurry_up )
    {
        /* hurry up mode needs noise reduction, even small */
        p_sys->i_noise_reduction = 1;
    }

292
    var_Get( p_enc, ENC_CFG_PREFIX "strict-rc", &val );
293
    p_sys->b_strict_rc = val.b_bool;
294
    var_Get( p_enc, ENC_CFG_PREFIX "rc-buffer-size", &val );
295
    p_sys->i_rc_buffer_size = val.i_int;
296
    var_Get( p_enc, ENC_CFG_PREFIX "rc-buffer-aggressivity", &val );
297 298
    p_sys->f_rc_buffer_aggressivity = val.f_float;

299
    var_Get( p_enc, ENC_CFG_PREFIX "i-quant-factor", &val );
300 301
    p_sys->f_i_quant_factor = val.f_float;

302
    var_Get( p_enc, ENC_CFG_PREFIX "noise-reduction", &val );
303 304
    p_sys->i_noise_reduction = val.i_int;

305
    var_Get( p_enc, ENC_CFG_PREFIX "mpeg4-matrix", &val );
306 307
    p_sys->b_mpeg4_matrix = val.b_bool;

gbazin's avatar
gbazin committed
308 309 310 311
    var_Get( p_enc, ENC_CFG_PREFIX "qscale", &val );
    if( val.f_float < 0.01 || val.f_float > 255.0 ) val.f_float = 0;
    p_sys->i_quality = (int)(FF_QP2LAMBDA * val.f_float + 0.5);

312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332
    var_Get( p_enc, ENC_CFG_PREFIX "hq", &val );
    if( val.psz_string && *val.psz_string )
    {
        if( !strcmp( val.psz_string, "rd" ) )
            p_sys->i_hq = FF_MB_DECISION_RD;
        else if( !strcmp( val.psz_string, "bits" ) )
            p_sys->i_hq = FF_MB_DECISION_BITS;
        else if( !strcmp( val.psz_string, "simple" ) )
            p_sys->i_hq = FF_MB_DECISION_SIMPLE;
        else
            p_sys->i_hq = FF_MB_DECISION_RD;
    }
    if( val.psz_string ) free( val.psz_string );

    var_Get( p_enc, ENC_CFG_PREFIX "qmin", &val );
    p_sys->i_qmin = val.i_int;
    var_Get( p_enc, ENC_CFG_PREFIX "qmax", &val );
    p_sys->i_qmax = val.i_int;
    var_Get( p_enc, ENC_CFG_PREFIX "trellis", &val );
    p_sys->b_trellis = val.b_bool;

gbazin's avatar
gbazin committed
333 334 335 336
    var_Get( p_enc, ENC_CFG_PREFIX "strict", &val );
    if( val.i_int < - 1 || val.i_int > 1 ) val.i_int = 0;
    p_context->strict_std_compliance = val.i_int;

337 338 339 340 341 342 343 344 345 346 347 348 349
    var_Get( p_enc, ENC_CFG_PREFIX "lumi-masking", &val );
    p_sys->f_lumi_masking = val.f_float;
    var_Get( p_enc, ENC_CFG_PREFIX "dark-masking", &val );
    p_sys->f_dark_masking = val.f_float;
    var_Get( p_enc, ENC_CFG_PREFIX "p-masking", &val );
    p_sys->f_p_masking = val.f_float;
    var_Get( p_enc, ENC_CFG_PREFIX "border-masking", &val );
    p_sys->f_border_masking = val.f_float;
    var_Get( p_enc, ENC_CFG_PREFIX "luma-elim-threshold", &val );
    p_sys->i_luma_elim = val.i_int;
    var_Get( p_enc, ENC_CFG_PREFIX "chroma-elim-threshold", &val );
    p_sys->i_chroma_elim = val.i_int;

gbazin's avatar
 
gbazin committed
350 351
    if( p_enc->fmt_in.i_cat == VIDEO_ES )
    {
352 353
        int i_aspect_num, i_aspect_den;

354 355 356 357 358 359 360 361
        if( !p_enc->fmt_in.video.i_width || !p_enc->fmt_in.video.i_height )
        {
            msg_Warn( p_enc, "invalid size %ix%i", p_enc->fmt_in.video.i_width,
                      p_enc->fmt_in.video.i_height );
            free( p_sys );
            return VLC_EGENERIC;
        }

gbazin's avatar
 
gbazin committed
362 363
        p_context->width = p_enc->fmt_in.video.i_width;
        p_context->height = p_enc->fmt_in.video.i_height;
gbazin's avatar
 
gbazin committed
364

365 366 367 368
#if LIBAVCODEC_BUILD >= 4754
        p_context->time_base.num = p_enc->fmt_in.video.i_frame_rate_base;
        p_context->time_base.den = p_enc->fmt_in.video.i_frame_rate;
#else
gbazin's avatar
 
gbazin committed
369 370
        p_context->frame_rate = p_enc->fmt_in.video.i_frame_rate;
        p_context->frame_rate_base= p_enc->fmt_in.video.i_frame_rate_base;
371
#endif
gbazin's avatar
 
gbazin committed
372

373 374 375 376 377 378 379 380
        /* Defaults from ffmpeg.c */
        p_context->qblur = 0.5;
        p_context->qcompress = 0.5;
        p_context->b_quant_offset = 1.25;
        p_context->b_quant_factor = 1.25;
        p_context->i_quant_offset = 0.0;
        p_context->i_quant_factor = -0.8;

381 382 383
        p_context->lumi_masking = p_sys->f_lumi_masking;
        p_context->dark_masking = p_sys->f_dark_masking;
        p_context->p_masking = p_sys->f_p_masking;
sigmunau's avatar
sigmunau committed
384
#if LIBAVCODEC_BUILD >= 4741
385
        p_context->border_masking = p_sys->f_border_masking;
sigmunau's avatar
sigmunau committed
386
#endif
387 388 389
        p_context->luma_elim_threshold = p_sys->i_luma_elim;
        p_context->chroma_elim_threshold = p_sys->i_chroma_elim;

390 391
        if( p_sys->i_key_int > 0 )
            p_context->gop_size = p_sys->i_key_int;
392
        p_context->max_b_frames =
393
            __MAX( __MIN( p_sys->i_b_frames, FF_MAX_B_FRAMES ), 0 );
394 395
        p_context->b_frame_strategy = 0;

gbazin's avatar
 
gbazin committed
396
#if LIBAVCODEC_BUILD >= 4687
397
        av_reduce( &i_aspect_num, &i_aspect_den,
398 399
                   p_enc->fmt_in.video.i_aspect,
                   VOUT_ASPECT_FACTOR, 1 << 30 /* something big */ );
400 401 402 403
        av_reduce( &p_context->sample_aspect_ratio.num,
                   &p_context->sample_aspect_ratio.den,
                   i_aspect_num * (int64_t)p_context->height,
                   i_aspect_den * (int64_t)p_context->width, 1 << 30 );
gbazin's avatar
 
gbazin committed
404
#else
gbazin's avatar
 
gbazin committed
405 406
        p_context->aspect_ratio = ((float)p_enc->fmt_in.video.i_aspect) /
            VOUT_ASPECT_FACTOR;
gbazin's avatar
 
gbazin committed
407
#endif
gbazin's avatar
 
gbazin committed
408

gbazin's avatar
 
gbazin committed
409 410
        p_sys->p_buffer_out = malloc( AVCODEC_MAX_VIDEO_FRAME_SIZE );

411
        p_enc->fmt_in.i_codec = VLC_FOURCC('I','4','2','0');
412
        p_context->pix_fmt = E_(GetFfmpegChroma)( p_enc->fmt_in.i_codec );
413
#if LIBAVCODEC_BUILD >= 4714
414 415 416 417 418 419 420 421 422 423
        if( p_codec->pix_fmts )
        {
            const enum PixelFormat *p = p_codec->pix_fmts;
            for( ; *p != -1; p++ )
            {
                if( *p == p_context->pix_fmt ) break;
            }
            if( *p == -1 ) p_context->pix_fmt = p_codec->pix_fmts[0];
            p_enc->fmt_in.i_codec = E_(GetVlcChroma)( p_context->pix_fmt );
        }
424 425 426
#else
        p_enc->fmt_in.i_codec = E_(GetVlcChroma)( p_context->pix_fmt );
#endif
Christophe Massiot's avatar
Christophe Massiot committed
427

428
        if ( p_sys->b_strict_rc )
Christophe Massiot's avatar
Christophe Massiot committed
429 430
        {
            p_context->rc_max_rate = p_enc->fmt_out.i_bitrate;
431
            p_context->rc_buffer_size = p_sys->i_rc_buffer_size;
432 433 434
            /* This is from ffmpeg's ffmpeg.c : */
            p_context->rc_initial_buffer_occupancy
                = p_sys->i_rc_buffer_size * 3/4;
435
            p_context->rc_buffer_aggressivity = p_sys->f_rc_buffer_aggressivity;
436 437
        }

438 439
        if ( p_sys->f_i_quant_factor != 0.0 )
            p_context->i_quant_factor = p_sys->f_i_quant_factor;
440

gbazin's avatar
 
gbazin committed
441
#if LIBAVCODEC_BUILD >= 4690
442
        p_context->noise_reduction = p_sys->i_noise_reduction;
gbazin's avatar
 
gbazin committed
443
#endif
444

445
        if ( p_sys->b_mpeg4_matrix )
446 447 448
        {
            p_context->intra_matrix = ff_mpeg4_default_intra_matrix;
            p_context->inter_matrix = ff_mpeg4_default_non_intra_matrix;
Christophe Massiot's avatar
Christophe Massiot committed
449 450
        }

451
        if ( p_sys->b_pre_me )
Christophe Massiot's avatar
Christophe Massiot committed
452 453 454 455
        {
            p_context->pre_me = 1;
            p_context->me_pre_cmp = FF_CMP_CHROMA;
        }
456

457
        if ( p_sys->b_interlace )
458
        {
459 460
            if ( p_context->height <= 280 )
            {
461 462 463 464
                if ( p_context->height != 16 || p_context->width != 16 )
                    msg_Warn( p_enc,
                        "disabling interlaced video because height=%d <= 280",
                        p_context->height );
465 466 467 468
            }
            else
            {
                p_context->flags |= CODEC_FLAG_INTERLACED_DCT;
469
#if LIBAVCODEC_BUILD >= 4698
470 471 472
                if ( p_sys->b_interlace_me )
                    p_context->flags |= CODEC_FLAG_INTERLACED_ME;
            }
473 474 475
#endif
        }

476
        if ( p_sys->b_trellis )
477 478
            p_context->flags |= CODEC_FLAG_TRELLIS_QUANT;

479
        if ( p_sys->i_qmin > 0 && p_sys->i_qmin == p_sys->i_qmax )
480 481
            p_context->flags |= CODEC_FLAG_QSCALE;

gbazin's avatar
 
gbazin committed
482
#if LIBAVCODEC_BUILD >= 4702
483 484
        if ( p_enc->i_threads >= 1 )
            p_context->thread_count = p_enc->i_threads;
gbazin's avatar
 
gbazin committed
485
#endif
486

487 488
        if( p_sys->i_vtolerance > 0 )
            p_context->bit_rate_tolerance = p_sys->i_vtolerance;
489

490 491 492 493
        if( p_sys->i_qmin > 0 )
            p_context->mb_qmin = p_context->qmin = p_sys->i_qmin;
        if( p_sys->i_qmax > 0 )
            p_context->mb_qmax = p_context->qmax = p_sys->i_qmax;
494 495
        p_context->max_qdiff = 3;

496
        p_context->mb_decision = p_sys->i_hq;
gbazin's avatar
gbazin committed
497 498 499 500 501 502 503 504

        if( p_sys->i_quality )
        {
            p_context->flags |= CODEC_FLAG_QSCALE;
#if LIBAVCODEC_BUILD >= 4668
            p_context->global_quality = p_sys->i_quality;
#endif
        }
gbazin's avatar
 
gbazin committed
505 506 507
    }
    else if( p_enc->fmt_in.i_cat == AUDIO_ES )
    {
508 509 510 511
        /* work around bug in libmp3lame encoding */
        if( i_codec_id == CODEC_ID_MP3 && p_enc->fmt_in.audio.i_channels > 2 )
            p_enc->fmt_in.audio.i_channels = 2;

gbazin's avatar
 
gbazin committed
512 513 514 515 516 517
        p_enc->fmt_in.i_codec  = AOUT_FMT_S16_NE;
        p_context->sample_rate = p_enc->fmt_in.audio.i_rate;
        p_context->channels    = p_enc->fmt_in.audio.i_channels;
    }

    /* Misc parameters */
gbazin's avatar
 
gbazin committed
518
    p_context->bit_rate = p_enc->fmt_out.i_bitrate;
gbazin's avatar
 
gbazin committed
519

gbazin's avatar
 
gbazin committed
520
    if( i_codec_id == CODEC_ID_RAWVIDEO )
gbazin's avatar
 
gbazin committed
521
    {
522 523
        /* XXX: hack: Force same codec (will be handled by transcode) */
        p_enc->fmt_in.i_codec = p_enc->fmt_out.i_codec;
gbazin's avatar
 
gbazin committed
524
        p_context->pix_fmt = E_(GetFfmpegChroma)( p_enc->fmt_in.i_codec );
gbazin's avatar
 
gbazin committed
525 526 527 528 529 530 531
    }

    /* Make sure we get extradata filled by the encoder */
    p_context->extradata_size = 0;
    p_context->extradata = NULL;
    p_context->flags |= CODEC_FLAG_GLOBAL_HEADER;

532
    vlc_mutex_lock( lockval.p_address );
gbazin's avatar
 
gbazin committed
533
    if( avcodec_open( p_context, p_codec ) )
gbazin's avatar
 
gbazin committed
534
    {
535
        vlc_mutex_unlock( lockval.p_address );
536 537 538
        if( p_enc->fmt_in.i_cat == AUDIO_ES &&
             (p_context->channels > 2 || i_codec_id == CODEC_ID_MP2
               || i_codec_id == CODEC_ID_MP3) )
gbazin's avatar
 
gbazin committed
539
        {
540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584
            if( p_context->channels > 2 )
            {
                p_context->channels = 2;
                p_enc->fmt_in.audio.i_channels = 2; // FIXME
                msg_Warn( p_enc, "stereo mode selected (codec limitation)" );
            }

            if( i_codec_id == CODEC_ID_MP2 || i_codec_id == CODEC_ID_MP3 )
            {
                int i_frequency, i;

                for ( i_frequency = 0; i_frequency < 6; i_frequency++ )
                {
                    if ( p_enc->fmt_out.audio.i_rate
                            == mpa_freq_tab[i_frequency] )
                        break;
                }
                if ( i_frequency == 6 )
                {
                    msg_Err( p_enc, "MPEG audio doesn't support frequency=%d",
                             p_enc->fmt_out.audio.i_rate );
                    free( p_sys );
                    return VLC_EGENERIC;
                }

                for ( i = 1; i < 14; i++ )
                {
                    if ( p_enc->fmt_out.i_bitrate / 1000
                          <= mpa_bitrate_tab[i_frequency / 3][i] )
                        break;
                }
                if ( p_enc->fmt_out.i_bitrate / 1000
                      != mpa_bitrate_tab[i_frequency / 3][i] )
                {
                    msg_Warn( p_enc,
                              "MPEG audio doesn't support bitrate=%d, using %d",
                              p_enc->fmt_out.i_bitrate,
                              mpa_bitrate_tab[i_frequency / 3][i] * 1000 );
                    p_enc->fmt_out.i_bitrate =
                        mpa_bitrate_tab[i_frequency / 3][i] * 1000;
                    p_context->bit_rate = p_enc->fmt_out.i_bitrate;
                }
            }

            p_context->codec = NULL;
585
            vlc_mutex_lock( lockval.p_address );
gbazin's avatar
 
gbazin committed
586 587
            if( avcodec_open( p_context, p_codec ) )
            {
588
                vlc_mutex_unlock( lockval.p_address );
gbazin's avatar
 
gbazin committed
589
                msg_Err( p_enc, "cannot open encoder" );
590
                free( p_sys );
gbazin's avatar
 
gbazin committed
591 592 593 594 595 596
                return VLC_EGENERIC;
            }
        }
        else
        {
            msg_Err( p_enc, "cannot open encoder" );
597
            free( p_sys );
gbazin's avatar
 
gbazin committed
598 599
            return VLC_EGENERIC;
        }
gbazin's avatar
 
gbazin committed
600
    }
601
    vlc_mutex_unlock( lockval.p_address );
gbazin's avatar
 
gbazin committed
602

gbazin's avatar
 
gbazin committed
603
    p_enc->fmt_out.i_extra = p_context->extradata_size;
604 605 606 607 608 609
    if( p_enc->fmt_out.i_extra )
    {
        p_enc->fmt_out.p_extra = malloc( p_enc->fmt_out.i_extra );
        memcpy( p_enc->fmt_out.p_extra, p_context->extradata,
                p_enc->fmt_out.i_extra );
    }
gbazin's avatar
 
gbazin committed
610 611
    p_context->flags &= ~CODEC_FLAG_GLOBAL_HEADER;

gbazin's avatar
 
gbazin committed
612 613
    if( p_enc->fmt_in.i_cat == AUDIO_ES )
    {
614
        p_sys->p_buffer_out = malloc( 2 * AVCODEC_MAX_AUDIO_FRAME_SIZE );
gbazin's avatar
 
gbazin committed
615 616 617 618
        p_sys->i_frame_size = p_context->frame_size * 2 * p_context->channels;
        p_sys->p_buffer = malloc( p_sys->i_frame_size );
    }

gbazin's avatar
 
gbazin committed
619 620
    msg_Dbg( p_enc, "found encoder %s", psz_namecodec );

gbazin's avatar
 
gbazin committed
621 622 623
    return VLC_SUCCESS;
}

624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697
/****************************************************************************
 * Ffmpeg threading system
 ****************************************************************************/
#if LIBAVCODEC_BUILD >= 4702
static int FfmpegThread( struct thread_context_t *p_context )
{
    while ( !p_context->b_die && !p_context->b_error )
    {
        vlc_mutex_lock( &p_context->lock );
        while ( !p_context->b_work && !p_context->b_die && !p_context->b_error )
        {
            vlc_cond_wait( &p_context->cond, &p_context->lock );
        }
        p_context->b_work = 0;
        vlc_mutex_unlock( &p_context->lock );
        if ( p_context->b_die || p_context->b_error )
            break;

        if ( p_context->pf_func )
        {
            p_context->i_ret = p_context->pf_func( p_context->p_context,
                                                   p_context->arg );
        }

        vlc_mutex_lock( &p_context->lock );
        p_context->b_done = 1;
        vlc_cond_signal( &p_context->cond );
        vlc_mutex_unlock( &p_context->lock );
    }

    return 0;
}

static int FfmpegExecute( AVCodecContext *s,
                          int (*pf_func)(AVCodecContext *c2, void *arg2),
                          void **arg, int *ret, int count )
{
    struct thread_context_t ** pp_contexts =
                         (struct thread_context_t **)s->thread_opaque;
    int i;

    /* Note, we can be certain that this is not called with the same
     * AVCodecContext by different threads at the same time */
    for ( i = 0; i < count; i++ )
    {
        vlc_mutex_lock( &pp_contexts[i]->lock );
        pp_contexts[i]->arg = arg[i];
        pp_contexts[i]->pf_func = pf_func;
        pp_contexts[i]->i_ret = 12345;
        pp_contexts[i]->b_work = 1;
        vlc_cond_signal( &pp_contexts[i]->cond );
        vlc_mutex_unlock( &pp_contexts[i]->lock );
    }
    for ( i = 0; i < count; i++ )
    {
        vlc_mutex_lock( &pp_contexts[i]->lock );
        while ( !pp_contexts[i]->b_done )
        {
            vlc_cond_wait( &pp_contexts[i]->cond, &pp_contexts[i]->lock );
        }
        pp_contexts[i]->b_done = 0;
        pp_contexts[i]->pf_func = NULL;
        vlc_mutex_unlock( &pp_contexts[i]->lock );

        if ( ret )
        {
            ret[i] = pp_contexts[i]->i_ret;
        }
    }

    return 0;
}
#endif

gbazin's avatar
 
gbazin committed
698 699 700 701 702 703 704 705
/****************************************************************************
 * EncodeVideo: the whole thing
 ****************************************************************************/
static block_t *EncodeVideo( encoder_t *p_enc, picture_t *p_pict )
{
    encoder_sys_t *p_sys = p_enc->p_sys;
    AVFrame frame;
    int i_out, i_plane;
706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737

#if LIBAVCODEC_BUILD >= 4702
    if ( !p_sys->b_inited && p_enc->i_threads >= 1 )
    {
        struct thread_context_t ** pp_contexts;
        int i;

        p_sys->b_inited = 1;
        pp_contexts = malloc( sizeof(struct thread_context_t *)
                                 * p_enc->i_threads );
        p_sys->p_context->thread_opaque = (void *)pp_contexts;

        for ( i = 0; i < p_enc->i_threads; i++ )
        {
            pp_contexts[i] = vlc_object_create( p_enc,
                                     sizeof(struct thread_context_t) );
            pp_contexts[i]->p_context = p_sys->p_context;
            vlc_mutex_init( p_enc, &pp_contexts[i]->lock );
            vlc_cond_init( p_enc, &pp_contexts[i]->cond );
            pp_contexts[i]->b_work = 0;
            pp_contexts[i]->b_done = 0;
            if ( vlc_thread_create( pp_contexts[i], "encoder", FfmpegThread,
                                    VLC_THREAD_PRIORITY_VIDEO, VLC_FALSE ) )
            {
                msg_Err( p_enc, "cannot spawn encoder thread, expect to die soon" );
                return NULL;
            }
        }

        p_sys->p_context->execute = FfmpegExecute;
    }
#endif
gbazin's avatar
 
gbazin committed
738

Laurent Aimar's avatar
Laurent Aimar committed
739
    memset( &frame, 0, sizeof( AVFrame ) );
gbazin's avatar
 
gbazin committed
740 741 742 743 744 745
    for( i_plane = 0; i_plane < p_pict->i_planes; i_plane++ )
    {
        frame.data[i_plane] = p_pict->p[i_plane].p_pixels;
        frame.linesize[i_plane] = p_pict->p[i_plane].i_pitch;
    }

746 747 748
    /* Let ffmpeg select the frame type */
    frame.pict_type = 0;

749
    frame.repeat_pict = p_pict->i_nb_fields - 2;
750 751 752

#if LIBAVCODEC_BUILD >= 4685
    frame.interlaced_frame = !p_pict->b_progressive;
753
    frame.top_field_first = !!p_pict->b_top_field_first;
754 755
#endif

756
#if LIBAVCODEC_BUILD < 4702
gbazin's avatar
 
gbazin committed
757
    /* Set the pts of the frame being encoded (segfaults with mpeg4!)*/
758 759 760
    if( p_enc->fmt_out.i_codec == VLC_FOURCC( 'm', 'p', 'g', 'v' ) ||
        p_enc->fmt_out.i_codec == VLC_FOURCC( 'm', 'p', '1', 'v' ) ||
        p_enc->fmt_out.i_codec == VLC_FOURCC( 'm', 'p', '2', 'v' ) )
761 762 763
#else
    if( 1 )
#endif
Christophe Massiot's avatar
Christophe Massiot committed
764
    {
765
        frame.pts = p_pict->date ? p_pict->date : (signed int) AV_NOPTS_VALUE;
766

767
        if ( p_sys->b_hurry_up && frame.pts != (signed int) AV_NOPTS_VALUE )
Christophe Massiot's avatar
Christophe Massiot committed
768
        {
769 770
            mtime_t current_date = mdate();

771
            if ( current_date + HURRY_UP_GUARD3 > frame.pts )
772 773 774 775 776 777 778
            {
                p_sys->p_context->mb_decision = FF_MB_DECISION_SIMPLE;
                p_sys->p_context->flags &= ~CODEC_FLAG_TRELLIS_QUANT;
                msg_Dbg( p_enc, "hurry up mode 3" );
            }
            else
            {
779
                p_sys->p_context->mb_decision = p_sys->i_hq;
780

781
                if ( current_date + HURRY_UP_GUARD2 > frame.pts )
782 783
                {
                    p_sys->p_context->flags &= ~CODEC_FLAG_TRELLIS_QUANT;
784
#if LIBAVCODEC_BUILD >= 4690
785
                    p_sys->p_context->noise_reduction = p_sys->i_noise_reduction
786
                         + (HURRY_UP_GUARD2 + current_date - frame.pts) / 500;
787 788 789 790 791
#endif
                    msg_Dbg( p_enc, "hurry up mode 2" );
                }
                else
                {
792
                    if ( p_sys->b_trellis )
793
                        p_sys->p_context->flags |= CODEC_FLAG_TRELLIS_QUANT;
794 795
#if LIBAVCODEC_BUILD >= 4690
                    p_sys->p_context->noise_reduction =
796
                        p_sys->i_noise_reduction;
797 798 799 800
#endif
                }
            }

801
            if ( current_date + HURRY_UP_GUARD1 > frame.pts )
802
            {
803 804
                frame.pict_type = FF_P_TYPE;
                /* msg_Dbg( p_enc, "hurry up mode 1 %lld", current_date + HURRY_UP_GUARD1 - frame.pts ); */
805
            }
Christophe Massiot's avatar
Christophe Massiot committed
806 807
        }
    }
gbazin's avatar
 
gbazin committed
808
    else
Christophe Massiot's avatar
Christophe Massiot committed
809
    {
810
        frame.pts = AV_NOPTS_VALUE;
Christophe Massiot's avatar
Christophe Massiot committed
811
    }
gbazin's avatar
 
gbazin committed
812

813
    if ( frame.pts != (signed int) AV_NOPTS_VALUE && frame.pts != 0 )
814 815 816
    {
        if ( p_sys->i_last_pts == frame.pts )
        {
gbazin's avatar
gbazin committed
817 818
            msg_Warn( p_enc, "almost fed libavcodec with two frames with the "
                      "same PTS (" I64Fd ")", frame.pts );
819 820
            return NULL;
        }
821 822 823 824 825 826 827
        else if ( p_sys->i_last_pts > frame.pts )
        {
            msg_Warn( p_enc, "almost fed libavcodec with a frame in the "
                      "past (current: " I64Fd ", last: "I64Fd")",
                      frame.pts, p_sys->i_last_pts );
            return NULL;
        }
828 829 830 831 832
        else
        {
            p_sys->i_last_pts = frame.pts;
        }
    }
gbazin's avatar
 
gbazin committed
833

gbazin's avatar
gbazin committed
834 835
    frame.quality = p_sys->i_quality;

836 837 838 839 840 841 842 843 844 845 846
    /* Ugly work-around for stupid libavcodec behaviour */
#if LIBAVCODEC_BUILD >= 4722
    p_sys->i_framenum++;
    p_sys->pi_delay_pts[p_sys->i_framenum % MAX_FRAME_DELAY] = frame.pts;
    frame.pts = p_sys->i_framenum * AV_TIME_BASE *
        p_enc->fmt_in.video.i_frame_rate_base;
    frame.pts += p_enc->fmt_in.video.i_frame_rate - 1;
    frame.pts /= p_enc->fmt_in.video.i_frame_rate;
#endif
    /* End work-around */

gbazin's avatar
 
gbazin committed
847 848
    i_out = avcodec_encode_video( p_sys->p_context, p_sys->p_buffer_out,
                                  AVCODEC_MAX_VIDEO_FRAME_SIZE, &frame );
Christophe Massiot's avatar
Christophe Massiot committed
849

gbazin's avatar
 
gbazin committed
850 851 852 853 854
    if( i_out > 0 )
    {
        block_t *p_block = block_New( p_enc, i_out );
        memcpy( p_block->p_buffer, p_sys->p_buffer_out, i_out );

855 856 857 858 859 860 861 862 863 864
        /* FIXME, 3-2 pulldown is not handled correctly */
        p_block->i_length = I64C(1000000) *
            p_enc->fmt_in.video.i_frame_rate_base /
                p_enc->fmt_in.video.i_frame_rate;

        if( !p_sys->p_context->max_b_frames || !p_sys->p_context->delay )
        {
            /* No delay -> output pts == input pts */
            p_block->i_pts = p_block->i_dts = p_pict->date;
        }
865
        else if( p_sys->p_context->coded_frame->pts != (signed int) AV_NOPTS_VALUE &&
gbazin's avatar
gbazin committed
866
            p_sys->p_context->coded_frame->pts != 0 &&
gbazin's avatar
 
gbazin committed
867 868 869
            p_sys->i_buggy_pts_detect != p_sys->p_context->coded_frame->pts )
        {
            p_sys->i_buggy_pts_detect = p_sys->p_context->coded_frame->pts;
870
            p_block->i_pts = p_sys->p_context->coded_frame->pts;
gbazin's avatar
 
gbazin committed
871

872 873 874 875 876 877
            /* Ugly work-around for stupid libavcodec behaviour */
#if LIBAVCODEC_BUILD >= 4722
            {
            int64_t i_framenum = p_block->i_pts *
                p_enc->fmt_in.video.i_frame_rate /
                p_enc->fmt_in.video.i_frame_rate_base / AV_TIME_BASE;
gbazin's avatar
 
gbazin committed
878

879 880 881 882 883 884 885
            p_block->i_pts = p_sys->pi_delay_pts[i_framenum % MAX_FRAME_DELAY];
            }
#endif
            /* End work-around */

            if( p_sys->p_context->coded_frame->pict_type != FF_I_TYPE &&
                p_sys->p_context->coded_frame->pict_type != FF_P_TYPE )
gbazin's avatar
 
gbazin committed
886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910
            {
                p_block->i_dts = p_block->i_pts;
            }
            else
            {
                if( p_sys->i_last_ref_pts )
                {
                    p_block->i_dts = p_sys->i_last_ref_pts;
                }
                else
                {
                    /* Let's put something sensible */
                    p_block->i_dts = p_block->i_pts;
                }

                p_sys->i_last_ref_pts = p_block->i_pts;
            }
        }
        else
        {
            /* Buggy libavcodec which doesn't update coded_frame->pts
             * correctly */
            p_block->i_dts = p_block->i_pts = p_pict->date;
        }

911 912 913 914 915 916 917 918 919 920 921 922 923
        switch ( p_sys->p_context->coded_frame->pict_type )
        {
        case FF_I_TYPE:
            p_block->i_flags |= BLOCK_FLAG_TYPE_I;
            break;
        case FF_P_TYPE:
            p_block->i_flags |= BLOCK_FLAG_TYPE_P;
            break;
        case FF_B_TYPE:
            p_block->i_flags |= BLOCK_FLAG_TYPE_B;
            break;
        }

gbazin's avatar
 
gbazin committed
924 925 926 927 928 929 930 931 932 933 934 935 936
        return p_block;
    }

    return NULL;
}

/****************************************************************************
 * EncodeAudio: the whole thing
 ****************************************************************************/
static block_t *EncodeAudio( encoder_t *p_enc, aout_buffer_t *p_aout_buf )
{
    encoder_sys_t *p_sys = p_enc->p_sys;
    block_t *p_block, *p_chain = NULL;
gbazin's avatar
 
gbazin committed
937

gbazin's avatar
 
gbazin committed
938 939 940 941 942 943
    char *p_buffer = p_aout_buf->p_buffer;
    int i_samples = p_aout_buf->i_nb_samples;
    int i_samples_delay = p_sys->i_samples_delay;

    p_sys->i_pts = p_aout_buf->start_date -
                (mtime_t)1000000 * (mtime_t)p_sys->i_samples_delay /
gbazin's avatar
 
gbazin committed
944
                (mtime_t)p_enc->fmt_in.audio.i_rate;
gbazin's avatar
 
gbazin committed
945 946 947 948 949 950 951 952 953 954 955

    p_sys->i_samples_delay += i_samples;

    while( p_sys->i_samples_delay >= p_sys->p_context->frame_size )
    {
        int16_t *p_samples;
        int i_out;

        if( i_samples_delay )
        {
            /* Take care of the left-over from last time */
gbazin's avatar
 
gbazin committed
956
            int i_delay_size = i_samples_delay * 2 *
gbazin's avatar
 
gbazin committed
957 958 959 960 961 962
                                 p_sys->p_context->channels;
            int i_size = p_sys->i_frame_size - i_delay_size;

            p_samples = (int16_t *)p_sys->p_buffer;
            memcpy( p_sys->p_buffer + i_delay_size, p_buffer, i_size );
            p_buffer -= i_delay_size;
gbazin's avatar
 
gbazin committed
963
            i_samples += i_samples_delay;
gbazin's avatar
 
gbazin committed
964 965 966 967 968 969 970 971 972 973
            i_samples_delay = 0;
        }
        else
        {
            p_samples = (int16_t *)p_buffer;
        }

        i_out = avcodec_encode_audio( p_sys->p_context, p_sys->p_buffer_out,
                                      2 * AVCODEC_MAX_AUDIO_FRAME_SIZE,
                                      p_samples );
gbazin's avatar
 
gbazin committed
974 975 976 977 978

#if 0
        msg_Warn( p_enc, "avcodec_encode_audio: %d", i_out );
#endif
        if( i_out < 0 ) break;
gbazin's avatar
 
gbazin committed
979 980 981

        p_buffer += p_sys->i_frame_size;
        p_sys->i_samples_delay -= p_sys->p_context->frame_size;
gbazin's avatar
 
gbazin committed
982 983 984
        i_samples -= p_sys->p_context->frame_size;

        if( i_out == 0 ) continue;
gbazin's avatar
 
gbazin committed
985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000

        p_block = block_New( p_enc, i_out );
        memcpy( p_block->p_buffer, p_sys->p_buffer_out, i_out );

        p_block->i_length = (mtime_t)1000000 *
            (mtime_t)p_sys->p_context->frame_size /
            (mtime_t)p_sys->p_context->sample_rate;

        p_block->i_dts = p_block->i_pts = p_sys->i_pts;

        /* Update pts */
        p_sys->i_pts += p_block->i_length;
        block_ChainAppend( &p_chain, p_block );
    }

    /* Backup the remaining raw samples */
gbazin's avatar
 
gbazin committed
1001
    if( i_samples )
gbazin's avatar
 
gbazin committed
1002
    {
gbazin's avatar
 
gbazin committed
1003 1004
        memcpy( p_sys->p_buffer + i_samples_delay * 2 *
                p_sys->p_context->channels, p_buffer,
gbazin's avatar
 
gbazin committed
1005 1006 1007 1008 1009 1010 1011
                i_samples * 2 * p_sys->p_context->channels );
    }

    return p_chain;
}

/*****************************************************************************
gbazin's avatar
 
gbazin committed
1012
 * CloseEncoder: ffmpeg encoder destruction
gbazin's avatar
 
gbazin committed
1013
 *****************************************************************************/
gbazin's avatar
 
gbazin committed
1014
void E_(CloseEncoder)( vlc_object_t *p_this )
gbazin's avatar
 
gbazin committed
1015 1016 1017
{
    encoder_t *p_enc = (encoder_t *)p_this;
    encoder_sys_t *p_sys = p_enc->p_sys;
1018 1019 1020
    vlc_value_t lockval;

    var_Get( p_enc->p_libvlc, "avcodec", &lockval );
gbazin's avatar
 
gbazin committed
1021

1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037
#if LIBAVCODEC_BUILD >= 4702
    if ( p_sys->b_inited && p_enc->i_threads >= 1 )
    {
        int i;
        struct thread_context_t ** pp_contexts =
                (struct thread_context_t **)p_sys->p_context->thread_opaque;
        for ( i = 0; i < p_enc->i_threads; i++ )
        {
            pp_contexts[i]->b_die = 1;
            vlc_cond_signal( &pp_contexts[i]->cond );
            vlc_thread_join( pp_contexts[i] );
            vlc_mutex_destroy( &pp_contexts[i]->lock );
            vlc_cond_destroy( &pp_contexts[i]->cond );
            vlc_object_destroy( pp_contexts[i] );
        }

1038
        free( pp_contexts );
1039 1040 1041
    }
#endif

1042
    vlc_mutex_lock( lockval.p_address );
gbazin's avatar
 
gbazin committed
1043
    avcodec_close( p_sys->p_context );
1044
    vlc_mutex_unlock( lockval.p_address );
1045
    av_free( p_sys-></