BufferRTOS.h

#pragma once
#include "AudioToolsConfig.h"
#include "AudioTools/CoreAudio/Buffers.h"
#include "AudioTools/CoreAudio/AudioLogger.h"
#include "AudioTools/CoreAudio/AudioBasic/Collections/Allocator.h"
 
#ifdef ESP32
#  include <freertos/stream_buffer.h>
#  include "freertos/FreeRTOS.h"
#elif defined(__linux__)
#else
#  include "FreeRTOS.h"
#  include "stream_buffer.h"
#endif
 
namespace audio_tools {
 
// #if ESP_IDF_VERSION_MAJOR >= 4
 
template <typename T>
class BufferRTOS : public BaseBuffer<T> {
 public:
  BufferRTOS(size_t streamBufferSize, size_t xTriggerLevel = 1,
             TickType_t writeMaxWait = portMAX_DELAY,
             TickType_t readMaxWait = portMAX_DELAY,
             Allocator &allocator = DefaultAllocator)
      : BaseBuffer<T>() {
    readWait = readMaxWait;
    writeWait = writeMaxWait;
    current_size_bytes = (streamBufferSize+1)  * sizeof(T);
    trigger_level = xTriggerLevel;
    p_allocator = &allocator;
 
    if (streamBufferSize > 0) {
      setup();
    }
  }
 
  ~BufferRTOS() { end(); }
 
  bool resize(size_t size) {
    bool result = true;
    int req_size_bytes = (size + 1)*sizeof(T);
    if (current_size_bytes != req_size_bytes) {
      end();
      current_size_bytes = req_size_bytes;
      result = setup();
    }
    return result;
  }
 
  void setReadMaxWait(TickType_t ticks) { readWait = ticks; }
 
  void setWriteMaxWait(TickType_t ticks) { writeWait = ticks; }
 
  void setWriteFromISR(bool active) { write_from_isr = active; }
 
  void setReadFromISR(bool active) { read_from_isr = active; }
 
  // reads a single value
  bool read(T &result) override {
    T data = 0;
    return readArray(&data, 1)==1;
  }
 
  // reads multiple values
  int readArray(T data[], int len) {
    if (xStreamBuffer == nullptr || data == nullptr || len == 0) return 0;
 
    if (read_from_isr) {
      xHigherPriorityTaskWoken = pdFALSE;
      int result = xStreamBufferReceiveFromISR(xStreamBuffer, (void *)data,
                                               sizeof(T) * len,
                                               &xHigherPriorityTaskWoken);
#ifdef ESP32X
      portYIELD_FROM_ISR();
#else
      portYIELD_FROM_ISR(xHigherPriorityTaskWoken);
#endif
      return result / sizeof(T);
    } else {
      return xStreamBufferReceive(xStreamBuffer, (void *)data, sizeof(T) * len,
                                  readWait) / sizeof(T);
    }
  }
 
  int writeArray(const T data[], int len) {
    LOGD("%s: %d", LOG_METHOD, len);
    if (xStreamBuffer == nullptr || data == nullptr || len == 0) return 0;
 
    if (write_from_isr) {
      xHigherPriorityTaskWoken = pdFALSE;
      int result =
          xStreamBufferSendFromISR(xStreamBuffer, (void *)data, sizeof(T) * len,
                                   &xHigherPriorityTaskWoken);
#ifdef ESP32X
      portYIELD_FROM_ISR();
#else
      portYIELD_FROM_ISR(xHigherPriorityTaskWoken);
#endif
      return result / sizeof(T);
    } else {
      return xStreamBufferSend(xStreamBuffer, (void *)data, sizeof(T) * len,
                               writeWait) / sizeof(T);
    }
  }
 
  // peeks the actual entry from the buffer
  bool peek(T &result) override {
    LOGE("peek not implemented");
    return false;
  }
 
  // checks if the buffer is full
  bool isFull() override {
    if (xStreamBuffer == nullptr) return true;
    return xStreamBufferIsFull(xStreamBuffer) == pdTRUE;
  }
 
  bool isEmpty() { 
    if (xStreamBuffer == nullptr) return true;
    return xStreamBufferIsEmpty(xStreamBuffer) == pdTRUE; 
  }
 
  // write add an entry to the buffer
  bool write(T data) override {
    int len = sizeof(T);
    return writeArray(&data, len) == len;
  }
 
  // clears the buffer
  void reset() override { 
    if (xStreamBuffer == nullptr) return;
    xStreamBufferReset(xStreamBuffer); 
  }
 
  // provides the number of entries that are available to read
  int available() override {
    if (xStreamBuffer == nullptr) return 0;
    return xStreamBufferBytesAvailable(xStreamBuffer) / sizeof(T);
  }
 
  // provides the number of entries that are available to write
  int availableForWrite() override {
    if (xStreamBuffer == nullptr) return 0;
    return xStreamBufferSpacesAvailable(xStreamBuffer) / sizeof(T);
  }
 
  // returns the address of the start of the physical read buffer
  T *address() override {
    LOGE("address() not implemented");
    return nullptr;
  }
 
  size_t size() { return current_size_bytes / sizeof(T); }
 
  operator bool() { return xStreamBuffer != nullptr && size()>0;}
 
 protected:
  StreamBufferHandle_t xStreamBuffer = nullptr;
  StaticStreamBuffer_t static_stream_buffer;
  uint8_t *p_data = nullptr;
  Allocator *p_allocator = nullptr;
  BaseType_t xHigherPriorityTaskWoken = pdFALSE;  // Initialised to pdFALSE.
  int readWait = portMAX_DELAY;
  int writeWait = portMAX_DELAY;
  bool read_from_isr = false;
  bool write_from_isr = false;
  size_t current_size_bytes = 0;
  size_t trigger_level = 0;
 
  bool setup() {
    if (current_size_bytes == 0) return true;
 
    // allocate data if necessary
    int size = (current_size_bytes + 1) * sizeof(T);
    if (p_data == nullptr) {
      p_data = (uint8_t *)p_allocator->allocate(size);
      // check allocation
      if (p_data == nullptr) {
        LOGE("allocate falied for %d bytes", size)
        return false;
      }
    }
 
 
    // create stream buffer if necessary
    if (xStreamBuffer == nullptr) {
      xStreamBuffer = xStreamBufferCreateStatic(current_size_bytes, trigger_level,
                                                p_data, &static_stream_buffer);
    }
    if (xStreamBuffer == nullptr) {
      LOGE("xStreamBufferCreateStatic failed");
      return false;
    }
    // make sure that the data is empty
    reset();
    return true;
  }
 
  void end() {
    if (xStreamBuffer != nullptr) vStreamBufferDelete(xStreamBuffer);
    if (p_data != nullptr) p_allocator->free(p_data);
    current_size_bytes = 0;
    p_data = nullptr;
    xStreamBuffer = nullptr;
  }
};
// #endif // ESP_IDF_VERSION_MAJOR >= 4
 
template <class T>
using SynchronizedBufferRTOS = BufferRTOS<T>;
 
}  // namespace audio_tools