45 lines
1.3 KiB
C++
45 lines
1.3 KiB
C++
//===- circular_raw_ostream.cpp - Implement circular_raw_ostream ----------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This implements support for circular buffered streams.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Support/circular_raw_ostream.h"
|
|
#include <algorithm>
|
|
using namespace llvm;
|
|
|
|
void circular_raw_ostream::write_impl(const char *Ptr, size_t Size) {
|
|
if (BufferSize == 0) {
|
|
TheStream->write(Ptr, Size);
|
|
return;
|
|
}
|
|
|
|
// Write into the buffer, wrapping if necessary.
|
|
while (Size != 0) {
|
|
unsigned Bytes =
|
|
std::min(unsigned(Size), unsigned(BufferSize - (Cur - BufferArray)));
|
|
memcpy(Cur, Ptr, Bytes);
|
|
Size -= Bytes;
|
|
Cur += Bytes;
|
|
if (Cur == BufferArray + BufferSize) {
|
|
// Reset the output pointer to the start of the buffer.
|
|
Cur = BufferArray;
|
|
Filled = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
void circular_raw_ostream::flushBufferWithBanner() {
|
|
if (BufferSize != 0) {
|
|
// Write out the buffer
|
|
TheStream->write(Banner, std::strlen(Banner));
|
|
flushBuffer();
|
|
}
|
|
}
|