From f6d06421090f88094aba76b72b04d614f54efafa Mon Sep 17 00:00:00 2001
From: jacqueline <me@jacqueline.id.au>
Date: Tue, 26 Sep 2023 11:23:53 +1000
Subject: Prepare for PMR with a memory_resource that understands
 heap_caps_malloc

---
 src/memory/include/memory_resource.hpp | 42 ++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 src/memory/include/memory_resource.hpp

(limited to 'src/memory/include/memory_resource.hpp')

diff --git a/src/memory/include/memory_resource.hpp b/src/memory/include/memory_resource.hpp
new file mode 100644
index 00000000..e1d12b2a
--- /dev/null
+++ b/src/memory/include/memory_resource.hpp
@@ -0,0 +1,42 @@
+/*
+ * Copyright 2023 jacqueline <me@jacqueline.id.au>
+ *
+ * SPDX-License-Identifier: GPL-3.0-only
+ */
+
+#pragma once
+
+#include <memory_resource>
+#include <string>
+
+#include <esp_heap_caps.h>
+#include <stdint.h>
+
+namespace memory {
+
+enum class Capabilities : uint32_t {
+  kDefault = MALLOC_CAP_DEFAULT,
+  kInternal = MALLOC_CAP_INTERNAL | MALLOC_CAP_8BIT,
+  kDma = MALLOC_CAP_DMA,
+  kSpiRam = MALLOC_CAP_SPIRAM,
+};
+
+class Resource : public std::pmr::memory_resource {
+ public:
+  explicit Resource(Capabilities caps) : caps_(caps) {}
+
+ private:
+  Capabilities caps_;
+
+  void* do_allocate(std::size_t bytes, std::size_t alignment) override;
+
+  void do_deallocate(void* p,
+                     std::size_t bytes,
+                     std::size_t alignment) override;
+
+  bool do_is_equal(const std::pmr::memory_resource& other) const noexcept override;
+};
+
+extern Resource kSpiRamResource;
+
+}  // namespace memory
-- 
cgit v1.2.3