mitkSegmentationInterpolationController.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 mitkSegmentationInterpolationController_h_Included
#define mitkSegmentationInterpolationController_h_Included

#include "mitkCommon.h"
#include "mitkImage.h"
#include <MitkSegmentationExports.h>

#include <itkImage.h>
#include <itkObjectFactory.h>

#include <map>
#include <vector>

namespace mitk
{
  class Image;

  class MITKSEGMENTATION_EXPORT SegmentationInterpolationController : public itk::Object
  {
  public:
    mitkClassMacroItkParent(SegmentationInterpolationController, itk::Object);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);

      static SegmentationInterpolationController *InterpolatorForImage(const Image *);

    void BlockModified(bool);

    void SetSegmentationVolume(const Image *segmentation);

    void SetReferenceVolume(const Image *segmentation);

    void SetChangedSlice(const Image *sliceDiff,
                         unsigned int sliceDimension,
                         unsigned int sliceIndex,
                         unsigned int timeStep);
    void SetChangedVolume(const Image *sliceDiff, unsigned int timeStep);

    Image::Pointer Interpolate(unsigned int sliceDimension,
                               unsigned int sliceIndex,
                               const mitk::PlaneGeometry *currentPlane,
                               unsigned int timeStep);

    void OnImageModified(const itk::EventObject &);

    void Activate2DInterpolation(bool);

    static SegmentationInterpolationController *GetInstance();

  protected:
    class MITKSEGMENTATION_EXPORT SetChangedSliceOptions
    {
    public:
      SetChangedSliceOptions(
        unsigned int sd, unsigned int si, unsigned int d0, unsigned int d1, unsigned int t, const void *pixels)
        : sliceDimension(sd), sliceIndex(si), dim0(d0), dim1(d1), timeStep(t), pixelData(pixels)
      {
      }

      unsigned int sliceDimension;
      unsigned int sliceIndex;
      unsigned int dim0;
      unsigned int dim1;
      unsigned int timeStep;
      const void *pixelData;
    };

    typedef std::vector<unsigned int> DirtyVectorType;
    // typedef std::vector< DirtyVectorType[3] > TimeResolvedDirtyVectorType; // cannot work with C++, so next line is
    // used for implementation
    typedef std::vector<std::vector<DirtyVectorType>> TimeResolvedDirtyVectorType;
    typedef std::map<const Image *, SegmentationInterpolationController *> InterpolatorMapType;

    SegmentationInterpolationController(); // purposely hidden
    ~SegmentationInterpolationController() override;

    template <typename DATATYPE>
    void ScanChangedSlice(const itk::Image<DATATYPE, 2> *, const SetChangedSliceOptions &options);

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

    template <typename DATATYPE>
    void ScanWholeVolume(const itk::Image<DATATYPE, 3> *, const Image *volume, unsigned int timeStep);

    void PrintStatus();

    TimeResolvedDirtyVectorType m_SegmentationCountInSlice;

    static InterpolatorMapType s_InterpolatorForImage;

    Image::ConstPointer m_Segmentation;
    Image::ConstPointer m_ReferenceImage;
    bool m_BlockModified;
    bool m_2DInterpolationActivated;
  };

} // namespace

#endif