v1.1.1

packages/artinchip/mpp/base/thread/mpp_thread.c

@@ -0,0 +1,140 @@
+/*
+ * Copyright (C) 2020-2024 ArtInChip Technology Co. Ltd
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Author: <che.jiang@artinchip.com>
+ * Desc: mpp thread interface
+ */
+
+#include "mpp_thread.h"
+#include "mpp_log.h"
+#include "mpp_mem.h"
+#include <inttypes.h>
+#include <string.h>
+#include <unistd.h>
+#include "rtconfig.h"
+
+#if defined(KERNEL_RTTHREAD) || \
+    defined(KERNEL_FREERTOS) || \
+    defined(KERNEL_BAREMETAL) || \
+    defined(KERNEL_RHINO) || \
+    defined(KERNEL_UCOS_II)
+#define MPP_USING_AIC_OSAL
+#endif
+
+
+#if defined(MPP_USING_AIC_OSAL)
+#include "aic_osal.h"
+struct mpp_sem_handle {
+    aicos_sem_t sem;
+};
+#else
+#include <pthread.h>
+struct mpp_sem_handle {
+    pthread_mutex_t mutex;
+    pthread_cond_t cond;
+    pthread_condattr_t condattr;
+    int need_signal;
+};
+#endif
+
+mpp_sem_t mpp_sem_create()
+{
+    struct mpp_sem_handle *h_sem =
+        (struct mpp_sem_handle *)mpp_alloc(sizeof(struct mpp_sem_handle));
+    if (h_sem == NULL)
+        return NULL;
+    memset(h_sem, 0, sizeof(struct mpp_sem_handle));
+
+#if defined(MPP_USING_AIC_OSAL)
+    h_sem->sem = aicos_sem_create(0);
+    if (!h_sem->sem) {
+        loge("aicos_sem_create err!!!\n");
+        return NULL;
+    }
+#else
+    h_sem->need_signal = 0;
+    if (pthread_mutex_init(&h_sem->mutex, NULL) != 0) {
+        loge("pthread_mutex_init err!!!\n");
+        return NULL;
+    }
+
+    pthread_condattr_init(&h_sem->condattr);
+    pthread_condattr_setclock(&h_sem->condattr, CLOCK_REALTIME);
+    pthread_cond_init(&h_sem->cond, &h_sem->condattr);
+#endif
+    return (mpp_sem_t)h_sem;
+}
+
+void mpp_sem_delete(mpp_sem_t sem)
+{
+    if (sem) {
+        struct mpp_sem_handle *h_sem = (struct mpp_sem_handle *)sem;
+#if defined(MPP_USING_AIC_OSAL)
+        aicos_sem_delete(h_sem->sem);
+        h_sem->sem = NULL;
+#else
+        pthread_condattr_destroy(&h_sem->condattr);
+        pthread_cond_destroy(&h_sem->cond);
+        pthread_mutex_destroy(&h_sem->mutex);
+#endif
+        mpp_free(sem);
+    }
+}
+
+int mpp_sem_signal(mpp_sem_t sem)
+{
+    if (!sem)
+        return -1;
+    struct mpp_sem_handle *h_sem = (struct mpp_sem_handle *)sem;
+#if defined(MPP_USING_AIC_OSAL)
+    aicos_sem_give(h_sem->sem);
+#else
+    pthread_mutex_lock(&h_sem->mutex);
+    if (h_sem->need_signal) {
+        pthread_cond_signal(&h_sem->cond);
+    }
+    pthread_mutex_unlock(&h_sem->mutex);
+#endif
+    return 0;
+}
+
+int mpp_sem_wait(mpp_sem_t sem, int64_t us)
+{
+    if (!sem)
+        return -1;
+
+    struct mpp_sem_handle *h_sem = (struct mpp_sem_handle *)sem;
+
+#if defined(MPP_USING_AIC_OSAL)
+    if (us == MPP_WAIT_FOREVER)
+        aicos_sem_take(h_sem->sem, AICOS_WAIT_FOREVER);
+    else
+        aicos_sem_take(h_sem->sem, (unsigned int)(us * 1000));
+    return 0;
+#else
+    int ret;
+    pthread_mutex_lock(&h_sem->mutex);
+    if (us == MPP_WAIT_FOREVER) {
+        h_sem->need_signal = 1;
+        ret = pthread_cond_wait(&h_sem->cond, &h_sem->mutex);
+        h_sem->need_signal = 0;
+    } else {
+        struct timespec out_time;
+        uint64_t tmp;
+        ret = clock_gettime(CLOCK_REALTIME, &out_time);
+        out_time.tv_sec += us / (1 * 1000 * 1000);
+        tmp = out_time.tv_nsec / 1000 + us % (1 * 1000 * 1000);
+        out_time.tv_sec += tmp / (1 * 1000 * 1000);
+        tmp = tmp % (1 * 1000 * 1000);
+        out_time.tv_nsec = tmp * 1000;
+        h_sem->need_signal = 1;
+        ret = pthread_cond_timedwait(&h_sem->cond, &h_sem->mutex, &out_time);
+        h_sem->need_signal = 0;
+    }
+
+    pthread_mutex_unlock(&h_sem->mutex);
+    return ret;
+#endif
+}