1 //===-- StreamWrapper.cpp -------------------------------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "StreamWrapper.h" 10 #include <cassert> 11 #include <fstream> 12 #include <iostream> 13 #include <memory> 14 #include <string> 15 16 namespace __llvm_libc { 17 namespace testutils { 18 outs()19StreamWrapper outs() { return {std::addressof(std::cout)}; } 20 operator <<(T t)21template <typename T> StreamWrapper &StreamWrapper::operator<<(T t) { 22 assert(os); 23 std::ostream &Stream = *reinterpret_cast<std::ostream *>(os); 24 Stream << t; 25 Stream.flush(); 26 return *this; 27 } 28 29 template StreamWrapper &StreamWrapper::operator<<<void *>(void *t); 30 template StreamWrapper &StreamWrapper::operator<<<const char *>(const char *t); 31 template StreamWrapper &StreamWrapper::operator<<<char *>(char *t); 32 template StreamWrapper &StreamWrapper::operator<<<char>(char t); 33 template StreamWrapper &StreamWrapper::operator<<<short>(short t); 34 template StreamWrapper &StreamWrapper::operator<<<int>(int t); 35 template StreamWrapper &StreamWrapper::operator<<<long>(long t); 36 template StreamWrapper &StreamWrapper::operator<<<long long>(long long t); 37 template StreamWrapper & 38 StreamWrapper::operator<<<unsigned char>(unsigned char t); 39 template StreamWrapper & 40 StreamWrapper::operator<<<unsigned short>(unsigned short t); 41 template StreamWrapper &StreamWrapper::operator<<<unsigned int>(unsigned int t); 42 template StreamWrapper & 43 StreamWrapper::operator<<<unsigned long>(unsigned long t); 44 template StreamWrapper & 45 StreamWrapper::operator<<<unsigned long long>(unsigned long long t); 46 template StreamWrapper &StreamWrapper::operator<<<bool>(bool t); 47 template StreamWrapper &StreamWrapper::operator<<<std::string>(std::string t); 48 template StreamWrapper &StreamWrapper::operator<<<float>(float t); 49 template StreamWrapper &StreamWrapper::operator<<<double>(double t); 50 OutputFileStream(const char * FN)51OutputFileStream::OutputFileStream(const char *FN) 52 : StreamWrapper(new std::ofstream(FN)) {} 53 ~OutputFileStream()54OutputFileStream::~OutputFileStream() { 55 delete reinterpret_cast<std::ofstream *>(os); 56 } 57 58 } // namespace testutils 59 } // namespace __llvm_libc 60