memory leak plugged

[wxWidgets.git] / src / common / datstrm.cpp

diff --git a/src/common/datstrm.cpp b/src/common/datstrm.cpp
index 980a280c137fad6de0d7a56705225cede7b596a6..d455a6dc4ea4182a2772a0536a775aec3f3a6434 100644 (file)
--- a/src/common/datstrm.cpp
+++ b/src/common/datstrm.cpp
@@ -26,36 +26,24 @@
 
 #include "wx/datstrm.h"
 
-wxDataStream::wxDataStream(istream& s)
-{
-  m_istream = &s;
-  m_ostream = NULL;
-}
-
-wxDataStream::wxDataStream(ostream& s)
-{
-  m_istream = NULL;
-  m_ostream = &s;
-}
+// ---------------------------------------------------------------------------
+// wxDataInputStream
+// ---------------------------------------------------------------------------
 
-wxDataStream::wxDataStream(iostream& s)
+wxDataInputStream::wxDataInputStream(wxInputStream& s)
+  : wxFilterInputStream(s)
 {
-  m_istream = &s;
-  m_ostream = &s;
 }
 
-wxDataStream::~wxDataStream()
+wxDataInputStream::~wxDataInputStream()
 {
 }
 
-unsigned long wxDataStream::Read32()
+unsigned long wxDataInputStream::Read32()
 {
   char buf[4];
 
-  if (!m_istream)
-    return 0;
-
-  m_istream->read(buf, 4);
+  Read(buf, 4);
 
   return (unsigned long)buf[0] |
          ((unsigned long)buf[1] << 8) |
@@ -63,113 +51,152 @@ unsigned long wxDataStream::Read32()
          ((unsigned long)buf[3] << 24);
 }
 
-unsigned short wxDataStream::Read16()
+unsigned short wxDataInputStream::Read16()
 {
   char buf[2];
 
-  if (!m_istream)
-    return 0;
-
-  m_istream->read(buf, 2);
+  Read(buf, 2);
 
   return (unsigned short)buf[0] |
          ((unsigned short)buf[1] << 8);
 }
 
-unsigned char wxDataStream::Read8()
+unsigned char wxDataInputStream::Read8()
 {
   char buf;
 
-  if (!m_istream)
-    return 0;
-
-  m_istream->read(&buf, 1);
+  Read(&buf, 1);
   return (unsigned char)buf;
 }
 
 // Must be at global scope for VC++ 5
 extern "C" double ConvertFromIeeeExtended(const unsigned char *bytes);
 
-double wxDataStream::ReadDouble()
+double wxDataInputStream::ReadDouble()
 {
-#if USE_APPLE_IEEE
+#if wxUSE_APPLE_IEEE
   char buf[10];
 
-  if (!m_istream)
-    return 0.0;
-
-  m_istream->read(buf, 10);
+  Read(buf, 10);
   return ConvertFromIeeeExtended((unsigned char *)buf);
 #else
   return 0.0;
 #endif
 }
 
-wxString wxDataStream::ReadLine()
+wxString wxDataInputStream::ReadLine()
+{
+  char c, last_endl = 0;
+  bool end_line = FALSE;
+  wxString line;
+
+  while (!end_line) {
+    c = GetC();
+    switch (c) {
+    case '\n':
+      end_line = TRUE;
+      break;
+    case '\r':
+      last_endl = '\r';
+      break;
+    default:
+      if (last_endl == '\r') {
+        end_line = TRUE;
+        InputStreamBuffer()->WriteBack(c);
+        break;
+      }
+      line += c;
+      break;
+    } 
+  }
+  return line;
+}
+
+wxString wxDataInputStream::ReadString()
 {
-  char i_strg[255];
+  wxString wx_string;
+  char *string;
+  unsigned long len;
 
-  if (!m_istream)
-    return "";
+  len = Read32();
+  string = new char[len+1];
 
-  m_istream->getline(i_strg, 255);
-  return i_strg;
+  Read(string, len);
+
+  string[len] = 0;
+  wx_string = string;
+  delete string;
+
+  return wx_string;
 }
 
-void wxDataStream::Write32(unsigned long i)
+// ---------------------------------------------------------------------------
+// wxDataOutputStream
+// ---------------------------------------------------------------------------
+
+wxDataOutputStream::wxDataOutputStream(wxOutputStream& s)
+ : wxFilterOutputStream(s)
 {
-  char buf[4];
+}
+
+wxDataOutputStream::~wxDataOutputStream()
+{
+}
 
-  if (!m_ostream)
-    return;
+void wxDataOutputStream::Write32(unsigned long i)
+{
+  char buf[4];
 
   buf[0] = i & 0xff;
   buf[1] = (i >> 8) & 0xff;
   buf[2] = (i >> 16) & 0xff;
   buf[3] = (i >> 24) & 0xff;
-  m_ostream->write(buf, 4);
+  Write(buf, 4);
 }
 
-void wxDataStream::Write16(unsigned short i)
+void wxDataOutputStream::Write16(unsigned short i)
 {
   char buf[2];
 
-  if (!m_ostream)
-    return;
-
   buf[0] = i & 0xff;
   buf[1] = (i >> 8) & 0xff;
-  m_ostream->write(buf, 2);
+  Write(buf, 2);
 }
 
-void wxDataStream::Write8(unsigned char i)
+void wxDataOutputStream::Write8(unsigned char i)
 {
-  if (!m_ostream)
-    return;
-
-  m_ostream->write(&i, 1);
+  Write(&i, 1);
 }
 
-void wxDataStream::WriteLine(const wxString& line)
+void wxDataOutputStream::WriteLine(const wxString& line)
 {
+#ifdef __WXMSW__
+  wxString tmp_string = line + "\r\n";
+#else
   wxString tmp_string = line + '\n';
+#endif
 
-  if (!m_ostream)
-    return;
+  Write((const char *) tmp_string, tmp_string.Length());
+}
 
-  m_ostream->write((const char *) tmp_string, tmp_string.Length());
+void wxDataOutputStream::WriteString(const wxString& string)
+{
+  Write32(string.Length());
+  Write((const char *) string, string.Length());
 }
 
 // Must be at global scope for VC++ 5
 extern "C" void ConvertToIeeeExtended(double num, unsigned char *bytes);
 
-void wxDataStream::WriteDouble(double d)
+void wxDataOutputStream::WriteDouble(double d)
 {
   char buf[10];
 
-  if (!m_ostream)
-    return;
-
+#if wxUSE_APPLE_IEEE
   ConvertToIeeeExtended(d, (unsigned char *)buf);
-  m_ostream->write(buf, 10);
+#else
+#  pragma warning "wxDataStream::WriteDouble() not using IeeeExtended - will not work!"
+ buf[0] = '\0';
+#endif
+  Write(buf, 10);
 }