src/dlgreconstruct.cpp

   1 /*****************************************************************************
   2 ** FILE IDENTIFICATION
   3 **
   4 **   Name:          dlgreconstruct.cpp
   5 **   Purpose:       Reconstruction Animation Dialog
   6 **   Programmer:    Kevin Rosenberg
   7 **   Date Started:  August 2000
   8 **
   9 **  This is part of the CTSim program
  10 **  Copyright (C) 1983-2000 Kevin Rosenberg
  11 **
  12 **  $Id: dlgreconstruct.cpp,v 1.8 2000/12/17 22:30:34 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 "dlgreconstruct.h"
  30 #endif
  31
  32 // For compilers that support precompilation, includes "wx.h".
  33 #include "wx/wxprec.h"
  34
  35 #ifdef __BORLANDC__
  36     #pragma hdrstop
  37 #endif
  38
  39     #include "wx/utils.h"
  40     #include "wx/frame.h"
  41     #include "wx/button.h"
  42     #include "wx/stattext.h"
  43     #include "wx/layout.h"
  44     #include "wx/event.h"
  45     #include "wx/intl.h"
  46     #include "wx/settings.h"
  47     #include "wx/dcclient.h"
  48     #include "wx/timer.h"
  49     #include "wx/image.h"
  50
  51 #include "dlgreconstruct.h"
  52 #include <algorithm>
  53 #include "ct.h"
  54
  55
  56 static const int LAYOUT_X_MARGIN = 4;
  57 static const int LAYOUT_Y_MARGIN = 4;
  58 \r
  59 \r
  60 const int ReconstructDialog::ID_BTN_PAUSE = 19998;\r
  61 const int ReconstructDialog::ID_BTN_STEP = 19999;\r
  62 const int ReconstructDialog::MAX_IMAGE_X = 400;\r
  63 const int ReconstructDialog::MAX_IMAGE_Y = 400;\r
  64 \r
  65
  66 BEGIN_EVENT_TABLE(ReconstructDialog, wxDialog)
  67    EVT_BUTTON(wxID_CANCEL, ReconstructDialog::OnCancel)
  68    EVT_BUTTON(ID_BTN_PAUSE, ReconstructDialog::OnPause)
  69    EVT_BUTTON(ID_BTN_STEP, ReconstructDialog::OnStep)
  70    EVT_CLOSE(ReconstructDialog::OnClose)
  71    EVT_PAINT(ReconstructDialog::OnPaint)
  72 END_EVENT_TABLE()
  73
  74 IMPLEMENT_CLASS(ReconstructDialog, wxDialog)
  75
  76
  77 ReconstructDialog::ReconstructDialog (Reconstructor& rReconstruct, const Projections& rProj, ImageFile& rIF, const int iTrace, wxWindow *parent)
  78     : wxDialog(parent, -1, "Reconstruction", wxDefaultPosition), m_rReconstructor(rReconstruct), m_rProjections(rProj), m_rImageFile(rIF), m_pSGPDriver(NULL), m_pSGP(NULL), m_iTrace(iTrace), m_pDC(NULL), m_btnAbort(0), m_btnPause(0), m_btnStep(0)
  79 {
  80     m_state = Continue;
  81     m_iLastView = -1;
  82     m_parentTop = parent;
  83     while ( m_parentTop && m_parentTop->GetParent() )
  84         m_parentTop = m_parentTop->GetParent();
  85
  86     m_btnAbort = new wxButton(this, wxID_CANCEL, _("Cancel"));
  87     wxLayoutConstraints* c = new wxLayoutConstraints;
  88     c->right.SameAs(this, wxRight, 2*LAYOUT_X_MARGIN);
  89     c->bottom.SameAs(this, wxBottom, 2*LAYOUT_Y_MARGIN);
  90
  91     wxSize sizeBtn = wxButton::GetDefaultSize();
  92     c->width.Absolute(sizeBtn.x);
  93     c->height.Absolute(sizeBtn.y);
  94
  95     m_btnAbort->SetConstraints(c);
  96
  97     m_btnPause = new wxButton (this, ID_BTN_PAUSE, wxString("Pause"));
  98     wxLayoutConstraints* cPause = new wxLayoutConstraints;
  99     cPause->right.SameAs(this, wxRight, 3*LAYOUT_X_MARGIN + sizeBtn.x);
 100     cPause->bottom.SameAs(this, wxBottom, 2*LAYOUT_Y_MARGIN);
 101     cPause->width.Absolute(sizeBtn.x);
 102     cPause->height.Absolute(sizeBtn.y);
 103     m_btnPause->SetConstraints(cPause);
 104
 105     m_btnStep = new wxButton (this, ID_BTN_STEP, wxString("Step"));
 106     wxLayoutConstraints* cStep = new wxLayoutConstraints;
 107     cStep->right.SameAs(this, wxRight, 5*LAYOUT_X_MARGIN + sizeBtn.x * 2);
 108     cStep->bottom.SameAs(this, wxBottom, 2*LAYOUT_Y_MARGIN);
 109     cStep->width.Absolute(sizeBtn.x);
 110     cStep->height.Absolute(sizeBtn.y);
 111     m_btnStep->SetConstraints(cStep);
 112
 113     SetAutoLayout(TRUE);
 114     Layout();
 115
 116     m_nxGraph = 500;
 117     m_nyGraph = 500;
 118     wxSize sizeDlg (m_nxGraph, m_nyGraph);
 119     m_nxImage = m_rImageFile.nx();
 120     if (m_nxImage > MAX_IMAGE_X)
 121       m_nxImage = MAX_IMAGE_X;
 122     m_nyImage = m_rImageFile.ny();
 123     if (m_nyImage > MAX_IMAGE_Y)
 124       m_nyImage = MAX_IMAGE_Y;
 125
 126     sizeDlg.x += m_nxImage;
 127     sizeDlg.y = max (sizeDlg.y, m_nyImage);
 128
 129     m_iClientX = sizeDlg.x;
 130     m_iClientY = sizeDlg.y;
 131     SetClientSize (sizeDlg);
 132
 133     Centre(wxCENTER_FRAME | wxBOTH);\r
 134 \r
 135     if ( m_parentTop )\r
 136       m_parentTop->Enable(FALSE);\r
 137 \r
 138     Show(TRUE);\r
 139     Enable(TRUE); // enable this window\r
 140 \r
 141     m_bitmap.Create (m_iClientX, m_iClientY); // save a copy of screen
 142     int x, y;\r
 143     this->GetClientSize(&x, &y);\r
 144 #if OLD\r
 145     m_pDC = new wxMemoryDC;
 146     m_pDC->SelectObject (m_bitmap);       // in memoryDC
 147     m_pSGPDriver = new SGPDriver (dynamic_cast<wxDC*>(m_pDC), x, y);\r
 148 #else\r
 149     m_pDC = dynamic_cast<wxDC*> (new wxClientDC (this));\r
 150     m_pSGPDriver = new SGPDriver (m_pDC, x, y);\r
 151 #endif\r
 152     m_pSGP = new SGP (*m_pSGPDriver);
 153
 154     wxYield();     // Update the display
 155
 156     m_pSGPDriver->idWX()->SetFont(*wxSWISS_FONT);
 157 #ifdef __WXMAC__
 158     MacUpdateImmediately();
 159 #endif
 160 }
 161
 162 void
 163 ReconstructDialog::showView (int iViewNumber, bool bBackprojectView)
 164 {
 165   if ( iViewNumber < m_rProjections.nView() ) {
 166     m_iLastView = iViewNumber;
 167     m_pSGP->eraseWindow();
 168
 169     char szProgress [256];
 170     snprintf (szProgress, sizeof(szProgress), "Reconstructing View %d (%.1f%%)", iViewNumber, 100 * iViewNumber / static_cast<double>(m_rProjections.nView()));
 171     m_pSGP->setViewport (0, 0, 1, 1);
 172     m_pSGP->setWindow (0, 0, 1, 1);
 173     m_pSGP->setTextColor (C_LTRED, -1);
 174     double dCharHeight = m_pSGP->getCharHeight();
 175     m_pSGP->setTextSize (dCharHeight * 2);
 176     m_pSGP->moveAbs(0., m_pSGP->getCharHeight());
 177     m_pSGP->drawText (szProgress);
 178     m_pSGP->setTextSize (dCharHeight);
 179
 180     m_pSGP->setViewport (0.0, 0.1, 0.66, 1.);
 181     m_rReconstructor.reconstructView (iViewNumber, 1, m_pSGP, bBackprojectView);
 182
 183     ImageFileArrayConst v = m_rImageFile.getArray();
 184     int xBase = m_nxGraph;
 185     int yBase = 0;
 186     if (m_nyGraph > m_nyImage)
 187         yBase = (m_nyGraph - m_nyImage) / 2;
 188     double minValue = v[0][0];
 189     double maxValue = v[0][0];
 190     for (int ix = 0; ix < m_nxImage; ix++) {
 191         for (int iy = 0; iy < m_nyImage; iy++) {
 192             double dPixel = v[ix][iy];
 193             if (dPixel < minValue)
 194                 minValue = dPixel;
 195             else if (dPixel > maxValue)
 196                 maxValue = dPixel;
 197         }
 198     }
 199     unsigned char* imageData = new unsigned char [m_nxImage * m_nyImage * 3];
 200     double dScale = 255 / (maxValue - minValue);
 201     for (int ix2 = 0; ix2 < m_nxImage; ix2++) {
 202         for (int iy = 0; iy < m_nyImage; iy++) {
 203             double dPixel = v[ix2][iy];
 204             dPixel = (dPixel - minValue) * dScale;
 205             int intensity = nearest<int>(dPixel);
 206             intensity = clamp (intensity, 0, 255);
 207             int baseAddr = ((m_nyImage - 1 - iy) * m_nxImage + ix2) * 3;
 208             imageData[baseAddr] = imageData[baseAddr+1] = imageData[baseAddr+2] = intensity;
 209         }
 210     }
 211     wxImage image (m_nxImage, m_nyImage, imageData, true);
 212     wxBitmap bitmap = image.ConvertToBitmap();
 213     m_pSGP->getDriver().idWX()->DrawBitmap(bitmap, xBase, yBase, false);
 214     delete imageData;
 215
 216     Refresh();
 217   }
 218 }
 219
 220 bool
 221 ReconstructDialog::reconstructView (int iViewNumber)
 222 {
 223   if (iViewNumber <= m_iLastView)  // have already done this view
 224     return true;
 225
 226   if (iViewNumber < m_rProjections.nView()) {
 227     ::wxYield();        // update the display
 228     showView (iViewNumber, true);
 229     ::wxYield();        // update the display
 230     if (m_iTrace >= Trace::TRACE_PLOT) {
 231       ::wxUsleep(250);
 232     }
 233   } else {
 234     m_state = Finished;    // so that we return TRUE below and
 235   }                        // that [Cancel] handler knew what to do
 236
 237 #ifdef __WXMAC__
 238   MacUpdateImmediately();
 239 #endif
 240
 241   ::wxYield();        // update the display
 242   return m_state != Cancelled;
 243 }
 244
 245
 246 // EVENT HANDLERS
 247
 248 void ReconstructDialog::OnCancel (wxCommandEvent& event)
 249 {
 250   if ( m_state == Finished ) {
 251     // this means that the count down is already finished and we're being
 252     // shown as a modal dialog - so just let the default handler do the job
 253     event.Skip();
 254   } else {
 255     // request to cancel was received, the next time Update() is called we
 256     // will handle it
 257     m_state = Cancelled;
 258
 259     // update the button state immediately so that the user knows that the
 260     // request has been noticed
 261     m_btnAbort->Disable();
 262   }
 263 }
 264
 265
 266 void
 267 ReconstructDialog::OnPause (wxCommandEvent& event)
 268 {
 269   if ( m_state == Finished ) {
 270     // this means that the count down is already finished and we're being
 271     // shown as a modal dialog - so just let the default handler do the job
 272     event.Skip();
 273   } else {
 274       if (m_state == Continue) {\r
 275 #ifndef OLD\r
 276           m_memoryDC.SelectObject (m_bitmap);       // in memoryDC\r
 277           m_pSGP->setDC (&m_memoryDC);\r
 278           m_memoryDC.SetFont (*wxSWISS_FONT);\r
 279           showView (m_iLastView, false);\r
 280           m_pSGP->setDC (m_pDC);\r
 281           m_memoryDC.SelectObject(wxNullBitmap);\r
 282 #endif
 283         m_state = Paused;
 284         m_btnPause->SetLabel (wxString("Resume"));
 285       } else if (m_state == Paused) {
 286         m_state = Continue;
 287         m_btnPause->SetLabel (wxString("Pause"));
 288       }
 289   }
 290 }
 291
 292 void
 293 ReconstructDialog::OnStep (wxCommandEvent& event)
 294 {
 295   if ( m_state == Finished ) {
 296     event.Skip();
 297   } else {
 298     if (m_state == Continue) {\r
 299 #ifndef OLD\r
 300           m_memoryDC.SelectObject (m_bitmap);       // in memoryDC\r
 301           m_pSGP->setDC (&m_memoryDC);\r
 302           m_memoryDC.SetFont (*wxSWISS_FONT);\r
 303       m_rReconstructor.reconstructView (m_iLastView, 1, m_pSGP, false);\r
 304           m_pSGP->setDC (m_pDC);\r
 305           m_memoryDC.SelectObject(wxNullBitmap);\r
 306 #endif
 307       m_state = Paused;
 308       m_btnPause->SetLabel (wxString("Resume"));
 309     } else if (m_state == Paused) {
 310       reconstructView (m_iLastView + 1);
 311     }
 312   }
 313 }
 314
 315 void ReconstructDialog::OnClose(wxCloseEvent& event)
 316 {
 317     if ( m_state == Uncancellable )
 318       event.Veto(TRUE);    // can't close this dialog
 319     else if ( m_state == Finished )
 320       event.Skip(); // let the default handler close the window as we already terminated
 321     else
 322       m_state = Cancelled;          // next Update() will notice it
 323 }
 324
 325 void
 326 ReconstructDialog::OnPaint (wxPaintEvent& event)
 327 {
 328   wxPaintDC paintDC (this);\r
 329   if (m_state == Paused)
 330     paintDC.DrawBitmap(m_bitmap, 0, 0, false);
 331 }
 332
 333
 334 /////////////////////////////////////////////////////
 335 // destruction
 336
 337 ReconstructDialog::~ReconstructDialog()
 338 {
 339   if ( m_parentTop )
 340     m_parentTop->Enable(TRUE);
 341
 342   delete m_pSGP;
 343   delete m_pSGPDriver;
 344   delete m_pDC;
 345 }
 346