mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
ef8825bfe6
LLVM has removed llvm::Optional, move over to std::optional. Also clang-format to fix up all the renamed #includes.
135 lines
4.4 KiB
C++
135 lines
4.4 KiB
C++
//===--- SILRemarkStreamer.h - Interface for streaming remarks --*- C++ -*-===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2020 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
/// \file
|
|
/// This file defines the interface used to stream SIL optimization diagnostics
|
|
/// through LLVM's RemarkStreamer interface.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef SWIFT_SIL_SILREMARKSTREAMER_H
|
|
#define SWIFT_SIL_SILREMARKSTREAMER_H
|
|
|
|
#include "swift/Basic/SourceManager.h"
|
|
#include "swift/SIL/OptimizationRemark.h"
|
|
#include "llvm/Remarks/Remark.h"
|
|
#include "llvm/Remarks/RemarkStreamer.h"
|
|
|
|
namespace swift {
|
|
|
|
class SILRemarkStreamer {
|
|
private:
|
|
enum class Owner {
|
|
SILModule,
|
|
LLVM,
|
|
} owner;
|
|
|
|
/// The underlying LLVM streamer.
|
|
///
|
|
/// If owned by a SILModule, this will be non-null.
|
|
std::unique_ptr<llvm::remarks::RemarkStreamer> streamer;
|
|
/// The owning LLVM context.
|
|
///
|
|
/// If owned by LLVM, this will be non-null.
|
|
llvm::LLVMContext *context;
|
|
|
|
/// The remark output stream used to record SIL remarks to a file.
|
|
std::unique_ptr<llvm::raw_fd_ostream> remarkStream;
|
|
|
|
// Source manager for resolving source locations.
|
|
const ASTContext &ctx;
|
|
/// Convert diagnostics into LLVM remark objects.
|
|
/// The lifetime of the members of the result is bound to the lifetime of
|
|
/// the SIL remarks.
|
|
template <typename RemarkT>
|
|
llvm::remarks::Remark
|
|
toLLVMRemark(const OptRemark::Remark<RemarkT> &remark) const;
|
|
|
|
SILRemarkStreamer(std::unique_ptr<llvm::remarks::RemarkStreamer> &&streamer,
|
|
std::unique_ptr<llvm::raw_fd_ostream> &&stream,
|
|
const ASTContext &Ctx);
|
|
|
|
public:
|
|
static std::unique_ptr<SILRemarkStreamer> create(SILModule &silModule);
|
|
|
|
public:
|
|
llvm::remarks::RemarkStreamer &getLLVMStreamer();
|
|
const llvm::remarks::RemarkStreamer &getLLVMStreamer() const;
|
|
|
|
const ASTContext &getASTContext() const { return ctx; }
|
|
|
|
public:
|
|
/// Perform a one-time ownership transfer to associate the underlying
|
|
/// \c llvm::remarks::RemarkStreamer with the given \c LLVMContext.
|
|
void intoLLVMContext(llvm::LLVMContext &Ctx) &;
|
|
|
|
public:
|
|
/// Emit a remark through the streamer.
|
|
template <typename RemarkT>
|
|
void emit(const OptRemark::Remark<RemarkT> &remark);
|
|
};
|
|
|
|
// Implementation for template member functions.
|
|
|
|
// OptRemark type -> llvm::remarks::Type
|
|
template <typename RemarkT> static llvm::remarks::Type toRemarkType() {
|
|
if (std::is_same<RemarkT, OptRemark::RemarkPassed>::value)
|
|
return llvm::remarks::Type::Passed;
|
|
if (std::is_same<RemarkT, OptRemark::RemarkMissed>::value)
|
|
return llvm::remarks::Type::Missed;
|
|
llvm_unreachable("Unknown remark type");
|
|
}
|
|
|
|
static inline std::optional<llvm::remarks::RemarkLocation>
|
|
toRemarkLocation(const SourceLoc &loc, const SourceManager &srcMgr) {
|
|
if (!loc.isValid())
|
|
return std::nullopt;
|
|
|
|
StringRef file = srcMgr.getDisplayNameForLoc(loc);
|
|
unsigned line, col;
|
|
std::tie(line, col) = srcMgr.getPresumedLineAndColumnForLoc(loc);
|
|
return llvm::remarks::RemarkLocation{file, line, col};
|
|
}
|
|
|
|
template <typename RemarkT>
|
|
llvm::remarks::Remark SILRemarkStreamer::toLLVMRemark(
|
|
const OptRemark::Remark<RemarkT> &optRemark) const {
|
|
llvm::remarks::Remark llvmRemark; // The result.
|
|
llvmRemark.RemarkType = toRemarkType<RemarkT>();
|
|
llvmRemark.PassName = optRemark.getPassName();
|
|
llvmRemark.RemarkName = optRemark.getIdentifier();
|
|
llvmRemark.FunctionName = optRemark.getDemangledFunctionName();
|
|
llvmRemark.Loc =
|
|
toRemarkLocation(optRemark.getLocation(), getASTContext().SourceMgr);
|
|
|
|
for (const OptRemark::Argument &arg : optRemark.getArgs()) {
|
|
llvmRemark.Args.emplace_back();
|
|
llvmRemark.Args.back().Key = arg.key.data;
|
|
llvmRemark.Args.back().Val = arg.val;
|
|
llvmRemark.Args.back().Loc =
|
|
toRemarkLocation(arg.loc, getASTContext().SourceMgr);
|
|
}
|
|
|
|
return llvmRemark;
|
|
}
|
|
|
|
template <typename RemarkT>
|
|
void SILRemarkStreamer::emit(const OptRemark::Remark<RemarkT> &optRemark) {
|
|
if (!getLLVMStreamer().matchesFilter(optRemark.getPassName()))
|
|
return;
|
|
|
|
return getLLVMStreamer().getSerializer().emit(toLLVMRemark(optRemark));
|
|
}
|
|
|
|
} // namespace swift
|
|
#endif
|