void do_init_encoder(struct x264lib_ctx *ctx, int width, int height, int initial_quality, int supports_csc_option)
{
ctx->quality = initial_quality;
ctx->supports_csc_option = supports_csc_option;
ctx->colour_sampling = get_x264_colour_sampling(ctx, initial_quality);
ctx->x264_quality = get_x264_quality(initial_quality);
ctx->csc_format = get_csc_format_for_x264_format(ctx->colour_sampling);
ctx->encoding_preset = 2;
ctx->preset = x264_preset_names[ctx->encoding_preset];
ctx->profile = get_profile_for_quality(initial_quality);
ctx->csc_algo = get_csc_algo_for_quality(initial_quality);
//printf("do_init_encoder(%p, %i, %i, %i, %i) colour_sampling=%i, x264_quality=%f, profile=%s\n", ctx, width, height, initial_quality, supports_csc_option, ctx->colour_sampling, ctx->x264_quality, ctx->profile);
x264_param_t param;
x264_param_default_preset(¶m, ctx->preset, "zerolatency");
param.i_threads = 1;
param.i_width = width;
param.i_height = height;
param.i_csp = ctx->colour_sampling;
param.rc.f_rf_constant = ctx->x264_quality;
param.i_log_level = 0;
x264_param_apply_profile(¶m, ctx->profile);
ctx->encoder = x264_encoder_open(¶m);
ctx->width = width;
ctx->height = height;
ctx->rgb2yuv = init_encoder_csc(ctx);
}
int H264Encoder::Prepare(const MediaDescription& desc) {
if ( desc.isVideo == false)
return -1;
int wid = desc.width;
int hei = desc.height;
// 0. building a default encoder parameters.
x264_param_default_preset(&x264_opt_, "ultrafast", "zerolatency");
x264_opt_.rc.i_rc_method = X264_RC_CRF;
x264_opt_.rc.i_bitrate = 512;
x264_opt_.i_nal_hrd = X264_NAL_HRD_CBR;
//x264_param_default(&opt);
// 1. Setting the fields of parameter struct
x264_opt_.i_width = wid;
x264_opt_.i_height = hei;
//opt.i_slice_count = 5;
//opt.b_intra_refresh = 1;
// 3. Prepare the output buffer and target file
x264_picture_alloc(&x264_picin_[0], X264_CSP_NV12, x264_opt_.i_width, x264_opt_.i_height);
x264_picture_alloc(&x264_picin_[1], X264_CSP_NV12, x264_opt_.i_width, x264_opt_.i_height);
x264_picture_alloc(&x264_picout_, X264_CSP_NV12, x264_opt_.i_width, x264_opt_.i_height);
ppIndex = -1;
// 4. Building the encoder handler
x264_hdl_ = x264_encoder_open(&x264_opt_);
x264_encoder_parameters(x264_hdl_, &x264_opt_);
return 0;
}
void *vc_open (int width, int height, double fps)
{
Ctx *ctx = new Ctx;
ctx->force_keyframe = 0;
// 设置编码属性
//x264_param_default(&ctx->param);
x264_param_default_preset(&ctx->param, "fast", "zerolatency");
//x264_param_apply_profile(&ctx->param, "baseline");
ctx->param.i_width = width;
ctx->param.i_height = height;
ctx->param.b_repeat_headers = 1; // 重复SPS/PPS 放到关键帧前面
ctx->param.b_cabac = 1;
ctx->param.i_threads = 1;
// ctx->param.b_intra_refresh = 1;
ctx->param.i_fps_num = (int)fps;
ctx->param.i_fps_den = 1;
//ctx->param.b_vfr_input = 1;
ctx->param.i_keyint_max = ctx->param.i_fps_num * 2;
//ctx->param.i_keyint_min = 1;
// rc
// ctx->param.rc.i_rc_method = X264_RC_CRF;
ctx->param.rc.i_bitrate = 50;
//ctx->param.rc.f_rate_tolerance = 0.1;
//ctx->param.rc.i_vbv_max_bitrate = ctx->param.rc.i_bitrate * 1.3;
//ctx->param.rc.f_rf_constant = 600;
//ctx->param.rc.f_rf_constant_max = ctx->param.rc.f_rf_constant * 1.3;
#ifdef DEBUG
ctx->param.i_log_level = X264_LOG_WARNING;
#else
ctx->param.i_log_level = X264_LOG_NONE;
#endif // release
ctx->x264 = x264_encoder_open(&ctx->param);
if (!ctx->x264) {
fprintf(stderr, "%s: x264_encoder_open err\n", __func__);
delete ctx;
return 0;
}
x264_picture_init(&ctx->picture);
ctx->picture.img.i_csp = X264_CSP_I420;
ctx->picture.img.i_plane = 3;
ctx->output = malloc(128*1024);
ctx->output_bufsize = 128*1024;
ctx->output_datasize = 0;
ctx->get_pts = first_pts;
ctx->info_valid = 0;
return ctx;
}
EncodeContext encode_context_create(int width, int height) {
EncodeContext context;
context.width = width;
context.height = height;
x264_param_t param;
x264_param_default_preset(¶m, "veryfast", "zerolatency");
param.i_threads = 1;
param.i_width = width;
param.i_height = height;
param.i_fps_num = 20; // the FPS doesn't matter
param.i_fps_den = 1;
// Intra refres:
param.i_keyint_max = 20; // once every twenty frames
param.b_intra_refresh = 1;
//Rate control:
param.rc.i_rc_method = X264_RC_CRF;
param.rc.f_rf_constant = 25;
param.rc.f_rf_constant_max = 35;
//For streaming:
param.b_repeat_headers = 1;
param.b_annexb = 1;
x264_param_apply_profile(¶m, "baseline");
context.encoder = x264_encoder_open(¶m);
context.converter = sws_getContext(width, height, PIX_FMT_RGB24,
width, height, PIX_FMT_YUV420P,
SWS_FAST_BILINEAR, NULL, NULL, NULL);
return context;
}
void H264Encoder::init_(const int wid, const int hei) {
// 0. building encoder parameters.
x264_param_default_preset(&x264_opt_, "ultrafast", "zerolatency");
x264_opt_.i_width = wid;
x264_opt_.i_height = hei;
x264_opt_.i_threads = 1;
x264_opt_.b_repeat_headers = 1;
x264_opt_.b_intra_refresh = 1;
x264_opt_.rc.i_rc_method = X264_RC_CQP;
x264_opt_.rc.i_qp_constant = 24;
x264_opt_.rc.i_qp_min = 24;
x264_opt_.rc.i_qp_max = 24;
//x264_param_default(&opt);
x264_param_apply_profile(&x264_opt_, "baseline");
// 1. Prepare the output buffer and target file
x264_picture_alloc(&x264_picin_, X264_CSP_NV12, x264_opt_.i_width, x264_opt_.i_height);
x264_picture_alloc(&x264_picout_, X264_CSP_NV12, x264_opt_.i_width, x264_opt_.i_height);
// 2. Building the encoder handler
x264_hdl_ = x264_encoder_open(&x264_opt_);
x264_encoder_parameters(x264_hdl_, &x264_opt_);
}
H264Encoder::H264Encoder(int width, int height, int fps)
: m_width(width), m_height(height), m_fps(fps)
{
m_stride = width*4;
/* Parametrize x264 for real-time */
x264_param_default_preset(&m_param, "veryfast", "zerolatency");
m_param.i_threads = 1;
m_param.i_width = width;
m_param.i_height = height;
m_param.i_fps_num = fps;
m_param.i_fps_den = 1;
// Intra refres:
m_param.i_keyint_max = fps;
m_param.b_intra_refresh = 1;
//Rate control:
m_param.rc.i_rc_method = X264_RC_CRF;
m_param.rc.f_rf_constant = 25;
m_param.rc.f_rf_constant_max = 35;
//For streaming:
m_param.b_repeat_headers = 1;
m_param.b_annexb = 1;
x264_param_apply_profile(&m_param, "baseline");
Init(&m_param);
}
HRESULT x264::init(int width, int height, int bitrate)
{
this->width = width;
this->height = height;
x264_param_t param;
x264_param_default_preset(¶m, "medium", "zerolatency");
// x264_param_apply_profile(¶m, "baseline");
param.i_frame_reference = 1;
param.i_width = width;
param.i_height = height;
param.i_fps_num = 24000;
param.i_fps_den = 1001;
param.i_csp = X264_CSP_I420;
param.i_keyint_max = 25;
//param.b_intra_refresh = 1;
param.b_cabac = 1;
param.b_annexb = 1;
param.rc.i_rc_method = X264_RC_ABR;
param.rc.i_bitrate = bitrate;
encoder = x264_encoder_open(¶m);
// init picture
x264_picture_alloc(&pic_in, X264_CSP_I420, width, height);
last_encode_time = timeGetTime();
return S_OK;
}
static bool reset_x264_params(struct obs_x264 *obsx264,
const char *preset, const char *tune)
{
int ret = x264_param_default_preset(&obsx264->params,
validate_preset(obsx264, preset),
validate(obsx264, tune, "tune", x264_tune_names));
return ret == 0;
}
VideoEncoder* video_encoder_init(int width, int height, int fpsNum, int fpsDen, int maxWidth){
printf("[arcade encoder init] %d x %d @ (%d / %d)\n",width,height,fpsNum,fpsDen);
VideoEncoder *enc = (VideoEncoder*) malloc(sizeof(VideoEncoder));
memset(enc,0,sizeof(VideoEncoder));
//TODO: this needs to become a parameter
enc->in_fmt = AV_PIX_FMT_BGRA;
enc->width = width;
enc->height = height;
enc->max_width = maxWidth;
printf("[arcade encoder] capping width at %dpx\n",enc->max_width);
float aspectRatio = (float) enc->width / (float) enc->height;
enc->out_width = enc->width > enc->max_width ? enc->max_width : enc->width;
enc->out_height = (int) ((float) enc->out_width / aspectRatio);
printf("[arcade encoder init]\n\t%d x %d --> %d x %d\n\t@ (%d / %d)\n",
enc->width,enc->height,
enc->out_width,enc->out_height,
fpsNum,fpsDen);
x264_param_t param;
x264_param_default_preset(¶m, "ultrafast", "zerolatency");
param.i_threads = 1;
param.i_width = enc->out_width;
param.i_height = enc->out_height;
param.i_fps_num = fpsNum;
param.i_fps_den = fpsDen;
// Intra refres:
param.i_keyint_max = 30000;
param.b_intra_refresh = 1;
//Rate control:
param.rc.i_rc_method = X264_RC_CRF;
param.rc.f_rf_constant = 25;
param.rc.f_rf_constant_max = 35;
//For streaming:
param.b_repeat_headers = 1;
param.b_annexb = 1;
x264_param_apply_profile(¶m, "baseline");
enc->encoder = x264_encoder_open(¶m);
x264_picture_alloc(&enc->pic_in, X264_CSP_I420, enc->out_width, enc->out_height);
enc->output_buffer_size = (int)sizeof(uint8_t)*4096*1024;
enc->output_buffer = malloc(enc->output_buffer_size);
enc->sws = sws_getContext(enc->width, enc->height, enc->in_fmt,
enc->out_width, enc->out_height, AV_PIX_FMT_YUV420P,
SWS_FAST_BILINEAR, NULL, NULL, NULL);
enc->num_frames = 0;
return enc;
}
void rtspStream::initH264Encoder(int width,int height,int fps,int bitRate)
{
frame_num = 0;
pX264Handle = NULL;
pX264Param = new x264_param_t;
assert(pX264Param);
m_nFPS = 25;
//* 配置参数
//* 使用默认参数，在这里因为我的是实时网络传输，所以我使用了zerolatency的选项，使用这个选项之后就不会有delayed_frames，如果你使用的不是这样的话，还需要在编码完成之后得到缓存的编码帧
x264_param_default_preset(pX264Param, "veryfast", "zerolatency");
//* cpuFlags
pX264Param->i_threads = X264_SYNC_LOOKAHEAD_AUTO;//* 取空缓冲区继续使用不死锁的保证.
//* 视频选项
pX264Param->i_width = width; //* 要编码的图像宽度.
pX264Param->i_height = height; //* 要编码的图像高度
pX264Param->i_frame_total = 0; //* 编码总帧数.不知道用0.
pX264Param->i_keyint_max = 10;
//* 流参数
pX264Param->i_bframe = 5;
pX264Param->b_open_gop = 0;
pX264Param->i_bframe_pyramid = 0;
pX264Param->i_bframe_adaptive = X264_B_ADAPT_TRELLIS;
//* Log参数，不需要打印编码信息时直接注释掉就行
// pX264Param->i_log_level = X264_LOG_DEBUG;
//* 速率控制参数
pX264Param->rc.i_bitrate = bitRate;//* 码率(比特率,单位Kbps)
//* muxing parameters
pX264Param->i_fps_den = 1; //* 帧率分母
pX264Param->i_fps_num = fps;//* 帧率分子
pX264Param->i_timebase_den = pX264Param->i_fps_num;
pX264Param->i_timebase_num = pX264Param->i_fps_den;
//* 设置Profile.使用Baseline profile
x264_param_apply_profile(pX264Param, x264_profile_names[0]);
pNals = NULL;
pPicIn = new x264_picture_t;
pPicOut = new x264_picture_t;
x264_picture_init(pPicOut);
x264_picture_alloc(pPicIn, X264_CSP_I420, pX264Param->i_width, pX264Param->i_height);
pPicIn->img.i_csp = X264_CSP_I420;
pPicIn->img.i_plane = 3;
//* 打开编码器句柄,通过x264_encoder_parameters得到设置给X264
//* 的参数.通过x264_encoder_reconfig更新X264的参数
pX264Handle = x264_encoder_open(pX264Param);
assert(pX264Handle);
pPicIn->img.plane[0] = PYUVBuf;
pPicIn->img.plane[1] = PYUVBuf + width *height;
pPicIn->img.plane[2] = PYUVBuf + width * height * 5 / 4;
pPicIn->img.plane[3] = 0;
}
void change_encoding_speed(struct x264lib_ctx *ctx, int increase)
{
x264_param_t param;
x264_encoder_parameters(ctx->encoder, ¶m);
ctx->encoding_preset -= increase;
if (ctx->encoding_preset < 0)
ctx->encoding_preset = 0;
if (ctx->encoding_preset > 5)
ctx->encoding_preset = 5;
x264_param_default_preset(¶m, x264_preset_names[ctx->encoding_preset], "zerolatency");
//printf("Setting encoding preset %s %d\n", x264_preset_names[ctx->encoding_preset], ctx->encoding_preset);
x264_param_apply_profile(¶m, "baseline");
x264_encoder_reconfig(ctx->encoder, ¶m);
}
void set_encoding_speed(struct x264lib_ctx *ctx, int pct)
{
x264_param_t param;
x264_encoder_parameters(ctx->encoder, ¶m);
int new_preset = 7-MAX(0, MIN(7, pct/12.5));
if (new_preset==ctx->encoding_preset)
return;
ctx->encoding_preset = new_preset;
//"tune" options: film, animation, grain, stillimage, psnr, ssim, fastdecode, zerolatency
//Multiple tunings can be used if separated by a delimiter in ",./-+"
//however multiple psy tunings cannot be used.
//film, animation, grain, stillimage, psnr, and ssim are psy tunings.
x264_param_default_preset(¶m, x264_preset_names[ctx->encoding_preset], "zerolatency");
x264_param_apply_profile(¶m, "baseline");
x264_encoder_reconfig(ctx->encoder, ¶m);
}
int CX264Encoder::set_param(x264_param_t* p)
{
int ret = -1;
ret = x264_param_default_preset(¶m, "ultrafast", NULL);
param.i_csp = X264_CSP_I420;// X264_CSP_I420;// X264_CSP_RGB;
param.i_width = width;
param.i_height = height;
param.b_vfr_input = 0;
param.b_repeat_headers = 1;
param.b_annexb = 1;
param.i_nal_hrd = X264_NAL_HRD_VBR;
ret = x264_param_apply_profile(¶m, "high");
return ret;
}
static void enc_preprocess(MSFilter *f){
EncData *d=(EncData*)f->data;
x264_param_t *params=&d->params;
d->packer=rfc3984_new();
rfc3984_set_mode(d->packer,d->mode);
rfc3984_enable_stap_a(d->packer,FALSE);
#if defined(__arm__) || defined(ANDROID)
if (x264_param_default_preset(params,"superfast"/*"ultrafast"*/,"zerolatency")) {
ms_error("Cannot apply default x264 configuration");
}
#else
x264_param_default(params);
#endif
params->i_threads=ms_get_cpu_count();
params->i_sync_lookahead=0;
params->i_width=d->vconf.vsize.width;
params->i_height=d->vconf.vsize.height;
params->i_fps_num=(int)d->vconf.fps;
params->i_fps_den=1;
params->i_slice_max_size=ms_get_payload_max_size()-100; /*-100 security margin*/
params->i_level_idc=13;
apply_bitrate(f);
params->rc.i_lookahead=0;
/*enable this by config ?*/
/*
params.i_keyint_max = (int)d->fps*d->keyframe_int;
params.i_keyint_min = (int)d->fps;
*/
params->b_repeat_headers=1;
params->b_annexb=0;
//these parameters must be set so that our stream is baseline
params->analyse.b_transform_8x8 = 0;
params->b_cabac = 0;
params->i_cqm_preset = X264_CQM_FLAT;
params->i_bframe = 0;
params->analyse.i_weighted_pred = X264_WEIGHTP_NONE;
d->enc=x264_encoder_open(params);
if (d->enc==NULL) ms_error("Fail to create x264 encoder.");
d->framenum=0;
video_starter_init(&d->starter);
}
void x264Encoder::Initilize()
{
x264_param_default_preset(¶meters, "veryfast", "zerolatency");
parameters.i_log_level = X264_LOG_INFO;
parameters.i_threads = 1;
parameters.i_width = 512 / 2;
parameters.i_height = 424 / 2;
parameters.i_fps_num = _fps;
parameters.i_fps_den = 1;
parameters.i_keyint_max = 15;
parameters.b_intra_refresh = 1;
parameters.rc.i_rc_method = X264_RC_CRF;
parameters.rc.i_vbv_buffer_size = 1000000;
parameters.rc.i_vbv_max_bitrate = 90000;
parameters.rc.f_rf_constant = 25;
parameters.rc.f_rf_constant_max = 35;
parameters.i_sps_id = 7;
// the following two value you should keep 1
parameters.b_repeat_headers = 1; // to get header before every I-Frame
parameters.b_annexb = 1; // put start code in front of nal. we will remove start code later
x264_param_apply_profile(¶meters, "baseline");
encoder = x264_encoder_open(¶meters);
x264_picture_alloc(&picture_in, X264_CSP_I420, 512 / 2, 424 / 2);
picture_in.i_type = X264_TYPE_AUTO;
picture_in.img.i_csp = X264_CSP_I420;
x264_picture_alloc(&picture_out, X264_CSP_I420, parameters.i_width, parameters.i_height);
picture_out.i_type = X264_TYPE_AUTO;
picture_out.img.i_csp = X264_CSP_I420;
// i have initilized my color space converter for BGR24 to YUV420 because my opencv
// video capture gives BGR24 image. You can initilize according to your input pixelFormat
convertContext = sws_getContext(
512 / 2,
424 / 2,
PIX_FMT_BGR24,
parameters.i_width,
parameters.i_height,
PIX_FMT_YUV420P,
SWS_FAST_BILINEAR,
NULL, NULL, NULL);
}
void encode_init()
{
// fill x264_param_t with default values and do CPU detection
x264_param_default(x264_encode.para);
if(x264_param_default_preset(x264_encode.para, capg.preset,
capg.tune) < 0)
CAP_DBG_EXIT("x264_param_default_preset error!\n");
// ricann debug
CAP_DBG("-------------------after x264_param_default_preset\n");
encode_printpara();
// Configure non-default params
// real frame rate is i_fps_num/i_fps_den
x264_encode.para->i_fps_num = capg.frame_rate;
x264_encode.para->i_width = capg.width;
x264_encode.para->i_height = capg.height;
x264_encode.para->i_keyint_max = capg.gop_size;
x264_encode.para->i_csp = x264_encode.colorspace;
x264_encode.para->b_vfr_input = 0;
x264_encode.para->b_repeat_headers = 1;
// ricann debug
CAP_DBG("-------------------after assign value\n");
encode_printpara();
x264_encode.pic->img.i_csp = x264_encode.colorspace;
x264_encode.pic->img.i_plane = 3;
x264_encode.pic->i_type = X264_TYPE_AUTO;
if(x264_param_apply_profile(x264_encode.para, capg.profile) < 0)
CAP_DBG_EXIT("x264_param_apply_profile error!\n");
x264_encode.handle = x264_encoder_open(x264_encode.para);
if(!x264_encode.handle)
CAP_DBG_EXIT("x264_encoder_open error!\n");
// ricann debug
CAP_DBG("-------------------after x264_encoder_open\n");
encode_printpara();
}
struct x264lib_ctx *init_encoder(int width, int height)
{
struct x264lib_ctx *ctx = malloc(sizeof(struct x264lib_ctx));
ctx->encoding_preset = 2;
x264_param_t param;
x264_param_default_preset(¶m, x264_preset_names[ctx->encoding_preset], "zerolatency");
param.i_threads = 1;
param.i_width = width;
param.i_height = height;
param.i_csp = X264_CSP_I420;
param.i_log_level = 0;
x264_param_apply_profile(¶m, "baseline");
ctx->encoder = x264_encoder_open(¶m);
ctx->width = width;
ctx->height = height;
ctx->rgb2yuv = sws_getContext(ctx->width, ctx->height, PIX_FMT_RGB24, ctx->width, ctx->height, PIX_FMT_YUV420P, SWS_SINC | SWS_ACCURATE_RND, NULL, NULL, NULL);
return ctx;
}
void init_encoder(void)
{
x264_param_default_preset(¶m, "veryfast", "zerolatency");
param.i_threads=1;
param.i_width=x264_width;
param.i_height=x264_height;
param.i_fps_num=fps;
param.i_fps_den=1;
param.i_keyint_max=25;
param.b_intra_refresh=1;
//If param->b_annexb is set, Annex-B bytestream with startcode.
//* This size is the size used in mp4/similar muxing; it is equal to ( i_payload-4 ) --> uint8_t *p_payload;
param.b_annexb=1;
x264_param_apply_profile(¶m,"baseline");
encoder=x264_encoder_open(¶m);
printf("in init_encoder function\n");
}
_declspec(dllexport) int __cdecl AllocEncoder(struct TranscoderContext* ctx) {
struct TranscoderOptions* options = ctx->options;
if (x264_picture_alloc(&pic_in, X264_CSP_I420, options->OutputWidth, options->OutputHeight) < 0) {
return 1;
}
fprintf(stdout, "Width: %d\n", options->InputWidth);
fprintf(stdout, "Height: %d\n", options->InputHeight);
NALBYTES = malloc(options->OutputWidth * options->OutputHeight * 4);
x264_param_t param;
x264_param_default_preset(¶m, "ultrafast", "zerolatency");
param.i_threads = 0;
param.i_width = options->OutputWidth;
param.i_height = options->OutputHeight;
param.rc.i_rc_method = X264_RC_CRF;
//param.rc.i_bitrate = 3000;
//param.rc.i_vbv_max_bitrate = 4000;
param.rc.f_rf_constant = 20;
fprintf(stdout, "rf_constant: %f\n", param.rc.f_rf_constant);
fprintf(stdout, "rf_constant_max: %f\n", param.rc.f_rf_constant_max);
//fprintf(stdout, "rf_constant_max: %f\n", param.rc.rf);
//param.i_keyint_max = 25;
param.i_keyint_max = 500;
//param.b_intra_refresh = 1;
param.b_repeat_headers = 1;
param.b_annexb = 1;
//param.i_log_level = -1;
x264_t* encoder = x264_encoder_open(¶m);
ctx->encoder = encoder;
convertCtx = sws_getContext(options->InputWidth, options->InputHeight, AV_PIX_FMT_RGBA, options->OutputWidth, options->OutputHeight, AV_PIX_FMT_YUV420P, SWS_FAST_BILINEAR, NULL, NULL, NULL);
return 0;
}
H264Exporter::H264Exporter(const exporter_settings& settings)
: _success{false}
, _settings(settings)
, _file{settings.path, std::ios::binary}
, _frame{0}
, _encoder{nullptr}
{
x264_param_t param;
// Best quality (0) -> "veryslow" (8); worst quality (4) -> "ultrafast" (0).
auto quality = std::to_string(2 * (4 - settings.quality));
if (x264_param_default_preset(¶m, quality.c_str(), "film") < 0) {
std::cerr << "couldn't get default preset" << std::endl;
return;
}
param.i_threads = settings.threads > 1 ? settings.threads - 1 : 1;
param.i_lookahead_threads = settings.threads > 1 ? 1 : 0;
param.i_width = settings.width;
param.i_height = settings.height;
param.i_fps_num = settings.fps;
param.i_fps_den = 1;
param.i_frame_total = settings.fps * settings.length;
param.i_keyint_min = 0;
param.i_keyint_max = settings.fps;
if (x264_param_apply_profile(¶m, "high") < 0) {
std::cerr << "couldn't get apply profile" << std::endl;
return;
}
_encoder = x264_encoder_open(¶m);
if (!_encoder) {
std::cerr << "couldn't create encoder" << std::endl;
return;
}
if (x264_picture_alloc(&_pic, X264_CSP_I420, _settings.width, _settings.height) < 0) {
std::cerr << "couldn't allocate picture" << std::endl;
return;
}
_success = true;
}
static void encode()
{
AVStream *st = st_h264;
AVFrame *frm = avcodec_alloc_frame();
int i, r, n;
decode_frame(frm, st, 1);
printf("gotit, startencode\n");
x264_param_t param;
av_seek_frame(ifc, st->index, 1000*200, 0);
x264_param_default_preset(¶m, "ultrafast", "");
param.i_log_level = X264_LOG_DEBUG;
param.i_width = frm->width;
param.i_height = frm->height;
param.i_csp = X264_CSP_I420;
x264_t *h = x264_encoder_open(¶m);
printf("264 %p\n", h);
FILE *fp = fopen("1.264", "wb+");
for (n = 0; n < 400; n++) {
decode_frame(frm, st, 0);
printf("pts=%lld", frm->pkt_pts);
if (n == 100) {
printf("info: %dx%d\n", frm->width, frm->height);
printf("info: %d,%d,%d\n",
frm->linesize[0],
frm->linesize[1],
frm->linesize[2]
);
FILE *fp;
fp = fopen("/tmp/Image.Y", "wb+");
fwrite(frm->data[0], frm->linesize[0]*frm->height, 1, fp);
fclose(fp);
fp = fopen("/tmp/Image.U", "wb+");
fwrite(frm->data[1], frm->linesize[1]*frm->height, 1, fp);
fclose(fp);
fp = fopen("/tmp/Image.V", "wb+");
fwrite(frm->data[2], frm->linesize[2]*frm->height, 1, fp);
fclose(fp);
exit(0);
}
x264_picture_t picin, picout;
memset(&picin, 0, sizeof(picin));
picin.img.i_stride[0] = frm->linesize[0];
picin.img.i_stride[1] = frm->linesize[1];
picin.img.i_stride[2] = frm->linesize[2];
printf("imginfo %d %d %d %dx%d\n",
frm->linesize[0],
frm->linesize[1],
frm->linesize[2],
param.i_width, param.i_height
);
picin.img.plane[0] = frm->data[0];
picin.img.plane[1] = frm->data[1];
picin.img.plane[2] = frm->data[2];
picin.img.i_csp = X264_CSP_I420;
picin.i_pts = frm->pkt_pts;
x264_nal_t *nal;
int i_nal;
r = x264_encoder_encode(h, &nal, &i_nal, &picin, &picout);
printf("##%d encode=%d\n", n, r);
if (r) {
printf("write frame\n");
AVPacket pkt;
pkt.stream_index = 0;
pkt.pts = frm->pkt_pts / 20;
pkt.dts = pkt.pts;
pkt.data = nal[0].p_payload;
pkt.size = r;
av_write_frame(ofc, &pkt);
avio_flush(ofc->pb);
fwrite(nal[0].p_payload, r, 1, fp);
}
}
fclose(fp);
}
void x264enc_set_param(const m_option_t* opt, char* arg)
{
static int initialized = 0;
int slow_firstpass = 0;
char *preset = NULL, *tune = NULL, *profile = NULL;
char *p, *copy, *name;
if (!initialized) {
x264_param_default(¶m);
initialized = 1;
}
if (!arg) {
mp_msg(MSGT_CFGPARSER, MSGL_ERR, "Option x264encopts: no options provided\n");
parse_error = 1;
return;
} else if (!*arg)
/* Empty arguments, just doing initialization of default parameters. */
return;
/* Step 1: look for initial preset/tune. */
copy = p = strdup(arg);
while ((name = strsep(©, ":"))) {
char *value = strpbrk(name, "=:");
if (!value)
continue;
*value++ = 0;
if (!strcasecmp(name, "preset"))
preset = value;
else if (!strcasecmp(name, "tune"))
tune = value;
}
if (x264_param_default_preset(¶m, preset, tune) < 0) {
mp_msg(MSGT_CFGPARSER, MSGL_ERR, "Option x264encopts: Invalid preset or tune.\n");
parse_error = 1;
}
free(p);
/* Step 2: explicit user overrides */
while ((name = strsep(&arg, ":")) && *name) {
int ret = 0;
char *value = strpbrk(name, "=:");
if (value)
*value++ = 0;
if (!strcasecmp(name, "profile"))
profile = value;
else if (!strcasecmp(name, "turbo")) {
mp_msg(MSGT_CFGPARSER, MSGL_WARN, "Option x264encopts: turbo option is deprecated; "
"use slow_firstpass to disable turbo\n");
if (value && *value == '0')
slow_firstpass = 1;
} else if (!strcasecmp(name, "slow_firstpass"))
slow_firstpass = 1;
else if (strcasecmp(name, "preset") && strcasecmp(name, "tune")) {
ret = x264_param_parse(¶m, name, value);
if (ret == X264_PARAM_BAD_NAME)
mp_msg(MSGT_CFGPARSER, MSGL_ERR, "Option x264encopts: Unknown suboption %s\n", name);
if (ret == X264_PARAM_BAD_VALUE)
mp_msg(MSGT_CFGPARSER, MSGL_ERR, "Option x264encopts: Bad argument %s=%s\n",
name, value ? value : "(null)");
}
/* mark this option as done, so it's not reparsed if there's another -x264encopts */
*name = 0;
parse_error |= ret;
}
/* Step 3: Apply fast first pass (turbo) options. */
if (!slow_firstpass)
x264_param_apply_fastfirstpass(¶m);
/* Step 4: enforce profile */
if (profile && x264_param_apply_profile(¶m, profile) < 0)
parse_error = 1;
}
int main(int argc, char** argv)
{
int listenfd, connfd;
int byterecv;
int bytesum = 0;
int ret = 0;
struct sockaddr_in servaddr;
char *buff;
int width, height;
x264_param_t param;
x264_picture_t pic;
x264_picture_t pic_out;
x264_t *h;
int i_frame = 0;
int i_frame_size;
x264_nal_t *nal;
int i_nal;
FILE *fout = NULL;
/* Get default params for preset/tuning */
if(x264_param_default_preset( ¶m, "medium", NULL ) < 0)
goto fail;
/* Configure non-default params */
param.i_csp = X264_CSP_I420;
param.i_width = WIDTH;
param.i_height = HEIGHT;
param.b_vfr_input = 0; //frame rate
param.b_repeat_headers = 1;
param.b_annexb = 1;
/* Apply profile restrictions. */
if(x264_param_apply_profile(¶m, "baseline") < 0)
goto fail;
if(x264_picture_alloc(&pic, param.i_csp, param.i_width, param.i_height) < 0)
goto fail;
#undef fail
#define fail fail2
h = x264_encoder_open(¶m);
if(!h)
goto fail;
printf("x264 encoder init successfully\n");
int luma_size = width * height;
int chroma_size = luma_size / 4;
fout = fopen("test.264", "wb+");
#undef fail
#define fail fail3
/* socket init*/
if((listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1){
printf("create socket error: %s(errno: %d)\n",strerror(errno),errno);
exit(0);
}
memset(&servaddr, 0, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = htons(6666);
if(bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1){
printf("bind socket error: %s(errno: %d)\n",strerror(errno),errno);
exit(0);
}
if(listen(listenfd, 10) == -1){
printf("listen socket error: %s(errno: %d)\n",strerror(errno),errno);
exit(0);
}
printf("======waiting for client's request======\n");
if((connfd = accept(listenfd, (struct sockaddr*)NULL, NULL)) == -1){
printf("accept socket error: %s(errno: %d)",strerror(errno),errno);
exit(0);
}
/* malloc 5k buffer for recv raw data from client*/
buff = (char *)malloc(5 * 1024 * 1024 * sizeof(char));
while(1) {
byterecv = recv(connfd, buff + bytesum, 32 * 1024, 0);
if (byterecv > 0) {
bytesum += byterecv;
/* got one frame here, encode it*/
if (bytesum >= RESOLUTION * 1.5) {
pic.img.plane[0] = buff;
pic.img.plane[1] = buff + luma_size;
pic.img.plane[2] = buff + luma_size + chroma_size;
pic.i_pts = i_frame;
i_frame ++;
i_frame_size = x264_encoder_encode(h, &nal, &i_nal, &pic, &pic_out);
if(i_frame_size < 0)
goto fail;
else if(i_frame_size) {
if(!fwrite( nal->p_payload, i_frame_size, 1, fout))
goto fail;
printf("encode frame %d\n", i_frame);
}
bytesum = bytesum - RESOLUTION * 1.5;
memcpy(buff, buff + (int)(RESOLUTION * 1.5), bytesum);
}
}
else if((byterecv < 0) && (errno == EAGAIN || errno == EINTR)) {
printf("coutinue\n");
continue;
}
else {
//printf("stop recv stream frome client, bytesum: %d, total: %d\n", bytesum, total);
break;
}
}
/* Flush delayed frames */
while(x264_encoder_delayed_frames(h)) {
i_frame_size = x264_encoder_encode(h, &nal, &i_nal, NULL, &pic_out);
if(i_frame_size < 0)
goto fail;
else if(i_frame_size) {
if(!fwrite( nal->p_payload, i_frame_size, 1, fout))
goto fail;
}
}
if (fout)
fclose(fout);
x264_encoder_close(h);
x264_picture_clean(&pic);
close(connfd);
close(listenfd);
return 0;
#undef fail
fail3:
x264_encoder_close(h);
fail2:
x264_picture_clean(&pic);
fail:
return -1;
}
int obe_populate_avc_encoder_params( obe_t *h, int input_stream_id, x264_param_t *param )
{
obe_int_input_stream_t *stream = get_input_stream( h, input_stream_id );
if( !stream )
{
fprintf( stderr, "Could not find stream \n" );
return -1;
}
if( stream->stream_type != STREAM_TYPE_VIDEO )
{
fprintf( stderr, "Stream type is not video \n" );
return -1;
}
if( !param )
{
fprintf( stderr, "Invalid parameter pointer \n" );
return -1;
}
if( h->obe_system == OBE_SYSTEM_TYPE_LOWEST_LATENCY || h->obe_system == OBE_SYSTEM_TYPE_LOW_LATENCY )
x264_param_default_preset( param, "veryfast", "zerolatency" );
else
x264_param_default( param );
param->b_deterministic = 0;
param->b_vfr_input = 0;
param->b_pic_struct = 1;
param->b_open_gop = 1;
param->rc.i_rc_method = X264_RC_ABR;
param->i_width = stream->width;
param->i_height = stream->height;
param->i_fps_num = stream->timebase_den;
param->i_fps_den = stream->timebase_num;
param->b_interlaced = stream->interlaced;
if( param->b_interlaced )
param->b_tff = stream->tff;
/* A reasonable default. x264 won't go higher than this parameter irrespective of speedcontrol */
if( h->obe_system == OBE_SYSTEM_TYPE_GENERIC )
param->i_frame_reference = 4;
if( stream->sar_num && stream->sar_den )
{
param->vui.i_sar_width = stream->sar_num;
param->vui.i_sar_height = stream->sar_den;
}
param->vui.i_overscan = 2;
if( ( param->i_fps_num == 25 || param->i_fps_num == 50 ) && param->i_fps_den == 1 )
{
param->vui.i_vidformat = 1; // PAL
param->vui.i_colorprim = 5; // BT.470-2 bg
param->vui.i_transfer = 5; // BT.470-2 bg
param->vui.i_colmatrix = 5; // BT.470-2 bg
param->i_keyint_max = param->i_fps_num == 50 ? 48 : 24;
}
else if( ( param->i_fps_num == 30000 || param->i_fps_num == 60000 ) && param->i_fps_den == 1001 )
{
param->vui.i_vidformat = 2; // NTSC
param->vui.i_colorprim = 6; // BT.601-6
param->vui.i_transfer = 6; // BT.601-6
param->vui.i_colmatrix = 6; // BT.601-6
param->i_keyint_max = param->i_fps_num / 1000;
}
else
{
param->vui.i_vidformat = 5; // undefined
param->vui.i_colorprim = 2; // undefined
param->vui.i_transfer = 2; // undefined
param->vui.i_colmatrix = 2; // undefined
}
/* Change to BT.709 for HD resolutions */
if( param->i_width >= 1280 && param->i_height >= 720 )
{
param->vui.i_colorprim = 1;
param->vui.i_transfer = 1;
param->vui.i_colmatrix = 1;
}
x264_param_apply_profile( param, X264_BIT_DEPTH == 10 ? "high10" : "high" );
param->i_nal_hrd = X264_NAL_HRD_FAKE_VBR;
param->b_aud = 1;
param->i_log_level = X264_LOG_INFO;
//param->rc.f_vbv_buffer_init = 0.1;
if( h->obe_system == OBE_SYSTEM_TYPE_GENERIC )
{
param->sc.f_speed = 1.0;
param->sc.b_alt_timer = 1;
if( param->i_width >= 1280 && param->i_height >= 720 )
param->sc.max_preset = 7; /* on the conservative side for HD */
else
param->sc.max_preset = 10;
param->rc.i_lookahead = param->i_keyint_max;
}
return 0;
}
static void enc_preprocess(MSFilter *f){
EncData *d=(EncData*)f->data;
x264_param_t *params=&d->params;
d->packer=rfc3984_new();
rfc3984_set_mode(d->packer,d->mode);
rfc3984_enable_stap_a(d->packer,FALSE);
#if defined(__arm__) || defined(ANDROID)
if (x264_param_default_preset(params,"superfast"/*"ultrafast"*/,"zerolatency")) {
#else
x264_param_default(params); {
#endif
ms_error("Cannot apply default x264 configuration");
};
params->i_threads=ms_get_cpu_count();
params->i_sync_lookahead=0;
params->i_width=d->vconf.vsize.width;
params->i_height=d->vconf.vsize.height;
params->i_fps_num=(int)d->vconf.fps;
params->i_fps_den=1;
params->i_slice_max_size=ms_get_payload_max_size()-100; /*-100 security margin*/
params->i_level_idc=13;
apply_bitrate(f);
params->rc.i_lookahead=0;
/*enable this by config ?*/
/*
params.i_keyint_max = (int)d->fps*d->keyframe_int;
params.i_keyint_min = (int)d->fps;
*/
params->b_repeat_headers=1;
params->b_annexb=0;
//these parameters must be set so that our stream is baseline
params->analyse.b_transform_8x8 = 0;
params->b_cabac = 0;
params->i_cqm_preset = X264_CQM_FLAT;
params->i_bframe = 0;
params->analyse.i_weighted_pred = X264_WEIGHTP_NONE;
d->enc=x264_encoder_open(params);
if (d->enc==NULL) ms_error("Fail to create x264 encoder.");
d->framenum=0;
video_starter_init(&d->starter);
}
static void x264_nals_to_msgb(x264_nal_t *xnals, int num_nals, MSQueue * nalus){
int i;
mblk_t *m;
/*int bytes;*/
for (i=0;i<num_nals;++i){
m=allocb(xnals[i].i_payload+10,0);
memcpy(m->b_wptr,xnals[i].p_payload+4,xnals[i].i_payload-4);
m->b_wptr+=xnals[i].i_payload-4;
if (xnals[i].i_type==7) {
ms_message("A SPS is being sent.");
}else if (xnals[i].i_type==8) {
ms_message("A PPS is being sent.");
}
ms_queue_put(nalus,m);
}
}
/**********************
* OpenCodec
* Abre el codec
***********************/
int H264RtspEncoder::OpenCodec()
{
x264_param_t params;
Log("-OpenCodec H264 [%dbps,%dfps]\n",bitrate,fps);
// Check
if (opened)
return Error("Codec already opened\n");
// Reset default values
x264_param_default(¶ms);
// Use a defulat preset
x264_param_default_preset(¶ms,"fast","zerolatency");
// Set log
params.pf_log = X264_log;
params.i_log_level = X264_LOG_WARNING;
// Set encoding context size
params.i_width = width;
params.i_height = height;
// Set parameters
params.i_keyint_max = intraPeriod;
params.i_keyint_min = 1;
params.b_cabac = 1;
params.i_frame_reference = 0;
params.rc.i_bitrate = bitrate / 1024;
params.rc.b_stat_write = 0;
params.i_slice_max_size = RTPPAYLOADSIZE;
params.b_sliced_threads = 0;
params.rc.i_lookahead = 0;
params.i_bframe = 0;
//params.b_annexb = 0;
params.b_repeat_headers = 1;
params.i_threads = 0;
params.b_vfr_input = 0;
params.rc.i_qp_min = qMin;
params.rc.i_qp_max = qMax;
params.rc.i_qp_step = qMax-qMin;
params.i_fps_num = fps;
params.i_fps_den = 2;
// Set profile level constrains
x264_param_apply_profile(¶ms,"baseline");
// Open encoder
enc = x264_encoder_open(¶ms);
//Check it is correct
if (!enc)
return Error("Could not open h264 codec\n");
// Clean pictures
memset(&pic,0,sizeof(x264_picture_t));
memset(&pic_out,0,sizeof(x264_picture_t));
//Set picture type
pic.i_type = X264_TYPE_AUTO;
// We are opened
opened=true;
// Exit
return 1;
}
bool X264Encoder::openX264Encoder()
{
this->closeX264Encoder();
if(!pParameter)
{
pParameter = (x264_param_t *)malloc(sizeof(x264_param_t));
if (!pParameter) {
this->closeX264Encoder();
return false;
}
memset(pParameter, 0, sizeof(x264_param_t));
}
int ret = x264_param_default_preset(pParameter, "ultrafast", "zerolatency");
if (ret != 0) {
this->closeX264Encoder();
return false;
}
pParameter->i_threads = 1;
pParameter->b_sliced_threads = 0;
pParameter->i_sync_lookahead = X264_SYNC_LOOKAHEAD_AUTO;
pParameter->i_width = width;
pParameter->i_height = height;
pParameter->i_frame_total = 0;
pParameter->b_deterministic = 1;
pParameter->i_frame_reference = 4;
pParameter->i_bframe = 0;
pParameter->i_bframe_pyramid = 0;
pParameter->i_bframe_adaptive = 0;
pParameter->b_intra_refresh = 0;
pParameter->i_csp = X264_CSP_I420;
pParameter->i_level_idc = 9;
pParameter->i_keyint_min = 10;
pParameter->i_keyint_max = 30;
pParameter->b_repeat_headers = 1;
pParameter->b_interlaced = 0;
pParameter->i_cqm_preset = X264_CQM_FLAT;
pParameter->psz_cqm_file = NULL;
pParameter->b_aud = 0;
pParameter->i_nal_hrd = X264_NAL_HRD_NONE;
pParameter->i_scenecut_threshold = 40;
pParameter->i_bframe_bias = 0;
pParameter->i_fps_num = i_fps;
pParameter->i_fps_den = 1;
pParameter->i_timebase_num = 1;
pParameter->i_timebase_den = 1000000;
pParameter->analyse.i_weighted_pred = 0;
pParameter->analyse.b_weighted_bipred = 0;
pParameter->analyse.b_chroma_me = 1;
pParameter->analyse.i_trellis = 1;
pParameter->analyse.i_subpel_refine = 4;
pParameter->analyse.b_transform_8x8 = 1;
pParameter->analyse.i_me_range = 8;
pParameter->analyse.i_me_method = X264_ME_UMH; //2
pParameter->analyse.i_direct_mv_pred = X264_DIRECT_PRED_TEMPORAL; //2
pParameter->analyse.intra = 0;
pParameter->analyse.inter = 0;
pParameter->b_cabac = 0;
pParameter->b_vfr_input = 0;
pParameter->rc.i_rc_method = X264_RC_ABR;//X264_RC_CQP;
pParameter->rc.f_qcompress = 0.6f; // 0.0 => cbr, 1.0 => constant qp
pParameter->rc.i_lookahead = 0;
pParameter->rc.b_mb_tree = 0;
pParameter->rc.i_qp_min = 10;
pParameter->rc.i_qp_max = 51;
pParameter->rc.i_qp_step = 3;
pParameter->rc.i_qp_constant = 10;
pParameter->rc.f_rf_constant = 10; // 1pass VBR, nominal QP
pParameter->rc.i_bitrate = 300;
pParameter->rc.i_vbv_max_bitrate = 300;
pParameter->rc.i_vbv_buffer_size = 300;
pParameter->rc.f_vbv_buffer_init = 0.6f;
pParameter->rc.f_rate_tolerance = 10 / 100.0f; // In CRF mode,maximum CRF as caused by VBV
if(x264_param_apply_profile(pParameter, "baseline"))
{
this->closeX264Encoder();
return false;
}
if (!x264EncoderHandle) {
x264EncoderHandle = x264_encoder_open(pParameter);
assert(x264EncoderHandle != NULL);
}
int nal_count = 0;
x264_nal_t* nals = NULL;
x264_encoder_headers(x264EncoderHandle, &nals, &nal_count);
assert(nal_count > 0);
for (int index = 0; index < nal_count; ++index)
{
if (nals[index].i_type == NAL_SPS) {
spslen = createNalBuffer(sps, nals[index].p_payload, nals[index].i_payload);
}
if (nals[index].i_type == NAL_PPS) {
ppslen = createNalBuffer(pps, nals[index].p_payload, nals[index].i_payload);
}
}
assert(spslen != 0);
assert(ppslen != 0);
if (!pOutput) {
pOutput = (x264_picture_t *)malloc(sizeof(x264_picture_t));
if (!pOutput) {
this->closeX264Encoder();
return false;
}
}
memset(pOutput, 0, sizeof(x264_picture_t));
return true;
}
X264Encoder(int fps, int width, int height, int quality, CTSTR preset, bool bUse444, ColorDescription &colorDesc, int maxBitrate, int bufferSize, bool bUseCFR)
{
curPreset = preset;
fps_ms = 1000/fps;
StringList paramList;
curProfile = AppConfig->GetString(TEXT("Video Encoding"), TEXT("X264Profile"), TEXT("high"));
BOOL bUseCustomParams = AppConfig->GetInt(TEXT("Video Encoding"), TEXT("UseCustomSettings"));
if(bUseCustomParams)
{
String strCustomParams = AppConfig->GetString(TEXT("Video Encoding"), TEXT("CustomSettings"));
strCustomParams.KillSpaces();
if(strCustomParams.IsValid())
{
Log(TEXT("Using custom x264 settings: \"%s\""), strCustomParams.Array());
strCustomParams.GetTokenList(paramList, ' ', FALSE);
for(UINT i=0; i<paramList.Num(); i++)
{
String &strParam = paramList[i];
if(!schr(strParam, '='))
continue;
String strParamName = strParam.GetToken(0, '=');
String strParamVal = strParam.GetTokenOffset(1, '=');
if(strParamName.CompareI(TEXT("preset")))
{
if(valid_x264_string(strParamVal, (const char**)x264_preset_names))
curPreset = strParamVal;
else
Log(TEXT("invalid preset: %s"), strParamVal.Array());
paramList.Remove(i--);
}
else if(strParamName.CompareI(TEXT("tune")))
{
if(valid_x264_string(strParamVal, (const char**)x264_tune_names))
curTune = strParamVal;
else
Log(TEXT("invalid tune: %s"), strParamVal.Array());
paramList.Remove(i--);
}
else if(strParamName.CompareI(TEXT("profile")))
{
if(valid_x264_string(strParamVal, (const char **)x264_profile_names))
curProfile = strParamVal;
else
Log(TEXT("invalid profile: %s"), strParamVal.Array());
paramList.Remove(i--);
}
}
}
}
zero(¶mData, sizeof(paramData));
LPSTR lpPreset = curPreset.CreateUTF8String();
LPSTR lpTune = curTune.CreateUTF8String();
if (x264_param_default_preset(¶mData, lpPreset, lpTune))
Log(TEXT("Failed to set x264 defaults: %s/%s"), curPreset.Array(), curTune.Array());
Free(lpTune);
Free(lpPreset);
this->width = width;
this->height = height;
paramData.b_deterministic = false;
bUseCBR = AppConfig->GetInt(TEXT("Video Encoding"), TEXT("UseCBR"), 1) != 0;
bPadCBR = AppConfig->GetInt(TEXT("Video Encoding"), TEXT("PadCBR"), 1) != 0;
this->bUseCFR = bUseCFR;
SetBitRateParams(maxBitrate, bufferSize);
if(bUseCBR)
{
if(bPadCBR) paramData.i_nal_hrd = X264_NAL_HRD_CBR;
paramData.rc.i_rc_method = X264_RC_ABR;
paramData.rc.f_rf_constant = 0.0f;
}
else
{
paramData.rc.i_rc_method = X264_RC_CRF;
paramData.rc.f_rf_constant = baseCRF+float(10-quality);
}
UINT keyframeInterval = AppConfig->GetInt(TEXT("Video Encoding"), TEXT("KeyframeInterval"), 0);
paramData.b_vfr_input = !bUseCFR;
paramData.i_width = width;
paramData.i_height = height;
paramData.vui.b_fullrange = colorDesc.fullRange;
paramData.vui.i_colorprim = colorDesc.primaries;
paramData.vui.i_transfer = colorDesc.transfer;
paramData.vui.i_colmatrix = colorDesc.matrix;
if (keyframeInterval)
paramData.i_keyint_max = fps*keyframeInterval;
paramData.i_fps_num = fps;
paramData.i_fps_den = 1;
paramData.i_timebase_num = 1;
paramData.i_timebase_den = 1000;
paramData.pf_log = get_x264_log;
paramData.i_log_level = X264_LOG_WARNING;
for(UINT i=0; i<paramList.Num(); i++)
{
String &strParam = paramList[i];
if(!schr(strParam, '='))
continue;
String strParamName = strParam.GetToken(0, '=');
String strParamVal = strParam.GetTokenOffset(1, '=');
if( strParamName.CompareI(TEXT("fps")) ||
strParamName.CompareI(TEXT("force-cfr")))
{
Log(TEXT("The custom x264 command '%s' is unsupported, use the application settings instead"), strParam.Array());
continue;
}
else
{
LPSTR lpParam = strParamName.CreateUTF8String();
LPSTR lpVal = strParamVal.CreateUTF8String();
if(x264_param_parse(¶mData, lpParam, lpVal) != 0)
Log(TEXT("The custom x264 command '%s' failed"), strParam.Array());
Free(lpParam);
Free(lpVal);
}
}
if(bUse444) paramData.i_csp = X264_CSP_I444;
else paramData.i_csp = X264_CSP_I420;
colorDesc.fullRange = paramData.vui.b_fullrange;
colorDesc.primaries = paramData.vui.i_colorprim;
colorDesc.transfer = paramData.vui.i_transfer;
colorDesc.matrix = paramData.vui.i_colmatrix;
if (curProfile)
{
LPSTR lpProfile = curProfile.CreateUTF8String();
if (x264_param_apply_profile (¶mData, lpProfile))
Log(TEXT("Failed to set x264 profile: %s"), curProfile.Array());
Free(lpProfile);
}
x264 = x264_encoder_open(¶mData);
if(!x264)
CrashError(TEXT("Could not initialize x264"));
Log(TEXT("------------------------------------------"));
Log(TEXT("%s"), GetInfoString().Array());
Log(TEXT("------------------------------------------"));
DataPacket packet;
GetHeaders(packet);
}
int init_param(x264_param_t * x264_param, int width, int height, int fps, int bitrate)
{
x264_param_default(x264_param);
x264_param_default_preset(x264_param, "medium", NULL );
x264_param_default_preset(x264_param, "veryfast", "zerolatency");
//x264_param_default_preset(x264_param_, "veryfast", NULL);
//x264_param_default_preset(x264_param_,"superfast","zerolatency");
// CPU flags
//x264_param_->i_threads = X264_THREADS_AUTO;
x264_param->i_threads = X264_SYNC_LOOKAHEAD_AUTO;
// Video Properties
x264_param->i_csp = X264_CSP_I420;
x264_param->i_width = width;
x264_param->i_height = height;
//x264_param_->i_frame_total = 0;
// Bitstream parameters
//x264_param_->i_bframe = 5;
//x264_param_->b_open_gop = 0;
//x264_param_->i_bframe_pyramid = 0;
//x264_param_->i_bframe_adaptive = X264_B_ADAPT_TRELLIS;
//x264_param_->i_keyint_min = 0;
//x264_param_->i_keyint_max = fps*10;
// Rate control parameters
//x264_param_->rc.i_qp_min = 0;
//x264_param_->rc.i_qp_max = 20;
//x264_param_->rc.i_qp_constant = 0;
//x264_param_->rc.i_bitrate = 1024 * 100;
//x264_param_->rc.i_rc_method = X264_RC_CRF;
//x264_param_->rc.f_rf_constant = 22.0+float(10-5);
// muxing parameters
x264_param->b_annexb = 1;
x264_param->b_repeat_headers = 1;
//x264_param_->i_fps_den = 1; //* 帧率分母
//x264_param_->i_fps_num = 10;//* 帧率分子
//x264_param_->i_timebase_den = x264_param_->i_fps_num;
//x264_param_->i_timebase_num = x264_param_->i_fps_den;
//x264_param_->b_vfr_input = 0;
// OBS
x264_param->i_fps_num = fps;
x264_param->i_keyint_max = fps * 2;
//x264_param_->b_deterministic = false;
//x264_param_->b_vfr_input = 0;
//x264_param_->i_fps_num =fps;//数值由客户端传下来，一般是15fps
//x264_param_->i_timebase_num = 1;
//x264_param_->i_timebase_den = 1000;
//x264_param_->rc.i_qp_min = 16;
//x264_param_->rc.i_qp_max = 20;
//x264_param_->rc.i_qp_constant = 16;
x264_param->rc.i_rc_method = X264_RC_CRF;
x264_param->rc.f_rf_constant = (float)bitrate;
x264_param_apply_profile(x264_param, "main");
}
// See https://gist.github.com/roxlu/0f61a499df75e64b764d for an older version of this, with some rate control tests
// @todo - we should check if the supplied settings are valid for the current profile.. e.g. bframes are not supported by the baseline profile
bool VideoEncoder::initializeX264() {
assert(settings.width > 0);
assert(settings.height > 0);
assert(settings.fps > 0);
int r = 0;
x264_param_t* p = ¶ms;
std::string preset = (settings.preset.size()) ? settings.preset : "superfast";
std::string tune = (settings.tune.size()) ? settings.tune : "zerolatency";
STREAMER_STATUS("x264 using preset: %s and tune: %s\n", preset.c_str(), tune.c_str());
r = x264_param_default_preset(p, preset.c_str(), tune.c_str());
if(r != 0) {
STREAMER_ERROR("error: cannot set the default preset on x264.\n");
return false;
}
p->i_threads = settings.threads;
p->i_width = settings.width;
p->i_height = settings.height;
p->i_fps_num = settings.fps;
p->i_fps_den = 1;
p->b_annexb = 0; // flv == no annexb, but strangely, when I disable it the generated flv cannot be played back, for raw h264 you'll need to set annexb to 1 when you want to play it in vlc (vlc isn't properly playing back flv)
p->rc.i_rc_method = X264_RC_ABR; // when you're limited to bandwidth you set the vbv_buffer_size and vbv_max_bitrate using the X264_RC_ABR rate control method. The vbv_buffer_size is a decoder option and tells the decoder how much data must be buffered before playback can start. When vbv_max_bitrate == vbv_buffer_size, then it will take one second before the playback might start. when vbv_buffer_size == vbv_max_bitrate * 0.5, it might start in 0.5 sec.
p->rc.i_bitrate = settings.bitrate;
p->rc.i_vbv_buffer_size = (settings.vbv_buffer_size < 0) ? p->rc.i_bitrate : settings.vbv_buffer_size;
p->rc.i_vbv_max_bitrate = (settings.vbv_max_bitrate < 0) ? p->rc.i_bitrate : settings.vbv_max_bitrate;;
if(settings.keyint_max > 0) {
p->i_keyint_max = settings.keyint_max;
}
if(settings.bframe > 0) {
p->i_bframe = settings.bframe;
}
if(settings.level_idc > 0) {
p->i_level_idc = settings.level_idc;
}
#if !defined(NDEBUG)
p->i_log_level = X264_LOG_DEBUG;
p->pf_log = videoencoder_x264_log;
#endif
if(settings.profile.size()) {
r = x264_param_apply_profile(p, settings.profile.c_str());
if(r != 0) {
STREAMER_ERROR("error: cannot set the baseline profile on x264.\n");
return false;
}
}
encoder = x264_encoder_open(p);
if(!encoder) {
STREAMER_ERROR("error: cannot create the encoder.\n");
return false;
}
x264_encoder_parameters(encoder, ¶ms);
print_x264_params(p);
return true;
}