itkSamplingSchemeQSpace.h

Go to the documentation of this file.

#ifndef __itkSamplingSchemeQSpace_h
#define __itkSamplingSchemeQSpace_h

#include "itkSamplingScheme3D.h"
#include "itkObject.h"

namespace itk 
{

template<class TPixelType = double> 
class ITK_EXPORT SamplingSchemeQSpace : public SamplingScheme3D<TPixelType>
{
public:
  typedef SamplingSchemeQSpace Self;
  typedef SamplingScheme3D<TPixelType> Superclass;
  typedef SmartPointer<Self> Pointer;
  typedef SmartPointer<const Self> ConstPointer;
  
  itkNewMacro(Self);

  itkTypeMacro(SamplingSchemeQSpace, SamplingScheme3D);
  
  typedef typename Superclass::STDVectorType          STDVectorType;
  typedef typename Superclass::STDVectorPointer       STDVectorPointer;
  typedef typename Superclass::IndexVectorType        IndexVectorType;
  typedef typename Superclass::Index2DVectorType      Index2DVectorType;
  typedef typename Superclass::Index2DVectorPointer   Index2DVectorPointer;
  
  itkSetMacro(BThresholdSingleShell, double);
  itkGetMacro(BThresholdSingleShell, double);

  void SetBVector(const STDVectorPointer bVec);
  itkGetMacro(BVector, STDVectorPointer);
  STDVectorPointer GetBVectorInShell(unsigned int shellIndex);

  void SetSamplingScheme3D(typename Superclass::Pointer scheme3D);
  
  void Clear();

  void ConvertBVectorToQVector();
  
  void ConvertQVectorToBVector();
  
  std::vector<STDVectorType> GroupBValues();
  
  void CorrectBValues();
  void CorrectRadiusValues() ITK_OVERRIDE;
  
  void RemoveSamplesNotIndexed();

protected:
  SamplingSchemeQSpace ();

  virtual ~SamplingSchemeQSpace ()
    {
    }
  
  typename LightObject::Pointer InternalClone() const ITK_OVERRIDE;
 
  void PrintSelf(std::ostream& os, Indent indent) const ITK_OVERRIDE;

  double m_BThresholdSingleShell;

  STDVectorPointer m_BVector;

private:
  SamplingSchemeQSpace(const Self &); //purposely not implemented
  void operator=(const Self &);  //purposely not implemented

  // std::vector<STDVectorType> GroupRadiusValues();
  // void CorrectRadiusValues();
};

}

#if !defined(ITK_MANUAL_INSTANTIATION) && !defined(__itkSamplingSchemeQSpace_hxx)
#include "itkSamplingSchemeQSpace.hxx"
#endif

#endif