1 /*****************************************************************************
4 ** Name: BackgroundSupr.cpp
5 ** Purpose: Background Supervisor classes
6 ** Programmer: Kevin Rosenberg
7 ** Date Started: February 2001
9 ** This is part of the CTSim program
10 ** Copyright (C) 1983-2001 Kevin Rosenberg
12 ** $Id: backgroundsupr.cpp,v 1.7 2001/02/25 19:24:01 kevin Exp $
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.
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.
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 ******************************************************************************/
28 #include "wx/wxprec.h"
38 #include "backgroundsupr.h"
39 #include "backgroundmgr.h"
41 #if defined(HAVE_CONFIG_H)
46 ////////////////////////////////////////////////////////////////////////////
48 // Class BackgroundSupervisor -- An event handler run by a SupervisorThread
50 ////////////////////////////////////////////////////////////////////////////
52 IMPLEMENT_DYNAMIC_CLASS(BackgroundSupervisor, wxEvtHandler)
53 BEGIN_EVENT_TABLE(BackgroundSupervisor, BackgroundSupervisor)
54 EVT_MENU(MSG_BACKGROUND_SUPERVISOR_CANCEL, BackgroundSupervisor::OnCancel)
55 EVT_MENU(MSG_WORKER_THREAD_FAIL, BackgroundSupervisor::OnWorkerFail)
56 EVT_MENU(MSG_WORKER_THREAD_DONE, BackgroundSupervisor::OnWorkerDone)
57 EVT_MENU(MSG_WORKER_THREAD_UNIT_TICK, BackgroundSupervisor::OnWorkerUnitTick)
58 EVT_MENU(MSG_DOCUMENT_ACK_REMOVE, BackgroundSupervisor::OnAckDocumentRemove)
63 BackgroundSupervisor::cancelSupervisor (BackgroundSupervisor* pSupervisor)
65 wxCommandEvent cancelEvent (wxEVT_COMMAND_MENU_SELECTED, MSG_BACKGROUND_SUPERVISOR_CANCEL);
66 wxPostEvent (pSupervisor, cancelEvent);
70 BackgroundSupervisor::BackgroundSupervisor (wxFrame* pParentFrame, wxDocument* pDocument, const char* const pszProcessTitle, int iTotalUnits)
71 : m_pParentFrame(pParentFrame), m_pDocument(pDocument), m_pDialogProgress(NULL), m_strProcessTitle(pszProcessTitle),
72 m_iTotalUnits(iTotalUnits), m_iNumThreads(0), m_bDone(false), m_bFail(false), m_bCancelled(false),
73 m_pTimer(NULL), m_bBackgroundTaskAdded(false), m_bWorkersDeleted(false),
76 m_iNumThreads = theApp->getNumberCPU();
79 m_vecpThreads.reserve (m_iNumThreads);
80 for (int iThread = 0; iThread < m_iNumThreads; iThread++)
81 m_vecpThreads[iThread] = NULL;
85 BackgroundSupervisor::~BackgroundSupervisor()
87 if (m_bBackgroundTaskAdded) {
88 wxCommandEvent doneEvent (wxEVT_COMMAND_MENU_SELECTED, MSG_BACKGROUND_SUPERVISOR_REMOVE);
89 doneEvent.SetClientData (this);
90 wxPostEvent (theApp->getBackgroundManager(), doneEvent);
91 wxPostEvent (m_pDocument, doneEvent);
94 while (m_bBackgroundTaskAdded) {
95 m_pMyThread->Sleep(50);
96 ProcessPendingEvents();
100 delete m_pDialogProgress;
104 BackgroundSupervisor::deleteWorkers()
106 wxCriticalSectionLocker lock (m_critsectThreads);
107 if (m_bWorkersDeleted)
110 for (int i = 0; i < m_iNumThreads; i++)
111 if (m_vecpThreads[i]) {
112 m_vecpThreads[i]->Delete(); // sends Destroy message to workers
115 while (m_iRunning > 0) {
116 m_pMyThread->Sleep(50);
117 ProcessPendingEvents();
120 m_bWorkersDeleted = true;
124 BackgroundSupervisor::start()
126 int iBaseUnits = m_iTotalUnits / m_iNumThreads;
127 int iExtraUnits = m_iTotalUnits % m_iNumThreads;
128 for (int iThread = 0; iThread < m_iNumThreads; iThread++) {
129 int iStartUnit = iThread * iBaseUnits;
130 int iNumUnits = iBaseUnits;
131 if (iThread < iExtraUnits)
133 m_vecpThreads[iThread] = createWorker (iThread, iStartUnit, iNumUnits);
134 if (! m_vecpThreads[iThread]) {
136 m_strFailMessage = "createWorker returned NULL [BackgroundSupervisor]";
139 if (m_vecpThreads[iThread]->Create () != wxTHREAD_NO_ERROR) {
141 m_strFailMessage = "Thread creation failed [BackgroundSupervisor]";
144 m_vecpThreads[iThread]->SetPriority (40);
149 m_pTimer = new Timer;
151 if (! theApp->getUseBackgroundTasks())
152 m_pDialogProgress = new wxProgressDialog (_T("Filtered Backprojection"), _T("Reconstruction Progress"),
153 m_iTotalUnits, m_pParentFrame, wxPD_CAN_ABORT | wxPD_AUTO_HIDE);
155 std::string strLabel (m_strProcessTitle);
157 strLabel += m_pParentFrame->GetTitle();
158 wxCommandEvent addTaskEvent (wxEVT_COMMAND_MENU_SELECTED, MSG_BACKGROUND_SUPERVISOR_ADD);
159 addTaskEvent.SetString (strLabel.c_str());
160 addTaskEvent.SetInt (m_iTotalUnits);
161 addTaskEvent.SetClientData (this);
162 wxPostEvent (theApp->getBackgroundManager(), addTaskEvent);
163 wxPostEvent (m_pDocument, addTaskEvent);
164 m_bBackgroundTaskAdded = true;
167 m_iRunning = m_iNumThreads;
170 for (int i = 0; i < m_iNumThreads; i++)
171 m_vecpThreads[i]->Run();
177 BackgroundSupervisor::OnCancel(wxCommandEvent& event)
185 BackgroundSupervisor::OnAckDocumentRemove(wxCommandEvent& event)
187 m_bBackgroundTaskAdded = false;
191 BackgroundSupervisor::OnWorkerUnitTick (wxCommandEvent& event)
196 if (theApp->getVerboseLogging())
197 *theApp->getLog() << "Units done: " << static_cast<int>(m_iUnitsDone) <<"\n";
200 if (m_pDialogProgress) {
201 if (! m_pDialogProgress->Update (m_iUnitsDone - 1)) {
202 wxCommandEvent dummy;
206 wxCommandEvent addTaskEvent (wxEVT_COMMAND_MENU_SELECTED, MSG_BACKGROUND_SUPERVISOR_UNIT_TICK);
207 addTaskEvent.SetInt (m_iUnitsDone - 1);
208 addTaskEvent.SetClientData (this);
209 wxPostEvent (theApp->getBackgroundManager(), addTaskEvent);
214 BackgroundSupervisor::OnWorkerDone (wxCommandEvent& event)
217 wxASSERT (m_iRunning >= 0);
220 if (theApp->getVerboseLogging()) {
222 msg.Printf("Background Supervisor: Thread finished. Remaining threads: %d\n", m_iRunning);
223 wxCommandEvent eventLog (wxEVT_COMMAND_MENU_SELECTED, MAINMENU_LOG_EVENT );
224 eventLog.SetString( msg );
225 wxPostEvent( theApp->getMainFrame(), eventLog ); // send log event
228 if (m_iRunning <= 0 && ! m_bCancelled) {
235 BackgroundSupervisor::OnWorkerFail (wxCommandEvent& event)
238 wxCommandEvent eventLog( wxEVT_COMMAND_MENU_SELECTED, MAINMENU_LOG_EVENT );
239 eventLog.SetString( event.GetString() );
240 wxPostEvent( theApp->getMainFrame(), eventLog ); // send log event
242 wxCommandEvent dummy;