itkSplitDWImageFilter.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 ITKSPLITDWIMAGEFILTER_H
#define ITKSPLITDWIMAGEFILTER_H

#include "itkImageToImageFilter.h"
#include <itkVectorImage.h>

namespace itk
{

template< class TInputImagePixelType,
          class TOutputImagePixelType >
class SplitDWImageFilter :
    public ImageToImageFilter< VectorImage<TInputImagePixelType, 3>,
    Image< TOutputImagePixelType, 4 > >
{
public:

  typedef SplitDWImageFilter  Self;
  typedef SmartPointer<Self>                      Pointer;
  typedef SmartPointer<const Self>                ConstPointer;

  typedef TInputImagePixelType                    InputPixelType;
  typedef TOutputImagePixelType                   OutputPixelType;

  typedef ImageToImageFilter< VectorImage<TInputImagePixelType, 3>,
      Image< TOutputImagePixelType, 4 > >         Superclass;

  typedef typename Superclass::InputImageType     InputImageType;
  //typedef typename Superclass::OutputImageType    OutputImageType;
  typedef Image< TOutputImagePixelType, 4 >       OutputImageType;
  typedef typename OutputImageType::RegionType    OutputImageRegionType;

  typedef std::vector< unsigned int > IndexListType;
  typedef std::map< unsigned int, IndexListType> BValueMapType;

  itkFactorylessNewMacro(Self)
  itkCloneMacro(Self)

  itkTypeMacro(SplitDWImageFilter, ImageToImageFilter);

  void SetExtractIndices( IndexListType list)
  {
    m_IndexList = list;
  }

  void SetExtractAll()
  {
    m_ExtractAllImages = true;
  }

  void SetExtractAllAboveThreshold( double b_threshold, BValueMapType map);

  void SetExtractSingleShell( double b_value, BValueMapType map, double tol );

  const IndexListType GetIndexList() const
  {
    return m_IndexList;
  }

protected:

  SplitDWImageFilter();
  virtual ~SplitDWImageFilter(){};

  void GenerateData();

  virtual void CopyInformation( const DataObject *data);

  virtual void GenerateOutputInformation();

  IndexListType m_IndexList;
  bool m_ExtractAllImages;
};

} //end namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkSplitDWImageFilter.txx"
#endif

#endif // ITKSPLITDWIMAGEFILTER_H