import

Kernel/SingleCore/os_countdownlatch.h

@@ -0,0 +1,85 @@
+#ifndef INCLUDED_OS_COUNTDOWNLATCH_H
+#define INCLUDED_OS_COUNTDOWNLATCH_H
+
+#ifndef INCLUDED_OS_MUTEX_H
+#include <os_mutex.h>
+#endif /*INCLUDED_OS_MUTEX_H*/
+
+#ifndef INCLUDED_OS_CONDV_H
+#include <os_condv.h>
+#endif /*INCLUDED_OS_CONDV_H*/
+
+/* ------------------------------------------------------------------------------------------------------------------ */
+/*  */
+
+typedef struct {
+    os_mutex_t lock;
+    os_condv_t condv;
+    os_int_t count;
+}os_countdownlatch_t;
+
+/* ------------------------------------------------------------------------------------------------------------------ */
+/*  */
+
+OS_STATIC_FORCE_INLINE
+void os_countdownlatch_init(os_countdownlatch_t* self, os_int_t cnt){
+    os_mutex_init(&self->lock);
+    os_condv_init(&self->condv);
+    self->count = cnt;
+}
+
+OS_STATIC_FORCE_INLINE
+void os_countdownlatch_destroy(os_countdownlatch_t* self){
+    os_condv_destroy(&self->condv);
+    os_mutex_destroy(&self->lock);
+}
+
+OS_STATIC_FORCE_INLINE
+void os_countdownlatch_await(os_countdownlatch_t* self){
+    os_mutex_lock(&self->lock);
+    while(self->count>0){
+        os_condv_wait(&self->condv, &self->lock);
+    }
+    os_mutex_unlock(&self->lock);
+}
+
+OS_STATIC_FORCE_INLINE
+os_err_t os_countdownlatch_timed_await(os_countdownlatch_t* self, os_tick_t ticks){
+    os_err_t err = OS_ERR_OK;
+    os_mutex_lock(&self->lock);
+    while(self->count>0){
+        err = os_condv_timed_wait(&self->condv, &self->lock, ticks);
+        if(err==OS_ERR_TIMEOUT){
+            break;
+        }
+    }
+    os_mutex_unlock(&self->lock);
+    return err;
+}
+
+OS_STATIC_FORCE_INLINE
+void os_countdownlatch_count_down(os_countdownlatch_t* self){
+    os_mutex_lock(&self->lock);
+    if(self->count>0){
+        self->count--;
+        if(self->count==0){
+            os_mutex_unlock(&self->lock);
+            os_condv_notify_all(&self->condv);
+            return;
+        }
+    }
+    os_mutex_unlock(&self->lock);
+}
+
+OS_STATIC_FORCE_INLINE
+os_int_t os_countdownlatch_get_count(os_countdownlatch_t* self){
+    os_int_t count;
+    os_mutex_lock(&self->lock);
+    count = self->count;
+    os_mutex_unlock(&self->lock);
+    return count;
+}
+
+
+
+#endif /*INCLUDED_OS_COUNTDOWNLATCH_H*/