mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 19:20:13 +01:00
Jon Ross-Perkins
2fef1cb713
Also makes the style guide explicitly comment on void, but this was the intent IIRC because it matches Carbon's `-> ()` (and "always" versus "except for void", which we definitely went back and forth on). Includes adjusting function pointers, which I definitely forget this syntax works sometimes. Excludes utils/tree_sitter/src/scanner.c because it claims to be C, but really we should probably fix that to be cpp.
65 lines
1.8 KiB
C++
65 lines
1.8 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_RAW_STRING_OSTREAM_H_
|
|
#define CARBON_COMMON_RAW_STRING_OSTREAM_H_
|
|
|
|
#include "common/check.h"
|
|
#include "common/ostream.h"
|
|
|
|
namespace Carbon {
|
|
|
|
// Implements streaming output with an underlying string. The string must always
|
|
// be taken prior to destruction.
|
|
//
|
|
// Versus `llvm::raw_string_ostream`:
|
|
// - Owns the underlying string.
|
|
// - Supports `pwrite` for compatibility with more stream uses in tests.
|
|
class RawStringOstream : public llvm::raw_pwrite_stream {
|
|
public:
|
|
explicit RawStringOstream() : llvm::raw_pwrite_stream(/*Unbuffered=*/true) {}
|
|
|
|
~RawStringOstream() override {
|
|
CARBON_CHECK(str_.empty(), "Expected to be emptied by TakeStr, have: {0}",
|
|
str_);
|
|
}
|
|
|
|
// Returns the streamed contents, clearing the stream back to empty.
|
|
auto TakeStr() -> std::string {
|
|
std::string result = std::move(str_);
|
|
clear();
|
|
return result;
|
|
}
|
|
|
|
// Clears the buffer, which can be helpful when destructing without
|
|
// necessarily calling `TakeStr()`.
|
|
auto clear() -> void { str_.clear(); }
|
|
|
|
auto empty() -> bool { return str_.empty(); }
|
|
auto size() -> size_t { return str_.size(); }
|
|
|
|
private:
|
|
auto current_pos() const -> uint64_t override { return str_.size(); }
|
|
|
|
auto pwrite_impl(const char* ptr, size_t size, uint64_t offset)
|
|
-> void override {
|
|
str_.replace(offset, size, ptr, size);
|
|
}
|
|
|
|
auto write_impl(const char* ptr, size_t size) -> void override {
|
|
str_.append(ptr, size);
|
|
}
|
|
|
|
auto reserveExtraSpace(uint64_t extra_size) -> void override {
|
|
str_.reserve(str_.size() + extra_size);
|
|
}
|
|
|
|
// The actual buffer.
|
|
std::string str_;
|
|
};
|
|
|
|
} // namespace Carbon
|
|
|
|
#endif // CARBON_COMMON_RAW_STRING_OSTREAM_H_
|