itkTractDensityImageFilter.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 __itkTractDensityImageFilter_h__
#define __itkTractDensityImageFilter_h__

#include <itkImageSource.h>
#include <itkImage.h>
#include <itkVectorContainer.h>
#include <itkRGBAPixel.h>
#include <mitkFiberBundle.h>

namespace itk{

template< class OutputImageType >
class TractDensityImageFilter : public ImageSource< OutputImageType >
{

public:
  typedef TractDensityImageFilter Self;
  typedef ProcessObject Superclass;
  typedef SmartPointer< Self > Pointer;
  typedef SmartPointer< const Self > ConstPointer;

  typedef typename OutputImageType::PixelType OutPixelType;

  itkFactorylessNewMacro(Self)
  itkCloneMacro(Self)
  itkTypeMacro( TractDensityImageFilter, ImageSource )

  itkSetMacro( UpsamplingFactor, float)                         
  itkGetMacro( UpsamplingFactor, float)                         
  itkSetMacro( InvertImage, bool)                               
  itkGetMacro( InvertImage, bool)                               
  itkSetMacro( BinaryOutput, bool)                              
  itkGetMacro( BinaryOutput, bool)                              
  itkSetMacro( OutputAbsoluteValues, bool)                      
  itkGetMacro( OutputAbsoluteValues, bool)                      
  itkSetMacro( UseImageGeometry, bool)                          
  itkGetMacro( UseImageGeometry, bool)                          
  itkSetMacro( FiberBundle, mitk::FiberBundle::Pointer)        
  itkSetMacro( InputImage, typename OutputImageType::Pointer)   
  itkSetMacro( UseTrilinearInterpolation, bool )
  itkSetMacro( DoFiberResampling, bool )

  void GenerateData();

protected:

  itk::Point<float, 3> GetItkPoint(double point[3]);

  TractDensityImageFilter();
  virtual ~TractDensityImageFilter();

  typename OutputImageType::Pointer m_InputImage;           
  mitk::FiberBundle::Pointer       m_FiberBundle;          
  float                             m_UpsamplingFactor;     
  bool                              m_InvertImage;          
  bool                              m_BinaryOutput;         
  bool                              m_UseImageGeometry;     
  bool                              m_OutputAbsoluteValues; 
  bool                              m_UseTrilinearInterpolation;
  bool                              m_DoFiberResampling;
};

}

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkTractDensityImageFilter.cpp"
#endif

#endif // __itkTractDensityImageFilter_h__