diff options
Diffstat (limited to 'src/profiling/SendTimelinePacket.hpp')
-rw-r--r-- | src/profiling/SendTimelinePacket.hpp | 65 |
1 files changed, 65 insertions, 0 deletions
diff --git a/src/profiling/SendTimelinePacket.hpp b/src/profiling/SendTimelinePacket.hpp new file mode 100644 index 0000000000..35ec24ffae --- /dev/null +++ b/src/profiling/SendTimelinePacket.hpp @@ -0,0 +1,65 @@ +// +// 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 |