r326: FFTW additions, filter image generation
[ctsim.git] / src / ctsim.cpp
index 0f7bc4e5490aabe4a5362b22cfebf97dd8a2c1b3..172cda95bc88c9bb6a14a556570a2a5dbe5e51fd 100644 (file)
@@ -1,15 +1,15 @@
 /*****************************************************************************
 ** FILE IDENTIFICATION
 **
-**   Name:          ctsim.h
-**   Purpose:       Header file for top-level routines of CTSim program
+**   Name:          ctsim.cpp
+**   Purpose:       Top-level routines of CTSim program
 **   Programmer:    Kevin Rosenberg
 **   Date Started:  July 2000
 **
 **  This is part of the CTSim program
 **  Copyright (C) 1983-2000 Kevin Rosenberg
 **
-**  $Id: ctsim.cpp,v 1.4 2000/07/13 07:01:59 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"
+\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"
+\r
+#if defined(HAVE_CONFIG_H)
+#include "config.h"
+#endif\r
+\r
+#if defined(HAVE_GETOPT_H) || defined(HAVE_GETOPT_LONG)
+#ifdef MSVC\r
+#define __STDC__ 1\r
+#endif\r
+#include "getopt.h"\r
+#ifdef MSVC\r
+#undef __STDC__\r
 #endif
-
-#if !wxUSE_DOC_VIEW_ARCHITECTURE
-#error You must set wxUSE_DOC_VIEW_ARCHITECTURE to 1 in setup.h!
 #endif
+\r
+static const char* rcsindent = "$Id: ctsim.cpp,v 1.23 2001/01/01 10:14:34 kevin Exp $";
 
-#include "wx/docview.h"
+class CTSimApp* theApp = NULL;
 
-#include "ctsim.h"
-#include "docs.h"
-#include "views.h"
+struct option CTSimApp::ctsimOptions[] = 
+{
+    {"help", 0, 0, O_HELP},
+    {"version", 0, 0, O_VERSION},
+    {0, 0, 0, 0}
+};
 
 IMPLEMENT_APP(CTSimApp)
 
-CTSimApp::CTSimApp(void)
+CTSimApp::CTSimApp()
   : m_docManager(NULL), m_pFrame(NULL)
 {
+    theApp = this;
 }
 
+#ifdef HAVE_SYS_TIME_H
+#include <sys/time.h>
+#endif
+
+#ifdef HAVE_SYS_RESOURCE_H
+#include <sys/resource.h>
+#endif
+
 bool
-CTSimApp::OnInit(void)
+CTSimApp::OnInit()
 {
+#ifdef HAVE_SETPRIORITY
+  setpriority (PRIO_PROCESS, 0, 15);  // set to low scheduling priority
+#endif
+
+    // process options
+    while (1) {
+      int c = getopt_long (argc, argv, "", ctsimOptions, NULL);
+      if (c == -1)
+       break;
+
+      switch (c) {
+      case O_VERSION:
+                 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 '?':
+       usage (argv[0]);
+       exit (0);
+      default:
+       usage (argv[0]);
+       exit (1);
+      }
+    }
+
     m_docManager = new wxDocManager;
     
-    (void) new wxDocTemplate (m_docManager, "ImageFile", "*.if", "", "if", "ImageFile doc", "ImageFile View", CLASSINFO(ImageFileDocument), CLASSINFO(ImageFileView));
+    new wxDocTemplate (m_docManager, "ImageFile", "*.if", "", "if", "ImageFile doc", "ImageFile View", CLASSINFO(ImageFileDocument), CLASSINFO(ImageFileView));
 
-    (void) new wxDocTemplate (m_docManager, "ProjectionFile", "*.pj", "", "pj", "ProjectionFile doc", "ProjectionFile View", CLASSINFO(ProjectionFileDocument), CLASSINFO(ProjectionFileView));
+    new wxDocTemplate (m_docManager, "ProjectionFile", "*.pj", "", "pj", "ProjectionFile doc", "ProjectionFile View", CLASSINFO(ProjectionFileDocument), CLASSINFO(ProjectionFileView));
+
+    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);
     
-    //// Make a menubar
-    wxMenu *file_menu = new wxMenu;
-    
-    //    file_menu->Append(wxID_NEW, "&New...");
-    file_menu->Append(wxID_OPEN, "&Open...");
-    
-    file_menu->AppendSeparator();
-    file_menu->Append(MAINMENU_FILE_EXIT, "E&xit");
-    
-    // A nice touch: a history of files visited. Use this menu.
-    m_docManager->FileHistoryUseMenu(file_menu);
-    
-    wxMenu *help_menu = new wxMenu;
-    help_menu->Append(MAINMENU_HELP_ABOUT, "&About");
-    
-    wxMenuBar *menu_bar = new wxMenuBar;
-    
-    menu_bar->Append(file_menu, "&File");
-    menu_bar->Append(help_menu, "&Help");
-    
-    m_pFrame->SetMenuBar(menu_bar);
-    
+    SetTopWindow (m_pFrame);
     m_pFrame->Centre(wxBOTH);
+
     m_pFrame->Show(true);
-    
-    SetTopWindow (m_pFrame);
+
+    for (int i = optind + 1; i <= argc; i++) {
+      wxString filename = argv [i - 1];
+      m_docManager->CreateDocument (filename, wxDOC_SILENT);
+    }
+
     return true;
 }
 
+void
+CTSimApp::usage(const char* program)
+{
+       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
-CTSimApp::OnExit(void)
+CTSimApp::OnExit()
 {
     delete m_docManager;
+#ifdef HAVE_DMALLOC
+    dmalloc_shutdown();
+#endif
     return 0;
 }
 
+wxString
+CTSimApp::getUntitledFilename()
+{
+  static int untitledNumber = 1;
+
+  wxString filename ("Untitled");
+  filename << untitledNumber++;
+
+  return (filename);
+}
+
 
 // Top-level window for CTSim
 
 IMPLEMENT_CLASS(MainFrame, wxDocParentFrame)
+
 BEGIN_EVENT_TABLE(MainFrame, wxDocParentFrame)
-    EVT_MENU(MAINMENU_HELP_ABOUT, MainFrame::OnAbout)
-    EVT_MENU(MAINMENU_FILE_EXIT, MainFrame::OnExit)
+  EVT_MENU(MAINMENU_HELP_ABOUT, MainFrame::OnAbout)
+  EVT_MENU(MAINMENU_HELP_CONTENTS, MainFrame::OnHelpContents)
+  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)
+  EVT_MENU(MAINMENU_WINDOW_BASE+2, MainFrame::OnWindowMenu2)
+  EVT_MENU(MAINMENU_WINDOW_BASE+3, MainFrame::OnWindowMenu3)
+  EVT_MENU(MAINMENU_WINDOW_BASE+4, MainFrame::OnWindowMenu4)
+  EVT_MENU(MAINMENU_WINDOW_BASE+5, MainFrame::OnWindowMenu5)
+  EVT_MENU(MAINMENU_WINDOW_BASE+6, MainFrame::OnWindowMenu6)
+  EVT_MENU(MAINMENU_WINDOW_BASE+7, MainFrame::OnWindowMenu7)
+  EVT_MENU(MAINMENU_WINDOW_BASE+8, MainFrame::OnWindowMenu8)
+  EVT_MENU(MAINMENU_WINDOW_BASE+9, MainFrame::OnWindowMenu9)
+  EVT_MENU(MAINMENU_WINDOW_BASE+10, MainFrame::OnWindowMenu10)
+  EVT_MENU(MAINMENU_WINDOW_BASE+11, MainFrame::OnWindowMenu11)
+  EVT_MENU(MAINMENU_WINDOW_BASE+12, MainFrame::OnWindowMenu12)
+  EVT_MENU(MAINMENU_WINDOW_BASE+13, MainFrame::OnWindowMenu13)
+  EVT_MENU(MAINMENU_WINDOW_BASE+14, MainFrame::OnWindowMenu14)
+  EVT_MENU(MAINMENU_WINDOW_BASE+15, MainFrame::OnWindowMenu15)
+  EVT_MENU(MAINMENU_WINDOW_BASE+16, MainFrame::OnWindowMenu16)
+  EVT_MENU(MAINMENU_WINDOW_BASE+17, MainFrame::OnWindowMenu17)
+  EVT_MENU(MAINMENU_WINDOW_BASE+18, MainFrame::OnWindowMenu18)
+  EVT_MENU(MAINMENU_WINDOW_BASE+19, MainFrame::OnWindowMenu19)
+  EVT_UPDATE_UI_RANGE(MAINMENU_WINDOW_BASE, MAINMENU_WINDOW_BASE+20, MainFrame::OnUpdateUI)
 END_EVENT_TABLE()
 
+
+
 MainFrame::MainFrame(wxDocManager *manager, wxFrame *frame, wxWindowID id, const wxString& title, const wxPoint& pos, const wxSize& size, const long type)
: wxDocParentFrame(manager, frame, id, title, pos, size, type)
 : wxDocParentFrame(manager, frame, id, title, pos, size, type), m_pLog(NULL)
 {
+    m_pLog = new wxTextCtrl (this, -1, "Log Window\n", wxPoint(0, 250), wxSize(100,50), wxTE_MULTILINE | wxTE_READONLY);
+    wxLog::SetActiveTarget(new wxLogTextCtrl(m_pLog));
     CreateStatusBar();
     SetStatusText ("Welcome to CTSim");
+
+    //// Make a menubar
+    wxMenu *file_menu = new wxMenu;
+    
+    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();
+    file_menu->Append(MAINMENU_FILE_EXIT, "E&xit");
+    
+    //  history of files visited
+    m_docManager->FileHistoryUseMenu(file_menu);
+    
+    m_pWindowMenu = new wxMenu;
+    m_pWindowMenu->UpdateUI (this);
+
+    wxMenu* help_menu = new wxMenu;
+    help_menu->Append(MAINMENU_HELP_CONTENTS, "&Contents");
+    help_menu->AppendSeparator();
+    help_menu->Append(MAINMENU_HELP_ABOUT, "&About");
+    
+    wxMenuBar* menu_bar = new wxMenuBar;
+    
+    menu_bar->Append(file_menu, "&File");
+    menu_bar->Append(m_pWindowMenu, "&Window");
+    menu_bar->Append(help_menu, "&Help");
+    
+    SetMenuBar(menu_bar);
+
+    for (int i = 0; i < MAX_WINDOW_MENUITEMS; i++) {
+      m_apWindowMenuItems[i] = new wxMenuItem (m_pWindowMenu, MAINMENU_WINDOW_BASE+i, wxString("<Empty>"));
+      m_pWindowMenu->Append (m_apWindowMenuItems[i]);
+      m_pWindowMenu->Enable (MAINMENU_WINDOW_BASE+i, false);
+    }
 }
 
 void 
 MainFrame::OnAbout(wxCommandEvent& WXUNUSED(event) )
+{\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) )
 {
-    wxMessageBox("CTSim\nAuthor: Kevin Rosenberg <kevin@rosenberg.net>\nUsage: ctsim", "About CTSim", wxOK | wxICON_INFORMATION, this);
+    wxMessageBox("No help available, refer to man pages of command-line tools");
 }
 
 void 
@@ -129,3 +346,101 @@ MainFrame::OnExit (wxCommandEvent& WXUNUSED(event) )
     Close(true);
 }
 
+void
+MainFrame::OnUpdateUI (wxUpdateUIEvent& rEvent)
+{
+  int iPos = 0;
+  wxList& rListDocs = m_docManager->GetDocuments();
+  wxNode* pNode = rListDocs.GetFirst();
+  while (iPos < MAX_WINDOW_MENUITEMS && pNode != NULL) {
+    wxDocument* pDoc = static_cast<wxDocument*>(pNode->GetData());
+    wxString strFilename = pDoc->GetFilename();
+    static_cast<wxMenuItemBase*>(m_apWindowMenuItems[iPos])->SetName (strFilename);
+    m_apWindowMenuData[iPos] = pDoc;
+    m_pWindowMenu->Enable (MAINMENU_WINDOW_BASE+iPos, true);
+    iPos++;
+    pNode = pNode->GetNext();
+  }
+  for (int i = iPos; i < MAX_WINDOW_MENUITEMS; i++) {
+    m_pWindowMenu->Enable (MAINMENU_WINDOW_BASE+i, false);
+    static_cast<wxMenuItemBase*>(m_apWindowMenuItems[i])->SetName (wxString("<Empty>"));
+    m_apWindowMenuData[i] = NULL;
+  }
+    
+}
+
+void 
+MainFrame::DoWindowMenu (int iMenuPosition, wxCommandEvent& event)
+{
+  if (wxDocument* pDoc = m_apWindowMenuData [iMenuPosition]) {
+    wxString strFilename = pDoc->GetFilename();
+    const wxView* pView = pDoc->GetFirstView();
+    if (pView) {
+      wxFrame* pFrame = pView->GetFrame();
+      pFrame->SetFocus();
+      pFrame->Raise();
+    }
+  }
+}
+
+void MainFrame::OnWindowMenu0 (wxCommandEvent& event)
+{ DoWindowMenu (0, event); }
+
+void MainFrame::OnWindowMenu1 (wxCommandEvent& event)
+{ DoWindowMenu (1, event); }
+
+void MainFrame::OnWindowMenu2 (wxCommandEvent& event)
+{ DoWindowMenu (2, event); }
+
+void MainFrame::OnWindowMenu3 (wxCommandEvent& event)
+{ DoWindowMenu (3, event); }
+
+void MainFrame::OnWindowMenu4 (wxCommandEvent& event)
+{ DoWindowMenu (4, event); }
+
+void MainFrame::OnWindowMenu5 (wxCommandEvent& event)
+{ DoWindowMenu (5, event); }
+
+void MainFrame::OnWindowMenu6 (wxCommandEvent& event)
+{ DoWindowMenu (6, event); }
+
+void MainFrame::OnWindowMenu7 (wxCommandEvent& event)
+{ DoWindowMenu (7, event); }
+
+void MainFrame::OnWindowMenu8 (wxCommandEvent& event)
+{ DoWindowMenu (8, event); }
+
+void MainFrame::OnWindowMenu9 (wxCommandEvent& event)
+{ DoWindowMenu (9, event); }
+
+void MainFrame::OnWindowMenu10 (wxCommandEvent& event)
+{ DoWindowMenu (10, event); }
+
+void MainFrame::OnWindowMenu11 (wxCommandEvent& event)
+{ DoWindowMenu (11, event); }
+
+void MainFrame::OnWindowMenu12 (wxCommandEvent& event)
+{ DoWindowMenu (12, event); }
+
+void MainFrame::OnWindowMenu13 (wxCommandEvent& event)
+{ DoWindowMenu (13, event); }
+
+void MainFrame::OnWindowMenu14 (wxCommandEvent& event)
+{ DoWindowMenu (14, event); }
+
+void MainFrame::OnWindowMenu15 (wxCommandEvent& event)
+{ DoWindowMenu (15, event); }
+
+void MainFrame::OnWindowMenu16 (wxCommandEvent& event)
+{ DoWindowMenu (16, event); }
+
+void MainFrame::OnWindowMenu17 (wxCommandEvent& event)
+{ DoWindowMenu (17, event); }
+
+void MainFrame::OnWindowMenu18 (wxCommandEvent& event)
+{ DoWindowMenu (18, event); }
+
+void MainFrame::OnWindowMenu19 (wxCommandEvent& event)
+{ DoWindowMenu (19, event); }
+
+