IVGCVSW-4391 Add Global/Backend Counter ID map

Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I943e02bc2026564e3a19d03e81ba75850d204497

1 files changed, 50 insertions, 0 deletions

diff --git a/src/profiling/CounterIdMap.cpp b/src/profiling/CounterIdMap.cpp
new file mode 100644
index 0000000000..a925fb9cb4
--- /dev/null
+++ b/src/profiling/CounterIdMap.cpp
@@ -0,0 +1,50 @@
+//
+// Copyright © 2020 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+#include "CounterIdMap.hpp"
+#include "armnn/BackendId.hpp"
+#include <map>
+#include <armnn/Exceptions.hpp>
+
+namespace armnn
+{
+namespace profiling
+{
+
+void CounterIdMap::RegisterMapping(uint16_t globalCounterId,
+                                   uint16_t backendCounterId,
+                                   const armnn::BackendId& backendId)
+{
+    std::pair<uint16_t, armnn::BackendId> backendIdPair(backendCounterId, backendId);
+    m_GlobalCounterIdMap[globalCounterId] = backendIdPair;
+    m_BackendCounterIdMap[backendIdPair] = globalCounterId;
+}
+
+uint16_t CounterIdMap::GetGlobalId(uint16_t backendCounterId, const armnn::BackendId& backendId)
+{
+    std::pair<uint16_t, armnn::BackendId> backendIdPair(backendCounterId, backendId);
+    auto it = m_BackendCounterIdMap.find(backendIdPair);
+    if (it == m_BackendCounterIdMap.end())
+    {
+        std::stringstream ss;
+        ss << "No Backend Counter [" << backendIdPair.second << ":" << backendIdPair.first << "] registered";
+        throw armnn::Exception(ss.str());
+    }
+    return it->second;
+}
+
+const std::pair<uint16_t, armnn::BackendId>& CounterIdMap::GetBackendId(uint16_t globalCounterId)
+{
+    auto it = m_GlobalCounterIdMap.find(globalCounterId);
+    if (it == m_GlobalCounterIdMap.end())
+    {
+        std::stringstream ss;
+        ss << "No Global Counter ID [" << globalCounterId << "] registered";
+        throw armnn::Exception(ss.str());
+    }
+    return it->second;
+}
+
+}    // namespace profiling
+}    // namespace armnn
\ No newline at end of file