| // |
| // Copyright © 2019 Arm Ltd. All rights reserved. |
| // SPDX-License-Identifier: MIT |
| // |
| |
| #pragma once |
| |
| #include "IPeriodicCounterCapture.hpp" |
| #include "Holder.hpp" |
| #include "Packet.hpp" |
| #include "SendCounterPacket.hpp" |
| #include "ICounterValues.hpp" |
| |
| #include <WallClockTimer.hpp> |
| |
| #include <atomic> |
| #include <chrono> |
| #include <mutex> |
| #include <thread> |
| |
| namespace armnn |
| { |
| |
| namespace profiling |
| { |
| |
| class PeriodicCounterCapture final : public IPeriodicCounterCapture |
| { |
| public: |
| PeriodicCounterCapture(const Holder& data, ISendCounterPacket& packet, const IReadCounterValues& readCounterValue) |
| : m_CaptureDataHolder(data) |
| , m_IsRunning(false) |
| , m_KeepRunning(false) |
| , m_ReadCounterValues(readCounterValue) |
| , m_SendCounterPacket(packet) |
| {} |
| ~PeriodicCounterCapture() { Stop(); } |
| |
| void Start() override; |
| void Stop() override; |
| bool IsRunning() const { return m_IsRunning; } |
| |
| private: |
| CaptureData ReadCaptureData(); |
| void Capture(const IReadCounterValues& readCounterValues); |
| |
| const Holder& m_CaptureDataHolder; |
| bool m_IsRunning; |
| std::atomic<bool> m_KeepRunning; |
| std::thread m_PeriodCaptureThread; |
| const IReadCounterValues& m_ReadCounterValues; |
| ISendCounterPacket& m_SendCounterPacket; |
| }; |
| |
| } // namespace profiling |
| |
| } // namespace armnn |