+/*****************************************************************************
+** FILE IDENTIFICATION
+**
+** Name: BackgroundSupr.cpp
+** Purpose: Background Supervisor classes
+** Programmer: Kevin Rosenberg
+** Date Started: February 2001
+**
+** This is part of the CTSim program
+** Copyright (C) 1983-2001 Kevin Rosenberg
+**
+** $Id: backgroundsupr.cpp,v 1.1 2001/02/25 06:32:12 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
+** published by the Free Software Foundation.
+**
+** This program is distributed in the hope that it will be useful,
+** but WITHOUT ANY WARRANTY; without even the implied warranty of
+** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+** GNU General Public License for more details.
+**
+** You should have received a copy of the GNU General Public License
+** along with this program; if not, write to the Free Software
+** Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+******************************************************************************/
+
+#include "wx/wxprec.h"
+
+#ifndef WX_PRECOMP
+#include "wx/wx.h"
+#endif
+
+#include "ct.h"
+#include "ctsim.h"
+#include "docs.h"
+#include "views.h"
+#include "backgroundsupr.h"
+#include "backgroundmgr.h"
+
+#if defined(HAVE_CONFIG_H)
+#include "config.h"
+#endif
+
+
+////////////////////////////////////////////////////////////////////////////
+//
+// Class BackgroundSupervisor -- An event handler run by a SupervisorThread
+//
+////////////////////////////////////////////////////////////////////////////
+
+IMPLEMENT_DYNAMIC_CLASS(BackgroundSupervisor, wxEvtHandler)
+BEGIN_EVENT_TABLE(BackgroundSupervisor, BackgroundSupervisor)
+EVT_MENU(MSG_BACKGROUND_SUPERVISOR_CANCEL, BackgroundSupervisor::OnCancel)
+EVT_MENU(MSG_WORKER_THREAD_FAIL, BackgroundSupervisor::OnWorkerFail)
+EVT_MENU(MSG_WORKER_THREAD_DONE, BackgroundSupervisor::OnWorkerDone)
+EVT_MENU(MSG_WORKER_THREAD_UNIT_TICK, BackgroundSupervisor::OnWorkerUnitTick)
+EVT_MENU(MSG_DOCUMENT_ACK_REMOVE, BackgroundSupervisor::OnAckDocumentRemove)
+END_EVENT_TABLE()
+
+// Static function
+void
+BackgroundSupervisor::cancelSupervisor (BackgroundSupervisor* pSupervisor)
+{
+ wxCommandEvent cancelEvent (wxEVT_COMMAND_MENU_SELECTED, MSG_BACKGROUND_SUPERVISOR_CANCEL);
+ wxPostEvent (pSupervisor, cancelEvent);
+}
+
+
+BackgroundSupervisor::BackgroundSupervisor (wxFrame* pParentFrame, wxDocument* pDocument, const char* const pszProcessTitle, int iTotalUnits)
+ : m_pParentFrame(pParentFrame), m_pDocument(pDocument), m_pDialogProgress(NULL), m_strProcessTitle(pszProcessTitle),
+ m_iTotalUnits(iTotalUnits), m_iNumThreads(0), m_bDone(false), m_bFail(false), m_bCancelled(false),
+ m_pTimer(NULL), m_bBackgroundTaskAdded(false),
+ wxEvtHandler()
+{
+ m_iNumThreads = theApp->getNumberCPU();
+ // ++m_iNumThreads;
+
+ m_vecpThreads.reserve (m_iNumThreads);
+ for (int iThread = 0; iThread < m_iNumThreads; iThread++)
+ m_vecpThreads[iThread] = NULL;
+
+}
+
+BackgroundSupervisor::~BackgroundSupervisor()
+{
+ if (m_bBackgroundTaskAdded) {
+ wxCommandEvent doneEvent (wxEVT_COMMAND_MENU_SELECTED, MSG_BACKGROUND_SUPERVISOR_REMOVE);
+ doneEvent.SetClientData (this);
+ wxPostEvent (theApp->getBackgroundManager(), doneEvent);
+ wxPostEvent (m_pDocument, doneEvent);
+ }
+
+ while (m_bBackgroundTaskAdded) {
+ Sleep(50);
+ ::wxYield();
+ }
+
+ delete m_pTimer;
+ delete m_pDialogProgress;
+}
+
+void
+BackgroundSupervisor::deleteAnyWorkers()
+{
+ m_critsectThreadContainer.Enter();
+ for (int i = 0; i < m_iNumThreads; i++)
+ if (m_vecpThreads[i]) {
+ m_vecpThreads[i]->Delete(); // sends Destroy message to workers
+ }
+ m_critsectThreadContainer.Leave();
+
+ while (m_iRunning > 0) {
+ Sleep(50);
+ ::wxYield();
+ }
+}
+
+
+bool
+BackgroundSupervisor::start()
+{
+ int iBaseUnits = m_iTotalUnits / m_iNumThreads;
+ int iExtraUnits = m_iTotalUnits % m_iNumThreads;
+ for (int iThread = 0; iThread < m_iNumThreads; iThread++) {
+ int iStartUnit = iThread * iBaseUnits;
+ int iNumUnits = iBaseUnits;
+ if (iThread < iExtraUnits)
+ ++iNumUnits;
+ m_vecpThreads[iThread] = createWorker (iThread, iStartUnit, iNumUnits);
+ if (! m_vecpThreads[iThread]) {
+ m_bFail = true;
+ m_strFailMessage = "createWorker returned NULL [BackgroundSupervisor]";
+ break;
+ }
+ if (m_vecpThreads[iThread]->Create () != wxTHREAD_NO_ERROR) {
+ m_bFail = true;
+ m_strFailMessage = "Thread creation failed [BackgroundSupervisor]";
+ break;
+ }
+ m_vecpThreads[iThread]->SetPriority (40);
+ }
+ if (m_bFail)
+ return false;
+
+ m_pTimer = new Timer;
+
+ if (! theApp->getUseBackgroundTasks())
+ m_pDialogProgress = new wxProgressDialog (_T("Filtered Backprojection"), _T("Reconstruction Progress"),
+ m_iTotalUnits, m_pParentFrame, wxPD_CAN_ABORT | wxPD_AUTO_HIDE);
+ else {
+ std::string strLabel (m_strProcessTitle);
+ strLabel += " ";
+ strLabel += m_pParentFrame->GetTitle();
+ wxCommandEvent addTaskEvent (wxEVT_COMMAND_MENU_SELECTED, MSG_BACKGROUND_SUPERVISOR_ADD);
+ addTaskEvent.SetString (strLabel.c_str());
+ addTaskEvent.SetInt (m_iTotalUnits);
+ addTaskEvent.SetClientData (this);
+ wxPostEvent (theApp->getBackgroundManager(), addTaskEvent);
+ wxPostEvent (m_pDocument, addTaskEvent);
+ m_bBackgroundTaskAdded = true;
+ }
+
+ m_iRunning = m_iNumThreads;
+ m_iUnitsDone = 0;
+
+ for (int i = 0; i < m_iNumThreads; i++)
+ m_vecpThreads[i]->Run();
+
+ return true;
+}
+
+void
+BackgroundSupervisor::OnCancel(wxCommandEvent& event)
+{
+ m_bCancelled = true;
+ deleteAnyWorkers();
+ m_bDone = true;
+}
+
+void
+BackgroundSupervisor::OnAckDocumentRemove(wxCommandEvent& event)
+{
+ m_bBackgroundTaskAdded = false;
+}
+
+void
+BackgroundSupervisor::OnWorkerUnitTick (wxCommandEvent& event)
+{
+ ++m_iUnitsDone;
+
+#ifdef DEBUG
+ if (theApp->getVerboseLogging())
+ *theApp->getLog() << "Units done: " << static_cast<int>(m_iUnitsDone) <<"\n";
+#endif
+
+ if (m_pDialogProgress) {
+ if (! m_pDialogProgress->Update (m_iUnitsDone - 1)) {
+ wxCommandEvent dummy;
+ OnCancel (dummy);
+ }
+ } else {
+ wxCommandEvent addTaskEvent (wxEVT_COMMAND_MENU_SELECTED, MSG_BACKGROUND_SUPERVISOR_UNIT_TICK);
+ addTaskEvent.SetInt (m_iUnitsDone - 1);
+ addTaskEvent.SetClientData (this);
+ wxPostEvent (theApp->getBackgroundManager(), addTaskEvent);
+ }
+}
+
+void
+BackgroundSupervisor::OnWorkerDone (wxCommandEvent& event)
+{
+ m_iRunning--;
+ wxASSERT (m_iRunning >= 0);
+ m_critsectThreadContainer.Enter();
+ m_vecpThreads[event.GetInt()] = NULL;
+ m_critsectThreadContainer.Leave();
+
+#ifdef DEBUG
+ if (theApp->getVerboseLogging()) {
+ wxString msg;
+ msg.Printf("Background Supervisor: Thread finished. Remaining threads: %d\n", m_iRunning);
+ wxCommandEvent eventLog (wxEVT_COMMAND_MENU_SELECTED, MAINMENU_LOG_EVENT );
+ eventLog.SetString( msg );
+ wxPostEvent( theApp->getMainFrame(), eventLog ); // send log event
+ }
+#endif
+ if (m_iRunning <= 0 && ! m_bCancelled)
+ onDone();
+}
+
+void
+BackgroundSupervisor::OnWorkerFail (wxCommandEvent& event)
+{
+ m_iRunning--;
+ m_critsectThreadContainer.Enter();
+ m_vecpThreads[event.GetInt()] = NULL;
+ m_critsectThreadContainer.Leave();
+ wxCommandEvent eventLog( wxEVT_COMMAND_MENU_SELECTED, MAINMENU_LOG_EVENT );
+ eventLog.SetString( event.GetString() );
+ wxPostEvent( theApp->getMainFrame(), eventLog ); // send log event
+
+ onDone();
+}
+