itkShortestPathCostFunctionTbss.h

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/
#ifndef __itkShortestPathCostFunctionTbss_h
#define __itkShortestPathCostFunctionTbss_h

#include "itkObject.h"
#include "itkObjectFactory.h"
#include "itkShapedNeighborhoodIterator.h"
#include "itkShortestPathCostFunction.h" // Superclass of Metrics
#include <itkImageRegionConstIterator.h>
#include <itkMacro.h>

namespace itk
{
  template <class TInputImageType>
  class ShortestPathCostFunctionTbss : public ShortestPathCostFunction<TInputImageType>
  {
  public:
    typedef ShortestPathCostFunctionTbss Self;
    typedef ShortestPathCostFunction<TInputImageType> Superclass;
    typedef SmartPointer<Self> Pointer;
    typedef SmartPointer<const Self> ConstPointer;
    typedef itk::ImageRegionConstIterator<TInputImageType> ConstIteratorType;
    typedef typename TInputImageType::IndexType IndexType;

    typedef itk::Image<float, 3> FloatImageType;

    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      itkTypeMacro(Self, Superclass);

    // \brief calculates the costs for going from p1 to p2
    virtual double GetCost(IndexType p1, IndexType p2);

    // \brief Initialize the metric
    virtual void Initialize();

    // \brief returns the minimal costs possible (needed for A*)
    virtual double GetMinCost();

    void SetThreshold(double t) { m_Threshold = t; }
  protected:
    ShortestPathCostFunctionTbss();

    virtual ~ShortestPathCostFunctionTbss(){};

    double m_Threshold;

  private:
  };

} // end namespace itk

#include "itkShortestPathCostFunctionTbss.txx"

#endif /* __itkShortestPathCostFunctionTbss_h */