summaryrefslogtreecommitdiffstats
path: root/libhb/hb.c
diff options
context:
space:
mode:
Diffstat (limited to 'libhb/hb.c')
-rw-r--r--libhb/hb.c12
1 files changed, 11 insertions, 1 deletions
diff --git a/libhb/hb.c b/libhb/hb.c
index 1e3686cf2..b1050e0e0 100644
--- a/libhb/hb.c
+++ b/libhb/hb.c
@@ -99,9 +99,19 @@ void hb_avcodec_init()
av_register_all();
}
-int hb_avcodec_open(AVCodecContext *avctx, AVCodec *codec)
+int hb_avcodec_open(AVCodecContext *avctx, AVCodec *codec, int thread_count)
{
int ret;
+
+ if ( ( thread_count == HB_FFMPEG_THREADS_AUTO || thread_count > 0 ) &&
+ ( codec->type == AVMEDIA_TYPE_VIDEO ) )
+ {
+ avctx->thread_count = ( thread_count == HB_FFMPEG_THREADS_AUTO ) ?
+ hb_get_cpu_count() / 2 + 1 : thread_count;
+ avctx->thread_type = FF_THREAD_FRAME|FF_THREAD_SLICE;
+ avctx->thread_safe_callbacks = 1;
+ }
+
ret = avcodec_open(avctx, codec);
return ret;
}