mitkAutoMLSegmentationWithPreviewTool.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 MITK_AUTO_ML_SEGMENTATION_WITH_PREVIEW_TOOL_H
#define MITK_AUTO_ML_SEGMENTATION_WITH_PREVIEW_TOOL_H

#include "mitkAutoSegmentationWithPreviewTool.h"
#include "mitkDataNode.h"
#include "mitkLabelSetImage.h"

#include <MitkSegmentationExports.h>

namespace mitk
{
  class MITKSEGMENTATION_EXPORT AutoMLSegmentationWithPreviewTool : public AutoSegmentationWithPreviewTool
  {
  public:
    mitkClassMacro(AutoMLSegmentationWithPreviewTool, AutoSegmentationWithPreviewTool);

    void Activated() override;
    void Deactivated() override;

    using SelectedLabelVectorType = std::vector<Label::PixelType>;
    void SetSelectedLabels(const SelectedLabelVectorType& regions);
    SelectedLabelVectorType GetSelectedLabels() const;

    const LabelSetImage* GetMLPreview() const;

  protected:
    AutoMLSegmentationWithPreviewTool();
    ~AutoMLSegmentationWithPreviewTool() = default;

    void UpdateCleanUp() override;
    void DoUpdatePreview(const Image* inputAtTimeStep, const Image* oldSegAtTimeStep, Image* previewImage, TimeStepType timeStep) override;

    virtual LabelSetImage::Pointer ComputeMLPreview(const Image* inputAtTimeStep, TimeStepType timeStep) = 0;

  private:
    template <typename TPixel, unsigned int VImageDimension>
    void CalculateMergedSimplePreview(const itk::Image<TPixel, VImageDimension>* mlImage, mitk::Image* segmentation, unsigned int timeStep);

    SelectedLabelVectorType m_SelectedLabels = {};

    // holds the multilabel result as a preview image
    mitk::DataNode::Pointer m_MLPreviewNode;
    TimeStepType m_LastMLTimeStep = 0;
  };
}
#endif