Allocator.h

Go to the documentation of this file.

#pragma once
#include <stdlib.h>
#include <memory>
#include <limits>
#include <assert.h>
#include <type_traits>
 
#ifdef ESP32
#include <esp_heap_caps.h>
#endif
#include "dlna_config.h"
 
namespace tiny_dlna {
 
template <typename T>
class AllocatorPSRAM {
 public:
  using value_type = T;
  using pointer = T*;
  using const_pointer = const T*;
  using reference = T&;
  using const_reference = const T&;
  using size_type = std::size_t;
  using difference_type = std::ptrdiff_t;
  using is_always_equal = std::true_type;
  using propagate_on_container_move_assignment = std::true_type;
 
  AllocatorPSRAM() noexcept {}
 
  template <typename U>
  AllocatorPSRAM(const AllocatorPSRAM<U>&) noexcept {}
 
  pointer allocate(size_type n) {
    if (n==0) return nullptr;
    // if (n > std::numeric_limits<size_type>::max() / sizeof(T))
    //     throw std::bad_alloc();
    // in Arduino excepitons are disabled!
    assert(n <= std::numeric_limits<size_type>::max() / sizeof(T));
 
#ifdef ESP32
    pointer p = static_cast<pointer>(
        heap_caps_malloc(n * sizeof(T), MALLOC_CAP_SPIRAM));
    if (p == nullptr) {
      p = static_cast<pointer>(malloc(n * sizeof(T)));
    }
    assert(p);
#else
    pointer p = static_cast<pointer>(malloc(n * sizeof(T)));
    if (!p) throw std::bad_alloc();
#endif
    return p;
  }
 
  void deallocate(pointer p, size_type) noexcept {
#ifdef ESP32
    if (p) heap_caps_free(p);
#else
    if (p) free(p);
#endif
  }
 
  template <typename U>
  struct rebind {
    using other = AllocatorPSRAM<U>;
  };
 
  // Allocators compare equal (stateless)
  template <typename U>
  bool operator==(const AllocatorPSRAM<U>&) const noexcept { return true; }
  template <typename U>
  bool operator!=(const AllocatorPSRAM<U>&) const noexcept { return false; }
};
 
// Alias for a std::string that stores its dynamic buffer in PSRAM
using stringPSRAM = std::basic_string<char, std::char_traits<char>, AllocatorPSRAM<char>>;
 
 
 
}  // namespace tiny_dlna