r326: FFTW additions, filter image generation
[ctsim.git] / src / ctsim.cpp
index 2d094d383e52e6e1198f82d9e204648f0e67a1bb..172cda95bc88c9bb6a14a556570a2a5dbe5e51fd 100644 (file)
@@ -9,7 +9,7 @@
 **  This is part of the CTSim program
 **  Copyright (C) 1983-2000 Kevin Rosenberg
 **
-**  $Id: ctsim.cpp,v 1.13 2000/09/07 14:29:05 kevin Exp $
+**  $Id: ctsim.cpp,v 1.23 2001/01/01 10:14:34 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
 **  along with this program; if not, write to the Free Software
 **  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 ******************************************************************************/
-
-// For compilers that support precompilation, includes "wx/wx.h".
-#include "wx/wxprec.h"
-
-#ifdef __BORLANDC__
-#pragma hdrstop
-#endif
-
-#ifndef WX_PRECOMP
-#include "wx/wx.h"
-#endif
-
-#if !wxUSE_DOC_VIEW_ARCHITECTURE
-#error You must set wxUSE_DOC_VIEW_ARCHITECTURE to 1 in setup.h!
-#endif
-
-#include "wx/docview.h"
+\r
+\r
+// For compilers that support precompilation, includes "wx/wx.h".\r
+#include "wx/wxprec.h"\r
+\r
+#ifdef __BORLANDC__\r
+#pragma hdrstop\r
+#endif\r
+\r
+#ifndef WX_PRECOMP\r
+#include "wx/wx.h"\r
+#endif\r
+\r
+#if !wxUSE_DOC_VIEW_ARCHITECTURE\r
+#error You must set wxUSE_DOC_VIEW_ARCHITECTURE to 1 in setup.h!\r
+#endif\r
+\r
+#include "ct.h"\r
 #include "ctsim.h"
 #include "docs.h"
 #include "views.h"
 #include "dialogs.h"
-#include "ctsupport.h"
+\r
 #if defined(HAVE_CONFIG_H)
 #include "config.h"
-#endif
+#endif\r
+\r
 #if defined(HAVE_GETOPT_H) || defined(HAVE_GETOPT_LONG)
-#include <getopt.h>
+#ifdef MSVC\r
+#define __STDC__ 1\r
+#endif\r
+#include "getopt.h"\r
+#ifdef MSVC\r
+#undef __STDC__\r
 #endif
-
-static const char* rcsindent = "$Id: ctsim.cpp,v 1.13 2000/09/07 14:29:05 kevin Exp $";
+#endif
+\r
+static const char* rcsindent = "$Id: ctsim.cpp,v 1.23 2001/01/01 10:14:34 kevin Exp $";
 
 class CTSimApp* theApp = NULL;
 
@@ -95,7 +103,12 @@ CTSimApp::OnInit()
 
       switch (c) {
       case O_VERSION:
-         cout << rcsindent << endl;
+                 std::cout << rcsindent << std::endl;\r
+#ifdef CTSIMVERSION\r
+                 std::cout << "Version: CTSIMVERSION" << std::endl;\r
+#elif defined(VERSION)\r
+                 std::cout << "Version: VERSION" << std::endl;\r
+#endif
          exit(0);
       case O_HELP:
       case '?':
@@ -115,6 +128,8 @@ CTSimApp::OnInit()
 
     new wxDocTemplate (m_docManager, "PhantomFile", "*.phm", "", "phm", "Phantom doc", "Phantom View", CLASSINFO(PhantomDocument), CLASSINFO(PhantomView));
 
+    new wxDocTemplate (m_docManager, "PlotFile", "*.plt", "", "plt", "Plot doc", "Plot View", CLASSINFO(PlotFileDocument), CLASSINFO(PlotFileView));
+
     //// Create the main frame window
     m_pFrame = new MainFrame(m_docManager, (wxFrame *) NULL, -1, "CTSim", wxPoint(0, 0), wxSize(500, 400), wxDEFAULT_FRAME_STYLE);
     
@@ -134,11 +149,11 @@ CTSimApp::OnInit()
 void
 CTSimApp::usage(const char* program)
 {
-    cout << "usage: " << fileBasename(program) << " [files-to-open...] [OPTIONS]\n";
-    cout << "Computed Tomography Simulator (Graphical Shell)\n";
-    cout << "\n";
-    cout << "  --version Display version\n";
-    cout << "  --help    Display this help message\n";
+       std::cout << "usage: " << fileBasename(program) << " [files-to-open...] [OPTIONS]\n";
+       std::cout << "Computed Tomography Simulator (Graphical Shell)\n";
+       std::cout << "\n";
+       std::cout << "  --version Display version\n";
+       std::cout << "  --help    Display this help message\n";
 }
 
 int
@@ -170,7 +185,8 @@ IMPLEMENT_CLASS(MainFrame, wxDocParentFrame)
 BEGIN_EVENT_TABLE(MainFrame, wxDocParentFrame)
   EVT_MENU(MAINMENU_HELP_ABOUT, MainFrame::OnAbout)
   EVT_MENU(MAINMENU_HELP_CONTENTS, MainFrame::OnHelpContents)
-  EVT_MENU(MAINMENU_FILE_CREATE_PHANTOM, MainFrame::OnCreatePhantom)
+  EVT_MENU(MAINMENU_FILE_CREATE_PHANTOM, MainFrame::OnCreatePhantom)\r
+  EVT_MENU(MAINMENU_FILE_CREATE_FILTER, MainFrame::OnCreateFilter)\r
   EVT_MENU(MAINMENU_FILE_EXIT, MainFrame::OnExit)
   EVT_MENU(MAINMENU_WINDOW_BASE, MainFrame::OnWindowMenu0)
   EVT_MENU(MAINMENU_WINDOW_BASE+1, MainFrame::OnWindowMenu1)
@@ -208,7 +224,8 @@ MainFrame::MainFrame(wxDocManager *manager, wxFrame *frame, wxWindowID id, const
     //// Make a menubar
     wxMenu *file_menu = new wxMenu;
     
-    file_menu->Append(MAINMENU_FILE_CREATE_PHANTOM, "Cr&eate Phantom...");
+    file_menu->Append(MAINMENU_FILE_CREATE_PHANTOM, "Cr&eate Phantom...");\r
+    file_menu->Append(MAINMENU_FILE_CREATE_FILTER, "Create &Filter...");\r
     file_menu->Append(wxID_OPEN, "&Open...");
     
     file_menu->AppendSeparator();
@@ -242,24 +259,81 @@ MainFrame::MainFrame(wxDocManager *manager, wxFrame *frame, wxWindowID id, const
 
 void 
 MainFrame::OnAbout(wxCommandEvent& WXUNUSED(event) )
-{
-    wxMessageBox("CTSim\nThe Open Source Computed Tomography Simulator\nAuthor: Kevin Rosenberg <kevin@rosenberg.net>\nUsage: ctsim [files-to-open..] [--help]", "About CTSim", wxOK | wxICON_INFORMATION, this);
-}
-
-void 
-MainFrame::OnCreatePhantom(wxCommandEvent& WXUNUSED(event))
-{
-    DialogGetPhantom dialogPhantom (this, Phantom::PHM_HERMAN);
-    int dialogReturn = dialogPhantom.ShowModal();
-    if (dialogReturn == wxID_OK) {
-      wxString selection (dialogPhantom.getPhantom());
-      *theApp->getLog() << "Selected phantom " << selection.c_str() << "\n";
-      wxString filename = selection + ".phm";
-      theApp->getDocManager()->CreateDocument(filename, wxDOC_SILENT);
-    }
-    
+{\r
+       wxString msg = "CTSim\nThe Open Source Computed Tomography Simulator\n";\r
+#ifdef CTSIMVERSION\r
+       msg += "Version ";\r
+       msg += CTSIMVERSION;\r
+       msg += "\n\n";\r
+#elif defined(VERSION)\r
+       msg << "Version: " <<  VERSION << "\n\n";\r
+#endif\r
+       msg += "Author: Kevin Rosenberg <kevin@rosenberg.net>\nUsage: ctsim [files-to-open..] [--help]";\r
+
+    wxMessageBox(msg, "About CTSim", wxOK | wxICON_INFORMATION, this);
 }
 
+void \r
+MainFrame::OnCreatePhantom(wxCommandEvent& WXUNUSED(event))\r
+{\r
+    DialogGetPhantom dialogPhantom (this, Phantom::PHM_HERMAN);\r
+    int dialogReturn = dialogPhantom.ShowModal();\r
+    if (dialogReturn == wxID_OK) {\r
+      wxString selection (dialogPhantom.getPhantom());\r
+      *theApp->getLog() << "Selected phantom " << selection.c_str() << "\n";\r
+      wxString filename = selection + ".phm";\r
+      theApp->getDocManager()->CreateDocument(filename, wxDOC_SILENT);\r
+    }\r
+    \r
+}\r
+\r
+void \r
+MainFrame::OnCreateFilter (wxCommandEvent& WXUNUSED(event))\r
+{\r
+  DialogGetFilterParameters dialogFilter (this, 256, 256, SignalFilter::FILTER_BANDLIMIT, 1., 10., SignalFilter::DOMAIN_SPATIAL);\r
+    int dialogReturn = dialogFilter.ShowModal();\r
+    if (dialogReturn == wxID_OK) {\r
+      wxString strFilter (dialogFilter.getFilterName());\r
+      wxString strDomain (dialogFilter.getDomainName());\r
+      unsigned int nx = dialogFilter.getXSize();\r
+      unsigned int ny = dialogFilter.getYSize();\r
+      double dBandwidth = dialogFilter.getBandwidth();\r
+      double dFilterParam= dialogFilter.getFilterParam();\r
+      *theApp->getLog() << "Selected filter " << strFilter.c_str() << ", domain " << strDomain.c_str() << ", filterParam " << dFilterParam << ", bandwidth " << dBandwidth << "\n";\r
+      wxString filename = "untitled.if";\r
+      ImageFileDocument* pFilterDoc = dynamic_cast<ImageFileDocument*>(theApp->getDocManager()->CreateDocument ("untitled.if", wxDOC_SILENT));\r
+      if (! pFilterDoc) {\r
+        sys_error (ERR_SEVERE, "Unable to create filter image");\r
+        return;\r
+      }\r
+      ImageFile& rIF = pFilterDoc->getImageFile();\r
+      rIF.setArraySize (nx, ny);\r
+      rIF.filterResponse (strDomain.c_str(), dBandwidth, strFilter.c_str(), dFilterParam);\r
+      if (theApp->getSetModifyNewDocs())\r
+        pFilterDoc->Modify (true);\r
+      pFilterDoc->UpdateAllViews();\r
+      pFilterDoc->GetFirstView()->OnUpdate (NULL, NULL);\r
+    }\r
+}\r
+\r
+void\r
+CTSimApp::getCompatibleImages (const ImageFileDocument* pIFDoc, std::vector<ImageFileDocument*>& vecIF)\r
+{\r
+  const ImageFile& rIF = pIFDoc->getImageFile();\r
+  unsigned int nx = rIF.nx();\r
+  unsigned int ny = rIF.ny();\r
+  wxList& rListDocs = m_docManager->GetDocuments();\r
+  for (wxNode* pNode = rListDocs.GetFirst(); pNode != NULL; pNode = pNode->GetNext()) {\r
+    wxDocument* pDoc = reinterpret_cast<wxDocument*>(pNode->GetData());\r
+    ImageFileDocument* pIFCompareDoc = dynamic_cast<ImageFileDocument*>(pDoc);\r
+    if (pIFCompareDoc && (pIFDoc != pIFCompareDoc)) {\r
+      const ImageFile& rCompareIF = pIFCompareDoc->getImageFile();\r
+      if (rCompareIF.nx() == nx && rCompareIF.ny() == ny)\r
+        vecIF.push_back (pIFCompareDoc);\r
+    }\r
+  }\r
+}\r
+\r
 void 
 MainFrame::OnHelpContents(wxCommandEvent& WXUNUSED(event) )
 {