mitkAutoCropImageFilter.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 __mitkAutoCropImageFilter_h_
#define __mitkAutoCropImageFilter_h_

#include "MitkAlgorithmsExtExports.h"
#include "mitkCommon.h"
#include "mitkImageTimeSelector.h"
#include "mitkSubImageSelector.h"

#include <itkImage.h>
#include <itkImageRegion.h>

namespace mitk
{
  class MITKALGORITHMSEXT_EXPORT AutoCropImageFilter : public SubImageSelector
  {
  public:
    typedef itk::ImageRegion<3> RegionType;

    mitkClassMacro(AutoCropImageFilter, SubImageSelector);

    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      itkGetConstMacro(BackgroundValue, float);
    itkSetMacro(BackgroundValue, float);

    itkGetConstMacro(MarginFactor, float);
    itkSetMacro(MarginFactor, float);

    itkGetMacro(CroppingRegion, RegionType);

    // Use this method to manually set a region
    void SetCroppingRegion(RegionType overrideRegion);

    virtual const PixelType GetOutputPixelType();

  protected:
    // default constructor
    AutoCropImageFilter();

    // default destructor
    virtual ~AutoCropImageFilter();

    // This method calculates the actual smallest box
    void ComputeNewImageBounds();

    // Crops the image using the itk::RegionOfInterestImageFilter and creates the new output image
    template <typename TPixel, unsigned int VImageDimension>
    void ITKCrop3DImage(itk::Image<TPixel, VImageDimension> *inputItkImage, unsigned int timestep);

    // Here, the output image is initialized by the input and the newly calculated region
    virtual void GenerateOutputInformation() override;

    // Purposely not implemented
    virtual void GenerateInputRequestedRegion() override;

    // Crops the image on all time steps
    virtual void GenerateData() override;

    float m_BackgroundValue;

    RegionType m_CroppingRegion;

    float m_MarginFactor;

    typedef itk::Image<float, 3> ImageType;
    typedef ImageType::Pointer ImagePointer;

    RegionType::SizeType m_RegionSize;
    RegionType::IndexType m_RegionIndex;

    mitk::ImageTimeSelector::Pointer m_TimeSelector;

    mitk::SlicedData::RegionType m_InputRequestedRegion;
    itk::TimeStamp m_TimeOfHeaderInitialization;

    bool m_OverrideCroppingRegion;
  };

} // namespace mitk

#endif