114 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			114 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			C++
		
	
	
	
//===- BlockPrinter.cpp - FDR Block Pretty Printer Implementation --------===//
 | 
						|
//
 | 
						|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 | 
						|
// See https://llvm.org/LICENSE.txt for license information.
 | 
						|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 | 
						|
//
 | 
						|
//===----------------------------------------------------------------------===//
 | 
						|
#include "llvm/XRay/BlockPrinter.h"
 | 
						|
 | 
						|
namespace llvm {
 | 
						|
namespace xray {
 | 
						|
 | 
						|
Error BlockPrinter::visit(BufferExtents &R) {
 | 
						|
  OS << "\n[New Block]\n";
 | 
						|
  CurrentState = State::Preamble;
 | 
						|
  return RP.visit(R);
 | 
						|
}
 | 
						|
 | 
						|
// Preamble printing.
 | 
						|
Error BlockPrinter::visit(NewBufferRecord &R) {
 | 
						|
  if (CurrentState == State::Start)
 | 
						|
    OS << "\n[New Block]\n";
 | 
						|
 | 
						|
  OS << "Preamble: \n";
 | 
						|
  CurrentState = State::Preamble;
 | 
						|
  return RP.visit(R);
 | 
						|
}
 | 
						|
 | 
						|
Error BlockPrinter::visit(WallclockRecord &R) {
 | 
						|
  CurrentState = State::Preamble;
 | 
						|
  return RP.visit(R);
 | 
						|
}
 | 
						|
 | 
						|
Error BlockPrinter::visit(PIDRecord &R) {
 | 
						|
  CurrentState = State::Preamble;
 | 
						|
  return RP.visit(R);
 | 
						|
}
 | 
						|
 | 
						|
// Metadata printing.
 | 
						|
Error BlockPrinter::visit(NewCPUIDRecord &R) {
 | 
						|
  if (CurrentState == State::Preamble)
 | 
						|
    OS << "\nBody:\n";
 | 
						|
  if (CurrentState == State::Function)
 | 
						|
    OS << "\nMetadata: ";
 | 
						|
  CurrentState = State::Metadata;
 | 
						|
  OS << " ";
 | 
						|
  auto E = RP.visit(R);
 | 
						|
  return E;
 | 
						|
}
 | 
						|
 | 
						|
Error BlockPrinter::visit(TSCWrapRecord &R) {
 | 
						|
  if (CurrentState == State::Function)
 | 
						|
    OS << "\nMetadata:";
 | 
						|
  CurrentState = State::Metadata;
 | 
						|
  OS << " ";
 | 
						|
  auto E = RP.visit(R);
 | 
						|
  return E;
 | 
						|
}
 | 
						|
 | 
						|
// Custom events will be rendered like "function" events.
 | 
						|
Error BlockPrinter::visit(CustomEventRecord &R) {
 | 
						|
  if (CurrentState == State::Metadata)
 | 
						|
    OS << "\n";
 | 
						|
  CurrentState = State::CustomEvent;
 | 
						|
  OS << "*  ";
 | 
						|
  auto E = RP.visit(R);
 | 
						|
  return E;
 | 
						|
}
 | 
						|
 | 
						|
Error BlockPrinter::visit(CustomEventRecordV5 &R) {
 | 
						|
  if (CurrentState == State::Metadata)
 | 
						|
    OS << "\n";
 | 
						|
  CurrentState = State::CustomEvent;
 | 
						|
  OS << "*  ";
 | 
						|
  auto E = RP.visit(R);
 | 
						|
  return E;
 | 
						|
}
 | 
						|
 | 
						|
Error BlockPrinter::visit(TypedEventRecord &R) {
 | 
						|
  if (CurrentState == State::Metadata)
 | 
						|
    OS << "\n";
 | 
						|
  CurrentState = State::CustomEvent;
 | 
						|
  OS << "*  ";
 | 
						|
  auto E = RP.visit(R);
 | 
						|
  return E;
 | 
						|
}
 | 
						|
 | 
						|
// Function call printing.
 | 
						|
Error BlockPrinter::visit(FunctionRecord &R) {
 | 
						|
  if (CurrentState == State::Metadata)
 | 
						|
    OS << "\n";
 | 
						|
  CurrentState = State::Function;
 | 
						|
  OS << "-  ";
 | 
						|
  auto E = RP.visit(R);
 | 
						|
  return E;
 | 
						|
}
 | 
						|
 | 
						|
Error BlockPrinter::visit(CallArgRecord &R) {
 | 
						|
  CurrentState = State::Arg;
 | 
						|
  OS << " : ";
 | 
						|
  auto E = RP.visit(R);
 | 
						|
  return E;
 | 
						|
}
 | 
						|
 | 
						|
Error BlockPrinter::visit(EndBufferRecord &R) {
 | 
						|
    CurrentState = State::End;
 | 
						|
    OS << " *** ";
 | 
						|
    auto E = RP.visit(R);
 | 
						|
    return E;
 | 
						|
}
 | 
						|
 | 
						|
} // namespace xray
 | 
						|
} // namespace llvm
 |