aboutsummaryrefslogtreecommitdiff
path: root/profiling/client/src/BufferManager.hpp
diff options
context:
space:
mode:
authorJim Flynn <jim.flynn@arm.com>2022-03-23 23:01:26 +0000
committerJim Flynn <jim.flynn@arm.com>2022-03-23 23:43:35 +0000
commit3e9bc19ad523361e6b18057849e30c0c48183915 (patch)
treeb7b012a9734ce39d054fc5d92302780fd838e5c8 /profiling/client/src/BufferManager.hpp
parent277618302d0f131eac0b6ac2015dd3eb09aa6ff9 (diff)
downloadarmnn-3e9bc19ad523361e6b18057849e30c0c48183915.tar.gz
IVGCVSW-6706 Create the libpipeClient library
Change-Id: I2368aade38ad3808fab55d8a86cd659d4e95d91e Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'profiling/client/src/BufferManager.hpp')
-rw-r--r--profiling/client/src/BufferManager.hpp76
1 files changed, 76 insertions, 0 deletions
diff --git a/profiling/client/src/BufferManager.hpp b/profiling/client/src/BufferManager.hpp
new file mode 100644
index 0000000000..0ab3e0e534
--- /dev/null
+++ b/profiling/client/src/BufferManager.hpp
@@ -0,0 +1,76 @@
+//
+// Copyright © 2019 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include "IBufferManager.hpp"
+#include "IConsumer.hpp"
+
+#include <condition_variable>
+#include <mutex>
+#include <vector>
+#include <queue>
+
+namespace arm
+{
+
+namespace pipe
+{
+
+class BufferManager : public IBufferManager
+{
+public:
+ BufferManager(unsigned int numberOfBuffers = 5, unsigned int maxPacketSize = 4096);
+
+ ~BufferManager() {}
+
+ IPacketBufferPtr Reserve(unsigned int requestedSize, unsigned int& reservedSize) override;
+
+ void Reset();
+
+ void Commit(IPacketBufferPtr& packetBuffer, unsigned int size, bool notifyConsumer = true) override;
+
+ void Release(IPacketBufferPtr& packetBuffer) override;
+
+ IPacketBufferPtr GetReadableBuffer() override;
+
+ void MarkRead(IPacketBufferPtr& packetBuffer) override;
+
+ /// Set Consumer on the buffer manager to be notified when there is a Commit
+ /// Can only be one consumer
+ void SetConsumer(IConsumer* consumer) override;
+
+ /// Notify the Consumer buffer can be read
+ void FlushReadList() override;
+
+private:
+ void Initialize();
+
+ // Maximum buffer size
+ unsigned int m_MaxBufferSize;
+ // Number of buffers
+ const unsigned int m_NumberOfBuffers;
+ const unsigned int m_MaxNumberOfBuffers;
+ unsigned int m_CurrentNumberOfBuffers;
+
+ // List of available packet buffers
+ std::vector<IPacketBufferPtr> m_AvailableList;
+
+ // List of readable packet buffers
+ std::queue<IPacketBufferPtr> m_ReadableList;
+
+ // Mutex for available packet buffer list
+ std::mutex m_AvailableMutex;
+
+ // Mutex for readable packet buffer list
+ std::mutex m_ReadableMutex;
+
+ // Consumer thread to notify packet is ready to read
+ IConsumer* m_Consumer = nullptr;
+};
+
+} // namespace pipe
+
+} // namespace arm