williamr@2: /*
williamr@2: * Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies). All rights reserved.
williamr@2: 
williamr@2: * Redistribution and use in source and binary forms, with or without 
williamr@2: * modification, are permitted provided that the following conditions are met:
williamr@2: 
williamr@2: * Redistributions of source code must retain the above copyright notice, this 
williamr@2: * list of conditions and the following disclaimer.
williamr@2: * Redistributions in binary form must reproduce the above copyright notice, 
williamr@2: * this list of conditions and the following disclaimer in the documentation 
williamr@2: * and/or other materials provided with the distribution.
williamr@2: * Neither the name of Nokia Corporation nor the names of its contributors 
williamr@2: * may be used to endorse or promote products derived from this software 
williamr@2: * without specific prior written permission.
williamr@2: 
williamr@2: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 
williamr@2: * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 
williamr@2: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 
williamr@2: * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE 
williamr@2: * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 
williamr@2: * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR 
williamr@2: * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 
williamr@2: * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 
williamr@2: * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 
williamr@2: * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
williamr@2: *
williamr@2: * Description:
williamr@2: *
williamr@2: */
williamr@2: 
williamr@2: #ifndef _STLP_STRING_IO_C
williamr@2: #define _STLP_STRING_IO_C
williamr@2: 
williamr@2: #ifndef _STLP_STRING_IO_H
williamr@2: # include <stl/_string_io.h>
williamr@2: #endif
williamr@2: 
williamr@2: #ifndef _STLP_INTERNAL_CTYPE_H
williamr@2: # include <stl/_ctype.h>
williamr@2: #endif
williamr@2: 
williamr@2: # ifdef _STLP_DEBUG
williamr@2: #  define basic_string _Nondebug_string
williamr@2: # endif
williamr@2: 
williamr@2: _STLP_BEGIN_NAMESPACE
williamr@2: 
williamr@2: # if defined (_STLP_OWN_IOSTREAMS)
williamr@2: #  define _STLP_USING_IO
williamr@2: # else
williamr@2: #  define _STLP_USING_IO _STLP_USING_VENDOR_STD
williamr@2: # endif
williamr@2: 
williamr@2: #if defined (_STLP_USE_NEW_IOSTREAMS)
williamr@2: 
williamr@2: template <class _CharT, class _Traits>
williamr@2: bool _STLP_CALL
williamr@2: __stlp_string_fill(basic_ostream<_CharT, _Traits>& __os,
williamr@2:                   basic_streambuf<_CharT, _Traits>* __buf,
williamr@2:                   size_t __n)
williamr@2: {
williamr@2:   _CharT __f = __os.fill();
williamr@2:   size_t __i;
williamr@2:   bool __ok = true;
williamr@2: 
williamr@2:   for (__i = 0; __i < __n; ++__i)
williamr@2:     __ok = __ok && !_Traits::eq_int_type(__buf->sputc(__f), _Traits::eof());
williamr@2:   return __ok;
williamr@2: }
williamr@2: 
williamr@2: template <class _CharT, class _Traits, class _Alloc>
williamr@2: basic_ostream<_CharT, _Traits>& _STLP_CALL
williamr@2: operator<<(basic_ostream<_CharT, _Traits>& __os, 
williamr@2:            const basic_string<_CharT,_Traits,_Alloc>& __s)
williamr@2: {
williamr@2: 
williamr@2:   _STLP_USING_IO
williamr@2:   typedef basic_ostream<_CharT, _Traits> __ostream;
williamr@2:   typename __ostream::sentry __sentry(__os);
williamr@2:   bool __ok = false;
williamr@2: 
williamr@2:   if (__sentry) {
williamr@2:     __ok = true;
williamr@2:     size_t __n = __s.size();
williamr@2:     size_t __pad_len = 0;
williamr@2:     const bool __left = (__os.flags() & __ostream::left) != 0;
williamr@2:     const size_t __w = __os.width(0);
williamr@2:     basic_streambuf<_CharT, _Traits>* __buf = __os.rdbuf();
williamr@2: 
williamr@2:     if (__n < __w) {
williamr@2:       __pad_len = __w - __n;
williamr@2:     }
williamr@2:     
williamr@2:     if (!__left)
williamr@2:       __ok = __stlp_string_fill(__os, __buf, __pad_len);    
williamr@2: 
williamr@2:     __ok = __ok && (__buf->sputn(__s.data(), streamsize(__n)) == streamsize(__n));
williamr@2: 
williamr@2:     if (__left)
williamr@2:       __ok = __ok && __stlp_string_fill(__os, __buf, __pad_len);
williamr@2:   }
williamr@2: 
williamr@2:   if (!__ok)
williamr@2:     __os.setstate(__ostream::failbit);
williamr@2: 
williamr@2:   return __os;
williamr@2: }
williamr@2:  
williamr@2: template <class _CharT, class _Traits, class _Alloc>
williamr@2: basic_istream<_CharT, _Traits>& _STLP_CALL 
williamr@2: operator>>(basic_istream<_CharT, _Traits>& __is,
williamr@2:            basic_string<_CharT,_Traits, _Alloc>& __s)
williamr@2: {
williamr@2:   _STLP_USING_IO
williamr@2:   typedef basic_istream<_CharT, _Traits> __istream;
williamr@2:   typename __istream::sentry __sentry(__is);
williamr@2: 
williamr@2:   if (__sentry) {
williamr@2:     basic_streambuf<_CharT, _Traits>* __buf = __is.rdbuf();
williamr@2:     typedef ctype<_CharT> _C_type;
williamr@2: 
williamr@2: #ifdef _STLP_OWN_IOSTREAMS
williamr@2:     //    const _C_type& _Ctype = use_facet<_C_type>(__loc);
williamr@2:     const _C_type& _Ctype = *(const _C_type*)__is._M_ctype_facet();
williamr@2: #else
williamr@2: # if defined (_STLP_MSVC) && (_STLP_MSVC <= 1200 ) || defined (__ICL)
williamr@2:     const locale& __loc = __is.getloc();
williamr@2:     const _C_type& _Ctype = use_facet(__loc , ( _C_type * ) 0, true);
williamr@2: # elif defined (__SUNPRO_CC)
williamr@2:     const locale& __loc = __is.getloc();
williamr@2:     const _C_type& _Ctype = use_facet(__loc , ( _C_type * ) 0);
williamr@2: # else
williamr@2:     const locale& __loc = __is.getloc();
williamr@2:     const _C_type& _Ctype = use_facet<_C_type>(__loc);
williamr@2: # endif
williamr@2: #endif
williamr@2:     __s.clear();
williamr@2:     size_t __n = __is.width(0);
williamr@2:     if (__n == 0)
williamr@2:       __n = __STATIC_CAST(size_t,-1);
williamr@2:     else
williamr@2:       __s.reserve(__n);
williamr@2:     
williamr@2: 
williamr@2:     while (__n-- > 0) {
williamr@2:       typename _Traits::int_type __c1 = __buf->sbumpc();
williamr@2:       if (_Traits::eq_int_type(__c1, _Traits::eof())) {
williamr@2:         __is.setstate(__istream::eofbit);
williamr@2:         break;
williamr@2:       }
williamr@2:       else {
williamr@2:         _CharT __c = _Traits::to_char_type(__c1);
williamr@2: 
williamr@2:         if (_Ctype.is(_C_type::space, __c)) {
williamr@2:           if (_Traits::eq_int_type(__buf->sputbackc(__c), _Traits::eof()))
williamr@2:             __is.setstate(__istream::failbit);
williamr@2:           break;
williamr@2:         }
williamr@2: #ifdef __SYMBIAN32__
williamr@2:         else if (__c == '\b') {
williamr@2:           __s.pop_back();
williamr@2:         }
williamr@2: #endif
williamr@2:         else
williamr@2:           __s.push_back(__c);
williamr@2:       }
williamr@2:     }
williamr@2:     
williamr@2:     // If we have read no characters, then set failbit.
williamr@2:     if (__s.size() == 0)
williamr@2:       __is.setstate(__istream::failbit);
williamr@2:   }
williamr@2:   else
williamr@2:     __is.setstate(__istream::failbit);
williamr@2: 
williamr@2:   return __is;
williamr@2: }
williamr@2: 
williamr@2: template <class _CharT, class _Traits, class _Alloc>    
williamr@2: basic_istream<_CharT, _Traits>& _STLP_CALL 
williamr@2: getline(basic_istream<_CharT, _Traits>& __is,
williamr@2:         basic_string<_CharT,_Traits,_Alloc>& __s,
williamr@2:         _CharT __delim)
williamr@2: {
williamr@2:   _STLP_USING_IO
williamr@2:   typedef basic_istream<_CharT, _Traits> __istream;
williamr@2:   size_t __nread = 0;
williamr@2:   typename basic_istream<_CharT, _Traits>::sentry __sentry(__is, true);
williamr@2:   if (__sentry) {
williamr@2:     basic_streambuf<_CharT, _Traits>* __buf = __is.rdbuf();
williamr@2:     __s.clear();
williamr@2: 
williamr@2:     while (__nread < __s.max_size()) {
williamr@2:       int __c1 = __buf->sbumpc();
williamr@2:       if (_Traits::eq_int_type(__c1, _Traits::eof())) {
williamr@2:         __is.setstate(__istream::eofbit);
williamr@2:         break;
williamr@2:       }
williamr@2:       else {
williamr@2:         ++__nread;
williamr@2:         _CharT __c = _Traits::to_char_type(__c1);
williamr@2:         if (!_Traits::eq(__c, __delim)) 
williamr@2:           __s.push_back(__c);
williamr@2:         else
williamr@2:           break;              // Character is extracted but not appended.
williamr@2:       }
williamr@2:     }
williamr@2:   }
williamr@2:   if (__nread == 0 || __nread >= __s.max_size())
williamr@2:     __is.setstate(__istream::failbit);
williamr@2: 
williamr@2:   return __is;
williamr@2: }
williamr@2: 
williamr@2: #elif ! defined ( _STLP_USE_NO_IOSTREAMS )
williamr@2: 
williamr@2: // (reg) For Watcom IO, _OSTREAM_DLL tells if ostream class is in .exe or in .dll
williamr@2: 
williamr@2: template <class _CharT, class _Traits, class _Alloc>
williamr@2: _OSTREAM_DLL&  _STLP_CALL operator<<(_OSTREAM_DLL& __os, 
williamr@2:                     const basic_string<_CharT,_Traits,_Alloc>& __s)
williamr@2: {
williamr@2:   _STLP_USING_IO
williamr@2:   streambuf* __buf = __os.rdbuf();
williamr@2:   if (__buf) {
williamr@2:     size_t __n = __s.size();
williamr@2:     size_t __pad_len = 0;
williamr@2:     const bool __left = (__os.flags() & ios::left) !=0;
williamr@2:     const size_t __w = __os.width();
williamr@2: 
williamr@2:     if (__n < __w) { 
williamr@2:       __pad_len = __w - __n; 
williamr@2:     } 
williamr@2:     
williamr@2:     if (!__left)
williamr@2:       __stlp_string_fill(__os, __buf, __pad_len);
williamr@2:   
williamr@2:     const size_t __nwritten = __buf->sputn(__s.data(), __n);
williamr@2: 
williamr@2:     if (__left)
williamr@2:       __stlp_string_fill(__os, __buf, __pad_len);
williamr@2: 
williamr@2:     if (__nwritten != __n)
williamr@2:       __os.clear(__os.rdstate() | ios::failbit);
williamr@2: 
williamr@2:     __os.width(0);
williamr@2:   }
williamr@2:   else
williamr@2:     __os.clear(__os.rdstate() | ios::badbit);
williamr@2: 
williamr@2:   return __os;
williamr@2: }
williamr@2: 
williamr@2: template <class _CharT, class _Traits, class _Alloc>
williamr@2: _ISTREAM_DLL& _STLP_CALL operator>>(_ISTREAM_DLL& __is, basic_string<_CharT,_Traits,_Alloc>& __s)
williamr@2: {
williamr@2:   _STLP_USING_IO
williamr@2:   if (!__is)
williamr@2:     return __is;
williamr@2: 
williamr@2:   streambuf* __buf = __is.rdbuf();
williamr@2:   if (__buf) {
williamr@2: 
williamr@2:     if (__is.flags() & ios::skipws) {
williamr@2:       //      _CharT __c;
williamr@2:       int __c;
williamr@2:       do {
williamr@2:         __c = __buf->sbumpc();
williamr@2:       }
williamr@2:       while (__c != EOF && isspace((unsigned char)__c));
williamr@2: 
williamr@2:       if (__c == EOF) {
williamr@2:         __is.clear(__is.rdstate() | ios::eofbit | ios::failbit);
williamr@2:       }
williamr@2:       else {
williamr@2: 	if (__buf->sputbackc(__c) == EOF)
williamr@2: 	  __is.clear(__is.rdstate() | ios::failbit);
williamr@2:       }
williamr@2:     }
williamr@2: 
williamr@2:     // If we arrive at end of file (or fail for some other reason) while
williamr@2:     // still discarding whitespace, then we don't try to read the string.
williamr@2:     if (__is) {
williamr@2:       __s.clear();
williamr@2: 
williamr@2:       size_t __n = __is.width();
williamr@2:       if (__n == 0)
williamr@2:         __n = __STATIC_CAST(size_t,-1);
williamr@2:       else
williamr@2:         __s.reserve(__n);
williamr@2: 
williamr@2:       while (__n-- > 0) {
williamr@2:         int __c1 = __buf->sbumpc();
williamr@2:         if (__c1 == EOF) {
williamr@2:           __is.clear(__is.rdstate() | ios::eofbit);
williamr@2:           break;
williamr@2:         }
williamr@2:         else {
williamr@2:           _CharT __c = _Traits::to_char_type(__c1);
williamr@2: 
williamr@2:           if (isspace((unsigned char) __c)) {
williamr@2:             if (__buf->sputbackc(__c) == EOF)
williamr@2:               __is.clear(__is.rdstate() | ios::failbit);
williamr@2:             break;
williamr@2:           }
williamr@2:           else
williamr@2:             __s.push_back(__c);
williamr@2:         }
williamr@2:       }
williamr@2:     
williamr@2:       // If we have read no characters, then set failbit.
williamr@2:       if (__s.size() == 0)
williamr@2:         __is.clear(__is.rdstate() | ios::failbit);
williamr@2:     }
williamr@2: 
williamr@2:     __is.width(0);
williamr@2:   }
williamr@2:   else                          // We have no streambuf.
williamr@2:     __is.clear(__is.rdstate() | ios::badbit);
williamr@2: 
williamr@2:   return __is;
williamr@2: }
williamr@2: 
williamr@2: template <class _CharT, class _Traits, class _Alloc>    
williamr@2: _ISTREAM_DLL& _STLP_CALL getline(_ISTREAM_DLL& __is,
williamr@2:                  basic_string<_CharT,_Traits,_Alloc>& __s,
williamr@2:                  _CharT __delim)
williamr@2: {
williamr@2:   _STLP_USING_IO
williamr@2:   streambuf* __buf = __is.rdbuf();
williamr@2:   if (__buf) {
williamr@2:     size_t __nread = 0;
williamr@2:     if (__is) {
williamr@2:       __s.clear();
williamr@2: 
williamr@2:       while (__nread < __s.max_size()) {
williamr@2:         int __c1 = __buf->sbumpc();
williamr@2:         if (__c1 == EOF) {
williamr@2:           __is.clear(__is.rdstate() | ios::eofbit);
williamr@2:           break;
williamr@2:         }
williamr@2:         else {
williamr@2:           ++__nread;
williamr@2:           _CharT __c = _Traits::to_char_type(__c1);
williamr@2:           if (!_Traits::eq(__c, __delim)) 
williamr@2:             __s.push_back(__c);
williamr@2:           else
williamr@2:             break;              // Character is extracted but not appended.
williamr@2:         }
williamr@2:       }
williamr@2:     }
williamr@2: 
williamr@2:     if (__nread == 0 || __nread >= __s.max_size())
williamr@2:       __is.clear(__is.rdstate() | ios::failbit);
williamr@2:   }
williamr@2:   else
williamr@2:     __is.clear(__is.rdstate() | ios::badbit);
williamr@2: 
williamr@2:   return __is;
williamr@2: }
williamr@2: 
williamr@2: # endif /* _STLP_NEW_IOSTREAMS */
williamr@2: 
williamr@2: _STLP_END_NAMESPACE
williamr@2: 
williamr@2: // # undef _STLP_USING_IO
williamr@2: # undef basic_string
williamr@2: 
williamr@2: #endif