diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2019-08-16 17:45:07 +0100 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2019-08-16 17:45:12 +0100 |
commit | 94d7915bef33ad59d1bdfa791490268c682c5359 (patch) | |
tree | 98a1732cd744bf79d4165e5745f95c1cae5075fe /src/profiling/CommandHandlerRegistry.hpp | |
parent | 8ab53f05071b020ab864b82d2edfefb28e427ec8 (diff) | |
download | armnn-94d7915bef33ad59d1bdfa791490268c682c5359.tar.gz |
IVGCVSW-3550 Create Command Handler Registry
Change-Id: I51e34068d79ba660ae2f16b22ad2bb8191d473fa
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Diffstat (limited to 'src/profiling/CommandHandlerRegistry.hpp')
-rw-r--r-- | src/profiling/CommandHandlerRegistry.hpp | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/src/profiling/CommandHandlerRegistry.hpp b/src/profiling/CommandHandlerRegistry.hpp new file mode 100644 index 0000000000..ba81f1790f --- /dev/null +++ b/src/profiling/CommandHandlerRegistry.hpp @@ -0,0 +1,36 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include "CommandHandlerFunctor.hpp" +#include "CommandHandlerKey.hpp" + +#include <boost/functional/hash.hpp> +#include <unordered_map> + +struct CommandHandlerHash +{ + std::size_t operator() (const CommandHandlerKey& commandHandlerKey) const + { + std::size_t seed = 0; + boost::hash_combine(seed, commandHandlerKey.GetPacketId()); + boost::hash_combine(seed, commandHandlerKey.GetVersion()); + return seed; + } +}; + +class CommandHandlerRegistry +{ +public: + CommandHandlerRegistry() = default; + + void RegisterFunctor(CommandHandlerFunctor* functor, uint32_t packetId, uint32_t version); + + CommandHandlerFunctor* GetFunctor(uint32_t packetId, uint32_t version) const; + +private: + std::unordered_map<CommandHandlerKey, CommandHandlerFunctor*, CommandHandlerHash> registry; +};
\ No newline at end of file |