cmtkMultiLevelOptimizer.h

Go to the documentation of this file.

/*
//
//  Copyright 1997-2009 Torsten Rohlfing
//
//  Copyright 2004-2010 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: 2398 $
//
//  $LastChangedDate: 2010-10-05 14:54:37 -0700 (Tue, 05 Oct 2010) $
//
//  $LastChangedBy: torstenrohlfing $
//
*/

#ifndef __cmtkMultiLevelOptimizer_h_included_
#define __cmtkMultiLevelOptimizer_h_included_

#include <cmtkconfig.h>

#include <Registration/cmtkOptimizer.h>
#include <Registration/cmtkOptimizerBase.h>

namespace
cmtk
{


class MultiLevelOptimizer : 
    public OptimizerBase
{
public:
  MultiLevelOptimizer( Optimizer::SmartPtr& optimizer )
    : m_Optimizer( optimizer ) {}

  void SetOptimizer( Optimizer::SmartPtr& optimizer )
  {
    this->m_Optimizer = optimizer;
  }

  virtual ~MultiLevelOptimizer() {}

  size_t AddFunctional( Functional::SmartPtr functional, const Types::Coordinate initialStepSize, const Types::Coordinate finalStepSize )
  {
    this->m_FunctionalList.push_back( FunctionalWithStepSizes::SmartPtr( new FunctionalWithStepSizes( functional, initialStepSize, finalStepSize ) ) );
    return this->m_FunctionalList.size();
  }
 
  virtual CallbackResult Optimize( CoordinateVector&, const Types::Coordinate = 1, const Types::Coordinate = 0 );

private:
  Optimizer::SmartPtr m_Optimizer;

  class FunctionalWithStepSizes
  {
  public:
    typedef FunctionalWithStepSizes Self;

    typedef SmartPointer<Self> SmartPtr;

    FunctionalWithStepSizes() :
      m_Functional( NULL ), m_InitialStepSize( 0.0 ), m_FinalStepSize( 0.0 ) 
    {}

    FunctionalWithStepSizes( Functional::SmartPtr functional, const Types::Coordinate initialStepSize, const Types::Coordinate finalStepSize ) :
      m_Functional( functional ), 
      m_InitialStepSize( initialStepSize ), 
      m_FinalStepSize( finalStepSize ) 
    {}

    Functional::SmartPtr m_Functional;

    Types::Coordinate m_InitialStepSize;

    Types::Coordinate m_FinalStepSize;
  };

  typedef std::list<FunctionalWithStepSizes::SmartPtr> FunctionalListType;

  FunctionalListType m_FunctionalList;
};


} // namespace cmtk

#endif // #ifndef __cmtkMultiLevelOptimizer_h_included_