freSetupOptimizationMultiThreadMetric.txx

Go to the documentation of this file.

/*=========================================================================

  Program:   F.R.E.E. - flexible registration evaluation engine
  Version:   v.1.0.0
  Date:      $Date: 2006/09/01 12:00:00 $
  Module:    $RCSfile: freSetupOptimizationMultiThreadMetric.txx,v $
  Language:  C++



  Copyright (c) 2007 Ralf o Floca (Department of Medical Informatics,
  Institute for Medical Biometry and Informatics, University of Heidelberg,
  Germany). All rights reserved.
  See FREECopyright.txt or http://www.mi.med.uni-hd.de/free/copyright.htm
  for details.

     This software is distributed WITHOUT ANY WARRANTY; without even 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#ifndef __freSetupOptimizationMultiThreadMetric_txx
#define __freSetupOptimizationMultiThreadMetric_txx

#include "freSetupOptimizationMultiThreadMetric.h"
#include "itkMultiThreader.h"

namespace FREE
{
template <typename TMonitor, typename TThread>
typename SetupOptimizationMultiThreadMetric<TMonitor, TThread>::DecomposedMeasureType
SetupOptimizationMultiThreadMetric<TMonitor, TThread>::
ComputeDecomposedValue( const ParametersType & parameters ) const
{
        if (this->m_smpAdaptations.IsNull()) throwExceptionMacro("Error. Cannot calculate metric value, adaptiation list is null.");
        if (this->m_Transform.IsNull()) throwExceptionMacro("Error. Cannot calculate metric value, no setup transform set.");

        if (this->m_Transform->GetNumberOfParameters() != parameters.size()) throwExceptionMacro("Error. Cannot calculate metric value, number of passed parameters is not equal to the parameter size of the transform. Passed parameter count: "<< parameters.size());

  this->InitializeValueComputation();

        m_Transform->SetParameters(parameters);
  Setup::Pointer smpTransformedSetup = m_Transform->GenerateTransformedSetup();

  typename MonitorType::Pointer smpMonitor = MonitorType::New();

  try
  {
    smpMonitor->fnOnEvaluationProgress = this->fnOnMonitorEvaluationProgress;
    smpMonitor->fnOnEvaluationDone = this->fnOnMonitorEvaluationDone;
    smpMonitor->fnOnNextAdaptation = this->fnOnMonitorNextAdaptation;
    smpMonitor->fnOnEvaluationFailed = this->fnOnMonitorEvaluationFailed;

    smpMonitor->SetGenericSetup(smpTransformedSetup);
    smpMonitor->SetAdaptationList(m_smpAdaptations);
    smpMonitor->SetMetricStatistic(&(this->GetMetricStatistic()));

    this->InitializeMonitor(*(smpMonitor.GetPointer()));
    smpMonitor->Initialize();
  }
  catchAllNPassMacro("Unknown error while initializing the monitor.");

  ThreadListType threadList;

  itk::MultiThreader::Pointer smpThreader = itk::MultiThreader::New();

  long lMaxThreadCount = this->m_NumberOfThreads;
  if (lMaxThreadCount<1) lMaxThreadCount = m_smpAdaptations->Size();
  
  smpThreader->SetNumberOfThreads(lMaxThreadCount);

  try
  {
    for (long index = 0; index<lMaxThreadCount; index++)
    { //create the threads, the id will be set via SetupOptimizationMetricThreadBase::ThreadExecution
      //and the thread info structure passed by the threader
      ThreadPointer smpThread = ThreadType::New();
      smpThread->SetMonitor(smpMonitor.GetPointer());

      this->InitializeThread(*(smpThread.GetPointer()));

      threadList.push_back(smpThread);
      smpThreader->SetMultipleMethod(index, Self::ThreadExecution, smpThread.GetPointer());
    }
  }
  catchAllNPassMacro("Unknown error while initializing the threads.");

  try
  {
    //Start with the work
    smpThreader->MultipleMethodExecute();
  }
  catchAllNPassMacro("Unknown error while executing threads to calculate the metric.");

  try
  {
    //threads are finished, compute measures
    return this->ComputeMeasure(*(smpMonitor.GetPointer()));
  }
  catchAllNPassMacro("Unknown error while computing the measure.");
};

template <typename TMonitor, typename TThread>
SetupOptimizationMultiThreadMetric<TMonitor, TThread>::
SetupOptimizationMultiThreadMetric()
{
  m_NumberOfThreads = 1;

  fnOnMonitorEvaluationProgress = ProgressEvent<Self>::New(this,&Self::OnMonitorEvaluationProgress);
  fnOnMonitorEvaluationDone = ProgressEvent<Self>::New(this,&Self::OnMonitorEvaluationDone);
  fnOnMonitorNextAdaptation = ProgressEvent<Self>::New(this,&Self::OnOnMonitorNextAdaptation);
  fnOnMonitorEvaluationFailed = ProgressEvent<Self>::New(this,&Self::OnOnMonitorEvaluationFailed);
};

template <typename TMonitor, typename TThread>
SetupOptimizationMultiThreadMetric<TMonitor, TThread>::
~SetupOptimizationMultiThreadMetric()
{
};

template <typename TMonitor, typename TThread>
void
SetupOptimizationMultiThreadMetric<TMonitor, TThread>::
OnMonitorEvaluationProgress(const long status, const std::string& sComment, void* pSender, long threadID)
{
  //pass the monitor event, but exchange the sende
  if (fnOnEvaluationProgress.IsNotNull()) fnOnEvaluationProgress->Execute(status,sComment,(void*)this, threadID);
  this->InvokeEvent( EvaluationDoneObserverEvent() );
};

template <typename TMonitor, typename TThread>
void
SetupOptimizationMultiThreadMetric<TMonitor, TThread>::
OnMonitorEvaluationDone(const long status, const std::string& sComment, void* pSender, long threadID)
{
  //pass the monitor event, but exchange the sende
  if (fnOnEvaluationDone.IsNotNull()) fnOnEvaluationDone->Execute(status,sComment,(void*)this, threadID);
  this->InvokeEvent( EvaluationDoneObserverEvent() );
};

template <typename TMonitor, typename TThread>
void
SetupOptimizationMultiThreadMetric<TMonitor, TThread>::
OnOnMonitorNextAdaptation(const long status, const std::string& sComment, void* pSender, long threadID)
{
  //pass the monitor event, but exchange the sende
  if (fnOnNextAdaptation.IsNotNull()) fnOnNextAdaptation->Execute(status,sComment,(void*)this, threadID);
  this->InvokeEvent( EvaluationDoneObserverEvent() );
};

template <typename TMonitor, typename TThread>
void
SetupOptimizationMultiThreadMetric<TMonitor, TThread>::
OnOnMonitorEvaluationFailed(const long status, const std::string& sComment, void* pSender, long threadID)
{
  //pass the monitor event, but exchange the sende
  if (fnOnEvaluationFailed.IsNotNull()) fnOnEvaluationFailed->Execute(status,sComment,(void*)this, threadID);
  this->InvokeEvent( EvaluationDoneObserverEvent() );
};

template <typename TMonitor, typename TThread>
ITK_THREAD_RETURN_TYPE
SetupOptimizationMultiThreadMetric<TMonitor, TThread>::
ThreadExecution( void *arg )
{
  ThreadType *pThread;
  int threadID;

  threadID = ((itk::MultiThreader::ThreadInfoStruct *)(arg))->ThreadID;

  pThread = (ThreadType *)(((itk::MultiThreader::ThreadInfoStruct *)(arg))->UserData);

  if (!pThread) throwStaticExceptionMacro("Error; user data for thread execution is missing");

  pThread->SetThreadID(threadID);
  pThread->Execute();

  return ITK_THREAD_RETURN_VALUE;
};

} // end namespace FREE

#endif

---