#ifndef LLVM_SUPPORT_STREAMS_H
#define LLVM_SUPPORT_STREAMS_H
#include <iosfwd>
namespace llvm {
void FlushStream(std::ostream &S);
template <typename StreamTy>
class BaseStream {
StreamTy *Stream;
public:
BaseStream() : Stream(0) {}
BaseStream(StreamTy &S) : Stream(&S) {}
BaseStream(StreamTy *S) : Stream(S) {}
StreamTy *stream() const { return Stream; }
inline BaseStream &operator << (std::ios_base &(*Func)(std::ios_base&)) {
if (Stream) *Stream << Func;
return *this;
}
inline BaseStream &operator << (StreamTy &(*Func)(StreamTy&)) {
if (Stream) *Stream << Func;
return *this;
}
void flush() {
if (Stream)
FlushStream(*Stream);
}
template <typename Ty>
BaseStream &operator << (const Ty &Thing) {
if (Stream) *Stream << Thing;
return *this;
}
template <typename Ty>
BaseStream &operator >> (Ty &Thing) {
if (Stream) *Stream >> Thing;
return *this;
}
template <typename Ty>
BaseStream &write(const Ty &A, unsigned N) {
if (Stream) Stream->write(A, N);
return *this;
}
operator StreamTy* () { return Stream; }
bool operator == (const StreamTy &S) { return &S == Stream; }
bool operator != (const StreamTy &S) { return !(*this == S); }
bool operator == (const BaseStream &S) { return S.Stream == Stream; }
bool operator != (const BaseStream &S) { return !(*this == S); }
};
typedef BaseStream<std::ostream> OStream;
typedef BaseStream<std::istream> IStream;
typedef BaseStream<std::stringstream> StringStream;
extern OStream cout;
extern OStream cerr;
extern IStream cin;
}
#endif