1 files changed, 75 insertions, 0 deletions
diff --git a/tests/profiling/gatordmock/DirectoryCaptureCommandHandler.hpp b/tests/profiling/gatordmock/DirectoryCaptureCommandHandler.hpp
new file mode 100644
index 0000000000..4cf96be741
--- /dev/null
+++ b/tests/profiling/gatordmock/DirectoryCaptureCommandHandler.hpp
@@ -0,0 +1,75 @@
+//
+// Copyright © 2019 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include "CounterDirectory.hpp"
+#include "GatordMockService.hpp"
+#include "MockUtils.hpp"
+
+
+#include "Packet.hpp"
+#include "CommandHandlerFunctor.hpp"
+#include "SendCounterPacket.hpp"
+#include "IPeriodicCounterCapture.hpp"
+
+
+#include <vector>
+#include <thread>
+#include <functional>
+
+namespace armnn
+{
+
+namespace gatordmock
+{
+
+class DirectoryCaptureCommandHandler : public profiling::CommandHandlerFunctor
+{
+
+public:
+
+    DirectoryCaptureCommandHandler(uint32_t familyId,
+                                   uint32_t packetId,
+                                   uint32_t version,
+                                   bool quietOperation = false)
+            : CommandHandlerFunctor(familyId, packetId, version)
+            , m_QuietOperation(quietOperation)
+            , m_CounterDirectoryCount(0)
+             {}
+
+    void operator()(const armnn::profiling::Packet &packet) override;
+
+    CounterDirectory GetCounterDirectory() const;
+    uint32_t GetCounterDirectoryCount() const;
+
+private:
+    void ParseData(const armnn::profiling::Packet &packet);
+
+    std::vector<CategoryRecord> ReadCategoryRecords(const unsigned char *const data,
+                                              uint32_t offset,
+                                              std::vector<uint32_t> categoryOffsets);
+
+    std::vector<CounterSetRecord> ReadCounterSetRecords(const unsigned char *const data,
+                                              uint32_t offset,
+                                              std::vector<uint32_t> eventRecordsOffsets);
+
+    std::vector<DeviceRecord> ReadDeviceRecords(const unsigned char *const data,
+                                              uint32_t offset,
+                                              std::vector<uint32_t> eventRecordsOffsets);
+
+    std::vector<EventRecord> ReadEventRecords(const unsigned char *const data,
+                                              uint32_t offset,
+                                              std::vector<uint32_t> eventRecordsOffsets);
+
+    CounterDirectory m_CounterDirectory;
+
+    bool m_QuietOperation;
+    std::atomic<uint32_t> m_CounterDirectoryCount;
+};
+
+} // namespace gatordmock
+
+} // namespace armnn