summaryrefslogtreecommitdiffstats
path: root/libhb
diff options
context:
space:
mode:
authorTim Walker <[email protected]>2016-02-17 00:55:54 +0100
committerJohn Stebbins <[email protected]>2016-12-17 07:28:51 -0800
commitac6c1ab005cad9c78bb427aad7be6c9af6d22816 (patch)
tree58c86cd4fc796b796a80b8778f43aabb155dedf8 /libhb
parent1c7db69a6766ef678e74628eedbc4ccfc60a61f8 (diff)
libhb: add code to support our QSV-accelerated encoder.
The files are copied verbatim from the patch we just removed, where they were known as: libavcodec/qsv.h libavcodec/qsv.c
Diffstat (limited to 'libhb')
-rw-r--r--libhb/qsv_libav.c646
-rw-r--r--libhb/qsv_libav.h494
2 files changed, 1140 insertions, 0 deletions
diff --git a/libhb/qsv_libav.c b/libhb/qsv_libav.c
new file mode 100644
index 000000000..2c46111ec
--- /dev/null
+++ b/libhb/qsv_libav.c
@@ -0,0 +1,646 @@
+/* ********************************************************************* *\
+
+Copyright (C) 2013 Intel Corporation. All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+- Redistributions of source code must retain the above copyright notice,
+this list of conditions and the following disclaimer.
+- Redistributions in binary form must reproduce the above copyright notice,
+this list of conditions and the following disclaimer in the documentation
+and/or other materials provided with the distribution.
+- Neither the name of Intel Corporation nor the names of its contributors
+may be used to endorse or promote products derived from this software
+without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY INTEL CORPORATION "AS IS" AND ANY EXPRESS OR
+IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+IN NO EVENT SHALL INTEL CORPORATION BE LIABLE FOR ANY DIRECT, INDIRECT,
+INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+\* ********************************************************************* */
+
+#include "qsv.h"
+
+#include "avcodec.h"
+#include "internal.h"
+
+int av_qsv_get_free_encode_task(av_qsv_list * tasks)
+{
+ int ret = MFX_ERR_NOT_FOUND;
+ int i = 0;
+ if (tasks)
+ for (i = 0; i < av_qsv_list_count(tasks); i++) {
+ av_qsv_task *task = av_qsv_list_item(tasks, i);
+ if (task->stage && task->stage->out.sync)
+ if (!(*task->stage->out.sync->p_sync)) {
+ ret = i;
+ break;
+ }
+ }
+ return ret;
+}
+
+int av_qsv_get_free_sync(av_qsv_space * space, av_qsv_context * qsv)
+{
+ int ret = -1;
+ int counter = 0;
+
+ while (1) {
+ for (int i = 0; i < space->sync_num; i++) {
+ if (!(*(space->p_syncp[i]->p_sync)) &&
+ 0 == space->p_syncp[i]->in_use ) {
+ if (i > space->sync_num_max_used)
+ space->sync_num_max_used = i;
+ ff_qsv_atomic_inc(&space->p_syncp[i]->in_use);
+ return i;
+ }
+ }
+#if HAVE_THREADS
+ if (++counter >= AV_QSV_REPEAT_NUM_DEFAULT) {
+#endif
+ av_log(NULL, AV_LOG_FATAL, "not enough to have %d sync point(s) allocated\n",
+ space->sync_num);
+ break;
+#if HAVE_THREADS
+ }
+ av_qsv_sleep(5);
+#endif
+ }
+ return ret;
+}
+
+int av_qsv_get_free_surface(av_qsv_space * space, av_qsv_context * qsv,
+ mfxFrameInfo * info, av_qsv_split part)
+{
+ int ret = -1;
+ int from = 0;
+ int up = space->surface_num;
+ int counter = 0;
+
+ while (1) {
+ from = 0;
+ up = space->surface_num;
+ if (part == QSV_PART_LOWER)
+ up /= 2;
+ if (part == QSV_PART_UPPER)
+ from = up / 2;
+
+ for (int i = from; i < up; i++) {
+ if (0 == space->p_surfaces[i]->Data.Locked) {
+ memcpy(&(space->p_surfaces[i]->Info), info,
+ sizeof(mfxFrameInfo));
+ if (i > space->surface_num_max_used)
+ space->surface_num_max_used = i;
+ return i;
+ }
+ }
+#if HAVE_THREADS
+ if (++counter >= AV_QSV_REPEAT_NUM_DEFAULT) {
+#endif
+ av_log(NULL, AV_LOG_FATAL,
+ "not enough to have %d surface(s) allocated\n", up);
+ break;
+#if HAVE_THREADS
+ }
+ av_qsv_sleep(5);
+#endif
+ }
+ return ret;
+}
+
+int ff_qsv_is_surface_in_pipe(mfxFrameSurface1 * p_surface, av_qsv_context * qsv)
+{
+ int ret = 0;
+ int a, b,i;
+ av_qsv_list *list = 0;
+ av_qsv_stage *stage = 0;
+
+ if (!p_surface)
+ return ret;
+ if (!qsv->pipes)
+ return ret;
+
+ for (a = 0; a < av_qsv_list_count(qsv->pipes); a++) {
+ list = av_qsv_list_item(qsv->pipes, a);
+ for (b = 0; b < av_qsv_list_count(list); b++) {
+ stage = av_qsv_list_item(list, b);
+ if (p_surface == stage->out.p_surface)
+ return (stage->type << 16) | 2;
+ if (p_surface == stage->in.p_surface)
+ return (stage->type << 16) | 1;
+ }
+ }
+ return ret;
+}
+
+int ff_qsv_is_sync_in_pipe(mfxSyncPoint * sync, av_qsv_context * qsv)
+{
+ int ret = 0;
+ int a, b;
+ av_qsv_list *list = 0;
+ av_qsv_stage *stage = 0;
+
+ if (!sync)
+ return ret;
+ if (!qsv->pipes)
+ return ret;
+
+ for (a = 0; a < av_qsv_list_count(qsv->pipes); a++) {
+ list = av_qsv_list_item(qsv->pipes, a);
+ for (b = 0; b < av_qsv_list_count(list); b++) {
+ stage = av_qsv_list_item(list, b);
+ if (sync == stage->out.sync->p_sync) {
+ return 1;
+ }
+ }
+ }
+ return ret;
+}
+
+av_qsv_stage *av_qsv_stage_init(void)
+{
+ av_qsv_stage *stage = av_mallocz(sizeof(av_qsv_stage));
+ return stage;
+}
+
+void av_qsv_stage_clean(av_qsv_stage ** stage)
+{
+ if ((*stage)->out.sync) {
+ if ((*stage)->out.sync->p_sync)
+ *(*stage)->out.sync->p_sync = 0;
+ if ((*stage)->out.sync->in_use > 0)
+ ff_qsv_atomic_dec(&(*stage)->out.sync->in_use);
+ (*stage)->out.sync = 0;
+ }
+ if ((*stage)->out.p_surface) {
+ (*stage)->out.p_surface = 0;
+
+ }
+ if ((*stage)->in.p_surface) {
+ (*stage)->in.p_surface = 0;
+ }
+
+ av_freep(stage);
+}
+
+void av_qsv_add_context_usage(av_qsv_context * qsv, int is_threaded)
+{
+ int is_active = 0;
+#if HAVE_THREADS
+ int mut_ret = 0;
+#endif
+
+ is_active = ff_qsv_atomic_inc(&qsv->is_context_active);
+ if (is_active == 1) {
+ memset(&qsv->mfx_session, 0, sizeof(mfxSession));
+ av_qsv_pipe_list_create(&qsv->pipes, is_threaded);
+
+ qsv->dts_seq = av_qsv_list_init(is_threaded);
+
+#if HAVE_THREADS
+ if (is_threaded) {
+ qsv->qts_seq_mutex = av_mallocz(sizeof(pthread_mutex_t));
+ if (qsv->qts_seq_mutex){
+ mut_ret = pthread_mutex_init(qsv->qts_seq_mutex, NULL);
+ if(mut_ret)
+ av_log(NULL, AV_LOG_ERROR, "pthread_mutex_init issue[%d] at %s\n",mut_ret,__FUNCTION__);
+ }
+
+ } else
+#endif
+ qsv->qts_seq_mutex = 0;
+ }
+}
+
+int av_qsv_context_clean(av_qsv_context * qsv)
+{
+ int is_active = 0;
+ mfxStatus sts = MFX_ERR_NONE;
+#if HAVE_THREADS
+ int mut_ret = 0;
+#endif
+
+ is_active = ff_qsv_atomic_dec(&qsv->is_context_active);
+
+ // spaces would have to be cleaned on the own,
+ // here we care about the rest, common stuff
+ if (is_active == 0) {
+
+ if (qsv->dts_seq) {
+ while (av_qsv_list_count(qsv->dts_seq))
+ av_qsv_dts_pop(qsv);
+
+ av_qsv_list_close(&qsv->dts_seq);
+ }
+#if HAVE_THREADS
+ if (qsv->qts_seq_mutex) {
+ mut_ret = pthread_mutex_destroy(qsv->qts_seq_mutex);
+ if(mut_ret)
+ av_log(NULL, AV_LOG_ERROR, "pthread_mutex_destroy issue[%d] at %s\n", mut_ret,__FUNCTION__);
+#endif
+ qsv->qts_seq_mutex = 0;
+#if HAVE_THREADS
+ }
+#endif
+
+ if (qsv->pipes)
+ av_qsv_pipe_list_clean(&qsv->pipes);
+
+ if (qsv->mfx_session) {
+ sts = MFXClose(qsv->mfx_session);
+ AV_QSV_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
+ qsv->mfx_session = 0;
+ }
+ }
+ return 0;
+}
+
+void av_qsv_pipe_list_create(av_qsv_list ** list, int is_threaded)
+{
+ if (!*list)
+ *list = av_qsv_list_init(is_threaded);
+}
+
+void av_qsv_pipe_list_clean(av_qsv_list ** list)
+{
+ av_qsv_list *stage;
+ int i = 0;
+ if (*list) {
+ for (i = av_qsv_list_count(*list); i > 0; i--) {
+ stage = av_qsv_list_item(*list, i - 1);
+ av_qsv_flush_stages(*list, &stage);
+ }
+ av_qsv_list_close(list);
+ }
+}
+
+void av_qsv_add_stagee(av_qsv_list ** list, av_qsv_stage * stage, int is_threaded)
+{
+ if (!*list)
+ *list = av_qsv_list_init(is_threaded);
+ av_qsv_list_add(*list, stage);
+}
+
+av_qsv_stage *av_qsv_get_last_stage(av_qsv_list * list)
+{
+ av_qsv_stage *stage = 0;
+ int size = 0;
+
+ av_qsv_list_lock(list);
+ size = av_qsv_list_count(list);
+ if (size > 0)
+ stage = av_qsv_list_item(list, size - 1);
+ av_qsv_list_unlock(list);
+
+ return stage;
+}
+
+void av_qsv_flush_stages(av_qsv_list * list, av_qsv_list ** item)
+{
+ int i = 0;
+ int x = 0;
+ av_qsv_stage *stage = 0;
+ av_qsv_list *to_remove_list = 0;
+ av_qsv_list *to_remove_atom_list = 0;
+ av_qsv_list *to_remove_atom = 0;
+
+ for (i = 0; i < av_qsv_list_count(*item); i++) {
+ stage = av_qsv_list_item(*item, i);
+ if(stage->pending){
+ if(!to_remove_list)
+ to_remove_list = av_qsv_list_init(0);
+ av_qsv_list_add(to_remove_list, stage->pending);
+ }
+ av_qsv_stage_clean(&stage);
+ // should actually remove from the list but ok...
+ }
+ av_qsv_list_rem(list, *item);
+ av_qsv_list_close(item);
+
+ if(to_remove_list){
+ for (i = av_qsv_list_count(to_remove_list); i > 0; i--){
+ to_remove_atom_list = av_qsv_list_item(to_remove_list, i-1);
+ for (x = av_qsv_list_count(to_remove_atom_list); x > 0; x--){
+ to_remove_atom = av_qsv_list_item(to_remove_atom_list, x-1);
+ av_qsv_flush_stages(list,&to_remove_atom);
+ }
+ }
+ av_qsv_list_close(&to_remove_list);
+ }
+}
+
+av_qsv_list *av_qsv_pipe_by_stage(av_qsv_list * list, av_qsv_stage * stage)
+{
+ av_qsv_list *item = 0;
+ av_qsv_stage *cur_stage = 0;
+ int i = 0;
+ int a = 0;
+ for (i = 0; i < av_qsv_list_count(list); i++) {
+ item = av_qsv_list_item(list, i);
+ for (a = 0; a < av_qsv_list_count(item); a++) {
+ cur_stage = av_qsv_list_item(item, a);
+ if (cur_stage == stage)
+ return item;
+ }
+ }
+ return 0;
+}
+
+// no duplicate of the same value, if end == 0 : working over full length
+void av_qsv_dts_ordered_insert(av_qsv_context * qsv, int start, int end,
+ int64_t dts, int iter)
+{
+ av_qsv_dts *cur_dts = 0;
+ av_qsv_dts *new_dts = 0;
+ int i = 0;
+#if HAVE_THREADS
+ int mut_ret = 0;
+#endif
+
+
+#if HAVE_THREADS
+ if (iter == 0 && qsv->qts_seq_mutex){
+ mut_ret = pthread_mutex_lock(qsv->qts_seq_mutex);
+ if(mut_ret)
+ av_log(NULL, AV_LOG_ERROR, "pthread_mutex_lock issue[%d] at %s\n",mut_ret, __FUNCTION__);
+ }
+#endif
+
+ if (end == 0)
+ end = av_qsv_list_count(qsv->dts_seq);
+
+ if (end <= start) {
+ new_dts = av_mallocz(sizeof(av_qsv_dts));
+ if( new_dts ) {
+ new_dts->dts = dts;
+ av_qsv_list_add(qsv->dts_seq, new_dts);
+ }
+ } else
+ for (i = end; i > start; i--) {
+ cur_dts = av_qsv_list_item(qsv->dts_seq, i - 1);
+ if (cur_dts->dts < dts) {
+ new_dts = av_mallocz(sizeof(av_qsv_dts));
+ if( new_dts ) {
+ new_dts->dts = dts;
+ av_qsv_list_insert(qsv->dts_seq, i, new_dts);
+ }
+ break;
+ } else if (cur_dts->dts == dts)
+ break;
+ }
+#if HAVE_THREADS
+ if (iter == 0 && qsv->qts_seq_mutex){
+ mut_ret = pthread_mutex_unlock(qsv->qts_seq_mutex);
+ if(mut_ret)
+ av_log(NULL, AV_LOG_ERROR, "pthread_mutex_unlock issue[%d] at %s\n",mut_ret, __FUNCTION__);
+ }
+#endif
+}
+
+void av_qsv_dts_pop(av_qsv_context * qsv)
+{
+ av_qsv_dts *item = 0;
+#if HAVE_THREADS
+ int mut_ret = 0;
+#endif
+
+#if HAVE_THREADS
+ if (qsv && qsv->qts_seq_mutex){
+ mut_ret = pthread_mutex_lock(qsv->qts_seq_mutex);
+ if(mut_ret)
+ av_log(NULL, AV_LOG_ERROR, "pthread_mutex_lock issue[%d] at %s\n",mut_ret, __FUNCTION__);
+ }
+#endif
+
+ if (av_qsv_list_count(qsv->dts_seq)) {
+ item = av_qsv_list_item(qsv->dts_seq, 0);
+ av_qsv_list_rem(qsv->dts_seq, item);
+ av_free(item);
+ }
+#if HAVE_THREADS
+ if (qsv && qsv->qts_seq_mutex){
+ mut_ret = pthread_mutex_unlock(qsv->qts_seq_mutex);
+ if(mut_ret)
+ av_log(NULL, AV_LOG_ERROR, "pthread_mutex_lock issue[%d] at %s\n",mut_ret, __FUNCTION__);
+ }
+#endif
+}
+
+
+av_qsv_list *av_qsv_list_init(int is_threaded)
+{
+ av_qsv_list *l;
+#if HAVE_THREADS
+ int mut_ret;
+#endif
+
+ l = av_mallocz(sizeof(av_qsv_list));
+ if (!l)
+ return 0;
+ l->items = av_mallocz(AV_QSV_JOB_SIZE_DEFAULT * sizeof(void *));
+ if (!l->items)
+ return 0;
+ l->items_alloc = AV_QSV_JOB_SIZE_DEFAULT;
+
+#if HAVE_THREADS
+ if (is_threaded) {
+ l->mutex = av_mallocz(sizeof(pthread_mutex_t));
+ if (l->mutex){
+ mut_ret = pthread_mutexattr_init(&l->mta);
+ if( mut_ret )
+ av_log(NULL, AV_LOG_ERROR, "pthread_mutexattr_init issue[%d] at %s\n",mut_ret, __FUNCTION__);
+ mut_ret = pthread_mutexattr_settype(&l->mta, PTHREAD_MUTEX_RECURSIVE /*PTHREAD_MUTEX_ERRORCHECK*/);
+ if( mut_ret )
+ av_log(NULL, AV_LOG_ERROR, "pthread_mutexattr_settype issue[%d] at %s\n",mut_ret, __FUNCTION__);
+ mut_ret = pthread_mutex_init(l->mutex, &l->mta);
+ if( mut_ret )
+ av_log(NULL, AV_LOG_ERROR, "pthread_mutex_init issue[%d] at %s\n",mut_ret, __FUNCTION__);
+ }
+ } else
+#endif
+ l->mutex = 0;
+ return l;
+}
+
+int av_qsv_list_count(av_qsv_list * l)
+{
+ int count;
+
+ av_qsv_list_lock(l);
+ count = l->items_count;
+ av_qsv_list_unlock(l);
+ return count;
+}
+
+int av_qsv_list_add(av_qsv_list * l, void *p)
+{
+ int pos = -1;
+
+ if (!p) {
+ return pos;
+ }
+
+ av_qsv_list_lock(l);
+
+ if (l->items_count == l->items_alloc) {
+ /* We need a bigger boat */
+ l->items_alloc += AV_QSV_JOB_SIZE_DEFAULT;
+ l->items = av_realloc(l->items, l->items_alloc * sizeof(void *));
+ }
+
+ l->items[l->items_count] = p;
+ pos = (l->items_count);
+ l->items_count++;
+
+ av_qsv_list_unlock(l);
+
+ return pos;
+}
+
+void av_qsv_list_rem(av_qsv_list * l, void *p)
+{
+ int i;
+
+ av_qsv_list_lock(l);
+
+ /* Find the item in the list */
+ for (i = 0; i < l->items_count; i++) {
+ if (l->items[i] == p) {
+ /* Shift all items after it sizeof( void * ) bytes earlier */
+ memmove(&l->items[i], &l->items[i + 1],
+ (l->items_count - i - 1) * sizeof(void *));
+
+ l->items_count--;
+ break;
+ }
+ }
+
+ av_qsv_list_unlock(l);
+}
+
+void *av_qsv_list_item(av_qsv_list * l, int i)
+{
+ void *ret = NULL;
+
+ if (i < 0)
+ return NULL;
+
+ av_qsv_list_lock(l);
+ if( i < l->items_count)
+ ret = l->items[i];
+ av_qsv_list_unlock(l);
+ return ret;
+}
+
+void av_qsv_list_insert(av_qsv_list * l, int pos, void *p)
+{
+
+ if (!p)
+ return;
+
+ av_qsv_list_lock(l);
+
+ if (l->items_count == l->items_alloc) {
+ l->items_alloc += AV_QSV_JOB_SIZE_DEFAULT;
+ l->items = av_realloc(l->items, l->items_alloc * sizeof(void *));
+ }
+
+ if (l->items_count != pos) {
+ memmove(&l->items[pos + 1], &l->items[pos],
+ (l->items_count - pos) * sizeof(void *));
+ }
+
+ l->items[pos] = p;
+ l->items_count--;
+
+ av_qsv_list_unlock(l);
+}
+
+void av_qsv_list_close(av_qsv_list ** _l)
+{
+ av_qsv_list *l = *_l;
+#if HAVE_THREADS
+ int mut_ret;
+#endif
+
+ av_qsv_list_lock(l);
+
+ av_free(l->items);
+
+#if HAVE_THREADS
+ if (l->mutex){
+ mut_ret = pthread_mutex_unlock(l->mutex);
+ if( mut_ret )
+ av_log(NULL, AV_LOG_ERROR, "pthread_mutex_unlock issue[%d] at %s\n",mut_ret, __FUNCTION__);
+ mut_ret = pthread_mutex_destroy(&l->mutex);
+ mut_ret = pthread_mutexattr_destroy(&l->mta);
+ }
+#endif
+ av_freep(_l);
+}
+
+int av_qsv_list_lock(av_qsv_list *l){
+ int ret = 0;
+#if HAVE_THREADS
+ if (l->mutex){
+ ret = pthread_mutex_lock(l->mutex);
+ if( ret )
+ av_log(NULL, AV_LOG_ERROR, "pthread_mutex_lock issue[%d] at %s\n",ret, __FUNCTION__);
+ }
+#endif
+ return ret;
+}
+
+int av_qsv_list_unlock(av_qsv_list *l){
+ int ret = 0;
+#if HAVE_THREADS
+ if (l->mutex){
+ ret = pthread_mutex_unlock(l->mutex);
+ if( ret )
+ av_log(NULL, AV_LOG_ERROR, "pthread_mutex_unlock issue[%d] at %s\n",ret, __FUNCTION__);
+ }
+#endif
+ return ret;
+}
+
+int av_is_qsv_available(mfxIMPL impl, mfxVersion * ver)
+{
+ mfxStatus sts = MFX_ERR_NONE;
+ mfxSession mfx_session;
+
+ memset(&mfx_session, 0, sizeof(mfxSession));
+ sts = MFXInit(impl, ver, &mfx_session);
+ if (sts >= 0)
+ MFXClose(mfx_session);
+ return sts;
+}
+
+void av_qsv_wait_on_sync(av_qsv_context *qsv, av_qsv_stage *stage)
+{
+ int iter = 0;
+ mfxStatus sts = MFX_ERR_NONE;
+ if( stage )
+ if(*stage->out.sync->p_sync){
+ while(1){
+ iter++;
+ sts = MFXVideoCORE_SyncOperation(qsv->mfx_session,*stage->out.sync->p_sync, AV_QSV_SYNC_TIME_DEFAULT);
+ if(MFX_WRN_IN_EXECUTION == sts){
+
+ if(iter>20)
+ AV_QSV_DEBUG_ASSERT(1, "Sync failed");
+
+ av_qsv_sleep(10);
+ continue;
+ }
+ AV_QSV_CHECK_RESULT(sts, MFX_ERR_NONE, sts);
+ break;
+ }
+ }
+} \ No newline at end of file
diff --git a/libhb/qsv_libav.h b/libhb/qsv_libav.h
new file mode 100644
index 000000000..971e6a4b3
--- /dev/null
+++ b/libhb/qsv_libav.h
@@ -0,0 +1,494 @@
+/* ********************************************************************* *\
+
+Copyright (C) 2013 Intel Corporation. All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+- Redistributions of source code must retain the above copyright notice,
+this list of conditions and the following disclaimer.
+- Redistributions in binary form must reproduce the above copyright notice,
+this list of conditions and the following disclaimer in the documentation
+and/or other materials provided with the distribution.
+- Neither the name of Intel Corporation nor the names of its contributors
+may be used to endorse or promote products derived from this software
+without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY INTEL CORPORATION "AS IS" AND ANY EXPRESS OR
+IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+IN NO EVENT SHALL INTEL CORPORATION BE LIABLE FOR ANY DIRECT, INDIRECT,
+INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+\* ********************************************************************* */
+
+#ifndef AVCODEC_QSV_H
+#define AVCODEC_QSV_H
+
+/**
+ * @file
+ * @ingroup lavc_codec_hwaccel_qsv
+ * Common header for QSV/MediaSDK acceleration
+ */
+
+/**
+ * @defgroup lavc_codec_hwaccel_qsv QSV/MediaSDK based Decode/Encode and VPP
+ * @ingroup lavc_codec_hwaccel
+ *
+ * As Intel Quick Sync Video (QSV) can decode/preprocess/encode with HW
+ * acceleration.
+ *
+ * Supported features:
+ * - access:
+ * - format AV_PIX_FMT_QSV_H264, AVCodec decoder based implementation
+ * - name "h264_qsv", avcodec_find_decoder_by_name( "h264_qsv")
+ * - IO Pattern:
+ * - Opaque memory: MFX_IOPATTERN_OUT_OPAQUE_MEMORY // Video memory is
+ * MFX_IMPL_HARDWARE or MFX_IMPL_AUTO and runtime support,
+ * otherwise: System Memory
+ * - System memory: MFX_IOPATTERN_OUT_SYSTEM_MEMORY
+ * - Allocators:
+ * - default allocator for System memory: MFX_MEMTYPE_SYSTEM_MEMORY
+ * - details:
+ * implementation as "per frame"
+ *
+ * TODO list:
+ * - access:
+ * - format AV_PIX_FMT_QSV_MPEG2
+ * - format AV_PIX_FMT_QSV_VC1
+ * - format AV_PIX_FMT_QSV, see "details" below
+ * - IO Pattern:
+ * - VIDEO_MEMORY // MFX_IOPATTERN_OUT_VIDEO_MEMORY
+ * - Allocators:
+ * - Video memory: MFX_MEMTYPE_VIDEO_MEMORY_DECODER_TARGET /
+ * MFX_MEMTYPE_VIDEO_MEMORY_PROCESSOR_TARGET
+ * - details:
+ * "per slice" support: AV_PIX_FMT_QSV with AVHWAccel based implementation
+ *
+ * Note av_qsv_config struct required to fill in via
+ * AVCodecContext.hwaccel_context
+ *
+ * As per frame, note AVFrame.data[2] (qsv_atom) used for frame atom id,
+ * data/linesize should be used together with SYSTEM_MEMORY and tested
+ *
+ * Note: Compilation would require:
+ * - Intel MediaSDK headers, Full SDK is avaialble from the original web site:
+ * http://software.intel.com/en-us/vcsource/tools/media-SDK
+ * Will be referenced as mfx/*.h (mfxdefs.h, mfxstructures.h, ... )
+ * and
+ * - Final application has to link against Intel MediaSDK dispatcher, available
+ * at MediaSDK as well
+ *
+ * Target OS: as per available dispatcher and driver support
+ *
+ * Implementation details:
+ * Provided struct av_qsv_context contain several struct av_qsv_space(s) for decode,
+ * VPP and encode.
+ * av_qsv_space just contain needed environment for the appropriate action.
+ * Based on this - pipeline (see pipes) will be build to pass details such as
+ * mfxFrameSurface1* and mfxSyncPoint* from one action to the next.
+ *
+ * Resources re-usage (av_qsv_flush_stages):
+ * av_qsv_context *qsv = (av_qsv_context *)video_codec_ctx->priv_data;
+ * av_qsv_list *pipe = (av_qsv_list *)video_frame->data[2];
+ * av_qsv_flush_stages( qsv->pipes, &pipe );
+ *
+ * DTS re-usage:
+ * av_qsv_dts_pop(qsv);
+ *
+ * for video,DX9/11 memory it has to be Unlock'ed as well
+ *
+ * Implementation is thread aware and uses synchronization point(s) from MediaSDK
+ * as per configuration.
+ *
+ * For the details of MediaSDK usage and options available - please refer to the
+ * available documentation at MediaSDK.
+ *
+ * Feature set used from MSDK is defined by AV_QSV_MSDK_VERSION_MAJOR and
+ * AV_QSV_MSDK_VERSION_MINOR
+ *
+ * @{
+ */
+
+#include <stdint.h>
+#include <string.h>
+#include "mfx/mfxvideo.h"
+#include "libavutil/mem.h"
+#include "libavutil/time.h"
+
+#ifdef HAVE_AV_CONFIG_H
+#include "config.h"
+#endif
+
+#if HAVE_THREADS
+#if defined (__GNUC__)
+#include <pthread.h>
+#define ff_qsv_atomic_inc(ptr) __sync_add_and_fetch(ptr,1)
+#define ff_qsv_atomic_dec(ptr) __sync_sub_and_fetch (ptr,1)
+#elif HAVE_WINDOWS_H // MSVC case
+#include <windows.h>
+#if HAVE_PTHREADS
+#include <pthread.h>
+#elif HAVE_W32THREADS
+#include "w32pthreads.h"
+#endif
+#define ff_qsv_atomic_inc(ptr) InterlockedIncrement(ptr)
+#define ff_qsv_atomic_dec(ptr) InterlockedDecrement (ptr)
+#else
+// targeting only for MinGW or MSVC
+#endif
+
+#else
+#define ff_qsv_atomic_inc(ptr) ((*ptr)++)
+#define ff_qsv_atomic_dec(ptr) ((*ptr)--)
+#endif
+
+
+// sleep is defined in milliseconds
+#define av_qsv_sleep(x) av_usleep((x)*1000)
+
+#define AV_QSV_ZERO_MEMORY(VAR) {memset(&VAR, 0, sizeof(VAR));}
+#define AV_QSV_ALIGN32(X) (((mfxU32)((X)+31)) & (~ (mfxU32)31))
+#define AV_QSV_ALIGN16(value) (((value + 15) >> 4) << 4)
+#ifndef AV_QSV_PRINT_RET_MSG
+#define AV_QSV_PRINT_RET_MSG(ERR) { av_log(NULL, AV_LOG_FATAL,"Error code %d,\t%s\t%d\n", ERR, __FUNCTION__, __LINE__); }
+#endif
+
+#ifndef AV_QSV_DEBUG_ASSERT
+#define AV_QSV_DEBUG_ASSERT(x,y) {if ((x)) {av_log(NULL, AV_LOG_FATAL,"\nASSERT: %s\n",y);};}
+#endif
+
+#define AV_QSV_CHECK_RESULT(P, X, ERR) {if ((X) > (P)) {AV_QSV_PRINT_RET_MSG(ERR); return ERR;}}
+#define AV_QSV_CHECK_POINTER(P, ERR) {if (!(P)) {AV_QSV_PRINT_RET_MSG(ERR); return ERR;}}
+#define AV_QSV_IGNORE_MFX_STS(P, X) {if ((X) == (P)) {P = MFX_ERR_NONE;}}
+
+#define AV_QSV_ID_BUFFER MFX_MAKEFOURCC('B','U','F','F')
+#define AV_QSV_ID_FRAME MFX_MAKEFOURCC('F','R','M','E')
+
+#define AV_QSV_SURFACE_NUM 80
+#define AV_QSV_SYNC_NUM AV_QSV_SURFACE_NUM*3/4
+#define AV_QSV_BUF_SIZE_DEFAULT 4096*2160*10
+#define AV_QSV_JOB_SIZE_DEFAULT 10
+#define AV_QSV_SYNC_TIME_DEFAULT 10000
+// see av_qsv_get_free_sync, av_qsv_get_free_surface , 100 if usleep(10*1000)(10ms) == 1 sec
+#define AV_QSV_REPEAT_NUM_DEFAULT 100
+#define AV_QSV_ASYNC_DEPTH_DEFAULT 4
+
+// version of MSDK/QSV API currently used
+#define AV_QSV_MSDK_VERSION_MAJOR 1
+#define AV_QSV_MSDK_VERSION_MINOR 3
+
+typedef enum AV_QSV_STAGE_TYPE {
+
+#define AV_QSV_DECODE_MASK 0x001
+ AV_QSV_DECODE = 0x001,
+
+#define AV_QSV_VPP_MASK 0x0F0
+ // "Mandatory VPP filter" , might be with "Hint-based VPP filters"
+ AV_QSV_VPP_DEFAULT = 0x010,
+ // "User Modules" etc
+ AV_QSV_VPP_USER = 0x020,
+
+#define av_QSV_ENCODE_MASK 0x100
+ AV_QSV_ENCODE = 0x100
+#define AV_QSV_ANY_MASK 0xFFF
+} AV_QSV_STAGE_TYPE;
+
+
+typedef struct av_qsv_list {
+ // practically pthread_mutex_t
+ void *mutex;
+#if HAVE_THREADS
+ pthread_mutexattr_t mta;
+#endif
+
+ void **items;
+ int items_alloc;
+
+ int items_count;
+} av_qsv_list;
+
+typedef struct av_qsv_sync {
+ mfxSyncPoint* p_sync;
+ int in_use;
+} av_qsv_sync;
+
+typedef struct av_qsv_stage {
+ AV_QSV_STAGE_TYPE type;
+ struct {
+ mfxBitstream *p_bs;
+ mfxFrameSurface1 *p_surface;
+ } in;
+ struct {
+ mfxBitstream *p_bs;
+ mfxFrameSurface1 *p_surface;
+ av_qsv_sync *sync;
+ } out;
+ av_qsv_list *pending;
+} av_qsv_stage;
+
+typedef struct av_qsv_task {
+ mfxBitstream *bs;
+ av_qsv_stage *stage;
+} av_qsv_task;
+
+
+typedef struct av_qsv_space {
+
+ uint8_t is_init_done;
+
+ AV_QSV_STAGE_TYPE type;
+
+ mfxVideoParam m_mfxVideoParam;
+
+ mfxFrameAllocResponse response;
+ mfxFrameAllocRequest request[2]; // [0] - in, [1] - out, if needed
+
+ mfxExtOpaqueSurfaceAlloc ext_opaque_alloc;
+ mfxExtBuffer **p_ext_params;
+ uint16_t p_ext_param_num;
+
+ uint16_t surface_num_max_used;
+ uint16_t surface_num;
+ mfxFrameSurface1 *p_surfaces[AV_QSV_SURFACE_NUM];
+
+ uint16_t sync_num_max_used;
+ uint16_t sync_num;
+ av_qsv_sync *p_syncp[AV_QSV_SYNC_NUM];
+
+ mfxBitstream bs;
+ uint8_t *p_buf;
+ size_t p_buf_max_size;
+
+ // only for encode and tasks
+ av_qsv_list *tasks;
+
+ av_qsv_list *pending;
+
+ // storage for allocations/mfxMemId*
+ mfxMemId *mids;
+} av_qsv_space;
+
+typedef struct av_qsv_context {
+ volatile int is_context_active;
+
+ mfxIMPL impl;
+ mfxSession mfx_session;
+ mfxVersion ver;
+
+ // decode
+ av_qsv_space *dec_space;
+ // encode
+ av_qsv_space *enc_space;
+ // vpp
+ av_qsv_list *vpp_space;
+
+ av_qsv_list *pipes;
+
+ // MediaSDK starting from API version 1.6 includes DecodeTimeStamp
+ // in addition to TimeStamp
+ // see also AV_QSV_MSDK_VERSION_MINOR , AV_QSV_MSDK_VERSION_MAJOR
+ av_qsv_list *dts_seq;
+
+ // practically pthread_mutex_t
+ void *qts_seq_mutex;
+
+ int is_anex;
+
+ void *qsv_config;
+
+} av_qsv_context;
+
+typedef enum {
+ QSV_PART_ANY = 0,
+ QSV_PART_LOWER,
+ QSV_PART_UPPER
+} av_qsv_split;
+
+typedef struct {
+ int64_t dts;
+} av_qsv_dts;
+
+typedef struct av_qsv_alloc_frame {
+ mfxU32 id;
+ mfxFrameInfo info;
+} av_qsv_alloc_frame;
+
+typedef struct av_qsv_alloc_buffer {
+ mfxU32 id;
+ mfxU32 nbytes;
+ mfxU16 type;
+} av_qsv_alloc_buffer;
+
+typedef struct av_qsv_allocators_space {
+ av_qsv_space *space;
+ mfxFrameAllocator frame_alloc;
+ mfxBufferAllocator buffer_alloc;
+} av_qsv_allocators_space;
+
+typedef struct av_qsv_config {
+ /**
+ * Set asynch depth of processing with QSV
+ * Format: 0 and more
+ *
+ * - encoding: Set by user.
+ * - decoding: Set by user.
+ */
+ int async_depth;
+
+ /**
+ * Range of numbers that indicate trade-offs between quality and speed.
+ * Format: from 1/MFX_TARGETUSAGE_BEST_QUALITY to 7/MFX_TARGETUSAGE_BEST_SPEED inclusive
+ *
+ * - encoding: Set by user.
+ * - decoding: unused
+ */
+ int target_usage;
+
+ /**
+ * Number of reference frames; if NumRefFrame = 0, this parameter is not specified.
+ * Format: 0 and more
+ *
+ * - encoding: Set by user.
+ * - decoding: unused
+ */
+ int num_ref_frame;
+
+ /**
+ * Distance between I- or P- key frames; if it is zero, the GOP structure is unspecified.
+ * Note: If GopRefDist = 1, there are no B-frames used.
+ *
+ * - encoding: Set by user.
+ * - decoding: unused
+ */
+ int gop_ref_dist;
+
+ /**
+ * Number of pictures within the current GOP (Group of Pictures); if GopPicSize=0,
+ * then the GOP size is unspecified. If GopPicSize=1, only I-frames are used.
+ *
+ * - encoding: Set by user.
+ * - decoding: unused
+ */
+ int gop_pic_size;
+
+ /**
+ * Set type of surfaces used with QSV
+ * Format: "IOPattern enum" of Media SDK
+ *
+ * - encoding: Set by user.
+ * - decoding: Set by user.
+ */
+ int io_pattern;
+
+ /**
+ * Set amount of additional surfaces might be needed
+ * Format: ammount of additional buffers(surfaces+syncs)
+ * to allocate in advance
+ *
+ * - encoding: Set by user.
+ * - decoding: Set by user.
+ */
+ int additional_buffers;
+
+ /**
+ * If pipeline should be sync.
+ * Format: wait time in milliseconds,
+ * AV_QSV_SYNC_TIME_DEFAULT/10000 might be a good value
+ *
+ * - encoding: Set by user.
+ * - decoding: Set by user.
+ */
+ int sync_need;
+
+ /**
+ * Type of implementation needed
+ *
+ * - encoding: Set by user.
+ * - decoding: Set by user.
+ */
+ int impl_requested;
+
+ /**
+ * if QSV usage is multithreaded.
+ * Format: Yes/No, 1/0
+ *
+ * - encoding: Set by user.
+ * - decoding: Set by user.
+ */
+ int usage_threaded;
+
+ /**
+ * if QSV use an external allocation (valid per session/mfxSession)
+ * Format: pointer to allocators, if default: 0
+ *
+ * note that:
+ * System Memory: can be used without provided and external allocator,
+ * meaning MediaSDK will use an internal one
+ * Video Memory: in this case - we must provide an external allocator
+ * Also, Media SDK session doesn't require external allocator if the application
+ * uses opaque memory
+ *
+ * Calls SetFrameAllocator/SetBufferAllocator
+ * (MFXVideoCORE_SetFrameAllocator/MFXVideoCORE_SetBufferAllocator)
+ * are to pass allocators to Media SDK
+ *
+ * - encoding: Set by user.
+ * - decoding: Set by user.
+ */
+ av_qsv_allocators_space *allocators;
+
+} av_qsv_config;
+
+#define ANEX_UNKNOWN 0
+#define ANEX_PREFIX 1
+#define ANEX_NO_PREFIX 2
+
+static const uint8_t ff_prefix_code[] = { 0x00, 0x00, 0x00, 0x01 };
+
+int av_qsv_get_free_sync(av_qsv_space *, av_qsv_context *);
+int av_qsv_get_free_surface(av_qsv_space *, av_qsv_context *, mfxFrameInfo *,
+ av_qsv_split);
+int av_qsv_get_free_encode_task(av_qsv_list *);
+
+int av_is_qsv_available(mfxIMPL, mfxVersion *);
+void av_qsv_wait_on_sync(av_qsv_context *, av_qsv_stage *);
+
+void av_qsv_add_context_usage(av_qsv_context *, int);
+
+void av_qsv_pipe_list_create(av_qsv_list **, int);
+void av_qsv_pipe_list_clean(av_qsv_list **);
+
+void av_qsv_add_stagee(av_qsv_list **, av_qsv_stage *, int);
+av_qsv_stage *av_qsv_get_last_stage(av_qsv_list *);
+av_qsv_list *av_qsv_pipe_by_stage(av_qsv_list *, av_qsv_stage *);
+void av_qsv_flush_stages(av_qsv_list *, av_qsv_list **);
+
+void av_qsv_dts_ordered_insert(av_qsv_context *, int, int, int64_t, int);
+void av_qsv_dts_pop(av_qsv_context *);
+
+av_qsv_stage *av_qsv_stage_init(void);
+void av_qsv_stage_clean(av_qsv_stage **);
+int av_qsv_context_clean(av_qsv_context *);
+
+int ff_qsv_is_sync_in_pipe(mfxSyncPoint *, av_qsv_context *);
+int ff_qsv_is_surface_in_pipe(mfxFrameSurface1 *, av_qsv_context *);
+
+av_qsv_list *av_qsv_list_init(int);
+int av_qsv_list_lock(av_qsv_list *);
+int av_qsv_list_unlock(av_qsv_list *);
+int av_qsv_list_add(av_qsv_list *, void *);
+void av_qsv_list_rem(av_qsv_list *, void *);
+void av_qsv_list_insert(av_qsv_list *, int, void *);
+void av_qsv_list_close(av_qsv_list **);
+
+int av_qsv_list_count(av_qsv_list *);
+void *av_qsv_list_item(av_qsv_list *, int);
+
+/* @} */
+
+#endif //AVCODEC_QSV_H