// Configuration Interface
//===--------------------------------------------------------------------===//
- /// SetBufferSize - Set the internal buffer size to the specified amount
- /// instead of the default.
- void SetBufferSize(size_t Size=4096);
+ /// SetBuffered - Set the stream to be buffered, with an automatically
+ /// determined buffer size.
+ void SetBuffered();
- size_t GetBufferSize() const {
+ /// SetBufferrSize - Set the stream to be buffered, using the
+ /// specified buffer size.
+ void SetBufferSize(size_t Size);
+
+ size_t GetBufferSize() {
+ // If we're supposed to be buffered but haven't actually gotten around
+ // to allocating the buffer yet, return the value that would be used.
+ if (!Unbuffered && !OutBufStart)
+ return preferred_buffer_size();
+
+ // Otherwise just return the size of the allocated buffer.
return OutBufEnd - OutBufStart;
}
- /// SetUnbuffered - Set the streams buffering status. When
- /// unbuffered the stream will flush after every write. This routine
+ /// SetUnbuffered - Set the stream to be unbuffered. When
+ /// unbuffered, the stream will flush after every write. This routine
/// will also flush the buffer immediately when the stream is being
/// set to unbuffered.
void SetUnbuffered();
virtual uint64_t current_pos() = 0;
protected:
+ /// preferred_buffer_size - Return an efficient buffer size for the
+ /// underlying output mechanism.
+ virtual size_t preferred_buffer_size();
+
/// error_detected - Set the flag indicating that an output error has
/// been encountered.
void error_detected() { Error = true; }
/// counting the bytes currently in the buffer.
virtual uint64_t current_pos() { return pos; }
+ /// preferred_buffer_size - Determine an efficient buffer size.
+ virtual size_t preferred_buffer_size();
+
public:
/// raw_fd_ostream - Open the specified file for writing. If an
/// error occurs, information about the error is put into ErrorInfo,
#include "llvm/Support/Compiler.h"
#include "llvm/Support/ErrorHandling.h"
#include <ostream>
+#include <sys/stat.h>
+#include <sys/types.h>
#if defined(HAVE_UNISTD_H)
# include <unistd.h>
// An out of line virtual method to provide a home for the class vtable.
void raw_ostream::handle() {}
+size_t raw_ostream::preferred_buffer_size() {
+ // BUFSIZ is intended to be a reasonable default.
+ return BUFSIZ;
+}
+
+void raw_ostream::SetBuffered() {
+ // Ask the subclass to determine an appropriate buffer size.
+ SetBufferSize(preferred_buffer_size());
+}
+
void raw_ostream::SetBufferSize(size_t Size) {
assert(Size >= 64 &&
"Buffer size must be somewhat large for invariants to hold");
}
if (!OutBufStart)
- SetBufferSize();
+ SetBuffered();
else
flush_nonempty();
}
return *this;
}
// Set up a buffer and start over.
- SetBufferSize();
+ SetBuffered();
return write(Ptr, Size);
}
// Write out the data in buffer-sized blocks until the remainder
return pos;
}
+size_t raw_fd_ostream::preferred_buffer_size() {
+#if !defined(_MSC_VER) // Windows reportedly doesn't have st_blksize.
+ assert(FD >= 0 && "File not yet open!");
+ struct stat statbuf;
+ if (fstat(FD, &statbuf) == 0)
+ return statbuf.st_blksize;
+ error_detected();
+#endif
+ return raw_ostream::preferred_buffer_size();
+}
+
raw_ostream &raw_fd_ostream::changeColor(enum Colors colors, bool bold,
bool bg) {
if (sys::Process::ColorNeedsFlush())