diff options
Diffstat (limited to 'src/profiling/RequestCounterDirectoryCommandHandler.hpp')
-rw-r--r-- | src/profiling/RequestCounterDirectoryCommandHandler.hpp | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/src/profiling/RequestCounterDirectoryCommandHandler.hpp b/src/profiling/RequestCounterDirectoryCommandHandler.hpp index a03300af48..02bf64d17a 100644 --- a/src/profiling/RequestCounterDirectoryCommandHandler.hpp +++ b/src/profiling/RequestCounterDirectoryCommandHandler.hpp @@ -8,6 +8,7 @@ #include "CommandHandlerFunctor.hpp" #include "ISendCounterPacket.hpp" #include "Packet.hpp" +#include "ProfilingStateMachine.hpp" namespace armnn { @@ -19,23 +20,25 @@ class RequestCounterDirectoryCommandHandler : public CommandHandlerFunctor { public: - RequestCounterDirectoryCommandHandler(uint32_t packetId, uint32_t version, + RequestCounterDirectoryCommandHandler(uint32_t packetId, + uint32_t version, ICounterDirectory& counterDirectory, - ISendCounterPacket& sendCounterPacket) - : CommandHandlerFunctor(packetId, version), - m_CounterDirectory(counterDirectory), - m_SendCounterPacket(sendCounterPacket) + ISendCounterPacket& sendCounterPacket, + ProfilingStateMachine& profilingStateMachine) + : CommandHandlerFunctor(packetId, version) + , m_CounterDirectory(counterDirectory) + , m_SendCounterPacket(sendCounterPacket) + , m_StateMachine(profilingStateMachine) {} void operator()(const Packet& packet) override; - private: - ICounterDirectory& m_CounterDirectory; + const ICounterDirectory& m_CounterDirectory; ISendCounterPacket& m_SendCounterPacket; + const ProfilingStateMachine& m_StateMachine; }; } // namespace profiling } // namespace armnn - |