src/ctsim.cpp

   1 /*****************************************************************************
   2 ** FILE IDENTIFICATION
   3 **
   4 **   Name:          ctsim.cpp
   5 **   Purpose:       Top-level routines of CTSim program
   6 **   Programmer:    Kevin Rosenberg
   7 **   Date Started:  July 2000
   8 **
   9 **  This is part of the CTSim program
  10 **  Copyright (C) 1983-2000 Kevin Rosenberg
  11 **
  12 **  $Id: ctsim.cpp,v 1.20 2000/12/22 04:18:00 kevin Exp $
  13 **
  14 **  This program is free software; you can redistribute it and/or modify
  15 **  it under the terms of the GNU General Public License (version 2) as
  16 **  published by the Free Software Foundation.
  17 **
  18 **  This program is distributed in the hope that it will be useful,
  19 **  but WITHOUT ANY WARRANTY; without even the implied warranty of
  20 **  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21 **  GNU General Public License for more details.
  22 **
  23 **  You should have received a copy of the GNU General Public License
  24 **  along with this program; if not, write to the Free Software
  25 **  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  26 ******************************************************************************/
  27 \r
  28 \r
  29 // For compilers that support precompilation, includes "wx/wx.h".\r
  30 #include "wx/wxprec.h"\r
  31 \r
  32 #ifdef __BORLANDC__\r
  33 #pragma hdrstop\r
  34 #endif\r
  35 \r
  36 #ifndef WX_PRECOMP\r
  37 #include "wx/wx.h"\r
  38 #endif\r
  39 \r
  40 #if !wxUSE_DOC_VIEW_ARCHITECTURE\r
  41 #error You must set wxUSE_DOC_VIEW_ARCHITECTURE to 1 in setup.h!\r
  42 #endif\r
  43 \r
  44 #include "ct.h"\r
  45 #include "ctsim.h"
  46 #include "docs.h"
  47 #include "views.h"
  48 #include "dialogs.h"
  49 \r
  50 #if defined(HAVE_CONFIG_H)
  51 #include "config.h"
  52 #endif\r
  53 \r
  54 #if defined(HAVE_GETOPT_H) || defined(HAVE_GETOPT_LONG)
  55 #ifdef MSVC\r
  56 #define __STDC__ 1\r
  57 #endif\r
  58 #include "getopt.h"\r
  59 #ifdef MSVC\r
  60 #undef __STDC__\r
  61 #endif
  62 #endif
  63 \r
  64 static const char* rcsindent = "$Id: ctsim.cpp,v 1.20 2000/12/22 04:18:00 kevin Exp $";
  65
  66 class CTSimApp* theApp = NULL;
  67
  68 struct option CTSimApp::ctsimOptions[] =
  69 {
  70     {"help", 0, 0, O_HELP},
  71     {"version", 0, 0, O_VERSION},
  72     {0, 0, 0, 0}
  73 };
  74
  75 IMPLEMENT_APP(CTSimApp)
  76
  77 CTSimApp::CTSimApp()
  78   : m_docManager(NULL), m_pFrame(NULL)
  79 {
  80     theApp = this;
  81 }
  82
  83 #ifdef HAVE_SYS_TIME_H
  84 #include <sys/time.h>
  85 #endif
  86
  87 #ifdef HAVE_SYS_RESOURCE_H
  88 #include <sys/resource.h>
  89 #endif
  90
  91 bool
  92 CTSimApp::OnInit()
  93 {
  94 #ifdef HAVE_SETPRIORITY
  95   setpriority (PRIO_PROCESS, 0, 15);  // set to low scheduling priority
  96 #endif
  97
  98     // process options
  99     while (1) {
 100       int c = getopt_long (argc, argv, "", ctsimOptions, NULL);
 101       if (c == -1)
 102         break;
 103
 104       switch (c) {
 105       case O_VERSION:
 106                   std::cout << rcsindent << std::endl;\r
 107 #ifdef CTSIMVERSION\r
 108                   std::cout << "Version: CTSIMVERSION" << std::endl;\r
 109 #elif defined(VERSION)\r
 110                   std::cout << "Version: VERSION" << std::endl;\r
 111 #endif
 112           exit(0);
 113       case O_HELP:
 114       case '?':
 115         usage (argv[0]);
 116         exit (0);
 117       default:
 118         usage (argv[0]);
 119         exit (1);
 120       }
 121     }
 122
 123     m_docManager = new wxDocManager;
 124
 125     new wxDocTemplate (m_docManager, "ImageFile", "*.if", "", "if", "ImageFile doc", "ImageFile View", CLASSINFO(ImageFileDocument), CLASSINFO(ImageFileView));
 126
 127     new wxDocTemplate (m_docManager, "ProjectionFile", "*.pj", "", "pj", "ProjectionFile doc", "ProjectionFile View", CLASSINFO(ProjectionFileDocument), CLASSINFO(ProjectionFileView));
 128
 129     new wxDocTemplate (m_docManager, "PhantomFile", "*.phm", "", "phm", "Phantom doc", "Phantom View", CLASSINFO(PhantomDocument), CLASSINFO(PhantomView));
 130
 131     new wxDocTemplate (m_docManager, "PlotFile", "*.plt", "", "plt", "Plot doc", "Plot View", CLASSINFO(PlotFileDocument), CLASSINFO(PlotFileView));
 132
 133     //// Create the main frame window
 134     m_pFrame = new MainFrame(m_docManager, (wxFrame *) NULL, -1, "CTSim", wxPoint(0, 0), wxSize(500, 400), wxDEFAULT_FRAME_STYLE);
 135
 136     SetTopWindow (m_pFrame);
 137     m_pFrame->Centre(wxBOTH);
 138
 139     m_pFrame->Show(true);
 140
 141     for (int i = optind + 1; i <= argc; i++) {
 142       wxString filename = argv [i - 1];
 143       m_docManager->CreateDocument (filename, wxDOC_SILENT);
 144     }
 145
 146     return true;
 147 }
 148
 149 void
 150 CTSimApp::usage(const char* program)
 151 {
 152         std::cout << "usage: " << fileBasename(program) << " [files-to-open...] [OPTIONS]\n";
 153         std::cout << "Computed Tomography Simulator (Graphical Shell)\n";
 154         std::cout << "\n";
 155         std::cout << "  --version Display version\n";
 156         std::cout << "  --help    Display this help message\n";
 157 }
 158
 159 int
 160 CTSimApp::OnExit()
 161 {
 162     delete m_docManager;
 163 #ifdef HAVE_DMALLOC
 164     dmalloc_shutdown();
 165 #endif
 166     return 0;
 167 }
 168
 169 wxString
 170 CTSimApp::getUntitledFilename()
 171 {
 172   static int untitledNumber = 1;
 173
 174   wxString filename ("Untitled");
 175   filename << untitledNumber++;
 176
 177   return (filename);
 178 }
 179
 180
 181 // Top-level window for CTSim
 182
 183 IMPLEMENT_CLASS(MainFrame, wxDocParentFrame)
 184
 185 BEGIN_EVENT_TABLE(MainFrame, wxDocParentFrame)
 186   EVT_MENU(MAINMENU_HELP_ABOUT, MainFrame::OnAbout)
 187   EVT_MENU(MAINMENU_HELP_CONTENTS, MainFrame::OnHelpContents)
 188   EVT_MENU(MAINMENU_FILE_CREATE_PHANTOM, MainFrame::OnCreatePhantom)
 189   EVT_MENU(MAINMENU_FILE_EXIT, MainFrame::OnExit)
 190   EVT_MENU(MAINMENU_WINDOW_BASE, MainFrame::OnWindowMenu0)
 191   EVT_MENU(MAINMENU_WINDOW_BASE+1, MainFrame::OnWindowMenu1)
 192   EVT_MENU(MAINMENU_WINDOW_BASE+2, MainFrame::OnWindowMenu2)
 193   EVT_MENU(MAINMENU_WINDOW_BASE+3, MainFrame::OnWindowMenu3)
 194   EVT_MENU(MAINMENU_WINDOW_BASE+4, MainFrame::OnWindowMenu4)
 195   EVT_MENU(MAINMENU_WINDOW_BASE+5, MainFrame::OnWindowMenu5)
 196   EVT_MENU(MAINMENU_WINDOW_BASE+6, MainFrame::OnWindowMenu6)
 197   EVT_MENU(MAINMENU_WINDOW_BASE+7, MainFrame::OnWindowMenu7)
 198   EVT_MENU(MAINMENU_WINDOW_BASE+8, MainFrame::OnWindowMenu8)
 199   EVT_MENU(MAINMENU_WINDOW_BASE+9, MainFrame::OnWindowMenu9)
 200   EVT_MENU(MAINMENU_WINDOW_BASE+10, MainFrame::OnWindowMenu10)
 201   EVT_MENU(MAINMENU_WINDOW_BASE+11, MainFrame::OnWindowMenu11)
 202   EVT_MENU(MAINMENU_WINDOW_BASE+12, MainFrame::OnWindowMenu12)
 203   EVT_MENU(MAINMENU_WINDOW_BASE+13, MainFrame::OnWindowMenu13)
 204   EVT_MENU(MAINMENU_WINDOW_BASE+14, MainFrame::OnWindowMenu14)
 205   EVT_MENU(MAINMENU_WINDOW_BASE+15, MainFrame::OnWindowMenu15)
 206   EVT_MENU(MAINMENU_WINDOW_BASE+16, MainFrame::OnWindowMenu16)
 207   EVT_MENU(MAINMENU_WINDOW_BASE+17, MainFrame::OnWindowMenu17)
 208   EVT_MENU(MAINMENU_WINDOW_BASE+18, MainFrame::OnWindowMenu18)
 209   EVT_MENU(MAINMENU_WINDOW_BASE+19, MainFrame::OnWindowMenu19)
 210   EVT_UPDATE_UI_RANGE(MAINMENU_WINDOW_BASE, MAINMENU_WINDOW_BASE+20, MainFrame::OnUpdateUI)
 211 END_EVENT_TABLE()
 212
 213
 214
 215 MainFrame::MainFrame(wxDocManager *manager, wxFrame *frame, wxWindowID id, const wxString& title, const wxPoint& pos, const wxSize& size, const long type)
 216   : wxDocParentFrame(manager, frame, id, title, pos, size, type), m_pLog(NULL)
 217 {
 218     m_pLog = new wxTextCtrl (this, -1, "Log Window\n", wxPoint(0, 250), wxSize(100,50), wxTE_MULTILINE | wxTE_READONLY);
 219     wxLog::SetActiveTarget(new wxLogTextCtrl(m_pLog));
 220     CreateStatusBar();
 221     SetStatusText ("Welcome to CTSim");
 222
 223     //// Make a menubar
 224     wxMenu *file_menu = new wxMenu;
 225
 226     file_menu->Append(MAINMENU_FILE_CREATE_PHANTOM, "Cr&eate Phantom...");
 227     file_menu->Append(wxID_OPEN, "&Open...");
 228
 229     file_menu->AppendSeparator();
 230     file_menu->Append(MAINMENU_FILE_EXIT, "E&xit");
 231
 232     //  history of files visited
 233     m_docManager->FileHistoryUseMenu(file_menu);
 234
 235     m_pWindowMenu = new wxMenu;
 236     m_pWindowMenu->UpdateUI (this);
 237
 238     wxMenu* help_menu = new wxMenu;
 239     help_menu->Append(MAINMENU_HELP_CONTENTS, "&Contents");
 240     help_menu->AppendSeparator();
 241     help_menu->Append(MAINMENU_HELP_ABOUT, "&About");
 242
 243     wxMenuBar* menu_bar = new wxMenuBar;
 244
 245     menu_bar->Append(file_menu, "&File");
 246     menu_bar->Append(m_pWindowMenu, "&Window");
 247     menu_bar->Append(help_menu, "&Help");
 248
 249     SetMenuBar(menu_bar);
 250
 251     for (int i = 0; i < MAX_WINDOW_MENUITEMS; i++) {
 252       m_apWindowMenuItems[i] = new wxMenuItem (m_pWindowMenu, MAINMENU_WINDOW_BASE+i, wxString("<Empty>"));
 253       m_pWindowMenu->Append (m_apWindowMenuItems[i]);
 254       m_pWindowMenu->Enable (MAINMENU_WINDOW_BASE+i, false);
 255     }
 256 }
 257
 258 void
 259 MainFrame::OnAbout(wxCommandEvent& WXUNUSED(event) )
 260 {\r
 261         wxString msg = "CTSim\nThe Open Source Computed Tomography Simulator\n";\r
 262 #ifdef CTSIMVERSION\r
 263         msg += "Version ";\r
 264         msg += CTSIMVERSION;\r
 265         msg += "\n\n";\r
 266 #elif defined(VERSION)\r
 267         msg << "Version: " <<  VERSION << "\n\n";\r
 268 #endif\r
 269         msg += "Author: Kevin Rosenberg <kevin@rosenberg.net>\nUsage: ctsim [files-to-open..] [--help]";\r
 270
 271     wxMessageBox(msg, "About CTSim", wxOK | wxICON_INFORMATION, this);
 272 }
 273
 274 void
 275 MainFrame::OnCreatePhantom(wxCommandEvent& WXUNUSED(event))
 276 {
 277     DialogGetPhantom dialogPhantom (this, Phantom::PHM_HERMAN);
 278     int dialogReturn = dialogPhantom.ShowModal();
 279     if (dialogReturn == wxID_OK) {
 280       wxString selection (dialogPhantom.getPhantom());
 281       *theApp->getLog() << "Selected phantom " << selection.c_str() << "\n";
 282       wxString filename = selection + ".phm";
 283       theApp->getDocManager()->CreateDocument(filename, wxDOC_SILENT);
 284     }
 285
 286 }
 287
 288 void\r
 289 CTSimApp::getCompatibleImages (const ImageFileDocument* pIFDoc, std::vector<ImageFileDocument*>& vecIF)\r
 290 {\r
 291   const ImageFile& rIF = pIFDoc->getImageFile();\r
 292   int nx = rIF.nx();\r
 293   int ny = rIF.ny();\r
 294   wxList& rListDocs = m_docManager->GetDocuments();\r
 295   std::vector<std::string> vecFilename;\r
 296   for (wxNode* pNode = rListDocs.GetFirst(); pNode != NULL; pNode = pNode->GetNext()) {\r
 297     wxDocument* pDoc = reinterpret_cast<wxDocument*>(pNode->GetData());\r
 298     ImageFileDocument* pIFCompareDoc = dynamic_cast<ImageFileDocument*>(pDoc);\r
 299     if (pIFCompareDoc && (pIFDoc != pIFCompareDoc)) {\r
 300       const ImageFile& rCompareIF = pIFCompareDoc->getImageFile();\r
 301       if (rCompareIF.nx() == nx && rCompareIF.ny() == ny) {\r
 302         std::string strFilename = pDoc->GetFilename();\r
 303         vecIF.push_back (pIFCompareDoc);\r
 304       }\r
 305     }\r
 306   }\r
 307 }\r
 308 \r
 309 void
 310 MainFrame::OnHelpContents(wxCommandEvent& WXUNUSED(event) )
 311 {
 312     wxMessageBox("No help available, refer to man pages of command-line tools");
 313 }
 314
 315 void
 316 MainFrame::OnExit (wxCommandEvent& WXUNUSED(event) )
 317 {
 318     Close(true);
 319 }
 320
 321 void
 322 MainFrame::OnUpdateUI (wxUpdateUIEvent& rEvent)
 323 {
 324   int iPos = 0;
 325   wxList& rListDocs = m_docManager->GetDocuments();
 326   wxNode* pNode = rListDocs.GetFirst();
 327   while (iPos < MAX_WINDOW_MENUITEMS && pNode != NULL) {
 328     wxDocument* pDoc = static_cast<wxDocument*>(pNode->GetData());
 329     wxString strFilename = pDoc->GetFilename();
 330     static_cast<wxMenuItemBase*>(m_apWindowMenuItems[iPos])->SetName (strFilename);
 331     m_apWindowMenuData[iPos] = pDoc;
 332     m_pWindowMenu->Enable (MAINMENU_WINDOW_BASE+iPos, true);
 333     iPos++;
 334     pNode = pNode->GetNext();
 335   }
 336   for (int i = iPos; i < MAX_WINDOW_MENUITEMS; i++) {
 337     m_pWindowMenu->Enable (MAINMENU_WINDOW_BASE+i, false);
 338     static_cast<wxMenuItemBase*>(m_apWindowMenuItems[i])->SetName (wxString("<Empty>"));
 339     m_apWindowMenuData[i] = NULL;
 340   }
 341
 342 }
 343
 344 void
 345 MainFrame::DoWindowMenu (int iMenuPosition, wxCommandEvent& event)
 346 {
 347   if (wxDocument* pDoc = m_apWindowMenuData [iMenuPosition]) {
 348     wxString strFilename = pDoc->GetFilename();
 349     const wxView* pView = pDoc->GetFirstView();
 350     if (pView) {
 351       wxFrame* pFrame = pView->GetFrame();
 352       pFrame->SetFocus();
 353       pFrame->Raise();
 354     }
 355   }
 356 }
 357
 358 void MainFrame::OnWindowMenu0 (wxCommandEvent& event)
 359 { DoWindowMenu (0, event); }
 360
 361 void MainFrame::OnWindowMenu1 (wxCommandEvent& event)
 362 { DoWindowMenu (1, event); }
 363
 364 void MainFrame::OnWindowMenu2 (wxCommandEvent& event)
 365 { DoWindowMenu (2, event); }
 366
 367 void MainFrame::OnWindowMenu3 (wxCommandEvent& event)
 368 { DoWindowMenu (3, event); }
 369
 370 void MainFrame::OnWindowMenu4 (wxCommandEvent& event)
 371 { DoWindowMenu (4, event); }
 372
 373 void MainFrame::OnWindowMenu5 (wxCommandEvent& event)
 374 { DoWindowMenu (5, event); }
 375
 376 void MainFrame::OnWindowMenu6 (wxCommandEvent& event)
 377 { DoWindowMenu (6, event); }
 378
 379 void MainFrame::OnWindowMenu7 (wxCommandEvent& event)
 380 { DoWindowMenu (7, event); }
 381
 382 void MainFrame::OnWindowMenu8 (wxCommandEvent& event)
 383 { DoWindowMenu (8, event); }
 384
 385 void MainFrame::OnWindowMenu9 (wxCommandEvent& event)
 386 { DoWindowMenu (9, event); }
 387
 388 void MainFrame::OnWindowMenu10 (wxCommandEvent& event)
 389 { DoWindowMenu (10, event); }
 390
 391 void MainFrame::OnWindowMenu11 (wxCommandEvent& event)
 392 { DoWindowMenu (11, event); }
 393
 394 void MainFrame::OnWindowMenu12 (wxCommandEvent& event)
 395 { DoWindowMenu (12, event); }
 396
 397 void MainFrame::OnWindowMenu13 (wxCommandEvent& event)
 398 { DoWindowMenu (13, event); }
 399
 400 void MainFrame::OnWindowMenu14 (wxCommandEvent& event)
 401 { DoWindowMenu (14, event); }
 402
 403 void MainFrame::OnWindowMenu15 (wxCommandEvent& event)
 404 { DoWindowMenu (15, event); }
 405
 406 void MainFrame::OnWindowMenu16 (wxCommandEvent& event)
 407 { DoWindowMenu (16, event); }
 408
 409 void MainFrame::OnWindowMenu17 (wxCommandEvent& event)
 410 { DoWindowMenu (17, event); }
 411
 412 void MainFrame::OnWindowMenu18 (wxCommandEvent& event)
 413 { DoWindowMenu (18, event); }
 414
 415 void MainFrame::OnWindowMenu19 (wxCommandEvent& event)
 416 { DoWindowMenu (19, event); }
 417
 418