Files
carbon-lang/common/ostream.h
T
Jon Ross-PerkinsandDana Jansens 3ce0df67bb Add Dump functions to Check, Parse, and Lex (#4669)
- Provide `Check::Dump(context, arg)` and similar.
- gdb and lldb should do contextual lookup, and `call Dump(*this,
Lex::TokenIndex::Invalid)` has been tested with gdb.
- Since this is only for debug, keeps the functions fully separated from
code.
- Uses alwayslink to ensure objects are correctly linked, even though
there are no calls.
- `-Wno-missing-prototypes` is needed when we don't have forward
declarations.
- Code is not linked in opt builds, using `#ifndef NDEBUG`.
- This probably could be doing something in BUILD files with a
`select()`, but the `#ifndef` seemed easier.

This is based on #4620, but uses free functions instead of member
functions.

Co-authored-by: Dana Jansens <danakj@orodu.net>

---------

Co-authored-by: danakj <danakj@orodu.net>
2024-12-12 20:51:02 +00:00

97 lines
3.2 KiB
C++

// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
// Exceptions. See /LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
#ifndef CARBON_COMMON_OSTREAM_H_
#define CARBON_COMMON_OSTREAM_H_
// Libraries should include this header instead of raw_ostream.
#include <concepts>
#include <ostream>
#include <type_traits>
#include "llvm/Support/Compiler.h"
#include "llvm/Support/raw_os_ostream.h"
#include "llvm/Support/raw_ostream.h" // IWYU pragma: export
namespace Carbon {
// CRTP base class for printable types. Children (DerivedT) must implement:
// - auto Print(llvm::raw_ostream& out) const -> void
template <typename DerivedT>
class Printable {
// Provides simple printing for debuggers.
LLVM_DUMP_METHOD void Dump() const {
static_cast<const DerivedT*>(this)->Print(llvm::errs());
}
// Supports printing to llvm::raw_ostream.
friend auto operator<<(llvm::raw_ostream& out, const DerivedT& obj)
-> llvm::raw_ostream& {
obj.Print(out);
return out;
}
// Supports printing to std::ostream.
friend auto operator<<(std::ostream& out, const DerivedT& obj)
-> std::ostream& {
llvm::raw_os_ostream raw_os(out);
obj.Print(raw_os);
return out;
}
// Allows GoogleTest and GoogleMock to print pointers by dereferencing them.
// This is important to allow automatic printing of arguments of mocked
// APIs.
friend auto PrintTo(DerivedT* p, std::ostream* out) -> void {
*out << static_cast<const void*>(p);
// Also print the object if non-null.
if (p) {
*out << " pointing to " << *p;
}
}
};
// Returns the result of printing the value.
template <typename T>
requires std::derived_from<T, Printable<T>>
inline auto PrintToString(const T& val) -> std::string {
std::string str;
llvm::raw_string_ostream stream(str);
stream << val;
return str;
}
} // namespace Carbon
namespace llvm {
// Injects an `operator<<` overload into the `llvm` namespace which detects LLVM
// types with `raw_ostream` overloads and uses that to map to a `std::ostream`
// overload. This allows LLVM types to be printed to `std::ostream` via their
// `raw_ostream` operator overloads, which is needed both for logging and
// testing.
//
// To make this overload be unusually low priority, it is designed to take even
// the `std::ostream` parameter as a template, and SFINAE disable itself unless
// that template parameter is derived from `std::ostream`. This ensures that an
// *explicit* operator will be preferred when provided. Some LLVM types may have
// this, and so we want to prioritize accordingly.
//
// It would be slightly cleaner for LLVM itself to provide this overload in
// `raw_os_ostream.h` so that we wouldn't need to inject into LLVM's namespace,
// but supporting `std::ostream` isn't a priority for LLVM so we handle it
// locally instead.
template <typename StreamT, typename ClassT>
requires std::derived_from<std::decay_t<StreamT>, std::ostream> &&
(!std::same_as<std::decay_t<ClassT>, raw_ostream>)
auto operator<<(StreamT& standard_out, const ClassT& value) -> StreamT& {
raw_os_ostream(standard_out) << value;
return standard_out;
}
} // namespace llvm
#endif // CARBON_COMMON_OSTREAM_H_