1 //===- BlockIndexer.cpp - FDR Block Indexing VIsitor ----------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // An implementation of the RecordVisitor which generates a mapping between a 11 // thread and a range of records representing a block. 12 // 13 //===----------------------------------------------------------------------===// 14 #include "llvm/XRay/BlockIndexer.h" 15 16 namespace llvm { 17 namespace xray { 18 19 Error BlockIndexer::visit(BufferExtents &) { return Error::success(); } 20 21 Error BlockIndexer::visit(WallclockRecord &R) { 22 CurrentBlock.Records.push_back(&R); 23 CurrentBlock.WallclockTime = &R; 24 return Error::success(); 25 } 26 27 Error BlockIndexer::visit(NewCPUIDRecord &R) { 28 CurrentBlock.Records.push_back(&R); 29 return Error::success(); 30 } 31 32 Error BlockIndexer::visit(TSCWrapRecord &R) { 33 CurrentBlock.Records.push_back(&R); 34 return Error::success(); 35 } 36 37 Error BlockIndexer::visit(CustomEventRecord &R) { 38 CurrentBlock.Records.push_back(&R); 39 return Error::success(); 40 } 41 42 Error BlockIndexer::visit(CallArgRecord &R) { 43 CurrentBlock.Records.push_back(&R); 44 return Error::success(); 45 } 46 47 Error BlockIndexer::visit(PIDRecord &R) { 48 CurrentBlock.ProcessID = R.pid(); 49 CurrentBlock.Records.push_back(&R); 50 return Error::success(); 51 } 52 53 Error BlockIndexer::visit(NewBufferRecord &R) { 54 if (!CurrentBlock.Records.empty()) 55 if (auto E = flush()) 56 return E; 57 58 CurrentBlock.ThreadID = R.tid(); 59 CurrentBlock.Records.push_back(&R); 60 return Error::success(); 61 } 62 63 Error BlockIndexer::visit(EndBufferRecord &R) { 64 CurrentBlock.Records.push_back(&R); 65 return Error::success(); 66 } 67 68 Error BlockIndexer::visit(FunctionRecord &R) { 69 CurrentBlock.Records.push_back(&R); 70 return Error::success(); 71 } 72 73 Error BlockIndexer::flush() { 74 Index::iterator It; 75 std::tie(It, std::ignore) = 76 Indices.insert({{CurrentBlock.ProcessID, CurrentBlock.ThreadID}, {}}); 77 It->second.push_back({CurrentBlock.ProcessID, CurrentBlock.ThreadID, 78 CurrentBlock.WallclockTime, 79 std::move(CurrentBlock.Records)}); 80 CurrentBlock.ProcessID = 0; 81 CurrentBlock.ThreadID = 0; 82 CurrentBlock.Records = {}; 83 CurrentBlock.WallclockTime = nullptr; 84 return Error::success(); 85 } 86 87 } // namespace xray 88 } // namespace llvm 89