//===- FDRRecordProducer.h - XRay FDR Mode Record Producer ----------------===//
 
//
 
// 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
 
//
 
//===----------------------------------------------------------------------===//
 
#ifndef LLVM_XRAY_FDRRECORDPRODUCER_H
 
#define LLVM_XRAY_FDRRECORDPRODUCER_H
 
 
 
#include "llvm/Support/Error.h"
 
#include "llvm/XRay/FDRRecords.h"
 
#include "llvm/XRay/XRayRecord.h"
 
#include <memory>
 
 
 
namespace llvm {
 
namespace xray {
 
 
 
class RecordProducer {
 
public:
 
  /// All producer implementations must yield either an Error or a non-nullptr
 
  /// unique_ptr<Record>.
 
  virtual Expected<std::unique_ptr<Record>> produce() = 0;
 
  virtual ~RecordProducer() = default;
 
};
 
 
 
class FileBasedRecordProducer : public RecordProducer {
 
  const XRayFileHeader &Header;
 
  DataExtractor &E;
 
  uint64_t &OffsetPtr;
 
  uint32_t CurrentBufferBytes = 0;
 
 
 
  // Helper function which gets the next record by speculatively reading through
 
  // the log, finding a buffer extents record.
 
  Expected<std::unique_ptr<Record>> findNextBufferExtent();
 
 
 
public:
 
  FileBasedRecordProducer(const XRayFileHeader &FH, DataExtractor &DE,
 
                          uint64_t &OP)
 
      : Header(FH), E(DE), OffsetPtr(OP) {}
 
 
 
  /// This producer encapsulates the logic for loading a File-backed
 
  /// RecordProducer hidden behind a DataExtractor.
 
  Expected<std::unique_ptr<Record>> produce() override;
 
};
 
 
 
} // namespace xray
 
} // namespace llvm
 
 
 
#endif // LLVM_XRAY_FDRRECORDPRODUCER_H