added ringbuffer class

srslte/include/srslte/utils/ringbuffer.h

@@ -0,0 +1,37 @@
+
+#ifndef RINGBUFFER_H
+#define RINGBUFFER_H
+
+#include "srslte/config.h"
+#include <pthread.h>
+#include <stdint.h>
+
+typedef struct {
+  uint8_t *buffer; 
+  int capacity; 
+  int count; 
+  int wpm; 
+  int rpm; 
+  pthread_mutex_t mutex; 
+  pthread_cond_t  cvar; 
+} srslte_ringbuffer_t; 
+
+
+SRSLTE_API int srslte_ringbuffer_init(srslte_ringbuffer_t *q, 
+                                       int capacity);
+
+SRSLTE_API void srslte_ringbuffer_free(srslte_ringbuffer_t *q, 
+                                       int capacity);
+
+SRSLTE_API int srslte_ringbuffer_write(srslte_ringbuffer_t *q, 
+                                       uint8_t *ptr, 
+                                       int nof_bytes);
+
+SRSLTE_API int srslte_ringbuffer_read(srslte_ringbuffer_t *q, 
+                                       uint8_t *ptr, 
+                                       int nof_bytes);
+
+
+#endif
+
+

srslte/lib/utils/ringbuffer.c

@@ -0,0 +1,85 @@
+
+#include <string.h>
+#include <stdlib.h>
+
+#include "srslte/utils/ringbuffer.h"
+#include "srslte/utils/vector.h"
+
+int srslte_ringbuffer_init(srslte_ringbuffer_t *q, int capacity)
+{
+  q->buffer = srslte_vec_malloc(capacity);
+  if (!q->buffer) {
+    return -1; 
+  }
+  q->capacity   = capacity; 
+  q->count      = 0; 
+  q->wpm        = 0; 
+  q->rpm        = 0; 
+  
+  pthread_mutex_init(&q->mutex, NULL); 
+  pthread_cond_init(&q->cvar, NULL);
+  
+  return 0; 
+}
+
+void srslte_ringbuffer_free(srslte_ringbuffer_t *q, int capacity)
+{
+  if (q) {
+    if (q->buffer) {
+      free(q->buffer);
+      q->buffer = NULL; 
+    }
+    pthread_mutex_destroy(&q->mutex); 
+    pthread_cond_destroy(&q->cvar);
+  }
+}
+
+int srslte_ringbuffer_write(srslte_ringbuffer_t *q, uint8_t *ptr, int nof_bytes)
+{  
+  int w_bytes = nof_bytes;
+  pthread_mutex_lock(&q->mutex);
+  if (q->count + w_bytes >= q->capacity) {
+    w_bytes = q->capacity - q->count; 
+    fprintf(stderr, "Buffer overrun: lost %d bytes\n", nof_bytes - w_bytes);
+  }
+  if (w_bytes > q->capacity - q->wpm) {
+    int x = q->capacity - q->wpm; 
+    memcpy(&q->buffer[q->wpm], ptr, x);    
+    memcpy(q->buffer, &ptr[x], w_bytes - x);    
+  } else {
+    memcpy(&q->buffer[q->wpm], ptr, w_bytes);    
+  }
+  q->wpm += w_bytes; 
+  if (q->wpm >= q->capacity) {
+    q->wpm -= q->capacity; 
+  }
+  q->count += w_bytes; 
+  pthread_cond_broadcast(&q->cvar);
+  pthread_mutex_unlock(&q->mutex);
+  return w_bytes; 
+}
+
+int srslte_ringbuffer_read(srslte_ringbuffer_t *q, uint8_t *ptr, int nof_bytes)
+{
+  pthread_mutex_lock(&q->mutex);
+  while(q->count < nof_bytes) {
+    pthread_cond_wait(&q->cvar, &q->mutex);
+  }
+  if (nof_bytes + q->rpm > q->capacity) {
+    int x = q->capacity - q->rpm; 
+    memcpy(ptr, &q->buffer[q->rpm], x);
+    memcpy(&ptr[x], q->buffer, nof_bytes - x);
+  } else {         
+    memcpy(ptr, &q->buffer[q->rpm], nof_bytes);
+  }
+  q->rpm += nof_bytes; 
+  if (q->rpm >= q->capacity) {
+    q->rpm -= q->capacity; 
+  }
+  q->count -= nof_bytes; 
+  pthread_mutex_unlock(&q->mutex);
+  return nof_bytes; 
+}
+
+
+