cmtkGroupwiseRegistrationRMIFunctional.h

Go to the documentation of this file.

/*
//
//  Copyright 1997-2009 Torsten Rohlfing
//
//  Copyright 2004-2011 SRI International
//
//  This file is part of the Computational Morphometry Toolkit.
//
//  http://www.nitrc.org/projects/cmtk/
//
//  The Computational Morphometry Toolkit is free software: you can
//  redistribute it and/or modify it under the terms of the GNU General Public
//  License as published by the Free Software Foundation, either version 3 of
//  the License, or (at your option) any later version.
//
//  The Computational Morphometry Toolkit 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 the Computational Morphometry Toolkit.  If not, see
//  <http://www.gnu.org/licenses/>.
//
//  $Revision: 2752 $
//
//  $LastChangedDate: 2011-01-17 11:33:31 -0800 (Mon, 17 Jan 2011) $
//
//  $LastChangedBy: torstenrohlfing $
//
*/

#ifndef __cmtkGroupwiseRegistrationRMIFunctional_h_included_
#define __cmtkGroupwiseRegistrationRMIFunctional_h_included_

#include <cmtkconfig.h>

#include <Registration/cmtkGroupwiseRegistrationFunctionalXformTemplate.h>

#include <System/cmtkSmartPtr.h>
#include <System/cmtkThreads.h>
#include <System/cmtkMutexLock.h>

#include <Base/cmtkUniformVolume.h>
#include <Base/cmtkXform.h>
#include <Base/cmtkMathUtil.h>
#include <Base/cmtkMatrix.h>

#include <vector>

namespace
cmtk
{


template<class TXform>
class GroupwiseRegistrationRMIFunctional : 
  public GroupwiseRegistrationFunctionalXformTemplate<TXform>
{
public:
  typedef GroupwiseRegistrationFunctionalXformTemplate<TXform> Superclass;

  typedef GroupwiseRegistrationRMIFunctional<TXform> Self;

  typedef SmartPointer<Self> SmartPtr;

  typedef TXform XformType;

  typedef typename XformType::SmartPtr XformPointer;

  GroupwiseRegistrationRMIFunctional();

  virtual ~GroupwiseRegistrationRMIFunctional();

  virtual void SetTemplateGrid( UniformVolume::SmartPtr& templateGrid , 
                                const int downsample = 1 , const bool useTemplateData = false  );

  virtual typename Self::ReturnType EvaluateWithGradient( CoordinateVector& v, CoordinateVector& g, const Types::Coordinate step = 1 );

  virtual typename Self::ReturnType Evaluate();
  
protected:
  typedef Matrix2D<typename Self::ReturnType> CovarianceMatrixType;

  typename Self::CovarianceMatrixType m_CovarianceMatrix;

  typedef std::vector<long int> SumsAndProductsVectorType;

  SumsAndProductsVectorType m_SumOfProductsMatrix;

  SumsAndProductsVectorType m_SumsVector;

  typename Self::ReturnType GetMetric
  ( const SumsAndProductsVectorType& sumOfProductsMatrix, 
    const SumsAndProductsVectorType& sumsVector,
    const unsigned int totalNumberOfSamples,
    typename Self::CovarianceMatrixType& covarianceMatrix ) const;

  std::vector<SumsAndProductsVectorType> m_ThreadSumOfProductsMatrix;

  std::vector<SumsAndProductsVectorType> m_ThreadSumsVector;

  unsigned int m_TotalNumberOfSamples;

  MutexLock m_MutexLock;

  virtual bool Wiggle();

private:
  typedef ThreadParameters<Self> ThreadParametersType;

  class EvaluateThreadParameters : 
    public ThreadParametersType
  {
  };
  
  static void EvaluateThread( void *const threadParameters, const size_t taskIdx, const size_t taskCnt, const size_t threadIdx, const size_t );

  static void EvaluateProbabilisticThread( void *const threadParameters, const size_t taskIdx, const size_t taskCnt, const size_t threadIdx, const size_t );
};


} // namespace cmtk

#endif // #ifndef __cmtkGroupwiseRegistrationRMIFunctional_h_included_