r189: *** empty log message ***
[ctsim.git] / src / views.cpp
1 /*****************************************************************************
2 ** FILE IDENTIFICATION
3 **
4 **   Name:          view.cpp
5 **   Purpose:       View & Canvas routines for 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: views.cpp,v 1.21 2000/09/04 09:06:46 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 __GNUG__
29 // #pragma implementation
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 #if !wxUSE_DOC_VIEW_ARCHITECTURE
44 #error You must set wxUSE_DOC_VIEW_ARCHITECTURE to 1 in setup.h!
45 #endif
46
47 #include "wx/image.h"
48 #include "wx/progdlg.h"
49
50 #include "ct.h"
51 #include "ctsim.h"
52 #include "docs.h"
53 #include "views.h"
54 #include "dialogs.h"
55 #include "dlgprojections.h"
56 #include "dlgreconstruct.h"
57 #include <sstream>
58 #include "backprojectors.h"
59 #include "reconstruct.h"
60 #include "timer.h"
61
62 // ImageFileCanvas
63
64 BEGIN_EVENT_TABLE(ImageFileCanvas, wxScrolledWindow)
65     EVT_MOUSE_EVENTS(ImageFileCanvas::OnMouseEvent)
66 END_EVENT_TABLE()
67
68
69 ImageFileCanvas::ImageFileCanvas (ImageFileView* v, wxFrame *frame, const wxPoint& pos, const wxSize& size, const long style)
70   : wxScrolledWindow(frame, -1, pos, size, style)
71 {
72     m_pView = v;
73 }
74
75 void 
76 ImageFileCanvas::OnDraw(wxDC& dc)
77 {
78     if (m_pView)
79         m_pView->OnDraw(& dc);
80 }
81
82 void 
83 ImageFileCanvas::OnMouseEvent(wxMouseEvent& event)
84 {
85     if (! m_pView)
86         return;
87     
88     wxClientDC dc(this);
89     PrepareDC(dc);
90     
91     wxPoint pt(event.GetLogicalPosition(dc));
92
93     if (event.LeftIsDown()) {
94         const ImageFile& rIF = m_pView->GetDocument()->getImageFile();
95         ImageFileArrayConst v = rIF.getArray();
96         int nx = rIF.nx();
97         int ny = rIF.ny();
98
99         if (pt.x >= 0 && pt.x < nx && pt.y >= 0 & pt.y < ny) {
100           ostringstream os;
101           os << "Image value (" << pt.x << "," << pt.y << ") = " << v[pt.x][ny - 1 - pt.y] << "\n";
102             *theApp->getLog() << os.str().c_str();
103         } else
104             *theApp->getLog() << "Mouse out of image range (" << pt.x << "," << pt.y << ")\n";
105             
106     }
107 }
108
109
110 // ImageFileView
111
112 IMPLEMENT_DYNAMIC_CLASS(ImageFileView, wxView)
113
114 BEGIN_EVENT_TABLE(ImageFileView, wxView)
115   EVT_MENU(IFMENU_FILE_PROPERTIES, ImageFileView::OnProperties)
116   EVT_MENU(IFMENU_VIEW_SCALE_MINMAX, ImageFileView::OnScaleMinMax)
117   EVT_MENU(IFMENU_VIEW_SCALE_AUTO, ImageFileView::OnScaleAuto)
118 END_EVENT_TABLE()
119
120 ImageFileView::ImageFileView(void) 
121   : wxView(), m_canvas(NULL), m_frame(NULL), m_bMinSpecified(false), m_bMaxSpecified(false)
122 {
123 }
124
125 ImageFileView::~ImageFileView(void)
126 {
127 }
128
129 void
130 ImageFileView::OnProperties (wxCommandEvent& event)
131 {
132   double min, max, mean, mode, median, stddev;
133   const ImageFile& rIF = GetDocument()->getImageFile();
134   if (rIF.nx() == 0 || rIF.ny() == 0)
135     *theApp->getLog() << "Properties: empty imagefile\n";
136   else {
137     const string& rFilename = rIF.getFilename();
138     rIF.statistics (min, max, mean, mode, median, stddev);
139     ostringstream os;
140     os << "file: " << rFilename << "\nmin: "<<min<<"\nmax: "<<max<<"\nmean: "<<mean<<"\nmode: "<<mode<<"\nstddev: "<<stddev << "\n";
141     *theApp->getLog() << os.str().c_str();
142     wxMessageDialog dialogMsg (m_frame, os.str().c_str(), "Imagefile Properties", wxOK | wxICON_INFORMATION);
143     dialogMsg.ShowModal();
144   }
145 }
146
147 void 
148 ImageFileView::OnScaleAuto (wxCommandEvent& event)
149 {
150     const ImageFile& rIF = GetDocument()->getImageFile();
151     DialogAutoScaleParameters dialogAutoScale (m_frame, rIF, m_dAutoScaleFactor);
152     int iRetVal = dialogAutoScale.ShowModal();
153     if (iRetVal == wxID_OK) {
154       m_bMinSpecified = true;
155       m_bMaxSpecified = true;
156       double dMin, dMax;
157       dialogAutoScale.getMinMax (&dMin, &dMax);
158       m_dMinPixel = dMin;
159       m_dMaxPixel = dMax;
160       m_dAutoScaleFactor = dialogAutoScale.getAutoScaleFactor();
161       OnUpdate (this, NULL);
162     }
163 }
164
165 void 
166 ImageFileView::OnScaleMinMax (wxCommandEvent& event)
167 {
168     const ImageFile& rIF = GetDocument()->getImageFile();
169     double min, max;
170     if (! m_bMinSpecified && ! m_bMaxSpecified)
171       rIF.getMinMax (min, max);
172
173     if (m_bMinSpecified)
174       min = m_dMinPixel;
175     if (m_bMaxSpecified)
176       max = m_dMaxPixel;
177
178     DialogGetImageMinMax dialogMinMax (m_frame, rIF, min, max);
179     int retVal = dialogMinMax.ShowModal();
180     if (retVal == wxID_OK) {
181       m_bMinSpecified = true;
182       m_bMaxSpecified = true;
183       m_dMinPixel = dialogMinMax.getMinimum();
184       m_dMaxPixel = dialogMinMax.getMaximum();
185       OnUpdate (this, NULL);
186     }
187 }
188
189
190 ImageFileCanvas* 
191 ImageFileView::CreateCanvas (wxView *view, wxFrame *parent)
192 {
193     ImageFileCanvas* pCanvas;
194     int width, height;
195     parent->GetClientSize(&width, &height);
196     
197     pCanvas = new ImageFileCanvas (dynamic_cast<ImageFileView*>(view), parent, wxPoint(0, 0), wxSize(width, height), 0);
198     
199     pCanvas->SetScrollbars(20, 20, 50, 50);
200     pCanvas->SetBackgroundColour(*wxWHITE);
201     pCanvas->Clear();
202     
203     return pCanvas;
204 }
205
206 wxFrame*
207 ImageFileView::CreateChildFrame(wxDocument *doc, wxView *view)
208 {
209     wxDocChildFrame *subframe = new wxDocChildFrame(doc, view, theApp->getMainFrame(), -1, "ImageFile Frame", wxPoint(-1, -1), wxSize(0, 0), wxDEFAULT_FRAME_STYLE);
210     
211     wxMenu *file_menu = new wxMenu;
212     
213     file_menu->Append(MAINMENU_FILE_CREATE_PHANTOM, "Cr&eate Phantom...");
214     file_menu->Append(wxID_OPEN, "&Open...");
215     file_menu->Append(wxID_SAVE, "&Save");
216     file_menu->Append(wxID_SAVEAS, "Save &As...");
217     file_menu->Append(wxID_CLOSE, "&Close");
218     
219     file_menu->AppendSeparator();
220     file_menu->Append(IFMENU_FILE_PROPERTIES, "P&roperties");
221
222     file_menu->AppendSeparator();
223     file_menu->Append(wxID_PRINT, "&Print...");
224     file_menu->Append(wxID_PRINT_SETUP, "Print &Setup...");
225     file_menu->Append(wxID_PREVIEW, "Print Pre&view");
226     
227     wxMenu *view_menu = new wxMenu;
228     view_menu->Append(IFMENU_VIEW_SCALE_MINMAX, "Display Scale &Set...");
229     view_menu->Append(IFMENU_VIEW_SCALE_AUTO, "Display Scale &Auto...");
230     
231     wxMenu *help_menu = new wxMenu;
232     help_menu->Append(MAINMENU_HELP_ABOUT, "&About");
233     
234     wxMenuBar *menu_bar = new wxMenuBar;
235     
236     menu_bar->Append(file_menu, "&File");
237     menu_bar->Append(view_menu, "&View");
238     menu_bar->Append(help_menu, "&Help");
239     
240     subframe->SetMenuBar(menu_bar);
241     
242     subframe->Centre(wxBOTH);
243     
244     return subframe;
245 }
246
247
248 bool 
249 ImageFileView::OnCreate(wxDocument *doc, long WXUNUSED(flags) )
250 {
251     m_frame = CreateChildFrame(doc, this);
252     SetFrame(m_frame);
253     
254     m_bMinSpecified = false;
255     m_bMaxSpecified = false;
256     m_dAutoScaleFactor = 1.;
257
258     int width, height;
259     m_frame->GetClientSize(&width, &height);
260     m_frame->SetTitle("ImageFileView");
261     m_canvas = CreateCanvas(this, m_frame);
262
263 #ifdef __X__
264     int x, y;  // X requires a forced resize
265     m_frame->GetSize(&x, &y);
266     m_frame->SetSize(-1, -1, x, y);
267 #endif
268
269     m_frame->Show(true);
270     Activate(true);
271     
272     return true;
273 }
274
275 void 
276 ImageFileView::OnDraw (wxDC* dc)
277 {
278   if (m_bitmap.Ok())
279     dc->DrawBitmap(m_bitmap, 0, 0, false);
280 }
281
282
283 void 
284 ImageFileView::OnUpdate(wxView *WXUNUSED(sender), wxObject *WXUNUSED(hint) )
285 {
286     const ImageFile& rIF = dynamic_cast<ImageFileDocument*>(GetDocument())->getImageFile();
287     ImageFileArrayConst v = rIF.getArray();
288     int nx = rIF.nx();
289     int ny = rIF.ny();
290     if (v != NULL && nx != 0 && ny != 0) {
291         if (! m_bMinSpecified || ! m_bMaxSpecified) {
292             double min, max;
293             rIF.getMinMax (min, max);
294             if (! m_bMinSpecified)
295                 m_dMinPixel = min;
296             if (! m_bMaxSpecified)
297                 m_dMaxPixel = max;
298         }
299         double scaleWidth = m_dMaxPixel - m_dMinPixel;
300     
301         unsigned char* imageData = new unsigned char [nx * ny * 3];
302         for (int ix = 0; ix < nx; ix++) {
303             for (int iy = 0; iy < ny; iy++) {
304                 double scaleValue = ((v[ix][iy] - m_dMinPixel) / scaleWidth) * 255;
305                 int intensity = static_cast<int>(scaleValue + 0.5);
306                 intensity = clamp (intensity, 0, 255);
307                 int baseAddr = ((ny - 1 - iy) * nx + ix) * 3;
308                 imageData[baseAddr] = imageData[baseAddr+1] = imageData[baseAddr+2] = intensity;
309             }
310         }
311         wxImage image (nx, ny, imageData, true);
312         m_bitmap = image.ConvertToBitmap();
313         delete imageData;
314         int xSize = nx;
315         int ySize = ny;
316         xSize = clamp (xSize, 0, 800);
317         ySize = clamp (ySize, 0, 800);
318         m_frame->SetClientSize (xSize, ySize);
319         m_canvas->SetScrollbars(20, 20, nx/20, ny/20);
320         m_canvas->SetBackgroundColour(*wxWHITE);
321     } 
322
323     if (m_canvas)
324         m_canvas->Refresh();
325 }
326
327 bool 
328 ImageFileView::OnClose (bool deleteWindow)
329 {
330     if (!GetDocument()->Close())
331         return false;
332
333     m_canvas->Clear();
334     m_canvas->m_pView = NULL;
335     m_canvas = NULL;
336     wxString s(theApp->GetAppName());
337     if (m_frame)
338       m_frame->SetTitle(s);
339     SetFrame(NULL);
340
341     Activate(false);
342     
343     if (deleteWindow) {
344         delete m_frame;
345         return true;
346     }
347     return true;
348 }
349
350
351
352 // PhantomCanvas
353
354 PhantomCanvas::PhantomCanvas (PhantomView* v, wxFrame *frame, const wxPoint& pos, const wxSize& size, const long style)
355   : wxScrolledWindow(frame, -1, pos, size, style)
356 {
357     m_pView = v;
358 }
359
360 void 
361 PhantomCanvas::OnDraw(wxDC& dc)
362 {
363     if (m_pView)
364         m_pView->OnDraw(& dc);
365 }
366
367
368 // PhantomView
369
370 IMPLEMENT_DYNAMIC_CLASS(PhantomView, wxView)
371
372 BEGIN_EVENT_TABLE(PhantomView, wxView)
373     EVT_MENU(PHMMENU_FILE_PROPERTIES, PhantomView::OnProperties)
374     EVT_MENU(PHMMENU_PROCESS_RASTERIZE, PhantomView::OnRasterize)
375     EVT_MENU(PHMMENU_PROCESS_PROJECTIONS, PhantomView::OnProjections)
376 END_EVENT_TABLE()
377
378 PhantomView::PhantomView(void) 
379   : wxView(), m_canvas(NULL), m_frame(NULL)
380 {
381     m_iDefaultNDet = 367;
382     m_iDefaultNView = 320;
383     m_iDefaultNSample = 2;
384     m_dDefaultRotation = 2;
385     m_dDefaultFocalLength = 2;
386     m_dDefaultFieldOfView = 1;
387     m_iDefaultGeometry = Scanner::GEOMETRY_PARALLEL;
388     m_iDefaultTrace = Trace::TRACE_NONE;
389 }
390
391 PhantomView::~PhantomView(void)
392 {
393 }
394
395 void
396 PhantomView::OnProperties (wxCommandEvent& event)
397 {
398   const int idPhantom = GetDocument()->getPhantomID();
399   const string& namePhantom = GetDocument()->getPhantomName();
400   ostringstream os;
401   os << "Phantom " << namePhantom << " (" << idPhantom << ")\n";
402   *theApp->getLog() << os.str().c_str();
403 #if DEBUG
404   const Phantom& rPhantom = GetDocument()->getPhantom();
405   rPhantom.print();
406 #endif
407 }
408
409
410 void
411 PhantomView::OnProjections (wxCommandEvent& event)
412 {
413   DialogGetProjectionParameters dialogProjection (m_frame, m_iDefaultNDet, m_iDefaultNView, m_iDefaultNSample, m_dDefaultRotation, m_dDefaultFocalLength, m_dDefaultFieldOfView, m_iDefaultGeometry, m_iDefaultTrace);
414   int retVal = dialogProjection.ShowModal();
415   if (retVal == wxID_OK) {
416     m_iDefaultNDet = dialogProjection.getNDet();
417     m_iDefaultNView = dialogProjection.getNView();
418     m_iDefaultNSample = dialogProjection.getNSamples();
419     m_iDefaultTrace = dialogProjection.getTrace();
420     m_dDefaultRotation = dialogProjection.getRotAngle();
421     m_dDefaultFocalLength = dialogProjection.getFocalLengthRatio();
422     m_dDefaultFieldOfView = dialogProjection.getFieldOfViewRatio();
423     wxString sGeometry = dialogProjection.getGeometry();
424     m_iDefaultGeometry = Scanner::convertGeometryNameToID (sGeometry.c_str());
425
426     if (m_iDefaultNDet > 0 && m_iDefaultNView > 0 && sGeometry != "") {
427       const Phantom& rPhantom = GetDocument()->getPhantom();
428       ProjectionFileDocument* pProjectionDoc = dynamic_cast<ProjectionFileDocument*>(theApp->getDocManager()->CreateDocument("untitled.pj", wxDOC_SILENT));
429       Projections& rProj = pProjectionDoc->getProjections();
430       Scanner theScanner (rPhantom, sGeometry.c_str(), m_iDefaultNDet, m_iDefaultNView, m_iDefaultNSample, m_dDefaultRotation, m_dDefaultFocalLength, m_dDefaultFieldOfView);
431       if (theScanner.fail()) {
432         *theApp->getLog() << "Failed making scanner: " << theScanner.failMessage().c_str() << "\n";
433         return;
434       }
435       rProj.initFromScanner (theScanner);
436       m_dDefaultRotation /= PI;  // convert back to PI units
437
438       if (m_iDefaultTrace > Trace::TRACE_CONSOLE) {
439         ProjectionsDialog dialogProjections (theScanner, rProj, rPhantom, m_iDefaultTrace, dynamic_cast<wxWindow*>(m_frame));
440         for (int iView = 0; iView < rProj.nView(); iView++) {
441           ::wxYield();
442           ::wxYield();
443           if (dialogProjections.isCancelled() || ! dialogProjections.projectView (iView)) {
444             pProjectionDoc->DeleteAllViews();
445             return;
446           }
447           ::wxYield();
448           ::wxYield();
449           while (dialogProjections.isPaused()) {
450               ::wxYield();
451               ::wxUsleep(50);
452           }
453         }
454       } else {
455         wxProgressDialog dlgProgress (wxString("Projection"), wxString("Projection Progress"), rProj.nView() + 1, m_frame, wxPD_CAN_ABORT);
456         for (int i = 0; i < rProj.nView(); i++) {
457           theScanner.collectProjections (rProj, rPhantom, i, 1, true, m_iDefaultTrace);
458           if (! dlgProgress.Update (i+1)) {
459             pProjectionDoc->DeleteAllViews();
460             return;
461           }
462         }
463       }
464
465       ostringstream os;
466       os << "Projections for " << rPhantom.name() << ": nDet=" << m_iDefaultNDet << ", nView=" << m_iDefaultNView << ", nSamples=" << m_iDefaultNSample << ", RotAngle=" << m_dDefaultRotation << ", FocalLengthRatio=" << m_dDefaultFocalLength << ", FieldOfViewRatio=" << m_dDefaultFieldOfView << ", Geometry=" << sGeometry.c_str() << "\n";
467       rProj.setRemark (os.str());
468       *theApp->getLog() << os.str().c_str();
469
470       ::wxYield();
471       pProjectionDoc->Modify(true);
472       pProjectionDoc->UpdateAllViews(this);
473       ::wxYield();
474       if (wxView* pView = pProjectionDoc->GetFirstView())
475         if (wxFrame* pFrame = pView->GetFrame()) {
476           pFrame->SetFocus();
477           pFrame->Raise();
478         }
479       m_frame->Lower();
480     }
481   }
482 }
483
484
485 void
486 PhantomView::OnRasterize (wxCommandEvent& event)
487 {
488   DialogGetRasterParameters dialogRaster (m_frame, 256, 256, 1);
489   int retVal = dialogRaster.ShowModal();
490   if (retVal == wxID_OK) {
491     int xSize = dialogRaster.getXSize();
492     int ySize = dialogRaster.getYSize();
493     int nSamples = dialogRaster.getNSamples();
494     if (nSamples < 1)
495       nSamples = 1;
496     if (xSize > 0 && ySize > 0) {
497       const Phantom& rPhantom = GetDocument()->getPhantom();
498       ImageFileDocument* pRasterDoc = dynamic_cast<ImageFileDocument*>(theApp->getDocManager()->CreateDocument("untitled.if", wxDOC_SILENT));
499       ImageFile& imageFile = pRasterDoc->getImageFile();
500
501       imageFile.setArraySize (xSize, ySize);
502       wxProgressDialog dlgProgress (wxString("Rasterize"), wxString("Rasterization Progress"), imageFile.nx() + 1, m_frame, wxPD_CAN_ABORT);
503       for (unsigned int i = 0; i < imageFile.nx(); i++) {
504         rPhantom.convertToImagefile (imageFile, nSamples, Trace::TRACE_NONE, i, 1, true);
505         if (! dlgProgress.Update(i+1)) {
506           pRasterDoc->DeleteAllViews();
507           return;
508         }
509       }
510       pRasterDoc->Modify(true);
511       pRasterDoc->UpdateAllViews(this);
512
513       ostringstream os;
514       os << "Rasterize Phantom " << rPhantom.name() << ": XSize=" << xSize << ", YSize=" << ySize << ", nSamples=" << nSamples << "\n";
515       *theApp->getLog() << os.str().c_str();
516     }
517   }
518 }
519
520
521 PhantomCanvas* 
522 PhantomView::CreateCanvas (wxView *view, wxFrame *parent)
523 {
524     PhantomCanvas* pCanvas;
525     int width, height;
526     parent->GetClientSize(&width, &height);
527     
528     pCanvas = new PhantomCanvas (dynamic_cast<PhantomView*>(view), parent, wxPoint(0, 0), wxSize(width, height), 0);
529     
530     pCanvas->SetBackgroundColour(*wxWHITE);
531     pCanvas->Clear();
532     
533     return pCanvas;
534 }
535
536 wxFrame*
537 PhantomView::CreateChildFrame(wxDocument *doc, wxView *view)
538 {
539     wxDocChildFrame *subframe = new wxDocChildFrame(doc, view, theApp->getMainFrame(), -1, "Phantom Frame", wxPoint(10, 10), wxSize(256, 256), wxDEFAULT_FRAME_STYLE);
540     
541     wxMenu *file_menu = new wxMenu;
542     
543     file_menu->Append(MAINMENU_FILE_CREATE_PHANTOM, "Cr&eate Phantom...");
544     file_menu->Append(wxID_OPEN, "&Open...");
545     file_menu->Append(wxID_CLOSE, "&Close");
546     
547     file_menu->AppendSeparator();
548     file_menu->Append(PHMMENU_FILE_PROPERTIES, "P&roperties");
549
550     file_menu->AppendSeparator();
551     file_menu->Append(wxID_PRINT, "&Print...");
552     file_menu->Append(wxID_PRINT_SETUP, "Print &Setup...");
553     file_menu->Append(wxID_PREVIEW, "Print Pre&view");
554     
555     wxMenu *process_menu = new wxMenu;
556     process_menu->Append(PHMMENU_PROCESS_RASTERIZE, "&Rasterize...");
557     process_menu->Append(PHMMENU_PROCESS_PROJECTIONS, "&Projections...");
558
559     wxMenu *help_menu = new wxMenu;
560     help_menu->Append(MAINMENU_HELP_CONTENTS, "&Contents");
561     help_menu->Append(MAINMENU_HELP_ABOUT, "&About");
562     
563     wxMenuBar *menu_bar = new wxMenuBar;
564     
565     menu_bar->Append(file_menu, "&File");
566     menu_bar->Append(process_menu, "&Process");
567     menu_bar->Append(help_menu, "&Help");
568     
569     subframe->SetMenuBar(menu_bar);
570     
571     subframe->Centre(wxBOTH);
572     
573     return subframe;
574 }
575
576
577 bool 
578 PhantomView::OnCreate(wxDocument *doc, long WXUNUSED(flags) )
579 {
580     m_frame = CreateChildFrame(doc, this);
581     SetFrame(m_frame);
582
583     int width, height;
584     m_frame->GetClientSize(&width, &height);
585     m_frame->SetTitle("PhantomView");
586     m_canvas = CreateCanvas(this, m_frame);
587
588 #ifdef __X__
589     int x, y;  // X requires a forced resize
590     m_frame->GetSize(&x, &y);
591     m_frame->SetSize(-1, -1, x, y);
592 #endif
593
594     m_frame->Show(true);
595     Activate(true);
596
597     return true;
598 }
599
600
601 void 
602 PhantomView::OnUpdate(wxView *WXUNUSED(sender), wxObject *WXUNUSED(hint) )
603 {
604     if (m_canvas)
605         m_canvas->Refresh();
606 }
607
608 bool 
609 PhantomView::OnClose (bool deleteWindow)
610 {
611     if (!GetDocument()->Close())
612         return false;
613
614     m_canvas->Clear();
615     m_canvas->m_pView = NULL;
616     m_canvas = NULL;
617     wxString s(wxTheApp->GetAppName());
618     if (m_frame)
619       m_frame->SetTitle(s);
620     SetFrame(NULL);
621
622     Activate(false);
623     
624     if (deleteWindow) {
625         delete m_frame;
626         return true;
627     }
628     return true;
629 }
630
631 void
632 PhantomView::OnDraw (wxDC* dc)
633 {
634   int xsize, ysize;
635   m_canvas->GetClientSize (&xsize, &ysize);
636   SGPDriver driver (dc, xsize, ysize);
637   SGP sgp (driver);
638   const Phantom& rPhantom = GetDocument()->getPhantom();
639   sgp.setColor (C_RED);
640   rPhantom.show (sgp);
641 }
642
643 // ProjectionCanvas
644
645 ProjectionFileCanvas::ProjectionFileCanvas (ProjectionFileView* v, wxFrame *frame, const wxPoint& pos, const wxSize& size, const long style)
646   : wxScrolledWindow(frame, -1, pos, size, style)
647 {
648     m_pView = v;
649 }
650
651 void 
652 ProjectionFileCanvas::OnDraw(wxDC& dc)
653 {
654     if (m_pView)
655         m_pView->OnDraw(& dc);
656 }
657
658 // ProjectionFileView
659
660 IMPLEMENT_DYNAMIC_CLASS(ProjectionFileView, wxView)
661
662 BEGIN_EVENT_TABLE(ProjectionFileView, wxView)
663     EVT_MENU(PJMENU_FILE_PROPERTIES, ProjectionFileView::OnProperties)
664     EVT_MENU(PJMENU_PROCESS_RECONSTRUCT, ProjectionFileView::OnReconstruct)
665 END_EVENT_TABLE()
666
667 ProjectionFileView::ProjectionFileView(void) 
668   : wxView(), m_canvas(NULL), m_frame(NULL)
669 {
670     m_iDefaultNX = 256;
671     m_iDefaultNY = 256;
672   m_iDefaultFilter = SignalFilter::FILTER_ABS_BANDLIMIT;
673   m_dDefaultFilterParam = 1.;
674 #if HAVE_FFTW
675   m_iDefaultFilterMethod = ProcessSignal::FILTER_METHOD_RFFTW;
676   m_iDefaultFilterGeneration = ProcessSignal::FILTER_GENERATION_INVERSE_FOURIER;
677 #else
678   m_iDefaultFilterMethod = ProcessSignal::FILTER_METHOD_CONVOLUTION;
679   m_iDefaultFilterGeneration = ProcessSignal::FILTER_GENERATION_DIRECT;
680 #endif
681   m_iDefaultZeropad = 1;
682   m_iDefaultBackprojector = Backprojector::BPROJ_IDIFF3;
683   m_iDefaultInterpolation = Backprojector::INTERP_LINEAR;
684   m_iDefaultInterpParam = 1;
685   m_iDefaultTrace = Trace::TRACE_NONE;
686 }
687
688 ProjectionFileView::~ProjectionFileView(void)
689 {
690 }
691
692 void
693 ProjectionFileView::OnProperties (wxCommandEvent& event)
694 {
695   const Projections& rProj = GetDocument()->getProjections();
696   ostringstream os;
697   rProj.printScanInfo(os);
698   *theApp->getLog() << os.str().c_str();
699   wxMessageDialog dialogMsg (m_frame, os.str().c_str(), "Projection File Properties", wxOK | wxICON_INFORMATION);
700   dialogMsg.ShowModal();
701 }
702
703
704 void
705 ProjectionFileView::OnReconstruct (wxCommandEvent& event)
706 {
707   DialogGetReconstructionParameters dialogReconstruction (m_frame, m_iDefaultNX, m_iDefaultNY, m_iDefaultFilter, m_dDefaultFilterParam, m_iDefaultFilterMethod, m_iDefaultFilterGeneration, m_iDefaultZeropad, m_iDefaultInterpolation, m_iDefaultInterpParam, m_iDefaultBackprojector, m_iDefaultTrace);
708
709   int retVal = dialogReconstruction.ShowModal();
710   if (retVal == wxID_OK) {
711     m_iDefaultNX = dialogReconstruction.getXSize();
712     m_iDefaultNY = dialogReconstruction.getYSize();
713     wxString optFilterName = dialogReconstruction.getFilterName();
714     m_iDefaultFilter = SignalFilter::convertFilterNameToID (optFilterName.c_str());
715     m_dDefaultFilterParam = dialogReconstruction.getFilterParam();
716     wxString optFilterMethodName = dialogReconstruction.getFilterMethodName();
717     m_iDefaultFilterMethod = ProcessSignal::convertFilterMethodNameToID(optFilterMethodName.c_str());
718     m_iDefaultZeropad = dialogReconstruction.getZeropad();
719     wxString optFilterGenerationName = dialogReconstruction.getFilterGenerationName();
720     m_iDefaultFilterGeneration = ProcessSignal::convertFilterGenerationNameToID (optFilterGenerationName.c_str());
721     wxString optInterpName = dialogReconstruction.getInterpName();
722     m_iDefaultInterpolation = Backprojector::convertInterpNameToID (optInterpName.c_str());
723     m_iDefaultInterpParam = dialogReconstruction.getInterpParam();
724     wxString optBackprojectName = dialogReconstruction.getBackprojectName();
725     m_iDefaultBackprojector = Backprojector::convertBackprojectNameToID (optBackprojectName.c_str());
726     m_iDefaultTrace = dialogReconstruction.getTrace();
727     if (m_iDefaultNX > 0 && m_iDefaultNY > 0) {
728       ImageFileDocument* pReconDoc = dynamic_cast<ImageFileDocument*>(theApp->getDocManager()->CreateDocument("untitled.if", wxDOC_SILENT));
729       ImageFile& imageFile = pReconDoc->getImageFile();
730       const Projections& rProj = GetDocument()->getProjections();
731       imageFile.setArraySize (m_iDefaultNX, m_iDefaultNY);
732       Timer timerRecon;
733
734       Reconstructor* pReconstruct = new Reconstructor (rProj, imageFile, optFilterName.c_str(), m_dDefaultFilterParam, optFilterMethodName.c_str(), m_iDefaultZeropad, optFilterGenerationName.c_str(), optInterpName.c_str(), m_iDefaultInterpParam, optBackprojectName.c_str(), m_iDefaultTrace);
735       if (m_iDefaultTrace > Trace::TRACE_CONSOLE) {
736         ReconstructDialog* pDlgReconstruct = new ReconstructDialog (*pReconstruct, rProj, imageFile, m_iDefaultTrace, m_frame);
737         for (int iView = 0; iView < rProj.nView(); iView++) {
738           ::wxYield();
739           ::wxYield();
740           if (pDlgReconstruct->isCancelled() || ! pDlgReconstruct->reconstructView (iView)) {
741             delete pDlgReconstruct;
742             delete pReconstruct;
743             pReconDoc->DeleteAllViews();
744             return;
745           }
746           ::wxYield();
747           ::wxYield();
748           while (pDlgReconstruct->isPaused()) {
749               ::wxYield();
750               ::wxUsleep(50);
751           }
752         }
753         delete pDlgReconstruct;
754       } else {
755         wxProgressDialog dlgProgress (wxString("Reconstruction"), wxString("Reconstruction Progress"), rProj.nView() + 1, m_frame, wxPD_CAN_ABORT);
756         for (int i = 0; i < rProj.nView(); i++) {
757           pReconstruct->reconstructView (i, 1);
758           if (! dlgProgress.Update(i + 1)) {
759             delete pReconstruct;
760             pReconDoc->DeleteAllViews();
761             return;
762           }
763         }
764       }
765       delete pReconstruct;
766       pReconDoc->Modify(true);
767       pReconDoc->UpdateAllViews(this);
768       ostringstream os;
769       os << "Reconstruct " << rProj.getFilename() << ": xSize=" << m_iDefaultNX << ", ySize=" << m_iDefaultNY << ", Filter=" << optFilterName.c_str() << ", FilterParam=" << m_dDefaultFilterParam << ", FilterMethod=" << optFilterMethodName.c_str() << ", FilterGeneration=" << optFilterGenerationName.c_str() << ", Zeropad=" << m_iDefaultZeropad << ", Interpolation=" << optInterpName.c_str() << ", InterpolationParam=" << m_iDefaultInterpParam << ", Backprojection=" << optBackprojectName.c_str() << "\n";
770       *theApp->getLog() << os.str().c_str();
771       imageFile.labelAdd (rProj.getLabel());
772       imageFile.labelAdd (Array2dFileLabel::L_HISTORY, os.str().c_str(), timerRecon.timerEnd());
773     }
774   }
775 }
776
777
778 ProjectionFileCanvas* 
779 ProjectionFileView::CreateCanvas (wxView *view, wxFrame *parent)
780 {
781     ProjectionFileCanvas* pCanvas;
782     int width, height;
783     parent->GetClientSize(&width, &height);
784     
785     pCanvas = new ProjectionFileCanvas (dynamic_cast<ProjectionFileView*>(view), parent, wxPoint(0, 0), wxSize(width, height), 0);
786     
787     pCanvas->SetScrollbars(20, 20, 50, 50);
788     pCanvas->SetBackgroundColour(*wxWHITE);
789     pCanvas->Clear();
790     
791     return pCanvas;
792 }
793
794 wxFrame*
795 ProjectionFileView::CreateChildFrame(wxDocument *doc, wxView *view)
796 {
797     wxDocChildFrame *subframe = new wxDocChildFrame(doc, view, theApp->getMainFrame(), -1, "Projection Frame", wxPoint(10, 10), wxSize(0, 0), wxDEFAULT_FRAME_STYLE);
798     
799     wxMenu *file_menu = new wxMenu;
800     
801     file_menu->Append(MAINMENU_FILE_CREATE_PHANTOM, "Cr&eate Phantom...");
802     file_menu->Append(wxID_OPEN, "&Open...");
803     file_menu->Append(wxID_SAVE, "&Save");
804     file_menu->Append(wxID_SAVEAS, "Save &As...");
805     file_menu->Append(wxID_CLOSE, "&Close");
806     
807     file_menu->AppendSeparator();
808     file_menu->Append(PJMENU_FILE_PROPERTIES, "P&roperties");
809
810     file_menu->AppendSeparator();
811     file_menu->Append(wxID_PRINT, "&Print...");
812     file_menu->Append(wxID_PRINT_SETUP, "Print &Setup...");
813     file_menu->Append(wxID_PREVIEW, "Print Pre&view");
814     
815     wxMenu *process_menu = new wxMenu;
816     process_menu->Append(PJMENU_PROCESS_RECONSTRUCT, "R&econstruct...");
817
818     wxMenu *help_menu = new wxMenu;
819     help_menu->Append(MAINMENU_HELP_CONTENTS, "&Contents");
820     help_menu->AppendSeparator();
821     help_menu->Append(MAINMENU_HELP_ABOUT, "&About");
822     
823     wxMenuBar *menu_bar = new wxMenuBar;
824     
825     menu_bar->Append(file_menu, "&File");
826     menu_bar->Append(process_menu, "&Process");
827     menu_bar->Append(help_menu, "&Help");
828     
829     subframe->SetMenuBar(menu_bar);
830     
831     subframe->Centre(wxBOTH);
832     
833     return subframe;
834 }
835
836
837 bool 
838 ProjectionFileView::OnCreate(wxDocument *doc, long WXUNUSED(flags) )
839 {
840     m_frame = CreateChildFrame(doc, this);
841     SetFrame(m_frame);
842     
843     int width, height;
844     m_frame->GetClientSize(&width, &height);
845     m_frame->SetTitle("ProjectionFileView");
846     m_canvas = CreateCanvas(this, m_frame);
847
848 #ifdef __X__
849     int x, y;  // X requires a forced resize
850     m_frame->GetSize(&x, &y);
851     m_frame->SetSize(-1, -1, x, y);
852 #endif
853
854     m_frame->Show(true);
855     Activate(true);
856     
857     return true;
858 }
859
860 void 
861 ProjectionFileView::OnDraw (wxDC* dc)
862 {
863     if (m_bitmap.Ok())
864         dc->DrawBitmap (m_bitmap, 0, 0, false);
865 }
866
867
868 void 
869 ProjectionFileView::OnUpdate(wxView *WXUNUSED(sender), wxObject *WXUNUSED(hint) )
870 {
871     const Projections& rProj = GetDocument()->getProjections();
872     const int nDet = rProj.nDet();
873     const int nView = rProj.nView();
874     if (nDet != 0 && nView != 0) {
875         const DetectorArray& detarray = rProj.getDetectorArray(0);
876         const DetectorValue* detval = detarray.detValues();
877         double min = detval[0];
878         double max = detval[0];
879         for (int iy = 0; iy < nView; iy++) {
880             const DetectorArray& detarray = rProj.getDetectorArray(iy);
881             const DetectorValue* detval = detarray.detValues();
882             for (int ix = 0; ix < nDet; ix++) {
883                 if (min > detval[ix])
884                     min = detval[ix];
885                 else if (max < detval[ix])
886                     max = detval[ix];
887             }
888         }
889
890         unsigned char* imageData = new unsigned char [nDet * nView * 3];
891         double scale = (max - min) / 255;
892         for (int iy = 0; iy < nView; iy++) {
893             const DetectorArray& detarray = rProj.getDetectorArray(iy);
894             const DetectorValue* detval = detarray.detValues();
895             for (int ix = 0; ix < nDet; ix++) {
896                 int intensity = static_cast<int>(((detval[ix] - min) / scale) + 0.5);
897                 intensity = clamp(intensity, 0, 255);
898                 int baseAddr = (iy * nDet + ix) * 3;
899                 imageData[baseAddr] = imageData[baseAddr+1] = imageData[baseAddr+2] = intensity;
900             }
901         }
902         wxImage image (nDet, nView, imageData, true);
903         m_bitmap = image.ConvertToBitmap();
904         delete imageData;
905         int xSize = nDet;
906         int ySize = nView;
907         xSize = clamp (xSize, 0, 800);
908         ySize = clamp (ySize, 0, 800);
909         m_frame->SetClientSize (xSize, ySize);
910         m_canvas->SetScrollbars (20, 20, nDet/20, nView/20);
911     }
912
913     if (m_canvas)
914         m_canvas->Refresh();
915 }
916
917 bool 
918 ProjectionFileView::OnClose (bool deleteWindow)
919 {
920     if (!GetDocument()->Close())
921         return false;
922
923     m_canvas->Clear();
924     m_canvas->m_pView = NULL;
925     m_canvas = NULL;
926     wxString s(wxTheApp->GetAppName());
927     if (m_frame)
928       m_frame->SetTitle(s);
929     SetFrame(NULL);
930
931     Activate(false);
932     
933     if (deleteWindow) {
934         delete m_frame;
935         return true;
936     }
937     return true;
938 }
939