111cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//===----------------------------------------------------------------------===// 211cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// 311cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// The LLVM Compiler Infrastructure 411cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// 511cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// This file is dual licensed under the MIT and the University of Illinois Open 611cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// Source Licenses. See LICENSE.TXT for details. 711cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// 811cd02dfb91661c65134cac258cf5924270e9d2Dan Albert//===----------------------------------------------------------------------===// 911cd02dfb91661c65134cac258cf5924270e9d2Dan Albert 1011cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// <ostream> 1111cd02dfb91661c65134cac258cf5924270e9d2Dan Albert 1211cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// template <class charT, class traits = char_traits<charT> > 1311cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// class basic_ostream; 1411cd02dfb91661c65134cac258cf5924270e9d2Dan Albert 1511cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// template <class charT, class traits> 1611cd02dfb91661c65134cac258cf5924270e9d2Dan Albert// basic_ostream<charT,traits>& ends(basic_ostream<charT,traits>& os); 1711cd02dfb91661c65134cac258cf5924270e9d2Dan Albert 1811cd02dfb91661c65134cac258cf5924270e9d2Dan Albert#include <ostream> 1911cd02dfb91661c65134cac258cf5924270e9d2Dan Albert#include <cassert> 2011cd02dfb91661c65134cac258cf5924270e9d2Dan Albert 2111cd02dfb91661c65134cac258cf5924270e9d2Dan Alberttemplate <class CharT> 2211cd02dfb91661c65134cac258cf5924270e9d2Dan Albertclass testbuf 2311cd02dfb91661c65134cac258cf5924270e9d2Dan Albert : public std::basic_streambuf<CharT> 2411cd02dfb91661c65134cac258cf5924270e9d2Dan Albert{ 2511cd02dfb91661c65134cac258cf5924270e9d2Dan Albert typedef std::basic_streambuf<CharT> base; 2611cd02dfb91661c65134cac258cf5924270e9d2Dan Albert std::basic_string<CharT> str_; 2711cd02dfb91661c65134cac258cf5924270e9d2Dan Albertpublic: 2811cd02dfb91661c65134cac258cf5924270e9d2Dan Albert testbuf() 2911cd02dfb91661c65134cac258cf5924270e9d2Dan Albert { 3011cd02dfb91661c65134cac258cf5924270e9d2Dan Albert } 3111cd02dfb91661c65134cac258cf5924270e9d2Dan Albert 3211cd02dfb91661c65134cac258cf5924270e9d2Dan Albert std::basic_string<CharT> str() const 3311cd02dfb91661c65134cac258cf5924270e9d2Dan Albert {return std::basic_string<CharT>(base::pbase(), base::pptr());} 3411cd02dfb91661c65134cac258cf5924270e9d2Dan Albert 3511cd02dfb91661c65134cac258cf5924270e9d2Dan Albertprotected: 3611cd02dfb91661c65134cac258cf5924270e9d2Dan Albert 3711cd02dfb91661c65134cac258cf5924270e9d2Dan Albert virtual typename base::int_type 3811cd02dfb91661c65134cac258cf5924270e9d2Dan Albert overflow(typename base::int_type __c = base::traits_type::eof()) 3911cd02dfb91661c65134cac258cf5924270e9d2Dan Albert { 4011cd02dfb91661c65134cac258cf5924270e9d2Dan Albert if (__c != base::traits_type::eof()) 4111cd02dfb91661c65134cac258cf5924270e9d2Dan Albert { 4211cd02dfb91661c65134cac258cf5924270e9d2Dan Albert int n = str_.size(); 4311cd02dfb91661c65134cac258cf5924270e9d2Dan Albert str_.push_back(__c); 4411cd02dfb91661c65134cac258cf5924270e9d2Dan Albert str_.resize(str_.capacity()); 4511cd02dfb91661c65134cac258cf5924270e9d2Dan Albert base::setp(const_cast<CharT*>(str_.data()), 4611cd02dfb91661c65134cac258cf5924270e9d2Dan Albert const_cast<CharT*>(str_.data() + str_.size())); 4711cd02dfb91661c65134cac258cf5924270e9d2Dan Albert base::pbump(n+1); 4811cd02dfb91661c65134cac258cf5924270e9d2Dan Albert } 4911cd02dfb91661c65134cac258cf5924270e9d2Dan Albert return __c; 5011cd02dfb91661c65134cac258cf5924270e9d2Dan Albert } 5111cd02dfb91661c65134cac258cf5924270e9d2Dan Albert}; 5211cd02dfb91661c65134cac258cf5924270e9d2Dan Albert 5311cd02dfb91661c65134cac258cf5924270e9d2Dan Albertint main() 5411cd02dfb91661c65134cac258cf5924270e9d2Dan Albert{ 5511cd02dfb91661c65134cac258cf5924270e9d2Dan Albert { 5611cd02dfb91661c65134cac258cf5924270e9d2Dan Albert testbuf<char> sb; 5711cd02dfb91661c65134cac258cf5924270e9d2Dan Albert std::ostream os(&sb); 5811cd02dfb91661c65134cac258cf5924270e9d2Dan Albert ends(os); 5911cd02dfb91661c65134cac258cf5924270e9d2Dan Albert assert(sb.str().size() == 1); 6011cd02dfb91661c65134cac258cf5924270e9d2Dan Albert assert(sb.str().back() == 0); 6111cd02dfb91661c65134cac258cf5924270e9d2Dan Albert assert(os.good()); 6211cd02dfb91661c65134cac258cf5924270e9d2Dan Albert } 6311cd02dfb91661c65134cac258cf5924270e9d2Dan Albert { 6411cd02dfb91661c65134cac258cf5924270e9d2Dan Albert testbuf<wchar_t> sb; 6511cd02dfb91661c65134cac258cf5924270e9d2Dan Albert std::wostream os(&sb); 6611cd02dfb91661c65134cac258cf5924270e9d2Dan Albert ends(os); 6711cd02dfb91661c65134cac258cf5924270e9d2Dan Albert assert(sb.str().size() == 1); 6811cd02dfb91661c65134cac258cf5924270e9d2Dan Albert assert(sb.str().back() == 0); 6911cd02dfb91661c65134cac258cf5924270e9d2Dan Albert assert(os.good()); 7011cd02dfb91661c65134cac258cf5924270e9d2Dan Albert } 7111cd02dfb91661c65134cac258cf5924270e9d2Dan Albert} 72