mitkBoundingShapeCropper.h

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

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

#ifndef BOUNDINGSHAPECROPPER_H
#define BOUNDINGSHAPECROPPER_H

#include "MitkBoundingShapeExports.h"
#include "mitkBoundingShapeCropper.h"
#include "mitkCommon.h"
#include "mitkGeometryData.h"
#include "mitkImageAccessByItk.h"
#include "mitkImageTimeSelector.h"
#include "mitkImageToImageFilter.h"

#include "itkImage.h"

namespace mitk
{
  //## @ingroup Process
  class MITKBOUNDINGSHAPE_EXPORT BoundingShapeCropper : public ImageToImageFilter
  {
  public:
    mitkClassMacro(BoundingShapeCropper, ImageToImageFilter);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);

      void SetGeometry(const mitk::GeometryData *geometry);
    // const mitk::GeometryData* GetGeometryData() const;

    itkSetMacro(OutsideValue, ScalarType);
    itkGetMacro(OutsideValue, ScalarType);
    itkSetMacro(UseWholeInputRegion, bool);
    itkGetMacro(UseWholeInputRegion, bool);

    itkSetMacro(CurrentTimeStep, ScalarType);
    itkGetMacro(CurrentTimeStep, ScalarType);
    itkSetMacro(UseCropTimeStepOnly, bool);
    itkGetMacro(UseCropTimeStepOnly, bool);

  protected:
    BoundingShapeCropper();
    ~BoundingShapeCropper() override;

    virtual const PixelType GetOutputPixelType();

    void GenerateInputRequestedRegion() override;
    void GenerateOutputInformation() override;
    void GenerateData() override;

    template <typename TPixel, unsigned int VImageDimension>
    void CutImage(itk::Image<TPixel, VImageDimension> *inputItkImage, int timeStep);

    virtual void ComputeData(mitk::Image *input3D, int boTimeStep);

    // virtual void ComputeData(mitk::LabelSetImage* image, int boTimeStep);

  private:
    mitk::GeometryData::Pointer m_Geometry;

    ScalarType m_OutsideValue;

    bool m_UseCropTimeStepOnly;

    int m_CurrentTimeStep;
    bool m_UseWholeInputRegion;
    mitk::ImageTimeSelector::Pointer m_InputTimeSelector;
    mitk::ImageTimeSelector::Pointer m_OutputTimeSelector;
    typedef itk::ImageRegion<5> RegionType;

    mitk::SlicedData::RegionType m_InputRequestedRegion;
    itk::TimeStamp m_TimeOfHeaderInitialization;
  };
} // namespace mitk

#endif /* BOUNDINGSHAPECROPPER_H */