X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/2049ba38adafa0ec146880de29f26e32dd69a125..742af0714edae1368d4fd24603a7522214c55745:/src/common/datstrm.cpp

diff --git a/src/common/datstrm.cpp b/src/common/datstrm.cpp
index df9095cdff..befc354784 100644
--- a/src/common/datstrm.cpp
+++ b/src/common/datstrm.cpp
@@ -17,198 +17,270 @@
 #include "wx/wxprec.h"
 
 #ifdef __BORLANDC__
-#pragma hdrstop
+  #pragma hdrstop
 #endif
 
-#ifndef WX_PRECOMP
-#include "wx/defs.h"
-#endif
+#if wxUSE_STREAMS
 
 #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)
+  : m_input(&s), m_be_order(FALSE)
 {
-  m_istream = &s;
-  m_ostream = &s;
 }
 
-wxDataStream::~wxDataStream()
+wxDataInputStream::~wxDataInputStream()
 {
 }
 
-unsigned long wxDataStream::Read32()
+wxUint32 wxDataInputStream::Read32()
 {
-  char buf[4];
-
-  if (!m_istream)
-    return 0;
+  wxUint32 i32;
 
-  m_istream->read(buf, 4);
+  m_input->Read(&i32, 4);
 
-  return (unsigned long)buf[0] |
-         ((unsigned long)buf[1] << 8) |
-         ((unsigned long)buf[2] << 16) |
-         ((unsigned long)buf[3] << 24);
+  if (m_be_order)
+    return wxUINT32_SWAP_ON_LE(i32);
+  else
+    return wxUINT32_SWAP_ON_BE(i32);
 }
 
-unsigned short wxDataStream::Read16()
+wxUint16 wxDataInputStream::Read16()
 {
-  char buf[2];
-
-  if (!m_istream)
-    return 0;
+  wxUint16 i16;
 
-  m_istream->read(buf, 2);
+  m_input->Read(&i16, 2);
 
-  return (unsigned short)buf[0] |
-         ((unsigned short)buf[1] << 8);
+  if (m_be_order)
+    return wxUINT16_SWAP_ON_LE(i16);
+  else
+    return wxUINT16_SWAP_ON_BE(i16);
 }
 
-unsigned char wxDataStream::Read8()
+wxUint8 wxDataInputStream::Read8()
 {
-  char buf;
+  wxUint8 buf;
 
-  if (!m_istream)
-    return 0;
-
-  m_istream->read(&buf, 1);
-  return (unsigned char)buf;
+  m_input->Read(&buf, 1);
+  return (wxUint8)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);
+  m_input->Read(buf, 10);
   return ConvertFromIeeeExtended((unsigned char *)buf);
 #else
   return 0.0;
 #endif
 }
 
-wxString wxDataStream::ReadLine()
-{
-  char i_strg[255];
-
-  if (!m_istream)
-    return "";
-
-  m_istream->getline(i_strg, 255);
-  return i_strg;
-}
-
-wxString wxDataStream::ReadString()
+wxString wxDataInputStream::ReadString()
 {
   wxString wx_string;
   char *string;
   unsigned long len;
 
-  if (!m_istream)
-    return "";
-
   len = Read32();
   string = new char[len+1];
 
-  m_istream->read(string, len);
+  m_input->Read(string, len);
 
   string[len] = 0;
   wx_string = string;
   delete string;
 
-  return wx_string; 
+  return wx_string;
+}
+  
+wxDataInputStream& wxDataInputStream::operator>>(wxString& s)
+{
+  s = ReadString();
+  return *this;
+}
+
+wxDataInputStream& wxDataInputStream::operator>>(wxInt8& c)
+{
+  c = (wxInt8)Read8();
+  return *this;
 }
 
-void wxDataStream::Write32(unsigned long i)
+wxDataInputStream& wxDataInputStream::operator>>(wxInt16& i)
 {
-  char buf[4];
+  i = (wxInt16)Read16();
+  return *this;
+}
 
-  if (!m_ostream)
-    return;
+wxDataInputStream& wxDataInputStream::operator>>(wxInt32& i)
+{
+  i = (wxInt32)Read32();
+  return *this;
+}
 
-  buf[0] = i & 0xff;
-  buf[1] = (i >> 8) & 0xff;
-  buf[2] = (i >> 16) & 0xff;
-  buf[3] = (i >> 24) & 0xff;
-  m_ostream->write(buf, 4);
+wxDataInputStream& wxDataInputStream::operator>>(wxUint8& c)
+{
+  c = Read8();
+  return *this;
 }
 
-void wxDataStream::Write16(unsigned short i)
+wxDataInputStream& wxDataInputStream::operator>>(wxUint16& i)
 {
-  char buf[2];
+  i = Read16();
+  return *this;
+}
 
-  if (!m_ostream)
-    return;
+wxDataInputStream& wxDataInputStream::operator>>(wxUint32& i)
+{
+  i = Read32();
+  return *this;
+}
 
-  buf[0] = i & 0xff;
-  buf[1] = (i >> 8) & 0xff;
-  m_ostream->write(buf, 2);
+wxDataInputStream& wxDataInputStream::operator>>(double& i)
+{
+  i = ReadDouble();
+  return *this;
 }
 
-void wxDataStream::Write8(unsigned char i)
+wxDataInputStream& wxDataInputStream::operator>>(float& f)
 {
-  if (!m_ostream)
-    return;
+  f = (float)ReadDouble();
+  return *this;
+}
+
+// ---------------------------------------------------------------------------
+// wxDataOutputStream
+// ---------------------------------------------------------------------------
 
-  m_ostream->write(&i, 1);
+wxDataOutputStream::wxDataOutputStream(wxOutputStream& s)
+  : m_output(&s), m_be_order(FALSE)
+{
 }
 
-void wxDataStream::WriteLine(const wxString& line)
+wxDataOutputStream::~wxDataOutputStream()
 {
-#ifdef __WXMSW__
-  wxString tmp_string = line + "\r\n";
-#else
-  wxString tmp_string = line + '\n';
-#endif
+}
+
+void wxDataOutputStream::Write32(wxUint32 i)
+{
+  wxUint32 i32;
+
+  if (m_be_order)
+    i32 = wxUINT32_SWAP_ON_LE(i);
+  else
+    i32 = wxUINT32_SWAP_ON_BE(i);
+  m_output->Write(&i32, 4);
+}
+
+void wxDataOutputStream::Write16(wxUint16 i)
+{
+  wxUint16 i16;
 
-  if (!m_ostream)
-    return;
+  if (m_be_order)
+    i16 = wxUINT16_SWAP_ON_LE(i);
+  else
+    i16 = wxUINT16_SWAP_ON_BE(i);
 
-  m_ostream->write((const char *) tmp_string, tmp_string.Length());
+  m_output->Write(&i16, 2);
 }
 
-void wxDataStream::WriteString(const wxString& string)
+void wxDataOutputStream::Write8(wxUint8 i)
 {
-  if (!m_ostream)
-    return;
+  m_output->Write(&i, 1);
+}
 
+void wxDataOutputStream::WriteString(const wxString& string)
+{
+  const wxWX2MBbuf buf = string.mb_str();
   Write32(string.Length());
-  m_ostream->write((const char *) string, string.Length());
+  m_output->Write(buf, string.Len());
 }
 
 // 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 USE_APPLE_IEEE
+#if wxUSE_APPLE_IEEE
   ConvertToIeeeExtended(d, (unsigned char *)buf);
 #else
-#	pragma warning "wxDataStream::WriteDouble() not using IeeeExtended - will not work!"
+#  pragma warning "wxDataOutputStream::WriteDouble() not using IeeeExtended - will not work!"
  buf[0] = '\0';
 #endif
-  m_ostream->write(buf, 10);
+  m_output->Write(buf, 10);
+}
+
+wxDataOutputStream& wxDataOutputStream::operator<<(const wxChar *string)
+{
+  Write32(wxStrlen(string));
+  m_output->Write((const char *)string, wxStrlen(string)*sizeof(wxChar));
+  return *this;
+}
+
+wxDataOutputStream& wxDataOutputStream::operator<<(wxString& string)
+{
+  WriteString(string);
+  return *this;
+}
+
+wxDataOutputStream& wxDataOutputStream::operator<<(wxInt8 c)
+{
+  Write8((wxUint8)c);
+  return *this;
+}
+
+wxDataOutputStream& wxDataOutputStream::operator<<(wxInt16 i)
+{
+  Write16((wxUint16)i);
+  return *this;
+}
+
+wxDataOutputStream& wxDataOutputStream::operator<<(wxInt32 i)
+{
+  Write32((wxUint32)i);
+  return *this;
+}
+
+wxDataOutputStream& wxDataOutputStream::operator<<(wxUint8 c)
+{
+  Write8(c);
+  return *this;
 }
+
+wxDataOutputStream& wxDataOutputStream::operator<<(wxUint16 i)
+{
+  Write16(i);
+  return *this;
+}
+
+wxDataOutputStream& wxDataOutputStream::operator<<(wxUint32 i)
+{
+  Write32(i);
+  return *this;
+}
+
+wxDataOutputStream& wxDataOutputStream::operator<<(double f)
+{
+  WriteDouble(f);
+  return *this;
+}
+
+wxDataOutputStream& wxDataOutputStream::operator<<(float f)
+{
+  WriteDouble((double)f);
+  return *this;
+}
+
+#endif
+  // wxUSE_STREAMS
+