mitkSliceBasedInterpolationController.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 mitkSliceBasedInterpolationController_h
#define mitkSliceBasedInterpolationController_h
 
#include "mitkLabelSetImage.h"
#include <MitkSegmentationExports.h>
 
#include <itkImage.h>
#include <itkObjectFactory.h>
 
#include <map>
#include <vector>
 
namespace mitk
{
  class Image;
 
  class MITKSEGMENTATION_EXPORT SliceBasedInterpolationController : public itk::Object
  {
  public:
    mitkClassMacroItkParent(SliceBasedInterpolationController, itk::Object);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
 
      static SliceBasedInterpolationController *InterpolatorForImage(const Image *);
 
    void SetWorkingImage(LabelSetImage *image);
 
    void SetReferenceImage(Image *image);
 
    void SetChangedSlice(const Image *image,
                         unsigned int sliceDimension,
                         unsigned int sliceIndex,
                         unsigned int timeStep);
 
    Image::Pointer Interpolate(unsigned int sliceDimension,
                               unsigned int sliceIndex,
                               const mitk::PlaneGeometry *currentPlane,
                               unsigned int timeStep);
 
    void ResetLabelCount();
 
  protected:
    class MITKSEGMENTATION_EXPORT SetChangedSliceOptions
    {
    public:
      SetChangedSliceOptions(unsigned int sd, unsigned int si, unsigned int d0, unsigned int d1, unsigned int t)
        : sliceDimension(sd), sliceIndex(si), dim0(d0), dim1(d1), timeStep(t)
      {
      }
 
      unsigned int sliceDimension;
      unsigned int sliceIndex;
      unsigned int dim0;
      unsigned int dim1;
      unsigned int timeStep;
      //        void* pixelData;
    };
 
    typedef std::vector<unsigned int> LabelCounterVectorType;
    typedef std::vector<LabelCounterVectorType> LabelCounterSliceVectorType;
    typedef std::vector<std::vector<LabelCounterSliceVectorType>> LabelCounterSliceTimeVectorType;
    typedef std::map<const Image *, SliceBasedInterpolationController *> InterpolatorMapType;
 
    SliceBasedInterpolationController(); // purposely hidden
    ~SliceBasedInterpolationController() override;
 
    template <typename PixelType>
    void ScanSliceITKProcessing(const itk::Image<PixelType, 2> *, const SetChangedSliceOptions &options);
 
    template <typename TPixel, unsigned int VImageDimension>
    void ScanImageITKProcessing(itk::Image<TPixel, VImageDimension> *, unsigned int timeStep);
 
    LabelCounterSliceTimeVectorType m_LabelCountInSlice;
 
    static InterpolatorMapType s_InterpolatorForImage;
 
    LabelSetImage::Pointer m_WorkingImage;
    Image::Pointer m_ReferenceImage;
  };
} // namespace
 
#endif