src/common/sstream.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Name:        common/sstream.cpp
   3 // Purpose:     string-based streams implementation
   4 // Author:      Vadim Zeitlin
   5 // Modified by: Ryan Norton (UTF8 UNICODE)
   6 // Created:     2004-09-19
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) 2004 Vadim Zeitlin <vadim@wxwindows.org>
   9 // Licence:     wxWindows licence
  10 ///////////////////////////////////////////////////////////////////////////////
  11
  12 // ============================================================================
  13 // declarations
  14 // ============================================================================
  15
  16 // ----------------------------------------------------------------------------
  17 // headers
  18 // ----------------------------------------------------------------------------
  19
  20 // For compilers that support precompilation, includes "wx.h".
  21 #include "wx/wxprec.h"
  22
  23 #ifdef __BORLANDC__
  24     #pragma hdrstop
  25 #endif
  26
  27 #if wxUSE_STREAMS
  28
  29 #include "wx/sstream.h"
  30
  31 // ============================================================================
  32 // wxStringInputStream implementation
  33 // ============================================================================
  34
  35 // ----------------------------------------------------------------------------
  36 // construction/destruction
  37 // ----------------------------------------------------------------------------
  38
  39 // TODO:  Do we want to include the null char in the stream?  If so then
  40 // just add +1 to m_len in the ctor
  41 wxStringInputStream::wxStringInputStream(const wxString& s)
  42 #if wxUSE_UNICODE
  43     : m_str(s), m_buf(wxMBConvUTF8().cWX2MB(s).release()), m_len(strlen(m_buf))
  44 #else
  45     : m_str(s), m_buf((char*)s.c_str()), m_len(s.length())
  46 #endif
  47 {
  48 #if wxUSE_UNICODE
  49     wxASSERT_MSG(m_buf != NULL, _T("Could not convert string to UTF8!"));
  50 #endif
  51     m_pos = 0;
  52 }
  53
  54 wxStringInputStream::~wxStringInputStream()
  55 {
  56 #if wxUSE_UNICODE
  57     // Note: wx[W]CharBuffer uses malloc()/free()
  58     free(m_buf);
  59 #endif
  60 }
  61
  62 // ----------------------------------------------------------------------------
  63 // getlength
  64 // ----------------------------------------------------------------------------
  65
  66 wxFileOffset wxStringInputStream::GetLength() const
  67 {
  68     return m_len;
  69 }
  70
  71 // ----------------------------------------------------------------------------
  72 // seek/tell
  73 // ----------------------------------------------------------------------------
  74
  75 wxFileOffset wxStringInputStream::OnSysSeek(wxFileOffset ofs, wxSeekMode mode)
  76 {
  77     switch ( mode )
  78     {
  79         case wxFromStart:
  80             // nothing to do, ofs already ok
  81             break;
  82
  83         case wxFromEnd:
  84             ofs += m_len;
  85             break;
  86
  87         case wxFromCurrent:
  88             ofs += m_pos;
  89             break;
  90
  91         default:
  92             wxFAIL_MSG( _T("invalid seek mode") );
  93             return wxInvalidOffset;
  94     }
  95
  96     if ( ofs < 0 || wx_static_cast(size_t, ofs) > m_len )
  97         return wxInvalidOffset;
  98
  99     m_pos = wx_static_cast(size_t, ofs);
 100
 101     return ofs;
 102 }
 103
 104 wxFileOffset wxStringInputStream::OnSysTell() const
 105 {
 106     return wx_static_cast(wxFileOffset, m_pos);
 107 }
 108
 109 // ----------------------------------------------------------------------------
 110 // actual IO
 111 // ----------------------------------------------------------------------------
 112
 113 size_t wxStringInputStream::OnSysRead(void *buffer, size_t size)
 114 {
 115     const size_t sizeMax = m_len - m_pos;
 116
 117     if ( size >= sizeMax )
 118     {
 119         if ( sizeMax == 0 )
 120         {
 121             m_lasterror = wxSTREAM_EOF;
 122             return 0;
 123         }
 124
 125         size = sizeMax;
 126     }
 127
 128     memcpy(buffer, m_buf + m_pos, size);
 129     m_pos += size;
 130
 131     return size;
 132 }
 133
 134 // ============================================================================
 135 // wxStringOutputStream implementation
 136 // ============================================================================
 137
 138 // ----------------------------------------------------------------------------
 139 // seek/tell
 140 // ----------------------------------------------------------------------------
 141
 142 wxFileOffset wxStringOutputStream::OnSysTell() const
 143 {
 144     return wx_static_cast(wxFileOffset, m_pos);
 145 }
 146
 147 // ----------------------------------------------------------------------------
 148 // actual IO
 149 // ----------------------------------------------------------------------------
 150
 151 size_t wxStringOutputStream::OnSysWrite(const void *buffer, size_t size)
 152 {
 153     const char *p = wx_static_cast(const char *, buffer);
 154
 155     // append the input buffer (may not be null terminated - thus
 156     // the literal length
 157     m_str->Append(wxString(p, m_conv, size));
 158
 159     // update position
 160     m_pos += size;
 161
 162     // return number of bytes actually written
 163     return size;
 164 }
 165
 166 #endif // wxUSE_STREAMS
 167