r418: no message
[ctsim.git] / 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.35 2001/01/18 21:30:22 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
28 #ifdef MSVC
29 #define strdup _strdup
30 #endif
31
32 // For compilers that support precompilation, includes "wx/wx.h".
33 #include "wx/wxprec.h"
34
35 #ifdef __BORLANDC__
36 #pragma hdrstop
37 #endif
38
39 #ifndef WX_PRECOMP
40 #include "wx/wx.h"
41 #endif
42
43 #include "wx/image.h"
44 #include "wx/filesys.h"
45 #include "wx/fs_zip.h"
46
47 #if !wxUSE_DOC_VIEW_ARCHITECTURE
48 #error You must set wxUSE_DOC_VIEW_ARCHITECTURE to 1 in setup.h!
49 #endif
50
51 #include "ct.h"
52 #include "ctsim.h"
53 #include "ctsim-map.h"
54 #include "docs.h"
55 #include "views.h"
56 #include "dialogs.h"
57
58 #if defined(HAVE_CONFIG_H)
59 #include "config.h"
60 #endif
61
62 #if defined(HAVE_GETOPT_H) || defined(HAVE_GETOPT_LONG)
63 #ifdef MSVC
64 #define __STDC__ 1
65 #endif
66 #include "getopt.h"
67 #ifdef MSVC
68 #undef __STDC__
69 #endif
70 #endif
71
72 static const char* rcsindent = "$Id: ctsim.cpp,v 1.35 2001/01/18 21:30:22 kevin Exp $";
73
74 struct option CTSimApp::ctsimOptions[] = 
75 {
76   {"help", 0, 0, O_HELP},
77   {"version", 0, 0, O_VERSION},
78   {0, 0, 0, 0}
79 };
80
81 IMPLEMENT_APP(CTSimApp)
82
83 CTSimApp::CTSimApp()
84 : m_docManager(NULL), m_pFrame(NULL)
85 {
86   theApp = this;
87 }
88
89 #ifdef HAVE_SYS_TIME_H
90 #include <sys/time.h>
91 #endif
92
93 #ifdef HAVE_SYS_RESOURCE_H
94 #include <sys/resource.h>
95 #endif
96
97 bool
98 CTSimApp::OnInit()
99 {
100 #ifdef HAVE_SETPRIORITY
101   setpriority (PRIO_PROCESS, 0, 15);  // set to low scheduling priority
102 #endif
103
104   g_bRunningWXWindows = true;
105   // process options
106   while (1) {
107     int c = getopt_long (argc, argv, "", ctsimOptions, NULL);
108     if (c == -1)
109       break;
110     
111     switch (c) {
112     case O_VERSION:
113       std::cout << rcsindent << std::endl;
114 #ifdef CTSIMVERSION
115       std::cout << "Version: CTSIMVERSION" << std::endl;
116 #elif defined(VERSION)
117       std::cout << "Version: VERSION" << std::endl;
118 #endif
119       exit(0);
120     case O_HELP:
121     case '?':
122       usage (argv[0]);
123       exit (0);
124     default:
125       usage (argv[0]);
126       exit (1);
127     }
128   }
129   
130   m_docManager = new wxDocManager;
131   
132   new wxDocTemplate (m_docManager, "ImageFile", "*.if", "", "if", "ImageFile doc", "ImageFile View", CLASSINFO(ImageFileDocument), CLASSINFO(ImageFileView));
133   new wxDocTemplate (m_docManager, "ProjectionFile", "*.pj", "", "pj", "ProjectionFile doc", "ProjectionFile View", CLASSINFO(ProjectionFileDocument), CLASSINFO(ProjectionFileView));
134   new wxDocTemplate (m_docManager, "PhantomFile", "*.phm", "", "phm", "Phantom doc", "Phantom View", CLASSINFO(PhantomDocument), CLASSINFO(PhantomView));
135   new wxDocTemplate (m_docManager, "PlotFile", "*.plt", "", "plt", "Plot doc", "Plot View", CLASSINFO(PlotFileDocument), CLASSINFO(PlotFileView));
136   
137 #if wxUSE_GIF
138     wxImage::AddHandler(new wxGIFHandler);     // Required for images in the online documentation
139 #endif
140
141 #if wxUSE_STREAMS && wxUSE_ZIPSTREAM && wxUSE_ZLIB
142       wxFileSystem::AddHandler(new wxZipFSHandler);     // Required for advanced HTML help
143 #endif
144
145   // Create the main frame window
146   m_pFrame = new MainFrame(m_docManager, (wxFrame *) NULL, -1, "CTSim", wxPoint(0, 0), wxSize(500, 400), wxDEFAULT_FRAME_STYLE);
147
148   SetTopWindow (m_pFrame);
149   m_pFrame->Centre(wxBOTH);
150   
151   m_pFrame->Show(true);
152   
153   for (int i = optind + 1; i <= argc; i++) {
154     wxString filename = argv [i - 1];
155     m_docManager->CreateDocument (filename, wxDOC_SILENT);
156   }
157
158   setIconForFrame (m_pFrame);
159
160   if (! m_pFrame->getHelpController().Initialize("ctsim"))
161     ::wxMessageBox ("Cannot initialize the help system", "Error");
162
163  
164   return true;
165 }
166
167
168 #include "./ctsim.xpm"
169 void
170 CTSimApp::setIconForFrame(wxFrame* pFrame)
171 {
172   wxIcon iconApp (ctsim16_xpm);
173
174   if (iconApp.Ok())
175     pFrame->SetIcon (iconApp);
176 }
177
178 void
179 CTSimApp::usage(const char* program)
180 {
181   std::cout << "usage: " << fileBasename(program) << " [files-to-open...] [OPTIONS]\n";
182   std::cout << "Computed Tomography Simulator (Graphical Shell)\n";
183   std::cout << "\n";
184   std::cout << "  --version Display version\n";
185   std::cout << "  --help    Display this help message\n";
186 }
187
188 int
189 CTSimApp::OnExit()
190 {
191   delete m_docManager;
192 #ifdef HAVE_DMALLOC
193   dmalloc_shutdown();
194 #endif
195   return 0;
196 }
197
198 wxString
199 CTSimApp::getUntitledFilename()
200 {
201   static int untitledNumber = 1;
202   
203   wxString filename ("Untitled");
204   filename << untitledNumber++;
205   
206   return (filename);
207 }
208
209
210 // Top-level window for CTSim
211
212 #if CTSIM_MDI
213 IMPLEMENT_CLASS(MainFrame, wxMDIParentFrame)
214
215 BEGIN_EVENT_TABLE(MainFrame, wxMDIParentFrame)
216 #else
217 IMPLEMENT_CLASS(MainFrame, wxDocParentFrame)
218
219 BEGIN_EVENT_TABLE(MainFrame, wxDocParentFrame)
220 #endif
221
222 EVT_MENU(MAINMENU_HELP_ABOUT, MainFrame::OnAbout)
223 EVT_MENU(MAINMENU_HELP_CONTENTS, MainFrame::OnHelpContents)
224 EVT_MENU(MAINMENU_HELP_TOPICS, MainFrame::OnHelpTopics)
225 EVT_MENU(MAINMENU_FILE_CREATE_PHANTOM, MainFrame::OnCreatePhantom)
226 EVT_MENU(MAINMENU_FILE_CREATE_FILTER, MainFrame::OnCreateFilter)
227 EVT_MENU(MAINMENU_FILE_EXIT, MainFrame::OnExit)
228 EVT_MENU(MAINMENU_WINDOW_BASE, MainFrame::OnWindowMenu0)
229 EVT_MENU(MAINMENU_WINDOW_BASE+1, MainFrame::OnWindowMenu1)
230 EVT_MENU(MAINMENU_WINDOW_BASE+2, MainFrame::OnWindowMenu2)
231 EVT_MENU(MAINMENU_WINDOW_BASE+3, MainFrame::OnWindowMenu3)
232 EVT_MENU(MAINMENU_WINDOW_BASE+4, MainFrame::OnWindowMenu4)
233 EVT_MENU(MAINMENU_WINDOW_BASE+5, MainFrame::OnWindowMenu5)
234 EVT_MENU(MAINMENU_WINDOW_BASE+6, MainFrame::OnWindowMenu6)
235 EVT_MENU(MAINMENU_WINDOW_BASE+7, MainFrame::OnWindowMenu7)
236 EVT_MENU(MAINMENU_WINDOW_BASE+8, MainFrame::OnWindowMenu8)
237 EVT_MENU(MAINMENU_WINDOW_BASE+9, MainFrame::OnWindowMenu9)
238 EVT_MENU(MAINMENU_WINDOW_BASE+10, MainFrame::OnWindowMenu10)
239 EVT_MENU(MAINMENU_WINDOW_BASE+11, MainFrame::OnWindowMenu11)
240 EVT_MENU(MAINMENU_WINDOW_BASE+12, MainFrame::OnWindowMenu12)
241 EVT_MENU(MAINMENU_WINDOW_BASE+13, MainFrame::OnWindowMenu13)
242 EVT_MENU(MAINMENU_WINDOW_BASE+14, MainFrame::OnWindowMenu14)
243 EVT_MENU(MAINMENU_WINDOW_BASE+15, MainFrame::OnWindowMenu15)
244 EVT_MENU(MAINMENU_WINDOW_BASE+16, MainFrame::OnWindowMenu16)
245 EVT_MENU(MAINMENU_WINDOW_BASE+17, MainFrame::OnWindowMenu17)
246 EVT_MENU(MAINMENU_WINDOW_BASE+18, MainFrame::OnWindowMenu18)
247 EVT_MENU(MAINMENU_WINDOW_BASE+19, MainFrame::OnWindowMenu19)
248 EVT_UPDATE_UI_RANGE(MAINMENU_WINDOW_BASE, MAINMENU_WINDOW_BASE+20, MainFrame::OnUpdateUI)
249 END_EVENT_TABLE()
250
251
252
253 #if CTSIM_MDI
254 MainFrame::MainFrame(wxDocManager *manager, wxFrame *frame, wxWindowID id, const wxString& title, const wxPoint& pos, const wxSize& size, const long type)
255 : wxMDIParentFrame(NULL,  id, title, pos, size, type), m_pLog(NULL)
256 #else
257 MainFrame::MainFrame(wxDocManager *manager, wxFrame *frame, wxWindowID id, const wxString& title, const wxPoint& pos, const wxSize& size, const long type)
258 : wxDocParentFrame(manager, frame, id, title, pos, size, type), m_pLog(NULL)
259 #endif
260 {
261   m_pLog = new wxTextCtrl (this, -1, "Log Window\n", wxPoint(0, 250), wxSize(100,50), wxTE_MULTILINE | wxTE_READONLY);
262   wxLog::SetActiveTarget(new wxLogTextCtrl(m_pLog));
263   CreateStatusBar();
264   SetStatusText ("Welcome to CTSim");
265   
266   //// Make a menubar
267   wxMenu *file_menu = new wxMenu;
268   
269   file_menu->Append(MAINMENU_FILE_CREATE_PHANTOM, "Cr&eate Phantom...\tCtrl-P");
270   file_menu->Append(MAINMENU_FILE_CREATE_FILTER, "Create &Filter...\tCtrl-F");
271   file_menu->Append(wxID_OPEN, "&Open...\tCtrl-O");
272   
273   file_menu->AppendSeparator();
274   file_menu->Append(MAINMENU_FILE_EXIT, "E&xit");
275   
276   //  history of files visited
277   theApp->getDocManager()->FileHistoryUseMenu(file_menu);
278   
279   m_pWindowMenu = new wxMenu;
280   m_pWindowMenu->UpdateUI (this);
281   
282   wxMenu* help_menu = new wxMenu;
283   help_menu->Append(MAINMENU_HELP_CONTENTS, "&Contents\tF1");
284   help_menu->Append(MAINMENU_HELP_TOPICS, "&Topics\tCtrl-T");
285   help_menu->Append(MAINMENU_HELP_ABOUT, "&About");
286   
287   wxMenuBar* menu_bar = new wxMenuBar;
288   
289   menu_bar->Append(file_menu, "&File");
290   menu_bar->Append(m_pWindowMenu, "&Window");
291   menu_bar->Append(help_menu, "&Help");
292   
293   SetMenuBar(menu_bar);
294   
295   for (int i = 0; i < MAX_WINDOW_MENUITEMS; i++) {
296     m_apWindowMenuItems[i] = new wxMenuItem (m_pWindowMenu, MAINMENU_WINDOW_BASE+i, wxString("<Empty>"));
297     m_pWindowMenu->Append (m_apWindowMenuItems[i]);
298     m_pWindowMenu->Enable (MAINMENU_WINDOW_BASE+i, false);
299   }
300   
301   m_iDefaultPhantomID = Phantom::PHM_HERMAN;
302   m_iDefaultFilterID = SignalFilter::FILTER_BANDLIMIT;
303   m_iDefaultFilterDomainID = SignalFilter::DOMAIN_FREQUENCY;
304   m_iDefaultFilterXSize = 256;
305   m_iDefaultFilterYSize = 256;
306   m_dDefaultFilterParam = 1.;
307   m_dDefaultFilterBandwidth = 1.;
308   m_dDefaultFilterInputScale = 1.;
309   m_dDefaultFilterOutputScale = 1.;
310   
311 }
312
313 void 
314 MainFrame::OnAbout(wxCommandEvent& WXUNUSED(event) )
315 {
316   wxString msg = "CTSim\nThe Open Source Computed Tomography Simulator\n";
317 #ifdef CTSIMVERSION
318   msg += "Version ";
319   msg += CTSIMVERSION;
320   msg += "\n\n";
321 #elif defined(VERSION)
322   msg << "Version: " <<  VERSION << "\n\n";
323 #endif
324   msg += "Author: Kevin Rosenberg <kevin@rosenberg.net>\nUsage: ctsim [files-to-open..] [--help]";
325   
326   wxMessageBox(msg, "About CTSim", wxOK | wxICON_INFORMATION, this);
327 }
328
329 void 
330 MainFrame::OnCreatePhantom(wxCommandEvent& event)
331 {
332   DialogGetPhantom dialogPhantom (this, m_iDefaultPhantomID);
333   int dialogReturn = dialogPhantom.ShowModal();
334   if (dialogReturn == wxID_OK) {
335     wxString selection (dialogPhantom.getPhantom());
336     *theApp->getLog() << "Selected phantom " << selection.c_str() << "\n";
337     wxString filename = selection + ".phm";
338     m_iDefaultPhantomID = Phantom::convertNameToPhantomID (selection.c_str());
339     theApp->getDocManager()->CreateDocument(filename, wxDOC_SILENT);
340   }
341   
342 }
343
344 void 
345 MainFrame::OnCreateFilter (wxCommandEvent& WXUNUSED(event))
346 {
347   DialogGetFilterParameters dialogFilter (this, m_iDefaultFilterXSize, m_iDefaultFilterYSize, m_iDefaultFilterID, m_dDefaultFilterParam, m_dDefaultFilterBandwidth, m_iDefaultFilterDomainID, m_dDefaultFilterInputScale, m_dDefaultFilterOutputScale);
348   int dialogReturn = dialogFilter.ShowModal();
349   if (dialogReturn == wxID_OK) {
350     wxString strFilter (dialogFilter.getFilterName());
351     wxString strDomain (dialogFilter.getDomainName());
352     m_iDefaultFilterID = SignalFilter::convertFilterNameToID (strFilter.c_str());
353     m_iDefaultFilterDomainID = SignalFilter::convertDomainNameToID (strDomain.c_str());
354     m_iDefaultFilterXSize = dialogFilter.getXSize();
355     m_iDefaultFilterYSize = dialogFilter.getYSize();
356     m_dDefaultFilterBandwidth = dialogFilter.getBandwidth();
357     m_dDefaultFilterParam= dialogFilter.getFilterParam();
358     m_dDefaultFilterInputScale = dialogFilter.getInputScale();
359     m_dDefaultFilterOutputScale = dialogFilter.getOutputScale();
360     std::ostringstream os;
361     os << "Generate Filter=" << strFilter.c_str() 
362       << ", size=(" << static_cast<int>(m_iDefaultFilterXSize) << "," << static_cast<int>(m_iDefaultFilterYSize) 
363       << "), domain=" << strDomain.c_str() << ", filterParam=" << m_dDefaultFilterParam << ", bandwidth=" << m_dDefaultFilterBandwidth 
364       << ", inputScale=" << m_dDefaultFilterInputScale << ", outputScale=" << m_dDefaultFilterOutputScale;
365     *theApp->getLog() << os.str().c_str() << "\n";
366     wxString filename = "untitled.if";
367     ImageFileDocument* pFilterDoc = dynamic_cast<ImageFileDocument*>(theApp->getDocManager()->CreateDocument ("untitled.if", wxDOC_SILENT));
368     if (! pFilterDoc) {
369       sys_error (ERR_SEVERE, "Unable to create filter image");
370       return;
371     }
372     ImageFile& rIF = pFilterDoc->getImageFile();
373     rIF.setArraySize (m_iDefaultFilterXSize, m_iDefaultFilterYSize);
374     rIF.filterResponse (strDomain.c_str(), m_dDefaultFilterBandwidth, strFilter.c_str(), m_dDefaultFilterParam, m_dDefaultFilterInputScale, m_dDefaultFilterOutputScale);
375     rIF.labelAdd (os.str().c_str());
376     if (theApp->getSetModifyNewDocs())
377       pFilterDoc->Modify (true);
378     pFilterDoc->UpdateAllViews();
379     pFilterDoc->GetFirstView()->OnUpdate (NULL, NULL);
380   }
381 }
382
383 void
384 CTSimApp::getCompatibleImages (const ImageFileDocument* pIFDoc, std::vector<ImageFileDocument*>& vecIF)
385 {
386   const ImageFile& rIF = pIFDoc->getImageFile();
387   unsigned int nx = rIF.nx();
388   unsigned int ny = rIF.ny();
389   wxList& rListDocs = m_docManager->GetDocuments();
390   for (wxNode* pNode = rListDocs.GetFirst(); pNode != NULL; pNode = pNode->GetNext()) {
391     wxDocument* pDoc = reinterpret_cast<wxDocument*>(pNode->GetData());
392     ImageFileDocument* pIFCompareDoc = dynamic_cast<ImageFileDocument*>(pDoc);
393     if (pIFCompareDoc && (pIFDoc != pIFCompareDoc)) {
394       const ImageFile& rCompareIF = pIFCompareDoc->getImageFile();
395       if (rCompareIF.nx() == nx && rCompareIF.ny() == ny)
396         vecIF.push_back (pIFCompareDoc);
397     }
398   }
399 }
400
401 void 
402 MainFrame::OnHelpTopics (wxCommandEvent& event)
403 {
404    showHelp (event.GetId());
405 }
406
407 void 
408 MainFrame::OnHelpContents (wxCommandEvent& event)
409 {
410    showHelp (event.GetId());
411 }
412
413 void
414 MainFrame::showHelp (int commandID)
415 {
416   m_help.LoadFile();
417
418   switch (commandID) {
419   case MAINMENU_HELP_CONTENTS:
420      m_help.DisplayContents ();
421      break;
422   case MAINMENU_HELP_TOPICS:
423      m_help.DisplaySection (introduction);
424      break;
425
426   default:
427     *getLog() << "Unknown help command # " << commandID << "\n";
428     break;
429   }
430 }
431
432 void 
433 MainFrame::OnExit (wxCommandEvent& WXUNUSED(event) )
434 {
435   Close(true);
436 }
437
438 void
439 MainFrame::OnUpdateUI (wxUpdateUIEvent& rEvent)
440 {
441   int iPos = 0;
442   wxList& rListDocs = theApp->getDocManager()->GetDocuments();
443   wxNode* pNode = rListDocs.GetFirst();
444   while (iPos < MAX_WINDOW_MENUITEMS && pNode != NULL) {
445     wxDocument* pDoc = static_cast<wxDocument*>(pNode->GetData());
446     wxString strFilename = pDoc->GetFilename();
447     static_cast<wxMenuItemBase*>(m_apWindowMenuItems[iPos])->SetName (strFilename);
448     m_apWindowMenuData[iPos] = pDoc;
449     m_pWindowMenu->Enable (MAINMENU_WINDOW_BASE+iPos, true);
450     iPos++;
451     pNode = pNode->GetNext();
452   }
453   for (int i = iPos; i < MAX_WINDOW_MENUITEMS; i++) {
454     m_pWindowMenu->Enable (MAINMENU_WINDOW_BASE+i, false);
455     static_cast<wxMenuItemBase*>(m_apWindowMenuItems[i])->SetName (wxString("<Empty>"));
456     m_apWindowMenuData[i] = NULL;
457   }
458   
459 }
460
461 void 
462 MainFrame::DoWindowMenu (int iMenuPosition, wxCommandEvent& event)
463 {
464   if (wxDocument* pDoc = m_apWindowMenuData [iMenuPosition]) {
465     wxString strFilename = pDoc->GetFilename();
466     const wxView* pView = pDoc->GetFirstView();
467     if (pView) {
468       wxFrame* pFrame = pView->GetFrame();
469       pFrame->SetFocus();
470       pFrame->Raise();
471     }
472   }
473 }
474
475 void MainFrame::OnWindowMenu0 (wxCommandEvent& event)
476 { DoWindowMenu (0, event); }
477
478 void MainFrame::OnWindowMenu1 (wxCommandEvent& event)
479 { DoWindowMenu (1, event); }
480
481 void MainFrame::OnWindowMenu2 (wxCommandEvent& event)
482 { DoWindowMenu (2, event); }
483
484 void MainFrame::OnWindowMenu3 (wxCommandEvent& event)
485 { DoWindowMenu (3, event); }
486
487 void MainFrame::OnWindowMenu4 (wxCommandEvent& event)
488 { DoWindowMenu (4, event); }
489
490 void MainFrame::OnWindowMenu5 (wxCommandEvent& event)
491 { DoWindowMenu (5, event); }
492
493 void MainFrame::OnWindowMenu6 (wxCommandEvent& event)
494 { DoWindowMenu (6, event); }
495
496 void MainFrame::OnWindowMenu7 (wxCommandEvent& event)
497 { DoWindowMenu (7, event); }
498
499 void MainFrame::OnWindowMenu8 (wxCommandEvent& event)
500 { DoWindowMenu (8, event); }
501
502 void MainFrame::OnWindowMenu9 (wxCommandEvent& event)
503 { DoWindowMenu (9, event); }
504
505 void MainFrame::OnWindowMenu10 (wxCommandEvent& event)
506 { DoWindowMenu (10, event); }
507
508 void MainFrame::OnWindowMenu11 (wxCommandEvent& event)
509 { DoWindowMenu (11, event); }
510
511 void MainFrame::OnWindowMenu12 (wxCommandEvent& event)
512 { DoWindowMenu (12, event); }
513
514 void MainFrame::OnWindowMenu13 (wxCommandEvent& event)
515 { DoWindowMenu (13, event); }
516
517 void MainFrame::OnWindowMenu14 (wxCommandEvent& event)
518 { DoWindowMenu (14, event); }
519
520 void MainFrame::OnWindowMenu15 (wxCommandEvent& event)
521 { DoWindowMenu (15, event); }
522
523 void MainFrame::OnWindowMenu16 (wxCommandEvent& event)
524 { DoWindowMenu (16, event); }
525
526 void MainFrame::OnWindowMenu17 (wxCommandEvent& event)
527 { DoWindowMenu (17, event); }
528
529 void MainFrame::OnWindowMenu18 (wxCommandEvent& event)
530 { DoWindowMenu (18, event); }
531
532 void MainFrame::OnWindowMenu19 (wxCommandEvent& event)
533 { DoWindowMenu (19, event); }
534
535