X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/2d1df0fc585f03cca4b194d1f9159b49a5711da6..880da677a495220275f81e0738a23da4e977e312:/samples/docview/doc.cpp

diff --git a/samples/docview/doc.cpp b/samples/docview/doc.cpp
index 1f1715d781..e093d82ae3 100644
--- a/samples/docview/doc.cpp
+++ b/samples/docview/doc.cpp
@@ -7,7 +7,7 @@
 // RCS-ID:      $Id$
 // Copyright:   (c) 1998 Julian Smart
 //              (c) 2008 Vadim Zeitlin
-// Licence:     wxWindows license
+// Licence:     wxWindows licence
 /////////////////////////////////////////////////////////////////////////////
 
 // ----------------------------------------------------------------------------
@@ -34,6 +34,7 @@
 #else
     #include "wx/txtstrm.h"
 #endif
+#include "wx/wfstream.h"
 
 #include "doc.h"
 #include "view.h"
@@ -44,37 +45,59 @@
 
 IMPLEMENT_DYNAMIC_CLASS(DrawingDocument, wxDocument)
 
-DocumentOstream& DrawingDocument::SaveObject(DocumentOstream& stream)
+DocumentOstream& DrawingDocument::SaveObject(DocumentOstream& ostream)
 {
-    wxDocument::SaveObject(stream);
+#if wxUSE_STD_IOSTREAM
+    DocumentOstream& stream = ostream;
+#else
+    wxTextOutputStream stream(ostream);
+#endif
+
+    wxDocument::SaveObject(ostream);
 
     const wxInt32 count = m_doodleSegments.size();
     stream << count << '\n';
 
     for ( int n = 0; n < count; n++ )
     {
-        m_doodleSegments[n].SaveObject(stream);
+        m_doodleSegments[n].SaveObject(ostream);
         stream << '\n';
     }
 
-    return stream;
+    return ostream;
 }
 
-DocumentIstream& DrawingDocument::LoadObject(DocumentIstream& stream)
+DocumentIstream& DrawingDocument::LoadObject(DocumentIstream& istream)
 {
-    wxDocument::LoadObject(stream);
+#if wxUSE_STD_IOSTREAM
+    DocumentIstream& stream = istream;
+#else
+    wxTextInputStream stream(istream);
+#endif
+
+    wxDocument::LoadObject(istream);
 
     wxInt32 count = 0;
     stream >> count;
+    if ( count < 0 )
+    {
+        wxLogWarning("Drawing document corrupted: invalid segments count.");
+#if wxUSE_STD_IOSTREAM
+        istream.clear(std::ios::badbit);
+#else
+        istream.Reset(wxSTREAM_READ_ERROR);
+#endif
+        return istream;
+    }
 
     for ( int n = 0; n < count; n++ )
     {
         DoodleSegment segment;
-        segment.LoadObject(stream);
+        segment.LoadObject(istream);
         m_doodleSegments.push_back(segment);
     }
 
-    return stream;
+    return istream;
 }
 
 void DrawingDocument::DoUpdate()
@@ -130,7 +153,7 @@ DocumentOstream& DoodleSegment::SaveObject(DocumentOstream& ostream)
             << line.y2 << '\n';
     }
 
-    return stream;
+    return ostream;
 }
 
 DocumentIstream& DoodleSegment::LoadObject(DocumentIstream& istream)
@@ -155,46 +178,121 @@ DocumentIstream& DoodleSegment::LoadObject(DocumentIstream& istream)
         m_lines.push_back(line);
     }
 
-    return stream;
+    return istream;
 }
 
 // ----------------------------------------------------------------------------
-// TextEditDocument implementation
+// wxTextDocument: wxDocument and wxTextCtrl married
 // ----------------------------------------------------------------------------
 
-IMPLEMENT_DYNAMIC_CLASS(TextEditDocument, wxDocument)
+IMPLEMENT_CLASS(wxTextDocument, wxDocument)
+
+bool wxTextDocument::OnCreate(const wxString& path, long flags)
+{
+    if ( !wxDocument::OnCreate(path, flags) )
+        return false;
+
+    // subscribe to changes in the text control to update the document state
+    // when it's modified
+    GetTextCtrl()->Connect
+    (
+        wxEVT_COMMAND_TEXT_UPDATED,
+        wxCommandEventHandler(wxTextDocument::OnTextChange),
+        NULL,
+        this
+    );
+
+    return true;
+}
 
 // Since text windows have their own method for saving to/loading from files,
 // we override DoSave/OpenDocument instead of Save/LoadObject
-bool TextEditDocument::DoSaveDocument(const wxString& filename)
+bool wxTextDocument::DoSaveDocument(const wxString& filename)
 {
-    return GetFirstView()->GetText()->SaveFile(filename);
+    return GetTextCtrl()->SaveFile(filename);
 }
 
-bool TextEditDocument::DoOpenDocument(const wxString& filename)
+bool wxTextDocument::DoOpenDocument(const wxString& filename)
 {
-    return GetFirstView()->GetText()->LoadFile(filename);
+    if ( !GetTextCtrl()->LoadFile(filename) )
+        return false;
+
+    // we're not modified by the user yet
+    Modify(false);
+
+    return true;
 }
 
-bool TextEditDocument::IsModified() const
+bool wxTextDocument::IsModified() const
 {
-    TextEditView* view = GetFirstView();
-    return wxDocument::IsModified() || (view && view->GetText()->IsModified());
+    wxTextCtrl* wnd = GetTextCtrl();
+    return wxDocument::IsModified() || (wnd && wnd->IsModified());
 }
 
-void TextEditDocument::Modify(bool modified)
+void wxTextDocument::Modify(bool modified)
 {
-    TextEditView* view = GetFirstView();
-
     wxDocument::Modify(modified);
 
-    if ( !modified && view && view->GetText() )
-        view->GetText()->DiscardEdits();
+    wxTextCtrl* wnd = GetTextCtrl();
+    if (wnd && !modified)
+    {
+        wnd->DiscardEdits();
+    }
+}
+
+void wxTextDocument::OnTextChange(wxCommandEvent& event)
+{
+    Modify(true);
+
+    event.Skip();
 }
 
-TextEditView* TextEditDocument::GetFirstView() const
+// ----------------------------------------------------------------------------
+// TextEditDocument implementation
+// ----------------------------------------------------------------------------
+
+IMPLEMENT_DYNAMIC_CLASS(TextEditDocument, wxDocument)
+
+wxTextCtrl* TextEditDocument::GetTextCtrl() const
 {
-    wxView* view = wxDocument::GetFirstView();
-    return view ? wxStaticCast(view, TextEditView) : NULL;
+    wxView* view = GetFirstView();
+    return view ? wxStaticCast(view, TextEditView)->GetText() : NULL;
 }
 
+// ----------------------------------------------------------------------------
+// ImageDocument and ImageDetailsDocument implementation
+// ----------------------------------------------------------------------------
+
+IMPLEMENT_DYNAMIC_CLASS(ImageDocument, wxDocument)
+
+bool ImageDocument::DoOpenDocument(const wxString& file)
+{
+    return m_image.LoadFile(file);
+}
+
+bool ImageDocument::OnOpenDocument(const wxString& filename)
+{
+    if ( !wxDocument::OnOpenDocument(filename) )
+        return false;
+
+    // we don't have a wxDocTemplate for the image details document as it's
+    // never created by wxWidgets automatically, instead just do it manually
+    ImageDetailsDocument * const docDetails = new ImageDetailsDocument(this);
+    docDetails->SetFilename(filename);
+
+    new ImageDetailsView(docDetails);
+
+    return true;
+}
+
+ImageDetailsDocument::ImageDetailsDocument(ImageDocument *parent)
+    : wxDocument(parent)
+{
+    const wxImage image = parent->GetImage();
+
+    m_size.x = image.GetWidth();
+    m_size.y = image.GetHeight();
+    m_numColours = image.CountColours();
+    m_type = image.GetType();
+    m_hasAlpha = image.HasAlpha();
+}