IVGCVSW-4391 Add Global/Backend Counter ID map
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I943e02bc2026564e3a19d03e81ba75850d204497
diff --git a/src/profiling/CounterIdMap.cpp b/src/profiling/CounterIdMap.cpp
new file mode 100644
index 0000000..a925fb9
--- /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