1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
//
// Copyright © 2019 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "TimelineDirectoryCaptureCommandHandler.hpp"
#include <ProfilingUtils.hpp>
#include <iostream>
#include <string>
using namespace armnn::profiling;
namespace armnn
{
namespace gatordmock
{
void TimelineDirectoryCaptureCommandHandler::ParseData(const armnn::profiling::Packet& packet)
{
uint32_t offset = 0;
if (packet.GetLength() < 8)
{
return;
}
const unsigned char* data = packet.GetData();
m_SwTraceHeader.m_StreamVersion = ReadUint8(data, offset);
offset += uint8_t_size;
m_SwTraceHeader.m_PointerBytes = ReadUint8(data, offset);
offset += uint8_t_size;
m_SwTraceHeader.m_ThreadIdBytes = ReadUint8(data, offset);
offset += uint8_t_size;
uint32_t numberOfDeclarations = profiling::ReadUint32(data, offset);
offset += uint32_t_size;
for (uint32_t declaration = 0; declaration < numberOfDeclarations; ++declaration)
{
m_SwTraceMessages.push_back(profiling::ReadSwTraceMessage(data, offset));
}
}
void TimelineDirectoryCaptureCommandHandler::Print()
{
std::string header;
header.append(profiling::CentreAlignFormatting("decl_id", 12));
header.append(" | ");
header.append(profiling::CentreAlignFormatting("decl_name", 20));
header.append(" | ");
header.append(profiling::CentreAlignFormatting("ui_name", 20));
header.append(" | ");
header.append(profiling::CentreAlignFormatting("arg_types", 16));
header.append(" | ");
header.append(profiling::CentreAlignFormatting("arg_names", 80));
header.append("\n");
std::cout << "\n" << "\n";
std::cout << profiling::CentreAlignFormatting("SW DIRECTORY", static_cast<int>(header.size()));
std::cout << "\n";
std::cout << std::string(header.size(), '=') << "\n";
std::cout << header;
for (const auto& swTraceMessage : m_SwTraceMessages)
{
std::string body;
body.append(profiling::CentreAlignFormatting(std::to_string(swTraceMessage.m_Id), 12));
body.append(" | ");
body.append(profiling::CentreAlignFormatting(swTraceMessage.m_Name, 20));
body.append(" | ");
body.append(profiling::CentreAlignFormatting(swTraceMessage.m_UiName, 20));
body.append(" | ");
std::string argTypes;
for (auto argType: swTraceMessage.m_ArgTypes)
{
argTypes += argType;
argTypes += " ";
}
body.append(profiling::CentreAlignFormatting(argTypes, 16));
body.append(" | ");
std::string argNames;
for (auto argName: swTraceMessage.m_ArgNames)
{
argNames += argName + " ";
}
body.append(profiling::CentreAlignFormatting(argNames, 80));
body.append("\n");
std::cout << std::string(body.size(), '-') << "\n";
std::cout << body;
}
}
void TimelineDirectoryCaptureCommandHandler::operator()(const profiling::Packet& packet)
{
ParseData(packet);
if (!m_QuietOperation)
{
Print();
}
}
} //namespace gatordmock
} //namespace armnn
|
