summaryrefslogtreecommitdiffstats
path: root/libhb/encavcodec.c
diff options
context:
space:
mode:
Diffstat (limited to 'libhb/encavcodec.c')
-rw-r--r--libhb/encavcodec.c84
1 files changed, 48 insertions, 36 deletions
diff --git a/libhb/encavcodec.c b/libhb/encavcodec.c
index 1ba1463a5..fb7f1adee 100644
--- a/libhb/encavcodec.c
+++ b/libhb/encavcodec.c
@@ -46,8 +46,8 @@ struct hb_work_private_s
hb_chapter_queue_t * chapter_queue;
- struct SwsContext * sws_context_to_nv12;
- hb_buffer_t * nv12_buf;
+ struct SwsContext * sws;
+ hb_buffer_t * sws_buf;
};
int encavcodecInit( hb_work_object_t *, hb_job_t * );
@@ -96,11 +96,6 @@ static const char * const h264_vt_profile_name[] =
"auto", "baseline", "main", "high", NULL
};
-static const char * const h265_vt_profile_name[] =
-{
- "auto", "main", NULL // "main10" not currently supported.
-};
-
static const char * const h26x_mf_preset_name[] =
{
"default", NULL
@@ -193,6 +188,10 @@ int encavcodecInit( hb_work_object_t * w, hb_job_t * job )
hb_log("encavcodecInit: H.265 (VideoToolbox)");
codec_name = "hevc_videotoolbox";
break;
+ case HB_VCODEC_FFMPEG_VT_H265_10BIT:
+ hb_log("encavcodecInit: H.265 10 bit (VideoToolbox)");
+ codec_name = "hevc_videotoolbox";
+ break;
case HB_VCODEC_FFMPEG_MF_H265:
hb_log("encavcodecInit: H.265 (MediaFoundation)");
codec_name = "hevc_mf";
@@ -437,7 +436,7 @@ int encavcodecInit( hb_work_object_t * w, hb_job_t * job )
}
hb_log( "encavcodec: encoding at QP %.2f", job->vquality );
}
- else if ( job->vcodec == HB_VCODEC_FFMPEG_VT_H264 || job->vcodec == HB_VCODEC_FFMPEG_VT_H265 )
+ else if ( job->vcodec == HB_VCODEC_FFMPEG_VT_H264 || job->vcodec == HB_VCODEC_FFMPEG_VT_H265 || job->vcodec == HB_VCODEC_FFMPEG_VT_H265_10BIT)
{
context->flags |= AV_CODEC_FLAG_QSCALE;
context->global_quality = job->vquality;
@@ -527,16 +526,30 @@ int encavcodecInit( hb_work_object_t * w, hb_job_t * job )
if (job->vcodec == HB_VCODEC_FFMPEG_VT_H265)
{
- // Set profile and level
- if (job->encoder_profile != NULL && *job->encoder_profile)
- {
- if (!strcasecmp(job->encoder_profile, "main"))
- av_dict_set(&av_opts, "profile", "main", 0);
- else if (!strcasecmp(job->encoder_profile, "main10"))
- av_dict_set(&av_opts, "profile", "main10", 0);
- }
+ av_dict_set(&av_opts, "profile", "main", 0);
+ context->max_b_frames = 16;
+ }
+ if (job->vcodec == HB_VCODEC_FFMPEG_VT_H265_10BIT)
+ {
+ av_dict_set(&av_opts, "profile", "main10", 0);
context->max_b_frames = 16;
+
+ if (job->pix_fmt != AV_PIX_FMT_P010LE)
+ {
+ pv->sws = hb_sws_get_context(
+ job->width, job->height,
+ context->pix_fmt,
+ job->width, job->height,
+ AV_PIX_FMT_P010LE,
+ SWS_LANCZOS|SWS_ACCURATE_RND,
+ SWS_CS_DEFAULT);
+
+ pv->sws_buf = hb_frame_buffer_init(
+ AV_PIX_FMT_P010LE, job->width, job->height);
+
+ context->pix_fmt = AV_PIX_FMT_P010LE;
+ }
}
if (job->vcodec == HB_VCODEC_FFMPEG_VCE_H264)
@@ -646,15 +659,15 @@ int encavcodecInit( hb_work_object_t * w, hb_job_t * job )
if (job->pix_fmt != AV_PIX_FMT_NV12)
{
- pv->sws_context_to_nv12 = hb_sws_get_context(
- job->width, job->height,
- context->pix_fmt,
- job->width, job->height,
- AV_PIX_FMT_NV12,
- SWS_LANCZOS|SWS_ACCURATE_RND,
- SWS_CS_DEFAULT);
-
- pv->nv12_buf = hb_frame_buffer_init(
+ pv->sws = hb_sws_get_context(
+ job->width, job->height,
+ context->pix_fmt,
+ job->width, job->height,
+ AV_PIX_FMT_NV12,
+ SWS_LANCZOS|SWS_ACCURATE_RND,
+ SWS_CS_DEFAULT);
+
+ pv->sws_buf = hb_frame_buffer_init(
AV_PIX_FMT_NV12, job->width, job->height);
context->pix_fmt = AV_PIX_FMT_NV12;
@@ -770,7 +783,7 @@ int encavcodecInit( hb_work_object_t * w, hb_job_t * job )
job->areBframes = 0;
if (context->has_b_frames > 0)
{
- if (job->vcodec == HB_VCODEC_FFMPEG_VT_H265)
+ if (job->vcodec == HB_VCODEC_FFMPEG_VT_H265 || job->vcodec == HB_VCODEC_FFMPEG_VT_H265_10BIT)
{
// VT appears to enable b-pyramid by default and there
// is no documented way of modifying this behaviour or
@@ -816,13 +829,13 @@ void encavcodecClose( hb_work_object_t * w )
}
hb_avcodec_free_context(&pv->context);
}
- if (pv->sws_context_to_nv12 != NULL)
+ if (pv->sws != NULL)
{
- sws_freeContext(pv->sws_context_to_nv12);
+ sws_freeContext(pv->sws);
}
- if (pv->nv12_buf != NULL)
+ if (pv->sws_buf != NULL)
{
- hb_buffer_close(&pv->nv12_buf);
+ hb_buffer_close(&pv->sws_buf);
}
if( pv->file )
{
@@ -984,14 +997,14 @@ static void Encode( hb_work_object_t *w, hb_buffer_t *in,
frame.linesize[1] = in->plane[1].stride;
frame.linesize[2] = in->plane[2].stride;
- if (pv->sws_context_to_nv12)
+ if (pv->sws)
{
uint8_t *srcs[] = { in->plane[0].data, in->plane[1].data, in->plane[2].data };
int srcs_stride[] = { in->plane[0].stride, in->plane[1].stride, in->plane[2].stride };
- uint8_t *dsts[] = { pv->nv12_buf->plane[0].data, pv->nv12_buf->plane[1].data, NULL };
- int dsts_stride[] = { pv->nv12_buf->plane[0].stride, pv->nv12_buf->plane[1].stride, 0 };
+ uint8_t *dsts[] = { pv->sws_buf->plane[0].data, pv->sws_buf->plane[1].data, NULL };
+ int dsts_stride[] = { pv->sws_buf->plane[0].stride, pv->sws_buf->plane[1].stride, 0 };
- sws_scale(pv->sws_context_to_nv12,
+ sws_scale(pv->sws,
(const uint8_t* const*)srcs, srcs_stride,
0, in->f.height, dsts, dsts_stride);
@@ -1208,6 +1221,7 @@ const char* const* hb_av_preset_get_names(int encoder)
case HB_VCODEC_FFMPEG_VT_H264:
case HB_VCODEC_FFMPEG_VT_H265:
+ case HB_VCODEC_FFMPEG_VT_H265_10BIT:
return h26x_vt_preset_name;
case HB_VCODEC_FFMPEG_MF_H264:
@@ -1229,8 +1243,6 @@ const char* const* hb_av_profile_get_names(int encoder)
return h265_nvenc_profile_names;
case HB_VCODEC_FFMPEG_VT_H264:
return h264_vt_profile_name;
- case HB_VCODEC_FFMPEG_VT_H265:
- return h265_vt_profile_name;
case HB_VCODEC_FFMPEG_MF_H264:
return h264_mf_profile_name;
case HB_VCODEC_FFMPEG_MF_H265: