]> git.saurik.com Git - wxWidgets.git/blobdiff - src/common/txtstrm.cpp
don't crash when loading images with verbose==false (patch 1449823)
[wxWidgets.git] / src / common / txtstrm.cpp
index 70ca315e43d15ca2e33284c23989344279f66d17..61260824e26937e3aca5ce46cad411d63c430f89 100644 (file)
@@ -6,13 +6,9 @@
 // Created:     28/06/98
 // RCS-ID:      $Id$
 // Copyright:   (c) Guilhem Lavaux
-// Licence:    wxWindows license
+// Licence:     wxWindows licence
 /////////////////////////////////////////////////////////////////////////////
 
-#ifdef __GNUG__
-#pragma implementation "txtstrm.h"
-#endif
-
 // For compilers that support precompilation, includes "wx.h".
 #include "wx/wxprec.h"
 
 // wxTextInputStream
 // ----------------------------------------------------------------------------
 
-wxTextInputStream::wxTextInputStream(wxInputStream& s)
-  : m_input(&s)
+#if wxUSE_UNICODE
+wxTextInputStream::wxTextInputStream(wxInputStream &s, const wxString &sep, wxMBConv& conv)
+  : m_input(s), m_separators(sep), m_conv(conv)
+{
+    memset((void*)m_lastBytes, 0, 10);
+}
+#else
+wxTextInputStream::wxTextInputStream(wxInputStream &s, const wxString &sep)
+  : m_input(s), m_separators(sep)
 {
+    memset((void*)m_lastBytes, 0, 10);
 }
+#endif
 
 wxTextInputStream::~wxTextInputStream()
 {
 }
 
-wxChar wxTextInputStream::NextNonWhiteSpace()
+void wxTextInputStream::UngetLast()
 {
-    wxChar c = (wxChar) 0;
-    for (;;)
+    size_t byteCount = 0;
+    while(m_lastBytes[byteCount]) // pseudo ANSI strlen (even for Unicode!)
+        byteCount++;
+    m_input.Ungetch(m_lastBytes, byteCount);
+    memset((void*)m_lastBytes, 0, 10);
+}
+
+wxChar wxTextInputStream::NextChar()
+{
+#if wxUSE_UNICODE
+    wxChar wbuf[2];
+    memset((void*)m_lastBytes, 0, 10);
+    for(size_t inlen = 0; inlen < 9; inlen++)
     {
-        c = m_input->GetC();
-       if (!m_input) return (wxChar) 0;
-       
-       if (c != _T('\n') &&
-           c != _T('\r') &&
-           c != _T('\t') &&
-           c != _T(' '))
-       {
-           return c;
-       }
+        // actually read the next character
+        m_lastBytes[inlen] = m_input.GetC();
+
+        if(m_input.LastRead() <= 0)
+            return wxEOT;
+
+        int retlen = (int) m_conv.MB2WC(wbuf, m_lastBytes, 2); // returns -1 for failure
+        if(retlen >= 0) // res == 0 could happen for '\0' char
+            return wbuf[0];
     }
+    // there should be no encoding which requires more than nine bytes for one character...
+    return wxEOT;
+#else
+    m_lastBytes[0] = m_input.GetC();
+
+    if(m_input.LastRead() <= 0)
+        return wxEOT;
+
+    return m_lastBytes[0];
+#endif
 
-    // this shouldn't happen
-    return (wxChar) 0;
 }
 
-void wxTextInputStream::SkipIfEndOfLine( wxChar c )
+wxChar wxTextInputStream::NextNonSeparators()
 {
-    if (c == _T('\n'))
+    for (;;)
     {
-        // eat on UNIX
-       return;
-    }
+        wxChar c = NextChar();
+        if (c == wxEOT) return (wxChar) 0;
 
-    if (c == _T('\r'))
-    {
-        // eat on both Mac and DOS
-       
-        wxChar c2 = m_input->GetC();
-        if (!m_input) return;
-       
-        if (c2 == _T('\n'))
-        {
-           // eat on DOS
-           return;
-       }
-        else
-       {
-           // Don't eat on Mac
-            m_input->Ungetch( c2 );
-       }
-    }
-    else
-    {
-        // no line terminator
-        m_input->Ungetch( c );
+        if (c != wxT('\n') &&
+            c != wxT('\r') &&
+            !m_separators.Contains(c))
+          return c;
     }
+
 }
 
-wxUint32 wxTextInputStream::Read32()
+bool wxTextInputStream::EatEOL(const wxChar &c)
 {
-    /* I only implemented a simple integer parser */
-    int sign;
-    wxInt32 i;
-
-    int c = NextNonWhiteSpace();
-    if (!m_input) return 0;
+    if (c == wxT('\n')) return true; // eat on UNIX
 
-    i = 0;
-    if (! (c == _T('-') || c == _T('+') || isdigit(c)) )
+    if (c == wxT('\r')) // eat on both Mac and DOS
     {
-        m_input->Ungetch(c);
-        return 0;
-    }
+        wxChar c2 = NextChar();
+        if(c2 == wxEOT) return true; // end of stream reached, had enough :-)
 
-    if (c == _T('-'))
-    {
-        sign = -1;
-        c = m_input->GetC();
-    } else
-    if (c == _T('+'))
-    {
-        sign = 1;
-        c = m_input->GetC();
-    } else
-    {
-        sign = 1;
+        if (c2 != wxT('\n')) UngetLast(); // Don't eat on Mac
+        return true;
     }
 
-    while (isdigit(c))
-    {
-        i = i*10 + (c - (int)_T('0'));
-        c = m_input->GetC();
-    }
+    return false;
+}
 
-    SkipIfEndOfLine( c );
+wxUint32 wxTextInputStream::Read32(int base)
+{
+    wxASSERT_MSG( !base || (base > 1 && base <= 36), _T("invalid base") );
+    if(!m_input) return 0;
 
-    i *= sign;
+    wxString word = ReadWord();
+    if(word.empty())
+        return 0;
+    return wxStrtoul(word.c_str(), 0, base);
+}
 
-    return (wxUint32)i;
+wxUint16 wxTextInputStream::Read16(int base)
+{
+    return (wxUint16)Read32(base);
 }
 
-wxUint16 wxTextInputStream::Read16()
+wxUint8 wxTextInputStream::Read8(int base)
 {
-    return (wxUint16)Read32();
+    return (wxUint8)Read32(base);
 }
 
-wxUint8 wxTextInputStream::Read8()
+wxInt32 wxTextInputStream::Read32S(int base)
 {
-    return (wxUint8)Read32();
+    wxASSERT_MSG( !base || (base > 1 && base <= 36), _T("invalid base") );
+    if(!m_input) return 0;
+
+    wxString word = ReadWord();
+    if(word.empty())
+        return 0;
+    return wxStrtol(word.c_str(), 0, base);
+}
+
+wxInt16 wxTextInputStream::Read16S(int base)
+{
+    return (wxInt16)Read32S(base);
+}
+
+wxInt8 wxTextInputStream::Read8S(int base)
+{
+    return (wxInt8)Read32S(base);
 }
 
 double wxTextInputStream::ReadDouble()
 {
-    /* I only implemented a simple float parser */
-    double f;
-    int sign;
+    if(!m_input) return 0;
+    wxString word = ReadWord();
+    if(word.empty())
+        return 0;
+    return wxStrtod(word.c_str(), 0);
+}
 
-    int c = NextNonWhiteSpace();
-    if (!m_input) return 0.0;
+wxString wxTextInputStream::ReadString()
+{
+    return ReadLine();
+}
 
-    f = 0.0;
-    if (! (c == _T('.') || c == _T('-') || c == _T('+') || isdigit(c)) )
-    {
-        m_input->Ungetch(c);
-        return 0.0;
-    }
+wxString wxTextInputStream::ReadLine()
+{
+    wxString line;
 
-    if (c == _T('-'))
+    while ( !m_input.Eof() )
     {
-        sign = -1;
-        c = m_input->GetC();
-    } else
-    if (c == _T('+'))
-    {
-        sign = 1;
-        c = m_input->GetC();
-    }
-    else
-    {
-        sign = 1;
-    }
+        wxChar c = NextChar();
+        if(c == wxEOT)
+            break;
 
-    while (isdigit(c))
-    {
-        f = f*10 + (c - _T('0'));
-        c = m_input->GetC();
-    }
+        if ( !m_input )
+            break;
 
-    if (c == _T('.'))
-    {
-        double f_multiplicator = (double) 0.1;
+        if (EatEOL(c))
+            break;
 
-        c = m_input->GetC();
+        line += c;
+    }
 
-        while (isdigit(c))
-       {
-            f += (c-_T('0'))*f_multiplicator;
-            f_multiplicator /= 10;
-            c = m_input->GetC();
-        }
+    return line;
+}
 
-        if (c == _T('e'))
-       {
-            double f_multiplicator = 0.0;
-            int i, e;
+wxString wxTextInputStream::ReadWord()
+{
+    wxString word;
 
-            c = m_input->GetC();
+    if ( !m_input )
+        return word;
 
-            switch (c)
-           {
-                case _T('-'): f_multiplicator = 0.1;  break;
-               case _T('+'): f_multiplicator = 10.0; break;
-           }
+    wxChar c = NextNonSeparators();
+    if ( !c )
+        return word;
 
-            e = Read8();  // why only max 256 ?
+    word += c;
 
-            for (i=0;i<e;i++)
-                f *= f_multiplicator;
-        }
-       else
-           SkipIfEndOfLine( c );
-    }
-    else
+    while ( !m_input.Eof() )
     {
-        m_input->Ungetch(c);
-    }
+        c = NextChar();
+        if(c == wxEOT)
+            break;
 
-    f *= sign;
+        if (m_separators.Contains(c))
+            break;
 
-    return f;
-}
+        if (EatEOL(c))
+            break;
 
-wxString wxTextInputStream::ReadString()
-{
-    wxChar c;
-    wxString line;
-
-    for (;;)
-    {
-        c = m_input->GetC();
-        if (!m_input) break;
-       
-        if (c == _T('\n'))
-        {
-           // eat on UNIX
-           break;
-       }
-       
-        if (c == _T('\r'))
-        {
-            // eat on both Mac and DOS
-       
-            wxChar c2 = m_input->GetC();
-           if (!m_input) break;
-       
-           if (c2 == _T('\n'))
-            {
-               // eat on DOS
-               break;
-           }
-            else
-           {
-               // Don't eat on Mac
-                m_input->Ungetch( c2 );
-               break;
-           }
-        }
-       
-        line += c;
+        word += c;
     }
 
-    return line;
+    return word;
 }
 
-wxTextInputStream& wxTextInputStream::operator>>(wxString& line)
+wxTextInputStream& wxTextInputStream::operator>>(wxString& word)
 {
-  line = ReadString();
-  return *this;
+    word = ReadWord();
+    return *this;
 }
 
-wxTextInputStream& wxTextInputStream::operator>>(wxChar& c)
+wxTextInputStream& wxTextInputStream::operator>>(char& c)
 {
-    wxChar c1 = m_input->GetC();
-    if (!m_input)
-    {
-        c = (wxChar) 0;
-        return *this;
-    }
+    c = m_input.GetC();
+    if(m_input.LastRead() <= 0) c = 0;
 
-    if (c1 == _T('\r'))
-    {
-        c = _T('\n');
-        wxChar c2 = m_input->GetC();
-       if (!m_input) return *this;
-       
-       if (c2 != _T('\n'))
-       {
-           // we are on a Mac
-            m_input->Ungetch( c2 );
-       }
-    }
-    else
-    {
-        c = c1;
-    }
+    if (EatEOL(c))
+        c = '\n';
+
+    return *this;
+}
+
+#if wxUSE_UNICODE && wxWCHAR_T_IS_REAL_TYPE
+
+wxTextInputStream& wxTextInputStream::operator>>(wchar_t& wc)
+{
+    wc = GetChar();
 
     return *this;
 }
 
+#endif // wxUSE_UNICODE
+
 wxTextInputStream& wxTextInputStream::operator>>(wxInt16& i)
 {
     i = (wxInt16)Read16();
@@ -348,19 +295,52 @@ wxTextInputStream& wxTextInputStream::operator>>(float& f)
     return *this;
 }
 
-wxTextOutputStream::wxTextOutputStream(wxOutputStream& s)
-  : m_output(&s)
+
+
+#if wxUSE_UNICODE
+wxTextOutputStream::wxTextOutputStream(wxOutputStream& s, wxEOL mode, wxMBConv& conv)
+  : m_output(s), m_conv(conv)
+#else
+wxTextOutputStream::wxTextOutputStream(wxOutputStream& s, wxEOL mode)
+  : m_output(s)
+#endif
 {
+    m_mode = mode;
+    if (m_mode == wxEOL_NATIVE)
+    {
+#if defined(__WXMSW__) || defined(__WXPM__)
+        m_mode = wxEOL_DOS;
+#elif defined(__WXMAC__) && !defined(__DARWIN__)
+        m_mode = wxEOL_MAC;
+#else
+        m_mode = wxEOL_UNIX;
+#endif
+    }
 }
 
 wxTextOutputStream::~wxTextOutputStream()
 {
 }
 
+void wxTextOutputStream::SetMode(wxEOL mode)
+{
+    m_mode = mode;
+    if (m_mode == wxEOL_NATIVE)
+    {
+#if defined(__WXMSW__) || defined(__WXPM__)
+        m_mode = wxEOL_DOS;
+#elif defined(__WXMAC__) && !defined(__DARWIN__)
+        m_mode = wxEOL_MAC;
+#else
+        m_mode = wxEOL_UNIX;
+#endif
+    }
+}
+
 void wxTextOutputStream::Write32(wxUint32 i)
 {
     wxString str;
-    str.Printf(_T("%u"), i);
+    str.Printf(wxT("%u"), i);
 
     WriteString(str);
 }
@@ -368,7 +348,7 @@ void wxTextOutputStream::Write32(wxUint32 i)
 void wxTextOutputStream::Write16(wxUint16 i)
 {
     wxString str;
-    str.Printf(_T("%u"), i);
+    str.Printf(wxT("%u"), (unsigned)i);
 
     WriteString(str);
 }
@@ -376,7 +356,7 @@ void wxTextOutputStream::Write16(wxUint16 i)
 void wxTextOutputStream::Write8(wxUint8 i)
 {
     wxString str;
-    str.Printf(_T("%u"), i);
+    str.Printf(wxT("%u"), (unsigned)i);
 
     WriteString(str);
 }
@@ -385,42 +365,62 @@ void wxTextOutputStream::WriteDouble(double d)
 {
     wxString str;
 
-    str.Printf(_T("%f"), d);
+    str.Printf(wxT("%f"), d);
     WriteString(str);
 }
 
 void wxTextOutputStream::WriteString(const wxString& string)
 {
-    for (size_t i = 0; i < string.Len(); i++)
+    size_t len = string.length();
+
+    wxString out;
+    out.reserve(len);
+
+    for ( size_t i = 0; i < len; i++ )
     {
-        wxChar c = string[i];
-        if (c == _T('\n'))
+        const wxChar c = string[i];
+        if ( c == wxT('\n') )
         {
-#if   defined(__WINDOWS__)
-            c = _T('\r');
-            m_output->Write( (const void*)(&c), sizeof(wxChar) );
-            c = _T('\n');
-            m_output->Write( (const void*)(&c), sizeof(wxChar) );
-#elif defined(__UNIX__)
-            c = _T('\n');
-            m_output->Write( (const void*)(&c), sizeof(wxChar) );
-#elif defined(__WXMAC__)
-            c = _T('\r');
-            m_output->Write( (const void*)(&c), sizeof(wxChar) );
-#elif   defined(__OS2__)
-            c = _T('\r');
-            m_output->Write( (const void*)(&c), sizeof(wxChar) );
-            c = _T('\n');
-            m_output->Write( (const void*)(&c), sizeof(wxChar) );
+            switch ( m_mode )
+            {
+                case wxEOL_DOS:
+                    out << _T("\r\n");
+                    continue;
+
+                case wxEOL_MAC:
+                    out << _T('\r');
+                    continue;
+
+                default:
+                    wxFAIL_MSG( _T("unknown EOL mode in wxTextOutputStream") );
+                    // fall through
+
+                case wxEOL_UNIX:
+                    // don't treat '\n' specially
+                    ;
+            }
+        }
+
+        out << c;
+    }
+
+    // We must not write the trailing NULL here
+#if wxUSE_UNICODE
+    wxCharBuffer buffer = m_conv.cWC2MB( out );
+    m_output.Write( (const char*) buffer, strlen( (const char*) buffer ) );
 #else
-            #error  "wxTextOutputStream: unsupported platform."
+    m_output.Write(out.c_str(), out.length() );
 #endif
-        }
-        else
-        {
-            m_output->Write( (const void*)(&c), sizeof(wxChar) );
-        }
-   }
+}
+
+wxTextOutputStream& wxTextOutputStream::PutChar(wxChar c)
+{
+#if wxUSE_UNICODE
+    WriteString( wxString(&c, m_conv, 1) );
+#else
+    WriteString( wxString(&c, wxConvLocal, 1) );
+#endif
+    return *this;
 }
 
 wxTextOutputStream& wxTextOutputStream::operator<<(const wxChar *string)
@@ -435,33 +435,57 @@ wxTextOutputStream& wxTextOutputStream::operator<<(const wxString& string)
     return *this;
 }
 
-wxTextOutputStream& wxTextOutputStream::operator<<(wxChar c)
+wxTextOutputStream& wxTextOutputStream::operator<<(char c)
 {
-    WriteString( wxString(c) );
+    WriteString( wxString::FromAscii(c) );
+
     return *this;
 }
 
+#if wxUSE_UNICODE && wxWCHAR_T_IS_REAL_TYPE
+
+wxTextOutputStream& wxTextOutputStream::operator<<(wchar_t wc)
+{
+    WriteString( wxString(&wc, m_conv, 1) );
+
+    return *this;
+}
+
+#endif // wxUSE_UNICODE
+
 wxTextOutputStream& wxTextOutputStream::operator<<(wxInt16 c)
 {
-    Write16( (wxUint16)c );
+    wxString str;
+    str.Printf(wxT("%d"), (signed int)c);
+    WriteString(str);
+
     return *this;
 }
 
 wxTextOutputStream& wxTextOutputStream::operator<<(wxInt32 c)
 {
-    Write32( (wxUint32)c );
+    wxString str;
+    str.Printf(wxT("%ld"), (signed long)c);
+    WriteString(str);
+
     return *this;
 }
 
 wxTextOutputStream& wxTextOutputStream::operator<<(wxUint16 c)
 {
-    Write16(c);
+    wxString str;
+    str.Printf(wxT("%u"), (unsigned int)c);
+    WriteString(str);
+
     return *this;
 }
 
 wxTextOutputStream& wxTextOutputStream::operator<<(wxUint32 c)
 {
-    Write32(c);
+    wxString str;
+    str.Printf(wxT("%lu"), (unsigned long)c);
+    WriteString(str);
+
     return *this;
 }
 
@@ -477,5 +501,10 @@ wxTextOutputStream& wxTextOutputStream::operator<<(float f)
     return *this;
 }
 
+wxTextOutputStream &endl( wxTextOutputStream &stream )
+{
+    return stream.PutChar(wxT('\n'));
+}
+
 #endif
   // wxUSE_STREAMS