ref: c1d630d5c80737c8b1b4dbe1c3d6c763c34583e1
dir: /vp9/decoder/vp9_job_queue.c/
/* * Copyright (c) 2018 The WebM project authors. All Rights Reserved. * * Use of this source code is governed by a BSD-style license * that can be found in the LICENSE file in the root of the source * tree. An additional intellectual property rights grant can be found * in the file PATENTS. All contributing project authors may * be found in the AUTHORS file in the root of the source tree. */ #include <assert.h> #include <string.h> #include "vpx/vpx_integer.h" #include "vp9/decoder/vp9_job_queue.h" void vp9_jobq_init(JobQueueRowMt *jobq, uint8_t *buf, size_t buf_size) { #if CONFIG_MULTITHREAD pthread_mutex_init(&jobq->mutex, NULL); pthread_cond_init(&jobq->cond, NULL); #endif jobq->buf_base = buf; jobq->buf_wr = buf; jobq->buf_rd = buf; jobq->buf_end = buf + buf_size; jobq->terminate = 0; } void vp9_jobq_reset(JobQueueRowMt *jobq) { #if CONFIG_MULTITHREAD pthread_mutex_lock(&jobq->mutex); #endif jobq->buf_wr = jobq->buf_base; jobq->buf_rd = jobq->buf_base; jobq->terminate = 0; #if CONFIG_MULTITHREAD pthread_mutex_unlock(&jobq->mutex); #endif } void vp9_jobq_deinit(JobQueueRowMt *jobq) { vp9_jobq_reset(jobq); #if CONFIG_MULTITHREAD pthread_mutex_destroy(&jobq->mutex); pthread_cond_destroy(&jobq->cond); #endif } void vp9_jobq_terminate(JobQueueRowMt *jobq) { #if CONFIG_MULTITHREAD pthread_mutex_lock(&jobq->mutex); #endif jobq->terminate = 1; #if CONFIG_MULTITHREAD pthread_cond_broadcast(&jobq->cond); pthread_mutex_unlock(&jobq->mutex); #endif } int vp9_jobq_queue(JobQueueRowMt *jobq, void *job, size_t job_size) { int ret = 0; #if CONFIG_MULTITHREAD pthread_mutex_lock(&jobq->mutex); #endif if (jobq->buf_end >= jobq->buf_wr + job_size) { memcpy(jobq->buf_wr, job, job_size); jobq->buf_wr = jobq->buf_wr + job_size; #if CONFIG_MULTITHREAD pthread_cond_signal(&jobq->cond); #endif ret = 0; } else { /* Wrap around case is not supported */ assert(0); ret = 1; } #if CONFIG_MULTITHREAD pthread_mutex_unlock(&jobq->mutex); #endif return ret; } int vp9_jobq_dequeue(JobQueueRowMt *jobq, void *job, size_t job_size, int blocking) { int ret = 0; #if CONFIG_MULTITHREAD pthread_mutex_lock(&jobq->mutex); #endif if (jobq->buf_end >= jobq->buf_rd + job_size) { while (1) { if (jobq->buf_wr >= jobq->buf_rd + job_size) { memcpy(job, jobq->buf_rd, job_size); jobq->buf_rd = jobq->buf_rd + job_size; ret = 0; break; } else { /* If all the entries have been dequeued, then break and return */ if (jobq->terminate == 1) { ret = 1; break; } if (blocking == 1) { #if CONFIG_MULTITHREAD pthread_cond_wait(&jobq->cond, &jobq->mutex); #endif } else { /* If there is no job available, * and this is non blocking call then return fail */ ret = 1; break; } } } } else { /* Wrap around case is not supported */ ret = 1; } #if CONFIG_MULTITHREAD pthread_mutex_unlock(&jobq->mutex); #endif return ret; }