| // |
| // Copyright © 2019 Arm Ltd. All rights reserved. |
| // SPDX-License-Identifier: MIT |
| // |
| |
| #pragma once |
| |
| #include "IBufferManager.hpp" |
| #include "ISendTimelinePacket.hpp" |
| #include "ProfilingUtils.hpp" |
| |
| #include <memory> |
| |
| namespace armnn |
| { |
| |
| namespace profiling |
| { |
| |
| class SendTimelinePacket : public ISendTimelinePacket |
| { |
| public: |
| SendTimelinePacket(IBufferManager& bufferManager) |
| : m_BufferManager(bufferManager) |
| , m_WriteBuffer(nullptr) |
| , m_Offset(0u) |
| , m_BufferSize(0u) |
| {} |
| |
| /// Commits the current buffer and reset the member variables |
| void Commit() override; |
| |
| /// Create and write a TimelineEntityBinaryPacket from the parameters to the buffer. |
| void SendTimelineEntityBinaryPacket(uint64_t profilingGuid) override; |
| |
| /// Create and write a TimelineEventBinaryPacket from the parameters to the buffer. |
| void SendTimelineEventBinaryPacket(uint64_t timestamp, uint32_t threadId, uint64_t profilingGuid) override; |
| |
| /// Create and write a TimelineEventClassBinaryPacket from the parameters to the buffer. |
| void SendTimelineEventClassBinaryPacket(uint64_t profilingGuid) override; |
| |
| /// Create and write a TimelineLabelBinaryPacket from the parameters to the buffer. |
| void SendTimelineLabelBinaryPacket(uint64_t profilingGuid, const std::string& label) override; |
| |
| /// Create and write a TimelineMessageDirectoryPackage in the buffer |
| void SendTimelineMessageDirectoryPackage() override; |
| |
| /// Create and write a TimelineRelationshipBinaryPacket from the parameters to the buffer. |
| virtual void SendTimelineRelationshipBinaryPacket(ProfilingRelationshipType relationshipType, |
| uint64_t relationshipGuid, |
| uint64_t headGuid, |
| uint64_t tailGuid) override; |
| private: |
| /// Reserves maximum packet size from buffer |
| void ReserveBuffer(); |
| |
| IBufferManager& m_BufferManager; |
| std::unique_ptr<IPacketBuffer> m_WriteBuffer; |
| unsigned int m_Offset; |
| unsigned int m_BufferSize; |
| }; |
| |
| } // namespace profiling |
| |
| } // namespace armnn |