r471: Fixed shutdown crash, added Revert

[ctsim.git] / src / docs.cpp

diff --git a/src/docs.cpp b/src/docs.cpp
index daef2f4dfc897f89e26047e6457a846083b4a0f3..03ad1138293dd42647bb123eedab51c0f4bac16c 100644 (file)
--- a/src/docs.cpp
+++ b/src/docs.cpp
@@ -9,7 +9,7 @@
 **  This is part of the CTSim program
 **  Copyright (c) 1983-2001 Kevin Rosenberg
 **
-**  $Id: docs.cpp,v 1.16 2001/01/28 19:10:18 kevin Exp $
+**  $Id: docs.cpp,v 1.18 2001/01/30 05:05:41 kevin Exp $
 **
 **  This program is free software; you can redistribute it and/or modify
 **  it under the terms of the GNU General Public License (version 2) as
@@ -57,7 +57,7 @@ IMPLEMENT_DYNAMIC_CLASS(ImageFileDocument, wxDocument)
 
 bool ImageFileDocument::OnSaveDocument(const wxString& filename)
 {
-  if (! m_imageFile.fileWrite (filename)) {
+  if (! m_pImageFile->fileWrite (filename)) {
     *theApp->getLog() << "Unable to write image file " << filename << "\n";
     return false;
   }
@@ -68,26 +68,32 @@ bool ImageFileDocument::OnSaveDocument(const wxString& filename)
 
 bool ImageFileDocument::OnOpenDocument(const wxString& filename)
 {
-  if (! m_imageFile.fileRead (filename.c_str())) {
+  if (! OnSaveModified())
+    return false;
+
+  if (! m_pImageFile->fileRead (filename.c_str())) {
     *theApp->getLog() << "Unable to read image file " << filename << "\n";
+    m_bBadFileOpen = true;
     return false;
   }
   *theApp->getLog() << "Read image file " << filename << "\n";
-  SetFilename(filename, true);
-  
+  SetFilename(filename, true);  
   Modify(false);
   UpdateAllViews();
-  GetFirstView()->OnUpdate (GetFirstView(), NULL);
-  
+  getView()->OnUpdate (getView(), NULL);
+  m_bBadFileOpen = false;
+
   return true;
 }
 
-bool ImageFileDocument::IsModified(void) const
+bool 
+ImageFileDocument::IsModified(void) const
 {
   return wxDocument::IsModified();
 }
 
-void ImageFileDocument::Modify(bool mod)
+void 
+ImageFileDocument::Modify(bool mod)
 {
   wxDocument::Modify(mod);
 }
@@ -98,13 +104,34 @@ ImageFileDocument::getView() const
   return dynamic_cast<ImageFileView*>(GetFirstView()); 
 }
 
+
+bool
+ImageFileDocument::Revert ()
+{
+  if (IsModified()) {
+    wxString msg ("Revert to saved ");
+    msg += GetFilename();
+    msg += "?";
+    wxMessageDialog dialog (getView()->getFrame(), msg, "Are you sure?", wxYES_NO | wxNO_DEFAULT);
+    if (dialog.ShowModal() == wxID_YES) {
+      *theApp->getLog() << "Reverting to saved " << GetFilename() << "\n";
+      Modify(false);
+      OnOpenDocument (GetFilename());
+    }
+  }
+  getView()->OnUpdate (getView(), NULL);
+  UpdateAllViews();
+
+  return true;
+}
+
 // ProjectionFileDocument
 
 IMPLEMENT_DYNAMIC_CLASS(ProjectionFileDocument, wxDocument)
 
 bool ProjectionFileDocument::OnSaveDocument(const wxString& filename)
 {
-  if (! m_projectionFile.write (filename.c_str())) {
+  if (! m_pProjectionFile->write (filename.c_str())) {
     *theApp->getLog() << "Unable to write projection file " << filename << "\n";
     return false;
   }
@@ -115,8 +142,12 @@ bool ProjectionFileDocument::OnSaveDocument(const wxString& filename)
 
 bool ProjectionFileDocument::OnOpenDocument(const wxString& filename)
 {
-  if (! m_projectionFile.read (filename.c_str())) {
+  if (! OnSaveModified())
+    return false;
+
+  if (! m_pProjectionFile->read (filename.c_str())) {
     *theApp->getLog() << "Unable to read projection file " << filename << "\n";
+    m_bBadFileOpen = true;
     return false;
   }
   *theApp->getLog() << "Read projection file " << filename << "\n";
@@ -124,6 +155,7 @@ bool ProjectionFileDocument::OnOpenDocument(const wxString& filename)
   Modify(false);
   UpdateAllViews();
   GetFirstView()->OnUpdate (GetFirstView(), NULL);
+  m_bBadFileOpen = false;
   
   return true;
 }
@@ -151,6 +183,9 @@ IMPLEMENT_DYNAMIC_CLASS(PhantomFileDocument, wxDocument)
 
 bool PhantomFileDocument::OnOpenDocument(const wxString& filename)
 {
+  if (! OnSaveModified())
+    return false;
+
   wxString myFilename = filename;
   if (wxFile::Exists (myFilename)) {
     m_phantom.createFromFile (myFilename);
@@ -163,12 +198,14 @@ bool PhantomFileDocument::OnOpenDocument(const wxString& filename)
   SetFilename (myFilename, true);
   if (m_phantom.fail()) {
     *theApp->getLog() << "Failure creating phantom " << myFilename << "\n";
+    m_bBadFileOpen = true;
     return false;
   }
   m_idPhantom = m_phantom.id();
   Modify(false);
   UpdateAllViews();
   GetFirstView()->OnUpdate (GetFirstView(), NULL);
+  m_bBadFileOpen = false;
   
   return true;
 }
@@ -219,8 +256,12 @@ bool PlotFileDocument::OnSaveDocument(const wxString& filename)
 
 bool PlotFileDocument::OnOpenDocument(const wxString& filename)
 {
+  if (! OnSaveModified())
+    return false;
+
   if (! m_plot.fileRead (filename.c_str())) {
     *theApp->getLog() << "Unable to read plot file " << filename << "\n";
+    m_bBadFileOpen = true;
     return false;
   }
   *theApp->getLog() << "Read plot file " << filename << "\n";
@@ -229,6 +270,7 @@ bool PlotFileDocument::OnOpenDocument(const wxString& filename)
   Modify (false);
   UpdateAllViews();
   GetFirstView()->OnUpdate (NULL, NULL);
+  m_bBadFileOpen = false;
   
   return true;
 }
@@ -273,12 +315,15 @@ bool TextFileDocument::OnOpenDocument(const wxString& filename)
 {
   TextFileView *view = getView();
   
-  if (! view->getTextCtrl()->LoadFile(filename))
+  if (! view->getTextCtrl()->LoadFile(filename)) {
+    m_bBadFileOpen = true;
     return false;
+  }
   
   SetFilename (filename, true);
   Modify (false);
   UpdateAllViews();
+  m_bBadFileOpen = false;
   return true;
 }
 
@@ -300,3 +345,10 @@ TextFileDocument::getView() const
 { 
   return dynamic_cast<TextFileView*>(GetFirstView()); 
 }
+
+wxTextCtrl* 
+TextFileDocument::getTextCtrl()
+{ 
+  return dynamic_cast<TextFileView*>(GetFirstView())->getTextCtrl(); 
+}
+