r273: *** empty log message ***
[ctsim.git] / src / dlgreconstruct.cpp
index 277b6ea8bd41c11e17579e80a773916f1581d6f4..7ec1d326d03135e7d0a372bc1e3b680d6a3f3804 100644 (file)
@@ -2,14 +2,14 @@
 ** FILE IDENTIFICATION
 **
 **   Name:          dlgreconstruct.cpp
-**   Purpose:       Projection Collection Animation Dialog
+**   Purpose:       Reconstruction Animation Dialog
 **   Programmer:    Kevin Rosenberg
 **   Date Started:  August 2000
 **
 **  This is part of the CTSim program
 **  Copyright (C) 1983-2000 Kevin Rosenberg
 **
-**  $Id: dlgreconstruct.cpp,v 1.2 2000/09/02 16:40:36 kevin Exp $
+**  $Id: dlgreconstruct.cpp,v 1.8 2000/12/17 22:30: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
@@ -36,7 +36,6 @@
     #pragma hdrstop
 #endif
 
-#ifndef WX_PRECOMP
     #include "wx/utils.h"
     #include "wx/frame.h"
     #include "wx/button.h"
     #include "wx/settings.h"
     #include "wx/dcclient.h"
     #include "wx/timer.h"
-#endif
+    #include "wx/image.h"
 
 #include "dlgreconstruct.h"
+#include <algorithm>
 #include "ct.h"
 
 
 static const int LAYOUT_X_MARGIN = 4;
 static const int LAYOUT_Y_MARGIN = 4;
+\r
+\r
+const int ReconstructDialog::ID_BTN_PAUSE = 19998;\r
+const int ReconstructDialog::ID_BTN_STEP = 19999;\r
+const int ReconstructDialog::MAX_IMAGE_X = 400;\r
+const int ReconstructDialog::MAX_IMAGE_Y = 400;\r
+\r
 
 BEGIN_EVENT_TABLE(ReconstructDialog, wxDialog)
    EVT_BUTTON(wxID_CANCEL, ReconstructDialog::OnCancel)
@@ -68,7 +75,7 @@ IMPLEMENT_CLASS(ReconstructDialog, wxDialog)
 
 
 ReconstructDialog::ReconstructDialog (Reconstructor& rReconstruct, const Projections& rProj, ImageFile& rIF, const int iTrace, wxWindow *parent)
-    : wxDialog(parent, -1, "Collect Projections"), 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)
+    : 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)
 {
     m_state = Continue;
     m_iLastView = -1;
@@ -106,33 +113,44 @@ ReconstructDialog::ReconstructDialog (Reconstructor& rReconstruct, const Project
     SetAutoLayout(TRUE);
     Layout();
 
-    wxSize sizeDlg (700,500);
-    if (sizeDlg.x != sizeDlg.y) {
-      sizeDlg.x = max(sizeDlg.x,sizeDlg.y);
-      sizeDlg.y = max(sizeDlg.x,sizeDlg.y);
-    }
+    m_nxGraph = 500;
+    m_nyGraph = 500;
+    wxSize sizeDlg (m_nxGraph, m_nyGraph);
+    m_nxImage = m_rImageFile.nx();
+    if (m_nxImage > MAX_IMAGE_X)
+      m_nxImage = MAX_IMAGE_X;
+    m_nyImage = m_rImageFile.ny();
+    if (m_nyImage > MAX_IMAGE_Y)
+      m_nyImage = MAX_IMAGE_Y;
+
+    sizeDlg.x += m_nxImage;
+    sizeDlg.y = max (sizeDlg.y, m_nyImage);
 
     m_iClientX = sizeDlg.x;
     m_iClientY = sizeDlg.y;
-    SetClientSize(sizeDlg);
-
+    SetClientSize (sizeDlg);
+
+    Centre(wxCENTER_FRAME | wxBOTH);\r
+\r
+    if ( m_parentTop )\r
+      m_parentTop->Enable(FALSE);\r
+\r
+    Show(TRUE);\r
+    Enable(TRUE); // enable this window\r
+\r
     m_bitmap.Create (m_iClientX, m_iClientY); // save a copy of screen
+    int x, y;\r
+    this->GetClientSize(&x, &y);\r
+#if OLD\r
     m_pDC = new wxMemoryDC;
     m_pDC->SelectObject (m_bitmap);       // in memoryDC
-    m_pDC->SetFont (*wxSWISS_FONT);
-    int x, y;
-    this->GetClientSize(&x, &y);
-    m_pSGPDriver = new SGPDriver (dynamic_cast<wxDC*>(m_pDC), x, y);
+    m_pSGPDriver = new SGPDriver (dynamic_cast<wxDC*>(m_pDC), x, y);\r
+#else\r
+    m_pDC = dynamic_cast<wxDC*> (new wxClientDC (this));\r
+    m_pSGPDriver = new SGPDriver (m_pDC, x, y);\r
+#endif\r
     m_pSGP = new SGP (*m_pSGPDriver);
 
-    Centre(wxCENTER_FRAME | wxBOTH);
-
-    if ( m_parentTop )
-      m_parentTop->Enable(FALSE);
-
-    Show(TRUE);
-    Enable(TRUE); // enable this window
-
     wxYield();     // Update the display
 
     m_pSGPDriver->idWX()->SetFont(*wxSWISS_FONT);
@@ -142,37 +160,75 @@ ReconstructDialog::ReconstructDialog (Reconstructor& rReconstruct, const Project
 }
 
 void
-ReconstructDialog::showView (int iViewNumber)
+ReconstructDialog::showView (int iViewNumber, bool bBackprojectView)
 {
   if ( iViewNumber < m_rProjections.nView() ) {
     m_iLastView = iViewNumber;
     m_pSGP->eraseWindow();
-    if (m_iTrace >= Trace::TRACE_PLOT)
-      m_pSGP->setViewport (0, 0, 1, 1);
+
     char szProgress [256];
     snprintf (szProgress, sizeof(szProgress), "Reconstructing View %d (%.1f%%)", iViewNumber, 100 * iViewNumber / static_cast<double>(m_rProjections.nView()));
-    double wText, hText;
+    m_pSGP->setViewport (0, 0, 1, 1);
+    m_pSGP->setWindow (0, 0, 1, 1);
     m_pSGP->setTextColor (C_LTRED, -1);
-    m_pSGP->getTextExtent (szProgress, &wText, &hText);
-    double xw_max, xw_min, yw_max, yw_min;
-    m_pSGP->getWindow (xw_min, yw_min, xw_max, yw_max);
-    m_pSGP->moveAbs ((xw_max - xw_min) / 2 + xw_min - wText / 2, yw_max - hText);
+    double dCharHeight = m_pSGP->getCharHeight();
+    m_pSGP->setTextSize (dCharHeight * 2);
+    m_pSGP->moveAbs(0., m_pSGP->getCharHeight());
     m_pSGP->drawText (szProgress);
-    if (m_iTrace >= Trace::TRACE_PLOT)
-      m_pSGP->setViewport (0, .1, 0.66, 1);
-    m_rReconstructor.reconstructView (iViewNumber, 1, m_pSGP);
+    m_pSGP->setTextSize (dCharHeight);
+
+    m_pSGP->setViewport (0.0, 0.1, 0.66, 1.);
+    m_rReconstructor.reconstructView (iViewNumber, 1, m_pSGP, bBackprojectView);
+
+    ImageFileArrayConst v = m_rImageFile.getArray();
+    int xBase = m_nxGraph;
+    int yBase = 0;
+    if (m_nyGraph > m_nyImage)
+       yBase = (m_nyGraph - m_nyImage) / 2;
+    double minValue = v[0][0];
+    double maxValue = v[0][0];
+    for (int ix = 0; ix < m_nxImage; ix++) {
+       for (int iy = 0; iy < m_nyImage; iy++) {
+           double dPixel = v[ix][iy];
+           if (dPixel < minValue)
+               minValue = dPixel;
+           else if (dPixel > maxValue)
+               maxValue = dPixel;
+       }
+    }
+    unsigned char* imageData = new unsigned char [m_nxImage * m_nyImage * 3];
+    double dScale = 255 / (maxValue - minValue);
+    for (int ix2 = 0; ix2 < m_nxImage; ix2++) {
+       for (int iy = 0; iy < m_nyImage; iy++) {
+           double dPixel = v[ix2][iy];
+           dPixel = (dPixel - minValue) * dScale;
+           int intensity = nearest<int>(dPixel);
+           intensity = clamp (intensity, 0, 255);
+           int baseAddr = ((m_nyImage - 1 - iy) * m_nxImage + ix2) * 3;
+           imageData[baseAddr] = imageData[baseAddr+1] = imageData[baseAddr+2] = intensity;
+       }
+    }
+    wxImage image (m_nxImage, m_nyImage, imageData, true);
+    wxBitmap bitmap = image.ConvertToBitmap();
+    m_pSGP->getDriver().idWX()->DrawBitmap(bitmap, xBase, yBase, false);
+    delete imageData;
+
+    Refresh();
   }
 }
 
 bool
 ReconstructDialog::reconstructView (int iViewNumber)
 {
-  ::wxYield();        // update the display
-  showView (iViewNumber);
-  ::wxYield();        // update the display
+  if (iViewNumber <= m_iLastView)  // have already done this view
+    return true;
+
   if (iViewNumber < m_rProjections.nView()) {
+    ::wxYield();        // update the display
+    showView (iViewNumber, true);
+    ::wxYield();        // update the display
     if (m_iTrace >= Trace::TRACE_PLOT) {
-      sleep(1);
+      ::wxUsleep(250);
     }
   } else {
     m_state = Finished;    // so that we return TRUE below and 
@@ -215,7 +271,15 @@ ReconstructDialog::OnPause (wxCommandEvent& event)
     // shown as a modal dialog - so just let the default handler do the job
     event.Skip();
   } else {
-      if (m_state == Continue) {
+      if (m_state == Continue) {\r
+#ifndef OLD\r
+         m_memoryDC.SelectObject (m_bitmap);       // in memoryDC\r
+         m_pSGP->setDC (&m_memoryDC);\r
+         m_memoryDC.SetFont (*wxSWISS_FONT);\r
+         showView (m_iLastView, false);\r
+         m_pSGP->setDC (m_pDC);\r
+         m_memoryDC.SelectObject(wxNullBitmap);\r
+#endif
        m_state = Paused;
        m_btnPause->SetLabel (wxString("Resume"));
       } else if (m_state == Paused) {
@@ -231,7 +295,15 @@ ReconstructDialog::OnStep (wxCommandEvent& event)
   if ( m_state == Finished ) {
     event.Skip();
   } else {
-    if (m_state == Continue) {
+    if (m_state == Continue) {\r
+#ifndef OLD\r
+         m_memoryDC.SelectObject (m_bitmap);       // in memoryDC\r
+         m_pSGP->setDC (&m_memoryDC);\r
+         m_memoryDC.SetFont (*wxSWISS_FONT);\r
+      m_rReconstructor.reconstructView (m_iLastView, 1, m_pSGP, false);\r
+         m_pSGP->setDC (m_pDC);\r
+         m_memoryDC.SelectObject(wxNullBitmap);\r
+#endif
       m_state = Paused;
       m_btnPause->SetLabel (wxString("Resume"));
     } else if (m_state == Paused) {
@@ -253,10 +325,9 @@ void ReconstructDialog::OnClose(wxCloseEvent& event)
 void
 ReconstructDialog::OnPaint (wxPaintEvent& event)
 {
-  wxPaintDC paintDC (this);
-  if (m_state == Paused) {
+  wxPaintDC paintDC (this);\r
+  if (m_state == Paused)
     paintDC.DrawBitmap(m_bitmap, 0, 0, false);
-  }
 }