freRegionImageMediaController.h

Go to the documentation of this file.

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

  Program:   F.R.E.E. - flexible registration evaluation engine
  Version:   v.1.0.0
  Date:      $Date: 2006/09/01 12:00:00 $
  Module:    $RCSfile: freRegionImageMediaController.h,v $
  Language:  C++



  Copyright (c) 2007 Ralf o Floca (Department of Medical Informatics,
  Institute for Medical Biometry and Informatics, University of Heidelberg,
  Germany). All rights reserved.
  See FREECopyright.txt or http://www.mi.med.uni-hd.de/free/copyright.htm
  for details.

     This software is distributed WITHOUT ANY WARRANTY; without even 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#ifndef __freRegionImageMediaController_h
#define __freRegionImageMediaController_h

#include "freIntensityImageMediaController.h"
#include "freImageIO.h"
#include "freImageTypes.h"

#include "itkNearestNeighborInterpolateImageFunction.h"

namespace FREE
{

        freControllerIDMacro(RegionImageMediaControllerBase, "RegionImageMediaControllerBase");
template <class TInternalPixel, class TExternalPixel, unsigned int VDimension>
class RegionImageMediaControllerBase : public IntensityImageMediaControllerBase< TInternalPixel, TExternalPixel, VDimension >
{
public:  
  typedef IntensityImageMediaControllerBase<TInternalPixel, TExternalPixel, VDimension> Superclass;
  typedef typename Superclass::ComponentType ComponentType;
  typedef typename Superclass::GenericComponentType GenericComponentType;
  typedef typename Superclass::GenericComponentPointer GenericComponentPointer;
  typedef typename Superclass::GenericMediaPointer GenericMediaPointer;
  typedef ComponentType ImageType;

        itkTypeMacro(RegionImageMediaControllerBase, IntensityImageMediaControllerBase);

  RegionImageMediaControllerBase()
  {
    //Profile settings
    this->UpdateControllerID(ControllerID::RegionImageMediaControllerBase);
    this->m_Description = "Base class for medias containing region/mask images.";
        };


  virtual ~RegionImageMediaControllerBase() {};

  virtual GenericMediaPointer ComputeActualizedMedia(GenericMediaType* pGenericMedia,
                                                     GenericMediaType* pGenericTransformationField) const
  {
    if (!pGenericMedia) throwCtrlExceptionMacro("","Passed media is NULL.");
    if (!pGenericTransformationField) throwCtrlExceptionMacro("","Passed transform field is NULL.");

    typename ImageType::Pointer smpCurrentImage;
    try
    {
      smpCurrentImage = dynamic_cast<ImageType*>(pGenericMedia);
    }
    catchAllNPassMacro("Error. Media that should be actualized seems not to be of correct type.");
    if (smpCurrentImage.IsNull()) throwCtrlExceptionMacro("","Media has wrong type. Cannot be casted.");

    typedef typename ImageTypes<VDimension>::TransformationFieldType TransformationFieldType;
    typename TransformationFieldType::Pointer smpTransfromField = static_cast< TransformationFieldType*>(pGenericTransformationField);

    typedef itk::WarpImageFilter< ImageType, ImageType, TransformationFieldType > WarpFilterType;

    typename WarpFilterType::Pointer smpWarpFilter = WarpFilterType::New();

    typedef itk::NearestNeighborInterpolateImageFunction<ImageType, ScalarType> InterpolatorType;

    typename InterpolatorType::Pointer interpolator = InterpolatorType::New();

    smpWarpFilter->SetInput(smpCurrentImage);
    smpWarpFilter->SetOutputSpacing(smpCurrentImage->GetSpacing());
    smpWarpFilter->SetOutputOrigin(smpCurrentImage->GetOrigin());
    smpWarpFilter->SetDeformationField(smpTransfromField);
    smpWarpFilter->SetInterpolator(interpolator);
    smpCurrentImage = smpWarpFilter->GetOutput();
    smpWarpFilter->Update();

    GenericMediaPointer smpResult = smpCurrentImage.GetPointer();

        return smpResult;
  };

  virtual GenericMediaPointer ComputeActualizedMediaByFunction(GenericMediaType* pGenericMedia,
                                                               GenericMediaType* pGenericTransformationFunction) const
  {
    if (!pGenericMedia) throwCtrlExceptionMacro("","Passed media is NULL.");
    if (!pGenericTransformationFunction) throwCtrlExceptionMacro("","Passed transform function is NULL.");

    typename ImageType::Pointer smpCurrentImage;
    try
    {
      smpCurrentImage = dynamic_cast<ImageType*>(pGenericMedia);
    }
    catchAllNPassMacro("Error. Media that should be actualized seems not to be of correct type.");
    if (smpCurrentImage.IsNull()) throwCtrlExceptionMacro("","Media has wrong type. Cannot be casted.");

    typedef TransformFunctionMedia<ScalarType, VDimension, VDimension> TransformationFunctionType;
    typename TransformationFunctionType::Pointer smpTransfromFunction = static_cast< TransformationFunctionType*>(pGenericTransformationFunction);

    typedef itk::ResampleImageFilter< ImageType, ImageType, ScalarType > ResampleFilterType;

    typename ResampleFilterType::Pointer smpResampleFilter = ResampleFilterType::New();

    typedef itk::NearestNeighborInterpolateImageFunction<ImageType, ScalarType> InterpolatorType;

    typename InterpolatorType::Pointer interpolator = InterpolatorType::New();

    smpResampleFilter->SetInput(smpCurrentImage);
    smpResampleFilter->SetOutputParametersFromImage(smpCurrentImage);
    smpResampleFilter->SetOutputStartIndex(smpCurrentImage->GetLargestPossibleRegion().GetIndex());
    smpResampleFilter->SetTransform(smpTransfromFunction->GetTransform());
    smpResampleFilter->SetInterpolator(interpolator);
    smpCurrentImage = smpResampleFilter->GetOutput();
    smpResampleFilter->Update();

    GenericMediaPointer smpResult = smpCurrentImage.GetPointer();

        return smpResult;
  };
};

freControllerIDMacro(RegionImage2DMediaController, "Region Image 2D Media");
class RegionImage2DMediaController : public RegionImageMediaControllerBase< InternalImagePixelType, LoadingPixelType, 2>
{
public:  
  typedef RegionImageMediaControllerBase<InternalImagePixelType, LoadingPixelType, 2> Superclass;
        typedef Superclass::ComponentType ComponentType;

        itkTypeMacro(RegionImage2DMediaController, RegionImageMediaControllerBase);

  RegionImage2DMediaController();

  virtual ~RegionImage2DMediaController();
};

freControllerIDMacro(RegionImage3DMediaController, "Region Image 3D Media");
class RegionImage3DMediaController : public RegionImageMediaControllerBase< InternalImagePixelType, LoadingPixelType, 3>
{
public:  
  typedef RegionImageMediaControllerBase<InternalImagePixelType, LoadingPixelType, 3> Superclass;
        typedef Superclass::ComponentType ComponentType;

        itkTypeMacro(RegionImage3DMediaController, RegionImageMediaControllerBase);

  RegionImage3DMediaController();

  virtual ~RegionImage3DMediaController();
};

} //end of namespace free

#endif

---