1//===- BlockPrinter.cpp - FDR Block Pretty Printer Implementation --------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8#include "llvm/XRay/BlockPrinter.h"
9
10namespace llvm {
11namespace xray {
12
13Error BlockPrinter::visit(BufferExtents &R) {
14 OS << "\n[New Block]\n";
15 CurrentState = State::Preamble;
16 return RP.visit(R);
17}
18
19// Preamble printing.
20Error BlockPrinter::visit(NewBufferRecord &R) {
21 if (CurrentState == State::Start)
22 OS << "\n[New Block]\n";
23
24 OS << "Preamble: \n";
25 CurrentState = State::Preamble;
26 return RP.visit(R);
27}
28
29Error BlockPrinter::visit(WallclockRecord &R) {
30 CurrentState = State::Preamble;
31 return RP.visit(R);
32}
33
34Error BlockPrinter::visit(PIDRecord &R) {
35 CurrentState = State::Preamble;
36 return RP.visit(R);
37}
38
39// Metadata printing.
40Error BlockPrinter::visit(NewCPUIDRecord &R) {
41 if (CurrentState == State::Preamble)
42 OS << "\nBody:\n";
43 if (CurrentState == State::Function)
44 OS << "\nMetadata: ";
45 CurrentState = State::Metadata;
46 OS << " ";
47 auto E = RP.visit(R);
48 return E;
49}
50
51Error BlockPrinter::visit(TSCWrapRecord &R) {
52 if (CurrentState == State::Function)
53 OS << "\nMetadata:";
54 CurrentState = State::Metadata;
55 OS << " ";
56 auto E = RP.visit(R);
57 return E;
58}
59
60// Custom events will be rendered like "function" events.
61Error BlockPrinter::visit(CustomEventRecord &R) {
62 if (CurrentState == State::Metadata)
63 OS << "\n";
64 CurrentState = State::CustomEvent;
65 OS << "* ";
66 auto E = RP.visit(R);
67 return E;
68}
69
70Error BlockPrinter::visit(CustomEventRecordV5 &R) {
71 if (CurrentState == State::Metadata)
72 OS << "\n";
73 CurrentState = State::CustomEvent;
74 OS << "* ";
75 auto E = RP.visit(R);
76 return E;
77}
78
79Error BlockPrinter::visit(TypedEventRecord &R) {
80 if (CurrentState == State::Metadata)
81 OS << "\n";
82 CurrentState = State::CustomEvent;
83 OS << "* ";
84 auto E = RP.visit(R);
85 return E;
86}
87
88// Function call printing.
89Error BlockPrinter::visit(FunctionRecord &R) {
90 if (CurrentState == State::Metadata)
91 OS << "\n";
92 CurrentState = State::Function;
93 OS << "- ";
94 auto E = RP.visit(R);
95 return E;
96}
97
98Error BlockPrinter::visit(CallArgRecord &R) {
99 CurrentState = State::Arg;
100 OS << " : ";
101 auto E = RP.visit(R);
102 return E;
103}
104
105Error BlockPrinter::visit(EndBufferRecord &R) {
106 CurrentState = State::End;
107 OS << " *** ";
108 auto E = RP.visit(R);
109 return E;
110}
111
112} // namespace xray
113} // namespace llvm
114