| //===- 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(); |
| } |
| } |