itkOuterProductVectorImageFilter.h

Go to the documentation of this file.

#ifndef __itkOuterProductVectorImageFilter_h
#define __itkOuterProductVectorImageFilter_h

#include "itkUnaryFunctorImageFilter.h"
#include "utlNDArray.h"

namespace itk
{
namespace Functor
{
template <class TInput, class TOutput=TInput>
class OuterProduct
{
public:
  typedef typename TInput::ValueType   InputValueType;
  typedef typename TOutput::ValueType  OutputValueType;
  typedef utl::NDArray<double, 1>      VectorType;

  OuterProduct()
  {
  }

  ~OuterProduct()
  {
  }

  bool operator!=(const OuterProduct & other) const
  {
    return !( *this == other );
  }

  bool operator==(const OuterProduct & other) const
  {
    return true;
  }

  inline TOutput operator()(const TInput & A)
  {
  int N = A.GetSize();
  TOutput out;
  out.SetSize( N*N );
  int kk=0;
  for ( int i = 0; i < N; ++i ) 
    {
    for ( int j = 0; j < N; ++j ) 
      {
      out[kk] = A[i]*A[j];
      kk++;
      }
    }

  return out;
  }

};
}

template <class TInputImage, class TOutputImage=TInputImage>
class ITK_EXPORT OuterProductVectorImageFilter :
  public
  UnaryFunctorImageFilter<TInputImage, TOutputImage,
                          Functor::OuterProduct<
                            typename TInputImage::PixelType,
                            typename TOutputImage::PixelType> >
{
public:
  typedef OuterProductVectorImageFilter Self;
  typedef UnaryFunctorImageFilter<
      TInputImage, TOutputImage,
      Functor::OuterProduct<
        typename TInputImage::PixelType, typename TOutputImage::PixelType> >             Superclass;

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

  itkNewMacro( Self );

  itkTypeMacro( OuterProductVectorImageFilter, UnaryFunctorImageFilter );

protected:
  OuterProductVectorImageFilter()
  {
  }

  virtual ~OuterProductVectorImageFilter()
  {
  }

  virtual void GenerateOutputInformation() ITK_OVERRIDE
    {
    Superclass::GenerateOutputInformation();
    typename TInputImage::ConstPointer inputPtr = this->GetInput();
    typename Superclass::OutputImagePointer outputPtr = this->GetOutput();
    int N = inputPtr->GetNumberOfComponentsPerPixel();
    outputPtr->SetNumberOfComponentsPerPixel(N*N);
    }


private:
  OuterProductVectorImageFilter(const Self &); // purposely not implemented
  void operator=(const Self &);                      // purposely not implemented
};
} // end namespace itk


#endif