r471: Fixed shutdown crash, added Revert
[ctsim.git] / src / docs.cpp
index 69b59544fc448f6afafff26201f94d69f5248170..03ad1138293dd42647bb123eedab51c0f4bac16c 100644 (file)
@@ -7,9 +7,9 @@
 **   Date Started:  July 2000
 **
 **  This is part of the CTSim program
-**  Copyright (C) 1983-2000 Kevin Rosenberg
+**  Copyright (c) 1983-2001 Kevin Rosenberg
 **
-**  $Id: docs.cpp,v 1.10 2001/01/02 05:34:57 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,38 +68,61 @@ bool ImageFileDocument::OnSaveDocument(const wxString& filename)
 
 bool ImageFileDocument::OnOpenDocument(const wxString& filename)
 {
-  if (filename == "untitled.if") {
-    wxString untitledFilename = theApp->getUntitledFilename();
-    SetFilename (untitledFilename, true);
-  } else {
-    if (! m_imageFile.fileRead (filename.c_str())) {
-      *theApp->getLog() << "Unable to read image file " << filename << "\n";
-      return false;
-    }
-    *theApp->getLog() << "Read image file " << filename << "\n";
-    SetFilename(filename, true);
+  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);  
   Modify(false);
-  UpdateAllViews();\r
-  GetFirstView()->OnUpdate (GetFirstView(), NULL);\r
+  UpdateAllViews();
+  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);
+  wxDocument::Modify(mod);
 }
 
-bool ImageFileDocument::OnCloseDocument ()
+ImageFileView* 
+ImageFileDocument::getView() const
+{ 
+  return dynamic_cast<ImageFileView*>(GetFirstView()); 
+}
+
+
+bool
+ImageFileDocument::Revert ()
 {
-  bool bReturn = wxDocument::OnCloseDocument();
-  return bReturn;
+  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
@@ -108,7 +131,7 @@ 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;
   }
@@ -119,20 +142,21 @@ bool ProjectionFileDocument::OnSaveDocument(const wxString& filename)
 
 bool ProjectionFileDocument::OnOpenDocument(const wxString& filename)
 {
-  if (filename == "untitled.pj") {
-    wxString untitledFilename = theApp->getUntitledFilename();
-    SetFilename (untitledFilename, true);
-  } else {
-    if (! m_projectionFile.read (filename.c_str())) {
-      *theApp->getLog() << "Unable to read projection file " << filename << "\n";
-      return false;
-    }
-    *theApp->getLog() << "Read projection file " << filename << "\n";
-    SetFilename(filename, true);
+  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";
+  SetFilename(filename, true);
   Modify(false);
-  UpdateAllViews();\r
-
+  UpdateAllViews();
+  GetFirstView()->OnUpdate (GetFirstView(), NULL);
+  m_bBadFileOpen = false;
+  
   return true;
 }
 
@@ -143,22 +167,25 @@ bool ProjectionFileDocument::IsModified(void) const
 
 void ProjectionFileDocument::Modify(bool mod)
 {
-    wxDocument::Modify(mod);
+  wxDocument::Modify(mod);
 }
 
 
-bool ProjectionFileDocument::OnCloseDocument ()
-{
-  bool bReturn = wxDocument::OnCloseDocument();
-  return bReturn;
+ProjectionFileView* 
+ProjectionFileDocument::getView() const
+{ 
+  return dynamic_cast<ProjectionFileView*>(GetFirstView()); 
 }
 
-// PhantomDocument
+// PhantomFileDocument
 
-IMPLEMENT_DYNAMIC_CLASS(PhantomDocument, wxDocument)
+IMPLEMENT_DYNAMIC_CLASS(PhantomFileDocument, wxDocument)
 
-bool PhantomDocument::OnOpenDocument(const wxString& filename)
+bool PhantomFileDocument::OnOpenDocument(const wxString& filename)
 {
+  if (! OnSaveModified())
+    return false;
+
   wxString myFilename = filename;
   if (wxFile::Exists (myFilename)) {
     m_phantom.createFromFile (myFilename);
@@ -171,86 +198,84 @@ bool PhantomDocument::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();\r
-
+  UpdateAllViews();
+  GetFirstView()->OnUpdate (GetFirstView(), NULL);
+  m_bBadFileOpen = false;
+  
   return true;
 }
 
-bool PhantomDocument::OnSaveDocument(const wxString& filename)\r
-{\r
-  if (! m_phantom.fileWrite (filename.c_str())) {\r
-    *theApp->getLog() << "Unable to write phantom file " << filename << "\n";\r
-    return false;\r
-  }\r
-  *theApp->getLog() << "Wrote phantom file " << filename << "\n";\r
-  Modify(false);\r
-  return true;\r
-}\r
-\r
-bool PhantomDocument::OnCloseDocument ()
+bool PhantomFileDocument::OnSaveDocument(const wxString& filename)
 {
-  bool bReturn = wxDocument::OnCloseDocument();
-  return bReturn;
+  if (! m_phantom.fileWrite (filename.c_str())) {
+    *theApp->getLog() << "Unable to write phantom file " << filename << "\n";
+    return false;
+  }
+  *theApp->getLog() << "Wrote phantom file " << filename << "\n";
+  Modify(false);
+  return true;
 }
 
-bool PhantomDocument::IsModified(void) const
+bool PhantomFileDocument::IsModified(void) const
 {
-  return wxDocument::IsModified();
+  return false;
 }
 
-void PhantomDocument::Modify(bool mod)
+void PhantomFileDocument::Modify(bool mod)
 {
-    wxDocument::Modify(mod);
+  wxDocument::Modify(mod);
 }
 
 
+PhantomFileView* 
+PhantomFileDocument::getView() const
+{ 
+  return dynamic_cast<PhantomFileView*>(GetFirstView()); 
+}
+
 // PlotFileDocument
 
 IMPLEMENT_DYNAMIC_CLASS(PlotFileDocument, wxDocument)
 
-bool PlotFileDocument::OnSaveDocument(const wxString& filename)\r
-{\r
-  m_namePlot = filename.c_str();\r
-  if (! m_plot.fileWrite (filename)) {\r
-    *theApp->getLog() << "Unable to write plot file " << filename << "\n";\r
-    return false;\r
-  }\r
-  *theApp->getLog() << "Wrote plot file " << filename << "\n";\r
-  Modify(false);\r
-  return true;\r
-}\r
-\r
-bool PlotFileDocument::OnOpenDocument(const wxString& filename)
+bool PlotFileDocument::OnSaveDocument(const wxString& filename)
 {
-  if (filename == "untitled.plt") {
-    wxString untitledFilename = theApp->getUntitledFilename();
-    SetFilename (untitledFilename, true);
-  } else {
-    if (! m_plot.fileRead (filename.c_str())) {
-      *theApp->getLog() << "Unable to read plot file " << filename << "\n";
-      return false;
-    }
-    *theApp->getLog() << "Read plot file " << filename << "\n";
-    SetFilename (filename, true);\r
-    m_namePlot = filename.c_str();
+  m_namePlot = filename.c_str();
+  if (! m_plot.fileWrite (filename)) {
+    *theApp->getLog() << "Unable to write plot file " << filename << "\n";
+    return false;
   }
-  Modify (false);
-  UpdateAllViews();\r
-  GetFirstView()->OnUpdate (NULL, NULL);\r
-
+  *theApp->getLog() << "Wrote plot file " << filename << "\n";
+  Modify(false);
   return true;
 }
 
-bool PlotFileDocument::OnCloseDocument ()
+bool PlotFileDocument::OnOpenDocument(const wxString& filename)
 {
-  bool bReturn = wxDocument::OnCloseDocument();
-  return bReturn;
+  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";
+  SetFilename (filename, true);
+  m_namePlot = filename.c_str();
+  Modify (false);
+  UpdateAllViews();
+  GetFirstView()->OnUpdate (NULL, NULL);
+  m_bBadFileOpen = false;
+  
+  return true;
 }
 
+
 bool PlotFileDocument::IsModified(void) const
 {
   return wxDocument::IsModified();
@@ -258,6 +283,72 @@ bool PlotFileDocument::IsModified(void) const
 
 void PlotFileDocument::Modify(bool mod)
 {
-    wxDocument::Modify(mod);
+  wxDocument::Modify(mod);
+}
+
+PlotFileView* 
+PlotFileDocument::getView() const
+{ 
+  return dynamic_cast<PlotFileView*>(GetFirstView()); 
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+// TextFileDocument
+//
+//////////////////////////////////////////////////////////////////////////
+
+IMPLEMENT_DYNAMIC_CLASS(TextFileDocument, wxDocument)
+
+// Since text windows have their own method for saving to/loading from files,
+// we override OnSave/OpenDocument instead of Save/LoadObject
+bool TextFileDocument::OnSaveDocument(const wxString& filename)
+{
+  TextFileView *view = getView();
+  if (! view->getTextCtrl()->SaveFile(filename))
+    return false;
+  Modify(false);
+  return true;
+}
+
+bool TextFileDocument::OnOpenDocument(const wxString& filename)
+{
+  TextFileView *view = getView();
+  
+  if (! view->getTextCtrl()->LoadFile(filename)) {
+    m_bBadFileOpen = true;
+    return false;
+  }
+  
+  SetFilename (filename, true);
+  Modify (false);
+  UpdateAllViews();
+  m_bBadFileOpen = false;
+  return true;
+}
+
+bool TextFileDocument::IsModified(void) const
+{
+  return false;
+  
+  TextFileView *view = getView();
+  
+  if (view)
+    return (wxDocument::IsModified() || view->getTextCtrl()->IsModified());
+  else
+    return wxDocument::IsModified();
+}
+
+
+TextFileView* 
+TextFileDocument::getView() const
+{ 
+  return dynamic_cast<TextFileView*>(GetFirstView()); 
+}
+
+wxTextCtrl* 
+TextFileDocument::getTextCtrl()
+{ 
+  return dynamic_cast<TextFileView*>(GetFirstView())->getTextCtrl(); 
 }