[libcamera-devel] [PATCH v7 1/3] libcamera: Move DmaHeap to HeapAllocator as a base class
Harvey Yang
chenghaoyang at chromium.org
Wed Aug 2 08:51:43 CEST 2023
As other components (like the WIP virtual pipeline handler) also need a
heap allocator, move DmaHeap from raspberry pi pipeline handler to a
general HeapAllocator as a base class.
Signed-off-by: Harvey Yang <chenghaoyang at chromium.org>
---
.../libcamera/internal/heap_allocator.h | 18 +-
include/libcamera/internal/meson.build | 1 +
src/libcamera/heap_allocator.cpp | 192 ++++++++++++++++++
src/libcamera/meson.build | 1 +
src/libcamera/pipeline/rpi/vc4/dma_heaps.cpp | 90 --------
src/libcamera/pipeline/rpi/vc4/meson.build | 5 +-
src/libcamera/pipeline/rpi/vc4/vc4.cpp | 11 +-
7 files changed, 209 insertions(+), 109 deletions(-)
rename src/libcamera/pipeline/rpi/vc4/dma_heaps.h => include/libcamera/internal/heap_allocator.h (57%)
create mode 100644 src/libcamera/heap_allocator.cpp
delete mode 100644 src/libcamera/pipeline/rpi/vc4/dma_heaps.cpp
diff --git a/src/libcamera/pipeline/rpi/vc4/dma_heaps.h b/include/libcamera/internal/heap_allocator.h
similarity index 57%
rename from src/libcamera/pipeline/rpi/vc4/dma_heaps.h
rename to include/libcamera/internal/heap_allocator.h
index 0a4a8d86..6620fa23 100644
--- a/src/libcamera/pipeline/rpi/vc4/dma_heaps.h
+++ b/include/libcamera/internal/heap_allocator.h
@@ -1,8 +1,9 @@
/* SPDX-License-Identifier: LGPL-2.1-or-later */
/*
* Copyright (C) 2020, Raspberry Pi Ltd
+ * Copyright (C) 2023, Google Inc.
*
- * dma_heaps.h - Helper class for dma-heap allocations.
+ * heap_allocator.h - Helper class for heap buffer allocations.
*/
#pragma once
@@ -13,20 +14,19 @@
namespace libcamera {
-namespace RPi {
+class Heap;
-class DmaHeap
+class HeapAllocator
{
public:
- DmaHeap();
- ~DmaHeap();
- bool isValid() const { return dmaHeapHandle_.isValid(); }
+ HeapAllocator();
+ ~HeapAllocator();
+
+ bool isValid() const;
UniqueFD alloc(const char *name, std::size_t size);
private:
- UniqueFD dmaHeapHandle_;
+ std::unique_ptr<Heap> heap_;
};
-} /* namespace RPi */
-
} /* namespace libcamera */
diff --git a/include/libcamera/internal/meson.build b/include/libcamera/internal/meson.build
index 7f1f3440..3030d039 100644
--- a/include/libcamera/internal/meson.build
+++ b/include/libcamera/internal/meson.build
@@ -27,6 +27,7 @@ libcamera_internal_headers = files([
'device_enumerator_udev.h',
'formats.h',
'framebuffer.h',
+ 'heap_allocator.h',
'ipa_manager.h',
'ipa_module.h',
'ipa_proxy.h',
diff --git a/src/libcamera/heap_allocator.cpp b/src/libcamera/heap_allocator.cpp
new file mode 100644
index 00000000..2a825d93
--- /dev/null
+++ b/src/libcamera/heap_allocator.cpp
@@ -0,0 +1,192 @@
+/* SPDX-License-Identifier: LGPL-2.1-or-later */
+/*
+ * Copyright (C) 2020, Raspberry Pi Ltd
+ * Copyright (C) 2023, Google Inc.
+ *
+ * heap_allocator.cpp - Helper class for heap buffer allocations.
+ */
+
+#include "libcamera/internal/heap_allocator.h"
+
+#include <array>
+#include <fcntl.h>
+#include <sys/ioctl.h>
+#include <unistd.h>
+
+#include <linux/dma-buf.h>
+#include <linux/dma-heap.h>
+
+#include <libcamera/base/log.h>
+
+/**
+ * \file libcamera/internal/heap_allocator.h
+ * \brief Internal heap buffer allocator support
+ */
+
+namespace libcamera {
+
+/*
+ * /dev/dma-heap/linux,cma is the dma-heap allocator, which allows dmaheap-cma
+ * to only have to worry about importing.
+ *
+ * Annoyingly, should the cma heap size be specified on the kernel command line
+ * instead of DT, the heap gets named "reserved" instead.
+ */
+static constexpr std::array<const char *, 3> dmaHeapNames = {
+ "/dev/dma_heap/linux,cma",
+ "/dev/dma_heap/reserved",
+ "/dev/dma_heap/system"
+};
+
+LOG_DEFINE_CATEGORY(HeapAllocator)
+
+/**
+ * \class Heap
+ * \brief The virtual base class for different implementations
+ */
+class Heap
+{
+public:
+ virtual ~Heap() = default;
+ bool isValid() const { return handle_.isValid(); }
+ virtual UniqueFD alloc(const char *name, std::size_t size) = 0;
+
+protected:
+ UniqueFD handle_;
+};
+
+/**
+ * \var Heap::handle_
+ * \brief A common UniqueFD to allocate buffers
+ */
+
+/**
+ * \fn Heap::isValid()
+ * \brief Return true if the heap is valid
+ */
+
+/**
+ * \fn Heap::alloc()
+ * \brief Return a valid UniqueFD if a heap buffer is allocated
+ * \param[in] name The name of the allocated buffer
+ * \param[in] size The size in byte of the allocated buffer
+ */
+
+/**
+ * \class DmaHeap
+ * \brief The derived class of Heap with DMA implementation
+ */
+class DmaHeap : public Heap
+{
+public:
+ DmaHeap();
+ ~DmaHeap();
+ UniqueFD alloc(const char *name, std::size_t size) override;
+};
+
+/**
+ * \brief Construct a DmaHeap with a list of |dmaHeapNames|
+ */
+DmaHeap::DmaHeap()
+{
+ for (const char *name : dmaHeapNames) {
+ int ret = ::open(name, O_RDWR | O_CLOEXEC);
+ if (ret < 0) {
+ ret = errno;
+ LOG(HeapAllocator, Debug) << "DmaHeap failed to open " << name << ": "
+ << strerror(ret);
+ continue;
+ }
+
+ handle_ = UniqueFD(ret);
+ break;
+ }
+
+ if (!handle_.isValid())
+ LOG(HeapAllocator, Error) << "DmaHeap could not open any dmaHeap device";
+}
+
+DmaHeap::~DmaHeap() = default;
+
+UniqueFD DmaHeap::alloc(const char *name, std::size_t size)
+{
+ int ret;
+
+ if (!name)
+ return {};
+
+ struct dma_heap_allocation_data alloc = {};
+
+ alloc.len = size;
+ alloc.fd_flags = O_CLOEXEC | O_RDWR;
+
+ ret = ::ioctl(handle_.get(), DMA_HEAP_IOCTL_ALLOC, &alloc);
+ if (ret < 0) {
+ LOG(HeapAllocator, Error) << "DmaHeap allocation failure for "
+ << name;
+ return {};
+ }
+
+ UniqueFD allocFd(alloc.fd);
+ ret = ::ioctl(allocFd.get(), DMA_BUF_SET_NAME, name);
+ if (ret < 0) {
+ LOG(HeapAllocator, Error) << "DmaHeap naming failure for "
+ << name;
+ return {};
+ }
+
+ return allocFd;
+}
+
+/**
+ * \class HeapAllocator
+ * \brief The allocator that allocates heap buffers
+ *
+ * HeapAllocator class is the interface for pipeline handlers to allocate heap
+ * buffers. It finds a usable heap implementation, if any, and users don't need
+ * to know the details. Users should check if the HeapAllocator instance is
+ * valid though before using it.
+ */
+
+/**
+ * \var HeapAllocator::heap_
+ * \brief The heap instance used to allocate heap buffers.
+ * Should be initialized in the c'tor.
+ */
+
+/**
+ * \brief Construct a HeapAllocator with DmaHeap implementation
+ */
+HeapAllocator::HeapAllocator()
+{
+ heap_ = std::make_unique<DmaHeap>();
+}
+
+HeapAllocator::~HeapAllocator() = default;
+
+/**
+ * \fn HeapAllocator::isValid()
+ * \brief Return true if the HeapAllocator works with a Heap implementation
+ */
+bool HeapAllocator::isValid() const
+{
+ return heap_->isValid();
+}
+
+/**
+ * \fn HeapAllocator::alloc()
+ * \brief Return a valid UniqueFD if a heap buffer is allocated
+ * \param[in] name The name of the allocated buffer
+ * \param[in] size The size in byte of the allocated buffer
+ */
+UniqueFD HeapAllocator::alloc(const char *name, std::size_t size)
+{
+ if (!isValid()) {
+ LOG(HeapAllocator, Fatal) << "Allocation attempted without allocator" << name;
+ return {};
+ }
+
+ return heap_->alloc(name, size);
+}
+
+} /* namespace libcamera */
diff --git a/src/libcamera/meson.build b/src/libcamera/meson.build
index b24f8296..967f08e6 100644
--- a/src/libcamera/meson.build
+++ b/src/libcamera/meson.build
@@ -22,6 +22,7 @@ libcamera_sources = files([
'framebuffer.cpp',
'framebuffer_allocator.cpp',
'geometry.cpp',
+ 'heap_allocator.cpp',
'ipa_controls.cpp',
'ipa_data_serializer.cpp',
'ipa_interface.cpp',
diff --git a/src/libcamera/pipeline/rpi/vc4/dma_heaps.cpp b/src/libcamera/pipeline/rpi/vc4/dma_heaps.cpp
deleted file mode 100644
index 317b1fc1..00000000
--- a/src/libcamera/pipeline/rpi/vc4/dma_heaps.cpp
+++ /dev/null
@@ -1,90 +0,0 @@
-/* SPDX-License-Identifier: LGPL-2.1-or-later */
-/*
- * Copyright (C) 2020, Raspberry Pi Ltd
- *
- * dma_heaps.h - Helper class for dma-heap allocations.
- */
-
-#include "dma_heaps.h"
-
-#include <array>
-#include <fcntl.h>
-#include <linux/dma-buf.h>
-#include <linux/dma-heap.h>
-#include <sys/ioctl.h>
-#include <unistd.h>
-
-#include <libcamera/base/log.h>
-
-/*
- * /dev/dma-heap/linux,cma is the dma-heap allocator, which allows dmaheap-cma
- * to only have to worry about importing.
- *
- * Annoyingly, should the cma heap size be specified on the kernel command line
- * instead of DT, the heap gets named "reserved" instead.
- */
-static constexpr std::array<const char *, 2> heapNames = {
- "/dev/dma_heap/linux,cma",
- "/dev/dma_heap/reserved"
-};
-
-namespace libcamera {
-
-LOG_DECLARE_CATEGORY(RPI)
-
-namespace RPi {
-
-DmaHeap::DmaHeap()
-{
- for (const char *name : heapNames) {
- int ret = ::open(name, O_RDWR | O_CLOEXEC, 0);
- if (ret < 0) {
- ret = errno;
- LOG(RPI, Debug) << "Failed to open " << name << ": "
- << strerror(ret);
- continue;
- }
-
- dmaHeapHandle_ = UniqueFD(ret);
- break;
- }
-
- if (!dmaHeapHandle_.isValid())
- LOG(RPI, Error) << "Could not open any dmaHeap device";
-}
-
-DmaHeap::~DmaHeap() = default;
-
-UniqueFD DmaHeap::alloc(const char *name, std::size_t size)
-{
- int ret;
-
- if (!name)
- return {};
-
- struct dma_heap_allocation_data alloc = {};
-
- alloc.len = size;
- alloc.fd_flags = O_CLOEXEC | O_RDWR;
-
- ret = ::ioctl(dmaHeapHandle_.get(), DMA_HEAP_IOCTL_ALLOC, &alloc);
- if (ret < 0) {
- LOG(RPI, Error) << "dmaHeap allocation failure for "
- << name;
- return {};
- }
-
- UniqueFD allocFd(alloc.fd);
- ret = ::ioctl(allocFd.get(), DMA_BUF_SET_NAME, name);
- if (ret < 0) {
- LOG(RPI, Error) << "dmaHeap naming failure for "
- << name;
- return {};
- }
-
- return allocFd;
-}
-
-} /* namespace RPi */
-
-} /* namespace libcamera */
diff --git a/src/libcamera/pipeline/rpi/vc4/meson.build b/src/libcamera/pipeline/rpi/vc4/meson.build
index cdb049c5..b2b79735 100644
--- a/src/libcamera/pipeline/rpi/vc4/meson.build
+++ b/src/libcamera/pipeline/rpi/vc4/meson.build
@@ -1,8 +1,5 @@
# SPDX-License-Identifier: CC0-1.0
-libcamera_sources += files([
- 'dma_heaps.cpp',
- 'vc4.cpp',
-])
+libcamera_sources += files(['vc4.cpp'])
subdir('data')
diff --git a/src/libcamera/pipeline/rpi/vc4/vc4.cpp b/src/libcamera/pipeline/rpi/vc4/vc4.cpp
index 018cf488..410ecfaf 100644
--- a/src/libcamera/pipeline/rpi/vc4/vc4.cpp
+++ b/src/libcamera/pipeline/rpi/vc4/vc4.cpp
@@ -12,12 +12,11 @@
#include <libcamera/formats.h>
#include "libcamera/internal/device_enumerator.h"
+#include "libcamera/internal/heap_allocator.h"
#include "../common/pipeline_base.h"
#include "../common/rpi_stream.h"
-#include "dma_heaps.h"
-
using namespace std::chrono_literals;
namespace libcamera {
@@ -87,7 +86,7 @@ public:
RPi::Device<Isp, 4> isp_;
/* DMAHEAP allocation helper. */
- RPi::DmaHeap dmaHeap_;
+ HeapAllocator heapAllocator_;
SharedFD lsTable_;
struct Config {
@@ -296,7 +295,7 @@ int PipelineHandlerVc4::platformRegister(std::unique_ptr<RPi::CameraData> &camer
{
Vc4CameraData *data = static_cast<Vc4CameraData *>(cameraData.get());
- if (!data->dmaHeap_.isValid())
+ if (!data->heapAllocator_.isValid())
return -ENOMEM;
MediaEntity *unicamImage = unicam->getEntityByName("unicam-image");
@@ -670,9 +669,9 @@ int Vc4CameraData::platformConfigureIpa(ipa::RPi::ConfigParams ¶ms)
{
params.ispControls = isp_[Isp::Input].dev()->controls();
- /* Allocate the lens shading table via dmaHeap and pass to the IPA. */
+ /* Allocate the lens shading table via heapAllocator and pass to the IPA. */
if (!lsTable_.isValid()) {
- lsTable_ = SharedFD(dmaHeap_.alloc("ls_grid", ipa::RPi::MaxLsGridSize));
+ lsTable_ = SharedFD(heapAllocator_.alloc("ls_grid", ipa::RPi::MaxLsGridSize));
if (!lsTable_.isValid())
return -ENOMEM;
--
2.41.0.585.gd2178a4bd4-goog
More information about the libcamera-devel
mailing list