mitkPlanarFigureMaskGenerator.h

Go to the documentation of this file.

#ifndef MITKPLANARFIGUREMASKGENERATOR
#define MITKPLANARFIGUREMASKGENERATOR

#include <MitkImageStatisticsExports.h>
#include <mitkImage.h>
#include <mitkPlanarFigure.h>
#include <itkImage.h>
#include <mitkMaskGenerator.h>
#include <vtkSmartPointer.h>
#include <itkVTKImageExport.h>
#include <itkVTKImageImport.h>
#include <vtkImageImport.h>
#include <vtkImageExport.h>

namespace mitk
{
class MITKIMAGESTATISTICS_EXPORT PlanarFigureMaskGenerator: public MaskGenerator
    {
    public:
    typedef PlanarFigureMaskGenerator           Self;
    typedef MaskGenerator                       Superclass;
    typedef itk::SmartPointer< Self >           Pointer;
    typedef itk::SmartPointer< const Self >     ConstPointer;

    itkNewMacro(Self)

    
    itkTypeMacro(PlanarFigureMaskGenerator, MaskGenerator)

    mitk::Image::Pointer GetMask();

    void SetPlanarFigure(mitk::PlanarFigure::Pointer planarFigure);

    mitk::Image::Pointer GetReferenceImage();

    void SetTimeStep(unsigned int timeStep);


    protected:
    PlanarFigureMaskGenerator():Superclass(){
        m_InternalMaskUpdateTime = 0;
        m_InternalMask = mitk::Image::New();
        m_ReferenceImage = nullptr;
    }


    private:
    void CalculateMask();

    template < typename TPixel, unsigned int VImageDimension >
    void InternalCalculateMaskFromPlanarFigure(
      const itk::Image< TPixel, VImageDimension > *image, unsigned int axis );

    mitk::Image::Pointer  extract2DImageSlice(unsigned int axis, unsigned int slice);

    bool GetPrincipalAxis(const BaseGeometry *geometry, Vector3D vector,
      unsigned int &axis );

    template <typename ITK_Exporter, typename VTK_Importer>
    void ConnectPipelines(ITK_Exporter exporter, vtkSmartPointer<VTK_Importer> importer)
    {
      importer->SetUpdateInformationCallback(exporter->GetUpdateInformationCallback());

      importer->SetPipelineModifiedCallback(exporter->GetPipelineModifiedCallback());
      importer->SetWholeExtentCallback(exporter->GetWholeExtentCallback());
      importer->SetSpacingCallback(exporter->GetSpacingCallback());
      importer->SetOriginCallback(exporter->GetOriginCallback());
      importer->SetScalarTypeCallback(exporter->GetScalarTypeCallback());

      importer->SetNumberOfComponentsCallback(exporter->GetNumberOfComponentsCallback());

      importer->SetPropagateUpdateExtentCallback(exporter->GetPropagateUpdateExtentCallback());
      importer->SetUpdateDataCallback(exporter->GetUpdateDataCallback());
      importer->SetDataExtentCallback(exporter->GetDataExtentCallback());
      importer->SetBufferPointerCallback(exporter->GetBufferPointerCallback());
      importer->SetCallbackUserData(exporter->GetCallbackUserData());
    }

    template <typename VTK_Exporter, typename ITK_Importer>
    void ConnectPipelines(vtkSmartPointer<VTK_Exporter> exporter, ITK_Importer importer)
    {
      importer->SetUpdateInformationCallback(exporter->GetUpdateInformationCallback());

      importer->SetPipelineModifiedCallback(exporter->GetPipelineModifiedCallback());
      importer->SetWholeExtentCallback(exporter->GetWholeExtentCallback());
      importer->SetSpacingCallback(exporter->GetSpacingCallback());
      importer->SetOriginCallback(exporter->GetOriginCallback());
      importer->SetScalarTypeCallback(exporter->GetScalarTypeCallback());

      importer->SetNumberOfComponentsCallback(exporter->GetNumberOfComponentsCallback());

      importer->SetPropagateUpdateExtentCallback(exporter->GetPropagateUpdateExtentCallback());
      importer->SetUpdateDataCallback(exporter->GetUpdateDataCallback());
      importer->SetDataExtentCallback(exporter->GetDataExtentCallback());
      importer->SetBufferPointerCallback(exporter->GetBufferPointerCallback());
      importer->SetCallbackUserData(exporter->GetCallbackUserData());
    }

    bool IsUpdateRequired() const;

    mitk::PlanarFigure::Pointer m_PlanarFigure;
    itk::Image<unsigned short, 2>::Pointer m_InternalITKImageMask2D;
    mitk::Image::Pointer m_InternalTimeSliceImage;
    mitk::Image::Pointer m_ReferenceImage;
    unsigned int m_PlanarFigureAxis;
    unsigned long m_InternalMaskUpdateTime;
    };
}

#endif // MITKPLANARFIGUREMASKGENERATOR