]> git.sesse.net Git - ffmpeg/blob - libavcodec/libx264.c
85f1996ab69d6eb221628b3f9f1ee959bd4300fa
[ffmpeg] / libavcodec / libx264.c
1 /*
2  * H.264 encoding using the x264 library
3  * Copyright (C) 2005  Mans Rullgard <mans@mansr.com>
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include "libavutil/eval.h"
23 #include "libavutil/internal.h"
24 #include "libavutil/opt.h"
25 #include "libavutil/mem.h"
26 #include "libavutil/pixdesc.h"
27 #include "libavutil/stereo3d.h"
28 #include "libavutil/intreadwrite.h"
29 #include "avcodec.h"
30 #include "internal.h"
31
32 #if defined(_MSC_VER)
33 #define X264_API_IMPORTS 1
34 #endif
35
36 #include <x264.h>
37 #include <float.h>
38 #include <math.h>
39 #include <stdio.h>
40 #include <stdlib.h>
41 #include <string.h>
42
43 typedef struct X264Context {
44     AVClass        *class;
45     x264_param_t    params;
46     x264_t         *enc;
47     x264_picture_t  pic;
48     uint8_t        *sei;
49     int             sei_size;
50     char *preset;
51     char *tune;
52     char *profile;
53     char *level;
54     int fastfirstpass;
55     char *wpredp;
56     char *x264opts;
57     float crf;
58     float crf_max;
59     int cqp;
60     int aq_mode;
61     float aq_strength;
62     char *psy_rd;
63     int psy;
64     int rc_lookahead;
65     int weightp;
66     int weightb;
67     int ssim;
68     int intra_refresh;
69     int bluray_compat;
70     int b_bias;
71     int b_pyramid;
72     int mixed_refs;
73     int dct8x8;
74     int fast_pskip;
75     int aud;
76     int mbtree;
77     char *deblock;
78     float cplxblur;
79     char *partitions;
80     int direct_pred;
81     int slice_max_size;
82     char *stats;
83     int nal_hrd;
84     int avcintra_class;
85     int motion_est;
86     int forced_idr;
87     int coder;
88     int a53_cc;
89     int b_frame_strategy;
90     int chroma_offset;
91     int scenechange_threshold;
92     int noise_reduction;
93
94     char *x264_params;
95 } X264Context;
96
97 static void X264_log(void *p, int level, const char *fmt, va_list args)
98 {
99     static const int level_map[] = {
100         [X264_LOG_ERROR]   = AV_LOG_ERROR,
101         [X264_LOG_WARNING] = AV_LOG_WARNING,
102         [X264_LOG_INFO]    = AV_LOG_INFO,
103         [X264_LOG_DEBUG]   = AV_LOG_DEBUG
104     };
105
106     if (level < 0 || level > X264_LOG_DEBUG)
107         return;
108
109     av_vlog(p, level_map[level], fmt, args);
110 }
111
112
113 static int encode_nals(AVCodecContext *ctx, AVPacket *pkt,
114                        const x264_nal_t *nals, int nnal)
115 {
116     X264Context *x4 = ctx->priv_data;
117     uint8_t *p;
118     int i, size = x4->sei_size, ret;
119
120     if (!nnal)
121         return 0;
122
123     for (i = 0; i < nnal; i++)
124         size += nals[i].i_payload;
125
126     if ((ret = ff_alloc_packet2(ctx, pkt, size, 0)) < 0)
127         return ret;
128
129     p = pkt->data;
130
131     /* Write the SEI as part of the first frame. */
132     if (x4->sei_size > 0 && nnal > 0) {
133         if (x4->sei_size > size) {
134             av_log(ctx, AV_LOG_ERROR, "Error: nal buffer is too small\n");
135             return -1;
136         }
137         memcpy(p, x4->sei, x4->sei_size);
138         p += x4->sei_size;
139         x4->sei_size = 0;
140         av_freep(&x4->sei);
141     }
142
143     for (i = 0; i < nnal; i++){
144         memcpy(p, nals[i].p_payload, nals[i].i_payload);
145         p += nals[i].i_payload;
146     }
147
148     return 1;
149 }
150
151 static int avfmt2_num_planes(int avfmt)
152 {
153     switch (avfmt) {
154     case AV_PIX_FMT_YUV420P:
155     case AV_PIX_FMT_YUVJ420P:
156     case AV_PIX_FMT_YUV420P9:
157     case AV_PIX_FMT_YUV420P10:
158     case AV_PIX_FMT_YUV444P:
159         return 3;
160
161     case AV_PIX_FMT_BGR0:
162     case AV_PIX_FMT_BGR24:
163     case AV_PIX_FMT_RGB24:
164         return 1;
165
166     default:
167         return 3;
168     }
169 }
170
171 static void reconfig_encoder(AVCodecContext *ctx, const AVFrame *frame)
172 {
173     X264Context *x4 = ctx->priv_data;
174     AVFrameSideData *side_data;
175
176
177   if (x4->avcintra_class < 0) {
178     if (x4->params.b_interlaced && x4->params.b_tff != frame->top_field_first) {
179
180         x4->params.b_tff = frame->top_field_first;
181         x264_encoder_reconfig(x4->enc, &x4->params);
182     }
183     if (x4->params.vui.i_sar_height*ctx->sample_aspect_ratio.num != ctx->sample_aspect_ratio.den * x4->params.vui.i_sar_width) {
184         x4->params.vui.i_sar_height = ctx->sample_aspect_ratio.den;
185         x4->params.vui.i_sar_width  = ctx->sample_aspect_ratio.num;
186         x264_encoder_reconfig(x4->enc, &x4->params);
187     }
188
189     if (x4->params.rc.i_vbv_buffer_size != ctx->rc_buffer_size / 1000 ||
190         x4->params.rc.i_vbv_max_bitrate != ctx->rc_max_rate    / 1000) {
191         x4->params.rc.i_vbv_buffer_size = ctx->rc_buffer_size / 1000;
192         x4->params.rc.i_vbv_max_bitrate = ctx->rc_max_rate    / 1000;
193         x264_encoder_reconfig(x4->enc, &x4->params);
194     }
195
196     if (x4->params.rc.i_rc_method == X264_RC_ABR &&
197         x4->params.rc.i_bitrate != ctx->bit_rate / 1000) {
198         x4->params.rc.i_bitrate = ctx->bit_rate / 1000;
199         x264_encoder_reconfig(x4->enc, &x4->params);
200     }
201
202     if (x4->crf >= 0 &&
203         x4->params.rc.i_rc_method == X264_RC_CRF &&
204         x4->params.rc.f_rf_constant != x4->crf) {
205         x4->params.rc.f_rf_constant = x4->crf;
206         x264_encoder_reconfig(x4->enc, &x4->params);
207     }
208
209     if (x4->params.rc.i_rc_method == X264_RC_CQP &&
210         x4->cqp >= 0 &&
211         x4->params.rc.i_qp_constant != x4->cqp) {
212         x4->params.rc.i_qp_constant = x4->cqp;
213         x264_encoder_reconfig(x4->enc, &x4->params);
214     }
215
216     if (x4->crf_max >= 0 &&
217         x4->params.rc.f_rf_constant_max != x4->crf_max) {
218         x4->params.rc.f_rf_constant_max = x4->crf_max;
219         x264_encoder_reconfig(x4->enc, &x4->params);
220     }
221   }
222
223     side_data = av_frame_get_side_data(frame, AV_FRAME_DATA_STEREO3D);
224     if (side_data) {
225         AVStereo3D *stereo = (AVStereo3D *)side_data->data;
226         int fpa_type;
227
228         switch (stereo->type) {
229         case AV_STEREO3D_CHECKERBOARD:
230             fpa_type = 0;
231             break;
232         case AV_STEREO3D_COLUMNS:
233             fpa_type = 1;
234             break;
235         case AV_STEREO3D_LINES:
236             fpa_type = 2;
237             break;
238         case AV_STEREO3D_SIDEBYSIDE:
239             fpa_type = 3;
240             break;
241         case AV_STEREO3D_TOPBOTTOM:
242             fpa_type = 4;
243             break;
244         case AV_STEREO3D_FRAMESEQUENCE:
245             fpa_type = 5;
246             break;
247 #if X264_BUILD >= 145
248         case AV_STEREO3D_2D:
249             fpa_type = 6;
250             break;
251 #endif
252         default:
253             fpa_type = -1;
254             break;
255         }
256
257         /* Inverted mode is not supported by x264 */
258         if (stereo->flags & AV_STEREO3D_FLAG_INVERT) {
259             av_log(ctx, AV_LOG_WARNING,
260                    "Ignoring unsupported inverted stereo value %d\n", fpa_type);
261             fpa_type = -1;
262         }
263
264         if (fpa_type != x4->params.i_frame_packing) {
265             x4->params.i_frame_packing = fpa_type;
266             x264_encoder_reconfig(x4->enc, &x4->params);
267         }
268     }
269 }
270
271 static int X264_frame(AVCodecContext *ctx, AVPacket *pkt, const AVFrame *frame,
272                       int *got_packet)
273 {
274     X264Context *x4 = ctx->priv_data;
275     x264_nal_t *nal;
276     int nnal, i, ret;
277     x264_picture_t pic_out = {0};
278     int pict_type;
279     AVFrameSideData *side_data;
280
281     x264_picture_init( &x4->pic );
282     x4->pic.img.i_csp   = x4->params.i_csp;
283     if (x264_bit_depth > 8)
284         x4->pic.img.i_csp |= X264_CSP_HIGH_DEPTH;
285     x4->pic.img.i_plane = avfmt2_num_planes(ctx->pix_fmt);
286
287     if (frame) {
288         for (i = 0; i < x4->pic.img.i_plane; i++) {
289             x4->pic.img.plane[i]    = frame->data[i];
290             x4->pic.img.i_stride[i] = frame->linesize[i];
291         }
292
293         x4->pic.i_pts  = frame->pts;
294
295         switch (frame->pict_type) {
296         case AV_PICTURE_TYPE_I:
297             x4->pic.i_type = x4->forced_idr >= 0 ? X264_TYPE_IDR
298                                                  : X264_TYPE_KEYFRAME;
299             break;
300         case AV_PICTURE_TYPE_P:
301             x4->pic.i_type = X264_TYPE_P;
302             break;
303         case AV_PICTURE_TYPE_B:
304             x4->pic.i_type = X264_TYPE_B;
305             break;
306         default:
307             x4->pic.i_type = X264_TYPE_AUTO;
308             break;
309         }
310         reconfig_encoder(ctx, frame);
311
312         if (x4->a53_cc) {
313             void *sei_data;
314             size_t sei_size;
315
316             ret = ff_alloc_a53_sei(frame, 0, &sei_data, &sei_size);
317             if (ret < 0) {
318                 av_log(ctx, AV_LOG_ERROR, "Not enough memory for closed captions, skipping\n");
319             } else if (sei_data) {
320                 x4->pic.extra_sei.payloads = av_mallocz(sizeof(x4->pic.extra_sei.payloads[0]));
321                 if (x4->pic.extra_sei.payloads == NULL) {
322                     av_log(ctx, AV_LOG_ERROR, "Not enough memory for closed captions, skipping\n");
323                     av_free(sei_data);
324                 } else {
325                     x4->pic.extra_sei.sei_free = av_free;
326
327                     x4->pic.extra_sei.payloads[0].payload_size = sei_size;
328                     x4->pic.extra_sei.payloads[0].payload = sei_data;
329                     x4->pic.extra_sei.num_payloads = 1;
330                     x4->pic.extra_sei.payloads[0].payload_type = 4;
331                 }
332             }
333         }
334     }
335
336     do {
337         if (x264_encoder_encode(x4->enc, &nal, &nnal, frame? &x4->pic: NULL, &pic_out) < 0)
338             return AVERROR_EXTERNAL;
339
340         ret = encode_nals(ctx, pkt, nal, nnal);
341         if (ret < 0)
342             return ret;
343     } while (!ret && !frame && x264_encoder_delayed_frames(x4->enc));
344
345     pkt->pts = pic_out.i_pts;
346     pkt->dts = pic_out.i_dts;
347
348
349     switch (pic_out.i_type) {
350     case X264_TYPE_IDR:
351     case X264_TYPE_I:
352         pict_type = AV_PICTURE_TYPE_I;
353         break;
354     case X264_TYPE_P:
355         pict_type = AV_PICTURE_TYPE_P;
356         break;
357     case X264_TYPE_B:
358     case X264_TYPE_BREF:
359         pict_type = AV_PICTURE_TYPE_B;
360         break;
361     default:
362         pict_type = AV_PICTURE_TYPE_NONE;
363     }
364 #if FF_API_CODED_FRAME
365 FF_DISABLE_DEPRECATION_WARNINGS
366     ctx->coded_frame->pict_type = pict_type;
367 FF_ENABLE_DEPRECATION_WARNINGS
368 #endif
369
370     pkt->flags |= AV_PKT_FLAG_KEY*pic_out.b_keyframe;
371     if (ret) {
372         ff_side_data_set_encoder_stats(pkt, (pic_out.i_qpplus1 - 1) * FF_QP2LAMBDA, NULL, 0, pict_type);
373
374 #if FF_API_CODED_FRAME
375 FF_DISABLE_DEPRECATION_WARNINGS
376         ctx->coded_frame->quality = (pic_out.i_qpplus1 - 1) * FF_QP2LAMBDA;
377 FF_ENABLE_DEPRECATION_WARNINGS
378 #endif
379     }
380
381     *got_packet = ret;
382     return 0;
383 }
384
385 static av_cold int X264_close(AVCodecContext *avctx)
386 {
387     X264Context *x4 = avctx->priv_data;
388
389     av_freep(&avctx->extradata);
390     av_freep(&x4->sei);
391
392     if (x4->enc) {
393         x264_encoder_close(x4->enc);
394         x4->enc = NULL;
395     }
396
397     return 0;
398 }
399
400 #define OPT_STR(opt, param)                                                   \
401     do {                                                                      \
402         int ret;                                                              \
403         if ((ret = x264_param_parse(&x4->params, opt, param)) < 0) { \
404             if(ret == X264_PARAM_BAD_NAME)                                    \
405                 av_log(avctx, AV_LOG_ERROR,                                   \
406                         "bad option '%s': '%s'\n", opt, param);               \
407             else                                                              \
408                 av_log(avctx, AV_LOG_ERROR,                                   \
409                         "bad value for '%s': '%s'\n", opt, param);            \
410             return -1;                                                        \
411         }                                                                     \
412     } while (0)
413
414 static int convert_pix_fmt(enum AVPixelFormat pix_fmt)
415 {
416     switch (pix_fmt) {
417     case AV_PIX_FMT_YUV420P:
418     case AV_PIX_FMT_YUVJ420P:
419     case AV_PIX_FMT_YUV420P9:
420     case AV_PIX_FMT_YUV420P10: return X264_CSP_I420;
421     case AV_PIX_FMT_YUV422P:
422     case AV_PIX_FMT_YUVJ422P:
423     case AV_PIX_FMT_YUV422P10: return X264_CSP_I422;
424     case AV_PIX_FMT_YUV444P:
425     case AV_PIX_FMT_YUVJ444P:
426     case AV_PIX_FMT_YUV444P9:
427     case AV_PIX_FMT_YUV444P10: return X264_CSP_I444;
428 #if CONFIG_LIBX264RGB_ENCODER
429     case AV_PIX_FMT_BGR0:
430         return X264_CSP_BGRA;
431     case AV_PIX_FMT_BGR24:
432         return X264_CSP_BGR;
433
434     case AV_PIX_FMT_RGB24:
435         return X264_CSP_RGB;
436 #endif
437     case AV_PIX_FMT_NV12:      return X264_CSP_NV12;
438     case AV_PIX_FMT_NV16:
439     case AV_PIX_FMT_NV20:      return X264_CSP_NV16;
440 #ifdef X264_CSP_NV21
441     case AV_PIX_FMT_NV21:      return X264_CSP_NV21;
442 #endif
443     };
444     return 0;
445 }
446
447 #define PARSE_X264_OPT(name, var)\
448     if (x4->var && x264_param_parse(&x4->params, name, x4->var) < 0) {\
449         av_log(avctx, AV_LOG_ERROR, "Error parsing option '%s' with value '%s'.\n", name, x4->var);\
450         return AVERROR(EINVAL);\
451     }
452
453 static av_cold int X264_init(AVCodecContext *avctx)
454 {
455     X264Context *x4 = avctx->priv_data;
456     AVCPBProperties *cpb_props;
457     int sw,sh;
458
459     if (avctx->global_quality > 0)
460         av_log(avctx, AV_LOG_WARNING, "-qscale is ignored, -crf is recommended.\n");
461
462 #if CONFIG_LIBX262_ENCODER
463     if (avctx->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
464         x4->params.b_mpeg2 = 1;
465         x264_param_default_mpeg2(&x4->params);
466     } else
467 #endif
468     x264_param_default(&x4->params);
469
470     x4->params.b_deblocking_filter         = avctx->flags & AV_CODEC_FLAG_LOOP_FILTER;
471
472     if (x4->preset || x4->tune)
473         if (x264_param_default_preset(&x4->params, x4->preset, x4->tune) < 0) {
474             int i;
475             av_log(avctx, AV_LOG_ERROR, "Error setting preset/tune %s/%s.\n", x4->preset, x4->tune);
476             av_log(avctx, AV_LOG_INFO, "Possible presets:");
477             for (i = 0; x264_preset_names[i]; i++)
478                 av_log(avctx, AV_LOG_INFO, " %s", x264_preset_names[i]);
479             av_log(avctx, AV_LOG_INFO, "\n");
480             av_log(avctx, AV_LOG_INFO, "Possible tunes:");
481             for (i = 0; x264_tune_names[i]; i++)
482                 av_log(avctx, AV_LOG_INFO, " %s", x264_tune_names[i]);
483             av_log(avctx, AV_LOG_INFO, "\n");
484             return AVERROR(EINVAL);
485         }
486
487     if (avctx->level > 0)
488         x4->params.i_level_idc = avctx->level;
489
490     x4->params.pf_log               = X264_log;
491     x4->params.p_log_private        = avctx;
492     x4->params.i_log_level          = X264_LOG_DEBUG;
493     x4->params.i_csp                = convert_pix_fmt(avctx->pix_fmt);
494
495     PARSE_X264_OPT("weightp", wpredp);
496
497     if (avctx->bit_rate) {
498         x4->params.rc.i_bitrate   = avctx->bit_rate / 1000;
499         x4->params.rc.i_rc_method = X264_RC_ABR;
500     }
501     x4->params.rc.i_vbv_buffer_size = avctx->rc_buffer_size / 1000;
502     x4->params.rc.i_vbv_max_bitrate = avctx->rc_max_rate    / 1000;
503     x4->params.rc.b_stat_write      = avctx->flags & AV_CODEC_FLAG_PASS1;
504     if (avctx->flags & AV_CODEC_FLAG_PASS2) {
505         x4->params.rc.b_stat_read = 1;
506     } else {
507         if (x4->crf >= 0) {
508             x4->params.rc.i_rc_method   = X264_RC_CRF;
509             x4->params.rc.f_rf_constant = x4->crf;
510         } else if (x4->cqp >= 0) {
511             x4->params.rc.i_rc_method   = X264_RC_CQP;
512             x4->params.rc.i_qp_constant = x4->cqp;
513         }
514
515         if (x4->crf_max >= 0)
516             x4->params.rc.f_rf_constant_max = x4->crf_max;
517     }
518
519     if (avctx->rc_buffer_size && avctx->rc_initial_buffer_occupancy > 0 &&
520         (avctx->rc_initial_buffer_occupancy <= avctx->rc_buffer_size)) {
521         x4->params.rc.f_vbv_buffer_init =
522             (float)avctx->rc_initial_buffer_occupancy / avctx->rc_buffer_size;
523     }
524
525     PARSE_X264_OPT("level", level);
526
527     if (avctx->i_quant_factor > 0)
528         x4->params.rc.f_ip_factor         = 1 / fabs(avctx->i_quant_factor);
529     if (avctx->b_quant_factor > 0)
530         x4->params.rc.f_pb_factor         = avctx->b_quant_factor;
531
532 #if FF_API_PRIVATE_OPT
533 FF_DISABLE_DEPRECATION_WARNINGS
534     if (avctx->chromaoffset >= 0)
535         x4->chroma_offset = avctx->chromaoffset;
536 FF_ENABLE_DEPRECATION_WARNINGS
537 #endif
538     if (x4->chroma_offset >= 0)
539         x4->params.analyse.i_chroma_qp_offset = x4->chroma_offset;
540
541     if (avctx->gop_size >= 0)
542         x4->params.i_keyint_max         = avctx->gop_size;
543     if (avctx->max_b_frames >= 0)
544         x4->params.i_bframe             = avctx->max_b_frames;
545
546 #if FF_API_PRIVATE_OPT
547 FF_DISABLE_DEPRECATION_WARNINGS
548     if (avctx->scenechange_threshold >= 0)
549         x4->scenechange_threshold = avctx->scenechange_threshold;
550 FF_ENABLE_DEPRECATION_WARNINGS
551 #endif
552     if (x4->scenechange_threshold >= 0)
553         x4->params.i_scenecut_threshold = x4->scenechange_threshold;
554
555     if (avctx->qmin >= 0)
556         x4->params.rc.i_qp_min          = avctx->qmin;
557     if (avctx->qmax >= 0)
558         x4->params.rc.i_qp_max          = avctx->qmax;
559     if (avctx->max_qdiff >= 0)
560         x4->params.rc.i_qp_step         = avctx->max_qdiff;
561     if (avctx->qblur >= 0)
562         x4->params.rc.f_qblur           = avctx->qblur;     /* temporally blur quants */
563     if (avctx->qcompress >= 0)
564         x4->params.rc.f_qcompress       = avctx->qcompress; /* 0.0 => cbr, 1.0 => constant qp */
565     if (avctx->refs >= 0)
566         x4->params.i_frame_reference    = avctx->refs;
567     else if (x4->level) {
568         int i;
569         int mbn = AV_CEIL_RSHIFT(avctx->width, 4) * AV_CEIL_RSHIFT(avctx->height, 4);
570         int level_id = -1;
571         char *tail;
572         int scale = X264_BUILD < 129 ? 384 : 1;
573
574         if (!strcmp(x4->level, "1b")) {
575             level_id = 9;
576         } else if (strlen(x4->level) <= 3){
577             level_id = av_strtod(x4->level, &tail) * 10 + 0.5;
578             if (*tail)
579                 level_id = -1;
580         }
581         if (level_id <= 0)
582             av_log(avctx, AV_LOG_WARNING, "Failed to parse level\n");
583
584         for (i = 0; i<x264_levels[i].level_idc; i++)
585             if (x264_levels[i].level_idc == level_id)
586                 x4->params.i_frame_reference = av_clip(x264_levels[i].dpb / mbn / scale, 1, x4->params.i_frame_reference);
587     }
588
589     if (avctx->trellis >= 0)
590         x4->params.analyse.i_trellis    = avctx->trellis;
591     if (avctx->me_range >= 0)
592         x4->params.analyse.i_me_range   = avctx->me_range;
593 #if FF_API_PRIVATE_OPT
594     FF_DISABLE_DEPRECATION_WARNINGS
595     if (avctx->noise_reduction >= 0)
596         x4->noise_reduction = avctx->noise_reduction;
597     FF_ENABLE_DEPRECATION_WARNINGS
598 #endif
599     if (x4->noise_reduction >= 0)
600         x4->params.analyse.i_noise_reduction = x4->noise_reduction;
601     if (avctx->me_subpel_quality >= 0)
602         x4->params.analyse.i_subpel_refine   = avctx->me_subpel_quality;
603 #if FF_API_PRIVATE_OPT
604 FF_DISABLE_DEPRECATION_WARNINGS
605     if (avctx->b_frame_strategy >= 0)
606         x4->b_frame_strategy = avctx->b_frame_strategy;
607 FF_ENABLE_DEPRECATION_WARNINGS
608 #endif
609     if (avctx->keyint_min >= 0)
610         x4->params.i_keyint_min = avctx->keyint_min;
611 #if FF_API_CODER_TYPE
612 FF_DISABLE_DEPRECATION_WARNINGS
613     if (avctx->coder_type >= 0)
614         x4->coder = avctx->coder_type == FF_CODER_TYPE_AC;
615 FF_ENABLE_DEPRECATION_WARNINGS
616 #endif
617     if (avctx->me_cmp >= 0)
618         x4->params.analyse.b_chroma_me = avctx->me_cmp & FF_CMP_CHROMA;
619
620     if (x4->aq_mode >= 0)
621         x4->params.rc.i_aq_mode = x4->aq_mode;
622     if (x4->aq_strength >= 0)
623         x4->params.rc.f_aq_strength = x4->aq_strength;
624     PARSE_X264_OPT("psy-rd", psy_rd);
625     PARSE_X264_OPT("deblock", deblock);
626     PARSE_X264_OPT("partitions", partitions);
627     PARSE_X264_OPT("stats", stats);
628     if (x4->psy >= 0)
629         x4->params.analyse.b_psy  = x4->psy;
630     if (x4->rc_lookahead >= 0)
631         x4->params.rc.i_lookahead = x4->rc_lookahead;
632     if (x4->weightp >= 0)
633         x4->params.analyse.i_weighted_pred = x4->weightp;
634     if (x4->weightb >= 0)
635         x4->params.analyse.b_weighted_bipred = x4->weightb;
636     if (x4->cplxblur >= 0)
637         x4->params.rc.f_complexity_blur = x4->cplxblur;
638
639     if (x4->ssim >= 0)
640         x4->params.analyse.b_ssim = x4->ssim;
641     if (x4->intra_refresh >= 0)
642         x4->params.b_intra_refresh = x4->intra_refresh;
643     if (x4->bluray_compat >= 0) {
644         x4->params.b_bluray_compat = x4->bluray_compat;
645         x4->params.b_vfr_input = 0;
646     }
647     if (x4->avcintra_class >= 0)
648 #if X264_BUILD >= 142
649         x4->params.i_avcintra_class = x4->avcintra_class;
650 #else
651         av_log(avctx, AV_LOG_ERROR,
652                "x264 too old for AVC Intra, at least version 142 needed\n");
653 #endif
654     if (x4->b_bias != INT_MIN)
655         x4->params.i_bframe_bias              = x4->b_bias;
656     if (x4->b_pyramid >= 0)
657         x4->params.i_bframe_pyramid = x4->b_pyramid;
658     if (x4->mixed_refs >= 0)
659         x4->params.analyse.b_mixed_references = x4->mixed_refs;
660     if (x4->dct8x8 >= 0)
661         x4->params.analyse.b_transform_8x8    = x4->dct8x8;
662     if (x4->fast_pskip >= 0)
663         x4->params.analyse.b_fast_pskip       = x4->fast_pskip;
664     if (x4->aud >= 0)
665         x4->params.b_aud                      = x4->aud;
666     if (x4->mbtree >= 0)
667         x4->params.rc.b_mb_tree               = x4->mbtree;
668     if (x4->direct_pred >= 0)
669         x4->params.analyse.i_direct_mv_pred   = x4->direct_pred;
670
671     if (x4->slice_max_size >= 0)
672         x4->params.i_slice_max_size =  x4->slice_max_size;
673
674     if (x4->fastfirstpass)
675         x264_param_apply_fastfirstpass(&x4->params);
676
677     /* Allow specifying the x264 profile through AVCodecContext. */
678     if (!x4->profile)
679         switch (avctx->profile) {
680         case FF_PROFILE_H264_BASELINE:
681             x4->profile = av_strdup("baseline");
682             break;
683         case FF_PROFILE_H264_HIGH:
684             x4->profile = av_strdup("high");
685             break;
686         case FF_PROFILE_H264_HIGH_10:
687             x4->profile = av_strdup("high10");
688             break;
689         case FF_PROFILE_H264_HIGH_422:
690             x4->profile = av_strdup("high422");
691             break;
692         case FF_PROFILE_H264_HIGH_444:
693             x4->profile = av_strdup("high444");
694             break;
695         case FF_PROFILE_H264_MAIN:
696             x4->profile = av_strdup("main");
697             break;
698         default:
699             break;
700         }
701
702     if (x4->nal_hrd >= 0)
703         x4->params.i_nal_hrd = x4->nal_hrd;
704
705     if (x4->motion_est >= 0) {
706         x4->params.analyse.i_me_method = x4->motion_est;
707 #if FF_API_MOTION_EST
708 FF_DISABLE_DEPRECATION_WARNINGS
709     } else {
710         if (avctx->me_method == ME_EPZS)
711             x4->params.analyse.i_me_method = X264_ME_DIA;
712         else if (avctx->me_method == ME_HEX)
713             x4->params.analyse.i_me_method = X264_ME_HEX;
714         else if (avctx->me_method == ME_UMH)
715             x4->params.analyse.i_me_method = X264_ME_UMH;
716         else if (avctx->me_method == ME_FULL)
717             x4->params.analyse.i_me_method = X264_ME_ESA;
718         else if (avctx->me_method == ME_TESA)
719             x4->params.analyse.i_me_method = X264_ME_TESA;
720 FF_ENABLE_DEPRECATION_WARNINGS
721 #endif
722     }
723
724     if (x4->coder >= 0)
725         x4->params.b_cabac = x4->coder;
726
727     if (x4->b_frame_strategy >= 0)
728         x4->params.i_bframe_adaptive = x4->b_frame_strategy;
729
730     if (x4->profile)
731         if (x264_param_apply_profile(&x4->params, x4->profile) < 0) {
732             int i;
733             av_log(avctx, AV_LOG_ERROR, "Error setting profile %s.\n", x4->profile);
734             av_log(avctx, AV_LOG_INFO, "Possible profiles:");
735             for (i = 0; x264_profile_names[i]; i++)
736                 av_log(avctx, AV_LOG_INFO, " %s", x264_profile_names[i]);
737             av_log(avctx, AV_LOG_INFO, "\n");
738             return AVERROR(EINVAL);
739         }
740
741     x4->params.i_width          = avctx->width;
742     x4->params.i_height         = avctx->height;
743     av_reduce(&sw, &sh, avctx->sample_aspect_ratio.num, avctx->sample_aspect_ratio.den, 4096);
744     x4->params.vui.i_sar_width  = sw;
745     x4->params.vui.i_sar_height = sh;
746     x4->params.i_timebase_den = avctx->time_base.den;
747     x4->params.i_timebase_num = avctx->time_base.num;
748     x4->params.i_fps_num = avctx->time_base.den;
749     x4->params.i_fps_den = avctx->time_base.num * avctx->ticks_per_frame;
750
751     x4->params.analyse.b_psnr = avctx->flags & AV_CODEC_FLAG_PSNR;
752
753     x4->params.i_threads      = avctx->thread_count;
754     if (avctx->thread_type)
755         x4->params.b_sliced_threads = avctx->thread_type == FF_THREAD_SLICE;
756
757     x4->params.b_interlaced   = avctx->flags & AV_CODEC_FLAG_INTERLACED_DCT;
758
759     x4->params.b_open_gop     = !(avctx->flags & AV_CODEC_FLAG_CLOSED_GOP);
760
761     x4->params.i_slice_count  = avctx->slices;
762
763     x4->params.vui.b_fullrange = avctx->pix_fmt == AV_PIX_FMT_YUVJ420P ||
764                                  avctx->pix_fmt == AV_PIX_FMT_YUVJ422P ||
765                                  avctx->pix_fmt == AV_PIX_FMT_YUVJ444P ||
766                                  avctx->color_range == AVCOL_RANGE_JPEG;
767
768     if (avctx->colorspace != AVCOL_SPC_UNSPECIFIED)
769         x4->params.vui.i_colmatrix = avctx->colorspace;
770     if (avctx->color_primaries != AVCOL_PRI_UNSPECIFIED)
771         x4->params.vui.i_colorprim = avctx->color_primaries;
772     if (avctx->color_trc != AVCOL_TRC_UNSPECIFIED)
773         x4->params.vui.i_transfer  = avctx->color_trc;
774
775     if (avctx->flags & AV_CODEC_FLAG_GLOBAL_HEADER)
776         x4->params.b_repeat_headers = 0;
777
778     if(x4->x264opts){
779         const char *p= x4->x264opts;
780         while(p){
781             char param[256]={0}, val[256]={0};
782             if(sscanf(p, "%255[^:=]=%255[^:]", param, val) == 1){
783                 OPT_STR(param, "1");
784             }else
785                 OPT_STR(param, val);
786             p= strchr(p, ':');
787             p+=!!p;
788         }
789     }
790
791     if (x4->x264_params) {
792         AVDictionary *dict    = NULL;
793         AVDictionaryEntry *en = NULL;
794
795         if (!av_dict_parse_string(&dict, x4->x264_params, "=", ":", 0)) {
796             while ((en = av_dict_get(dict, "", en, AV_DICT_IGNORE_SUFFIX))) {
797                 if (x264_param_parse(&x4->params, en->key, en->value) < 0)
798                     av_log(avctx, AV_LOG_WARNING,
799                            "Error parsing option '%s = %s'.\n",
800                             en->key, en->value);
801             }
802
803             av_dict_free(&dict);
804         }
805     }
806
807     // update AVCodecContext with x264 parameters
808     avctx->has_b_frames = x4->params.i_bframe ?
809         x4->params.i_bframe_pyramid ? 2 : 1 : 0;
810     if (avctx->max_b_frames < 0)
811         avctx->max_b_frames = 0;
812
813     avctx->bit_rate = x4->params.rc.i_bitrate*1000;
814
815     x4->enc = x264_encoder_open(&x4->params);
816     if (!x4->enc)
817         return AVERROR_EXTERNAL;
818
819     if (avctx->flags & AV_CODEC_FLAG_GLOBAL_HEADER) {
820         x264_nal_t *nal;
821         uint8_t *p;
822         int nnal, s, i;
823
824         s = x264_encoder_headers(x4->enc, &nal, &nnal);
825         avctx->extradata = p = av_mallocz(s + AV_INPUT_BUFFER_PADDING_SIZE);
826         if (!p)
827             return AVERROR(ENOMEM);
828
829         for (i = 0; i < nnal; i++) {
830             /* Don't put the SEI in extradata. */
831             if (nal[i].i_type == NAL_SEI) {
832                 av_log(avctx, AV_LOG_INFO, "%s\n", nal[i].p_payload+25);
833                 x4->sei_size = nal[i].i_payload;
834                 x4->sei      = av_malloc(x4->sei_size);
835                 if (!x4->sei)
836                     return AVERROR(ENOMEM);
837                 memcpy(x4->sei, nal[i].p_payload, nal[i].i_payload);
838                 continue;
839             }
840             memcpy(p, nal[i].p_payload, nal[i].i_payload);
841             p += nal[i].i_payload;
842         }
843         avctx->extradata_size = p - avctx->extradata;
844     }
845
846     cpb_props = ff_add_cpb_side_data(avctx);
847     if (!cpb_props)
848         return AVERROR(ENOMEM);
849     cpb_props->buffer_size = x4->params.rc.i_vbv_buffer_size * 1000;
850     cpb_props->max_bitrate = x4->params.rc.i_vbv_max_bitrate * 1000;
851     cpb_props->avg_bitrate = x4->params.rc.i_bitrate         * 1000;
852
853     return 0;
854 }
855
856 static const enum AVPixelFormat pix_fmts_8bit[] = {
857     AV_PIX_FMT_YUV420P,
858     AV_PIX_FMT_YUVJ420P,
859     AV_PIX_FMT_YUV422P,
860     AV_PIX_FMT_YUVJ422P,
861     AV_PIX_FMT_YUV444P,
862     AV_PIX_FMT_YUVJ444P,
863     AV_PIX_FMT_NV12,
864     AV_PIX_FMT_NV16,
865 #ifdef X264_CSP_NV21
866     AV_PIX_FMT_NV21,
867 #endif
868     AV_PIX_FMT_NONE
869 };
870 static const enum AVPixelFormat pix_fmts_9bit[] = {
871     AV_PIX_FMT_YUV420P9,
872     AV_PIX_FMT_YUV444P9,
873     AV_PIX_FMT_NONE
874 };
875 static const enum AVPixelFormat pix_fmts_10bit[] = {
876     AV_PIX_FMT_YUV420P10,
877     AV_PIX_FMT_YUV422P10,
878     AV_PIX_FMT_YUV444P10,
879     AV_PIX_FMT_NV20,
880     AV_PIX_FMT_NONE
881 };
882 #if CONFIG_LIBX264RGB_ENCODER
883 static const enum AVPixelFormat pix_fmts_8bit_rgb[] = {
884     AV_PIX_FMT_BGR0,
885     AV_PIX_FMT_BGR24,
886     AV_PIX_FMT_RGB24,
887     AV_PIX_FMT_NONE
888 };
889 #endif
890
891 static av_cold void X264_init_static(AVCodec *codec)
892 {
893     if (x264_bit_depth == 8)
894         codec->pix_fmts = pix_fmts_8bit;
895     else if (x264_bit_depth == 9)
896         codec->pix_fmts = pix_fmts_9bit;
897     else if (x264_bit_depth == 10)
898         codec->pix_fmts = pix_fmts_10bit;
899 }
900
901 #define OFFSET(x) offsetof(X264Context, x)
902 #define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
903 static const AVOption options[] = {
904     { "preset",        "Set the encoding preset (cf. x264 --fullhelp)",   OFFSET(preset),        AV_OPT_TYPE_STRING, { .str = "medium" }, 0, 0, VE},
905     { "tune",          "Tune the encoding params (cf. x264 --fullhelp)",  OFFSET(tune),          AV_OPT_TYPE_STRING, { 0 }, 0, 0, VE},
906     { "profile",       "Set profile restrictions (cf. x264 --fullhelp) ", OFFSET(profile),       AV_OPT_TYPE_STRING, { 0 }, 0, 0, VE},
907     { "fastfirstpass", "Use fast settings when encoding first pass",      OFFSET(fastfirstpass), AV_OPT_TYPE_BOOL, { .i64 = 1 }, 0, 1, VE},
908     {"level", "Specify level (as defined by Annex A)", OFFSET(level), AV_OPT_TYPE_STRING, {.str=NULL}, 0, 0, VE},
909     {"passlogfile", "Filename for 2 pass stats", OFFSET(stats), AV_OPT_TYPE_STRING, {.str=NULL}, 0, 0, VE},
910     {"wpredp", "Weighted prediction for P-frames", OFFSET(wpredp), AV_OPT_TYPE_STRING, {.str=NULL}, 0, 0, VE},
911     {"a53cc",          "Use A53 Closed Captions (if available)",          OFFSET(a53_cc),        AV_OPT_TYPE_BOOL,   {.i64 = 0}, 0, 1, VE},
912     {"x264opts", "x264 options", OFFSET(x264opts), AV_OPT_TYPE_STRING, {.str=NULL}, 0, 0, VE},
913     { "crf",           "Select the quality for constant quality mode",    OFFSET(crf),           AV_OPT_TYPE_FLOAT,  {.dbl = -1 }, -1, FLT_MAX, VE },
914     { "crf_max",       "In CRF mode, prevents VBV from lowering quality beyond this point.",OFFSET(crf_max), AV_OPT_TYPE_FLOAT, {.dbl = -1 }, -1, FLT_MAX, VE },
915     { "qp",            "Constant quantization parameter rate control method",OFFSET(cqp),        AV_OPT_TYPE_INT,    { .i64 = -1 }, -1, INT_MAX, VE },
916     { "aq-mode",       "AQ method",                                       OFFSET(aq_mode),       AV_OPT_TYPE_INT,    { .i64 = -1 }, -1, INT_MAX, VE, "aq_mode"},
917     { "none",          NULL,                              0, AV_OPT_TYPE_CONST, {.i64 = X264_AQ_NONE},         INT_MIN, INT_MAX, VE, "aq_mode" },
918     { "variance",      "Variance AQ (complexity mask)",   0, AV_OPT_TYPE_CONST, {.i64 = X264_AQ_VARIANCE},     INT_MIN, INT_MAX, VE, "aq_mode" },
919     { "autovariance",  "Auto-variance AQ",                0, AV_OPT_TYPE_CONST, {.i64 = X264_AQ_AUTOVARIANCE}, INT_MIN, INT_MAX, VE, "aq_mode" },
920 #if X264_BUILD >= 144
921     { "autovariance-biased", "Auto-variance AQ with bias to dark scenes", 0, AV_OPT_TYPE_CONST, {.i64 = X264_AQ_AUTOVARIANCE_BIASED}, INT_MIN, INT_MAX, VE, "aq_mode" },
922 #endif
923     { "aq-strength",   "AQ strength. Reduces blocking and blurring in flat and textured areas.", OFFSET(aq_strength), AV_OPT_TYPE_FLOAT, {.dbl = -1}, -1, FLT_MAX, VE},
924     { "psy",           "Use psychovisual optimizations.",                 OFFSET(psy),           AV_OPT_TYPE_BOOL,   { .i64 = -1 }, -1, 1, VE },
925     { "psy-rd",        "Strength of psychovisual optimization, in <psy-rd>:<psy-trellis> format.", OFFSET(psy_rd), AV_OPT_TYPE_STRING,  {0 }, 0, 0, VE},
926     { "rc-lookahead",  "Number of frames to look ahead for frametype and ratecontrol", OFFSET(rc_lookahead), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, INT_MAX, VE },
927     { "weightb",       "Weighted prediction for B-frames.",               OFFSET(weightb),       AV_OPT_TYPE_BOOL,   { .i64 = -1 }, -1, 1, VE },
928     { "weightp",       "Weighted prediction analysis method.",            OFFSET(weightp),       AV_OPT_TYPE_INT,    { .i64 = -1 }, -1, INT_MAX, VE, "weightp" },
929     { "none",          NULL, 0, AV_OPT_TYPE_CONST, {.i64 = X264_WEIGHTP_NONE},   INT_MIN, INT_MAX, VE, "weightp" },
930     { "simple",        NULL, 0, AV_OPT_TYPE_CONST, {.i64 = X264_WEIGHTP_SIMPLE}, INT_MIN, INT_MAX, VE, "weightp" },
931     { "smart",         NULL, 0, AV_OPT_TYPE_CONST, {.i64 = X264_WEIGHTP_SMART},  INT_MIN, INT_MAX, VE, "weightp" },
932     { "ssim",          "Calculate and print SSIM stats.",                 OFFSET(ssim),          AV_OPT_TYPE_BOOL,   { .i64 = -1 }, -1, 1, VE },
933     { "intra-refresh", "Use Periodic Intra Refresh instead of IDR frames.",OFFSET(intra_refresh),AV_OPT_TYPE_BOOL,   { .i64 = -1 }, -1, 1, VE },
934     { "bluray-compat", "Bluray compatibility workarounds.",               OFFSET(bluray_compat) ,AV_OPT_TYPE_BOOL,   { .i64 = -1 }, -1, 1, VE },
935     { "b-bias",        "Influences how often B-frames are used",          OFFSET(b_bias),        AV_OPT_TYPE_INT,    { .i64 = INT_MIN}, INT_MIN, INT_MAX, VE },
936     { "b-pyramid",     "Keep some B-frames as references.",               OFFSET(b_pyramid),     AV_OPT_TYPE_INT,    { .i64 = -1 }, -1, INT_MAX, VE, "b_pyramid" },
937     { "none",          NULL,                                  0, AV_OPT_TYPE_CONST, {.i64 = X264_B_PYRAMID_NONE},   INT_MIN, INT_MAX, VE, "b_pyramid" },
938     { "strict",        "Strictly hierarchical pyramid",       0, AV_OPT_TYPE_CONST, {.i64 = X264_B_PYRAMID_STRICT}, INT_MIN, INT_MAX, VE, "b_pyramid" },
939     { "normal",        "Non-strict (not Blu-ray compatible)", 0, AV_OPT_TYPE_CONST, {.i64 = X264_B_PYRAMID_NORMAL}, INT_MIN, INT_MAX, VE, "b_pyramid" },
940     { "mixed-refs",    "One reference per partition, as opposed to one reference per macroblock", OFFSET(mixed_refs), AV_OPT_TYPE_BOOL, { .i64 = -1}, -1, 1, VE },
941     { "8x8dct",        "High profile 8x8 transform.",                     OFFSET(dct8x8),        AV_OPT_TYPE_BOOL,   { .i64 = -1 }, -1, 1, VE},
942     { "fast-pskip",    NULL,                                              OFFSET(fast_pskip),    AV_OPT_TYPE_BOOL,   { .i64 = -1 }, -1, 1, VE},
943     { "aud",           "Use access unit delimiters.",                     OFFSET(aud),           AV_OPT_TYPE_BOOL,   { .i64 = -1 }, -1, 1, VE},
944     { "mbtree",        "Use macroblock tree ratecontrol.",                OFFSET(mbtree),        AV_OPT_TYPE_BOOL,   { .i64 = -1 }, -1, 1, VE},
945     { "deblock",       "Loop filter parameters, in <alpha:beta> form.",   OFFSET(deblock),       AV_OPT_TYPE_STRING, { 0 },  0, 0, VE},
946     { "cplxblur",      "Reduce fluctuations in QP (before curve compression)", OFFSET(cplxblur), AV_OPT_TYPE_FLOAT,  {.dbl = -1 }, -1, FLT_MAX, VE},
947     { "partitions",    "A comma-separated list of partitions to consider. "
948                        "Possible values: p8x8, p4x4, b8x8, i8x8, i4x4, none, all", OFFSET(partitions), AV_OPT_TYPE_STRING, { 0 }, 0, 0, VE},
949     { "direct-pred",   "Direct MV prediction mode",                       OFFSET(direct_pred),   AV_OPT_TYPE_INT,    { .i64 = -1 }, -1, INT_MAX, VE, "direct-pred" },
950     { "none",          NULL,      0,    AV_OPT_TYPE_CONST, { .i64 = X264_DIRECT_PRED_NONE },     0, 0, VE, "direct-pred" },
951     { "spatial",       NULL,      0,    AV_OPT_TYPE_CONST, { .i64 = X264_DIRECT_PRED_SPATIAL },  0, 0, VE, "direct-pred" },
952     { "temporal",      NULL,      0,    AV_OPT_TYPE_CONST, { .i64 = X264_DIRECT_PRED_TEMPORAL }, 0, 0, VE, "direct-pred" },
953     { "auto",          NULL,      0,    AV_OPT_TYPE_CONST, { .i64 = X264_DIRECT_PRED_AUTO },     0, 0, VE, "direct-pred" },
954     { "slice-max-size","Limit the size of each slice in bytes",           OFFSET(slice_max_size),AV_OPT_TYPE_INT,    { .i64 = -1 }, -1, INT_MAX, VE },
955     { "stats",         "Filename for 2 pass stats",                       OFFSET(stats),         AV_OPT_TYPE_STRING, { 0 },  0,       0, VE },
956     { "nal-hrd",       "Signal HRD information (requires vbv-bufsize; "
957                        "cbr not allowed in .mp4)",                        OFFSET(nal_hrd),       AV_OPT_TYPE_INT,    { .i64 = -1 }, -1, INT_MAX, VE, "nal-hrd" },
958     { "none",          NULL, 0, AV_OPT_TYPE_CONST, {.i64 = X264_NAL_HRD_NONE}, INT_MIN, INT_MAX, VE, "nal-hrd" },
959     { "vbr",           NULL, 0, AV_OPT_TYPE_CONST, {.i64 = X264_NAL_HRD_VBR},  INT_MIN, INT_MAX, VE, "nal-hrd" },
960     { "cbr",           NULL, 0, AV_OPT_TYPE_CONST, {.i64 = X264_NAL_HRD_CBR},  INT_MIN, INT_MAX, VE, "nal-hrd" },
961     { "avcintra-class","AVC-Intra class 50/100/200",                      OFFSET(avcintra_class),AV_OPT_TYPE_INT,     { .i64 = -1 }, -1, 200   , VE},
962     { "motion-est",   "Set motion estimation method",                     OFFSET(motion_est),    AV_OPT_TYPE_INT,    { .i64 = -1 }, -1, X264_ME_TESA, VE, "motion-est"},
963     { "dia",           NULL, 0, AV_OPT_TYPE_CONST, { .i64 = X264_ME_DIA },  INT_MIN, INT_MAX, VE, "motion-est" },
964     { "hex",           NULL, 0, AV_OPT_TYPE_CONST, { .i64 = X264_ME_HEX },  INT_MIN, INT_MAX, VE, "motion-est" },
965     { "umh",           NULL, 0, AV_OPT_TYPE_CONST, { .i64 = X264_ME_UMH },  INT_MIN, INT_MAX, VE, "motion-est" },
966     { "esa",           NULL, 0, AV_OPT_TYPE_CONST, { .i64 = X264_ME_ESA },  INT_MIN, INT_MAX, VE, "motion-est" },
967     { "tesa",          NULL, 0, AV_OPT_TYPE_CONST, { .i64 = X264_ME_TESA }, INT_MIN, INT_MAX, VE, "motion-est" },
968     { "forced-idr",   "If forcing keyframes, force them as IDR frames.",                                  OFFSET(forced_idr),  AV_OPT_TYPE_BOOL,   { .i64 = -1 }, -1, 1, VE },
969     { "coder",    "Coder type",                                           OFFSET(coder), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, 1, VE, "coder" },
970     { "default",          NULL, 0, AV_OPT_TYPE_CONST, { .i64 = -1 }, INT_MIN, INT_MAX, VE, "coder" },
971     { "cavlc",            NULL, 0, AV_OPT_TYPE_CONST, { .i64 = 0 },  INT_MIN, INT_MAX, VE, "coder" },
972     { "cabac",            NULL, 0, AV_OPT_TYPE_CONST, { .i64 = 1 },  INT_MIN, INT_MAX, VE, "coder" },
973     { "vlc",              NULL, 0, AV_OPT_TYPE_CONST, { .i64 = 0 },  INT_MIN, INT_MAX, VE, "coder" },
974     { "ac",               NULL, 0, AV_OPT_TYPE_CONST, { .i64 = 1 },  INT_MIN, INT_MAX, VE, "coder" },
975     { "b_strategy",   "Strategy to choose between I/P/B-frames",          OFFSET(b_frame_strategy), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, 2, VE },
976     { "chromaoffset", "QP difference between chroma and luma",            OFFSET(chroma_offset), AV_OPT_TYPE_INT, { .i64 = -1 }, INT_MIN, INT_MAX, VE },
977     { "sc_threshold", "Scene change threshold",                           OFFSET(scenechange_threshold), AV_OPT_TYPE_INT, { .i64 = -1 }, INT_MIN, INT_MAX, VE },
978     { "noise_reduction", "Noise reduction",                               OFFSET(noise_reduction), AV_OPT_TYPE_INT, { .i64 = -1 }, INT_MIN, INT_MAX, VE },
979
980     { "x264-params",  "Override the x264 configuration using a :-separated list of key=value parameters", OFFSET(x264_params), AV_OPT_TYPE_STRING, { 0 }, 0, 0, VE },
981     { NULL },
982 };
983
984 static const AVCodecDefault x264_defaults[] = {
985     { "b",                "0" },
986     { "bf",               "-1" },
987     { "flags2",           "0" },
988     { "g",                "-1" },
989     { "i_qfactor",        "-1" },
990     { "b_qfactor",        "-1" },
991     { "qmin",             "-1" },
992     { "qmax",             "-1" },
993     { "qdiff",            "-1" },
994     { "qblur",            "-1" },
995     { "qcomp",            "-1" },
996 //     { "rc_lookahead",     "-1" },
997     { "refs",             "-1" },
998 #if FF_API_PRIVATE_OPT
999     { "sc_threshold",     "-1" },
1000 #endif
1001     { "trellis",          "-1" },
1002 #if FF_API_PRIVATE_OPT
1003     { "nr",               "-1" },
1004 #endif
1005     { "me_range",         "-1" },
1006 #if FF_API_MOTION_EST
1007     { "me_method",        "-1" },
1008 #endif
1009     { "subq",             "-1" },
1010 #if FF_API_PRIVATE_OPT
1011     { "b_strategy",       "-1" },
1012 #endif
1013     { "keyint_min",       "-1" },
1014 #if FF_API_CODER_TYPE
1015     { "coder",            "-1" },
1016 #endif
1017     { "cmp",              "-1" },
1018     { "threads",          AV_STRINGIFY(X264_THREADS_AUTO) },
1019     { "thread_type",      "0" },
1020     { "flags",            "+cgop" },
1021     { "rc_init_occupancy","-1" },
1022     { NULL },
1023 };
1024
1025 #if CONFIG_LIBX264_ENCODER
1026 static const AVClass x264_class = {
1027     .class_name = "libx264",
1028     .item_name  = av_default_item_name,
1029     .option     = options,
1030     .version    = LIBAVUTIL_VERSION_INT,
1031 };
1032
1033 AVCodec ff_libx264_encoder = {
1034     .name             = "libx264",
1035     .long_name        = NULL_IF_CONFIG_SMALL("libx264 H.264 / AVC / MPEG-4 AVC / MPEG-4 part 10"),
1036     .type             = AVMEDIA_TYPE_VIDEO,
1037     .id               = AV_CODEC_ID_H264,
1038     .priv_data_size   = sizeof(X264Context),
1039     .init             = X264_init,
1040     .encode2          = X264_frame,
1041     .close            = X264_close,
1042     .capabilities     = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_AUTO_THREADS,
1043     .priv_class       = &x264_class,
1044     .defaults         = x264_defaults,
1045     .init_static_data = X264_init_static,
1046     .caps_internal    = FF_CODEC_CAP_INIT_THREADSAFE |
1047                         FF_CODEC_CAP_INIT_CLEANUP,
1048 };
1049 #endif
1050
1051 #if CONFIG_LIBX264RGB_ENCODER
1052 static const AVClass rgbclass = {
1053     .class_name = "libx264rgb",
1054     .item_name  = av_default_item_name,
1055     .option     = options,
1056     .version    = LIBAVUTIL_VERSION_INT,
1057 };
1058
1059 AVCodec ff_libx264rgb_encoder = {
1060     .name           = "libx264rgb",
1061     .long_name      = NULL_IF_CONFIG_SMALL("libx264 H.264 / AVC / MPEG-4 AVC / MPEG-4 part 10 RGB"),
1062     .type           = AVMEDIA_TYPE_VIDEO,
1063     .id             = AV_CODEC_ID_H264,
1064     .priv_data_size = sizeof(X264Context),
1065     .init           = X264_init,
1066     .encode2        = X264_frame,
1067     .close          = X264_close,
1068     .capabilities   = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_AUTO_THREADS,
1069     .priv_class     = &rgbclass,
1070     .defaults       = x264_defaults,
1071     .pix_fmts       = pix_fmts_8bit_rgb,
1072 };
1073 #endif
1074
1075 #if CONFIG_LIBX262_ENCODER
1076 static const AVClass X262_class = {
1077     .class_name = "libx262",
1078     .item_name  = av_default_item_name,
1079     .option     = options,
1080     .version    = LIBAVUTIL_VERSION_INT,
1081 };
1082
1083 AVCodec ff_libx262_encoder = {
1084     .name             = "libx262",
1085     .long_name        = NULL_IF_CONFIG_SMALL("libx262 MPEG2VIDEO"),
1086     .type             = AVMEDIA_TYPE_VIDEO,
1087     .id               = AV_CODEC_ID_MPEG2VIDEO,
1088     .priv_data_size   = sizeof(X264Context),
1089     .init             = X264_init,
1090     .encode2          = X264_frame,
1091     .close            = X264_close,
1092     .capabilities     = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_AUTO_THREADS,
1093     .priv_class       = &X262_class,
1094     .defaults         = x264_defaults,
1095     .pix_fmts         = pix_fmts_8bit,
1096     .caps_internal    = FF_CODEC_CAP_INIT_THREADSAFE |
1097                         FF_CODEC_CAP_INIT_CLEANUP,
1098 };
1099 #endif