src/threadrecon.cpp

   1 /*****************************************************************************
   2 ** FILE IDENTIFICATION
   3 **
   4 **   Name:          threadrecon.cpp
   5 **   Purpose:       Threaded reconstruction class
   6 **   Programmer:    Kevin Rosenberg
   7 **   Date Started:  February 2001
   8 **
   9 **  This is part of the CTSim program
  10 **  Copyright (C) 1983-2001 Kevin Rosenberg
  11 **
  12 **  $Id$
  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
  29 #include "wx/wxprec.h"
  30
  31 #ifndef WX_PRECOMP
  32 #include "wx/wx.h"
  33 #endif
  34
  35 #include "ct.h"
  36 #include "ctsim.h"
  37 #include "docs.h"
  38 #include "views.h"
  39 #include "threadrecon.h"
  40 #include "backgroundmgr.h"
  41 #include "backgroundsupr.h"
  42
  43 #ifdef HAVE_WXTHREADS
  44
  45
  46
  47
  48 /////////////////////////////////////////////////////////////////////
  49 //
  50 // Class ReconstructorSupervisorThread -- Thread for Background Supervisor
  51 //
  52 /////////////////////////////////////////////////////////////////////
  53
  54 ReconstructorSupervisorThread::ReconstructorSupervisorThread (ProjectionFileView* pProjView, int iNX, int iNY,
  55    const char* pszFilterName, double dFilterParam, const char* pszFilterMethod, int iZeropad,
  56    const char* pszFilterGenerationName, const char* pszInterpName, int iInterpParam,
  57    const char* pszBackprojectName, const char* const pszLabel, ReconstructionROI* pROI, bool bRebinToParallel)
  58 :   SupervisorThread(), m_pProjView(pProjView), m_iNX(iNX), m_iNY(iNY), m_strFilterName(pszFilterName), m_dFilterParam(dFilterParam),
  59   m_strFilterMethod(pszFilterMethod), m_iZeropad(iZeropad), m_strFilterGenerationName(pszFilterGenerationName),
  60   m_strInterpName(pszInterpName), m_iInterpParam(iInterpParam), m_strBackprojectName(pszBackprojectName),
  61   m_strLabel(pszLabel), m_reconROI(*pROI), m_bRebinToParallel(bRebinToParallel)
  62 {
  63 }
  64
  65 wxThread::ExitCode
  66 ReconstructorSupervisorThread::Entry()
  67 {
  68   Projections* pProj = &m_pProjView->GetDocument()->getProjections();
  69
  70   if (m_bRebinToParallel)
  71     pProj = pProj->interpolateToParallel();
  72
  73   ReconstructorSupervisor reconSupervisor (this, pProj, m_pProjView, m_iNX, m_iNY,
  74    m_strFilterName.c_str(), m_dFilterParam, m_strFilterMethod.c_str(), m_iZeropad, m_strFilterGenerationName.c_str(),
  75    m_strInterpName.c_str(), m_iInterpParam, m_strBackprojectName.c_str(), m_strLabel.c_str(), &m_reconROI);
  76
  77   reconSupervisor.start();
  78   while (! reconSupervisor.workersDone() && ! reconSupervisor.fail() && ! reconSupervisor.cancelled()) {
  79     Sleep(100);
  80   }
  81   if (reconSupervisor.fail())
  82   {
  83     wxString msg ("Error starting reconstructor supervisor: ");
  84     msg += reconSupervisor.getFailMessage().c_str();
  85     msg += "\n";
  86     wxCommandEvent eventLog (wxEVT_COMMAND_MENU_SELECTED, MAINMENU_LOG_EVENT );
  87     eventLog.SetString( msg );
  88     wxPostEvent( theApp->getMainFrame(), eventLog ); // send log event
  89   }
  90   if (! reconSupervisor.cancelled())
  91           reconSupervisor.onDone();
  92   reconSupervisor.deleteWorkers();
  93
  94   if (m_bRebinToParallel)
  95     delete pProj;
  96
  97   return static_cast<wxThread::ExitCode>(0);
  98 }
  99
 100 void
 101 ReconstructorSupervisorThread::OnExit()
 102 {
 103 }
 104
 105
 106 /////////////////////////////////////////////////////////////////////
 107 //
 108 // Class ReconstructorSupervisor -- A Background Supervisor
 109 //
 110 /////////////////////////////////////////////////////////////////////
 111
 112 ReconstructorSupervisor::ReconstructorSupervisor (SupervisorThread* pThread, Projections* pProj,
 113   ProjectionFileView* pProjView, int iImageNX, int iImageNY, const char* pszFilterName, double dFilterParam,
 114   const char* pszFilterMethod, int iZeropad, const char* pszFilterGenerationName,
 115   const char* pszInterpName, int iInterpParam, const char* pszBackprojectName, const char* const pszLabel,
 116   ReconstructionROI* pROI)
 117     : BackgroundSupervisor (pThread, pProjView->GetFrame(), pProjView->GetDocument(), "Reconstructing", pProjView->GetDocument()->getProjections().nView()),
 118       m_pProj(pProj), m_pProjView(pProjView), m_pProjDoc(pProjView->GetDocument()),
 119       m_iImageNX(iImageNX), m_iImageNY(iImageNY),
 120       m_pszFilterName(pszFilterName), m_dFilterParam(dFilterParam), m_pszFilterMethod(pszFilterMethod),
 121       m_iZeropad(iZeropad), m_pszFilterGenerationName(pszFilterGenerationName), m_pszInterpName(pszInterpName),
 122       m_iInterpParam(iInterpParam), m_pszBackprojectName(pszBackprojectName), m_pszLabel(pszLabel),
 123       m_pReconROI(pROI)
 124 {
 125   m_vecpChildImageFile.reserve (getNumWorkers());
 126   for (int iThread = 0; iThread < getNumWorkers(); iThread++) {
 127     m_vecpChildImageFile[iThread] = new ImageFile (m_iImageNX, m_iImageNY);
 128   }
 129
 130 }
 131
 132 ReconstructorSupervisor::~ReconstructorSupervisor()
 133 {
 134   for (int i = 0; i < getNumWorkers(); i++) {
 135       delete m_vecpChildImageFile[i];
 136       m_vecpChildImageFile[i] = NULL;
 137     }
 138 }
 139
 140 BackgroundWorkerThread*
 141 ReconstructorSupervisor::createWorker (int iThread, int iStartUnit, int iNumUnits)
 142 {
 143    ReconstructorWorker* pThread = new ReconstructorWorker (this, iThread, iStartUnit, iNumUnits);
 144    pThread->SetParameters (m_pProj, m_pProjView, m_vecpChildImageFile[iThread], m_pszFilterName,
 145      m_dFilterParam, m_pszFilterMethod, m_iZeropad, m_pszFilterGenerationName, m_pszInterpName,
 146      m_iInterpParam, m_pszBackprojectName, m_pReconROI);
 147
 148    return pThread;
 149 }
 150
 151 void
 152 ReconstructorSupervisor::onDone()
 153 {
 154   wxCriticalSection doneSection;
 155   wxCriticalSectionLocker critsect (doneSection);
 156
 157   ImageFile* pImageFile = getImageFile();
 158   pImageFile->labelAdd (m_pProj->getLabel());
 159   pImageFile->labelAdd (m_pszLabel, getTimerEnd());
 160
 161   wxCommandEvent eventLog (wxEVT_COMMAND_MENU_SELECTED, MAINMENU_LOG_EVENT );
 162   wxString msg (m_pszLabel);
 163   msg += "\n";
 164   eventLog.SetString( msg );
 165   wxPostEvent( theApp->getMainFrame(), eventLog ); // send log event
 166
 167   wxCommandEvent newImageEvent (wxEVT_COMMAND_MENU_SELECTED, NEW_IMAGEFILE_EVENT);
 168   newImageEvent.SetClientData (pImageFile);
 169   wxPostEvent (theApp->getMainFrame(), newImageEvent);
 170
 171   setDone();
 172 }
 173
 174
 175 ImageFile*
 176 ReconstructorSupervisor::getImageFile()
 177 {
 178   ImageFile* pImageFile = new ImageFile (m_iImageNX, m_iImageNY);
 179   pImageFile->arrayDataClear();
 180   ImageFileArray pArray = pImageFile->getArray();
 181
 182   int i;
 183   for (i = 0; i < getNumWorkers(); i++) {
 184     ImageFileArrayConst pChildArray = m_vecpChildImageFile[i]->getArray();
 185     for (int ix = 0; ix < m_iImageNX; ix++)
 186       for (int iy = 0; iy < m_iImageNY; iy++)
 187         pArray[ix][iy] += pChildArray[ix][iy];
 188   }
 189
 190   return (pImageFile);
 191 }
 192
 193
 194 /////////////////////////////////////////////////////////////////////
 195 //
 196 // Class ReconstructorWorker -- A worker thread
 197 //
 198 /////////////////////////////////////////////////////////////////////
 199
 200 void
 201 ReconstructorWorker::SetParameters (const Projections* pProj, ProjectionFileView* pProjView, ImageFile* pImageFile,
 202  const char* pszFilterName, double dFilterParam, const char* pszFilterMethod, int iZeropad,
 203  const char* pszFilterGenerationName, const char* pszInterpName, int iInterpParam,
 204  const char* pszBackprojectName, ReconstructionROI* pROI)
 205 {
 206    m_pProj = pProj;
 207    m_pProjView = pProjView;
 208    m_pImageFile = pImageFile;
 209    m_pszFilterName = pszFilterName;
 210    m_dFilterParam = dFilterParam;
 211    m_pszFilterMethod = pszFilterMethod;
 212    m_iZeropad = iZeropad;
 213    m_pszFilterGenerationName = pszFilterGenerationName;
 214    m_pszInterpName = pszInterpName;
 215    m_iInterpParam = iInterpParam;
 216    m_pszBackprojectName = pszBackprojectName;
 217    m_pReconROI = pROI;
 218 }
 219
 220 wxThread::ExitCode
 221 ReconstructorWorker::Entry ()
 222 {
 223   Reconstructor* pReconstructor = new Reconstructor (*m_pProj, *m_pImageFile, m_pszFilterName,
 224     m_dFilterParam, m_pszFilterMethod, m_iZeropad, m_pszFilterGenerationName, m_pszInterpName,
 225     m_iInterpParam, m_pszBackprojectName, Trace::TRACE_NONE, m_pReconROI, false);
 226
 227   bool bFail = pReconstructor->fail();
 228   std::string failMsg;
 229   if (bFail) {
 230       failMsg = "Unable to make reconstructor: ";
 231       failMsg += pReconstructor->failMessage().c_str();
 232       wxCommandEvent event( wxEVT_COMMAND_MENU_SELECTED, MAINMENU_LOG_EVENT );
 233       event.SetString( failMsg.c_str() );
 234       wxPostEvent( theApp->getMainFrame(), event );
 235   }
 236   else
 237   {
 238     wxCommandEvent eventProgress (wxEVT_COMMAND_MENU_SELECTED, BackgroundSupervisor::MSG_WORKER_THREAD_UNIT_TICK);
 239     for (int iUnit = 0; iUnit < m_iNumUnits; iUnit++) {
 240       if (TestDestroy()) {
 241 #ifdef DEBUG
 242         if (theApp->getVerboseLogging()) {
 243           wxString msg;
 244           msg.Printf("Worker thread: Received destroy message at work unit #%d\n", iUnit);
 245           wxCommandEvent event( wxEVT_COMMAND_MENU_SELECTED, MAINMENU_LOG_EVENT );
 246           event.SetString( msg );
 247           wxPostEvent( theApp->getMainFrame(), event );
 248         }
 249 #endif
 250         break;
 251       }
 252       pReconstructor->reconstructView (iUnit + m_iStartUnit, 1);
 253       m_pSupervisor->onWorkerUnitTick();
 254     }
 255     pReconstructor->postProcessing();
 256   }
 257   delete pReconstructor;
 258
 259   if (bFail) {
 260         m_pSupervisor->onWorkerFail (m_iThread, failMsg);
 261   } else {
 262         m_pSupervisor->onWorkerDone (m_iThread);
 263   }
 264
 265   while (! TestDestroy())
 266     Sleep(100);
 267
 268   return reinterpret_cast<wxThread::ExitCode>(0);
 269 }
 270
 271 void
 272 ReconstructorWorker::OnExit ()
 273 {
 274 }
 275
 276 #endif // HAVE_WXTHREADS