mitkROIBasedParameterFitImageGenerator.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_ROI_BASED_PARAMETER_FIT_IMAGE_GENERATOR_H_
#define __MITK_ROI_BASED_PARAMETER_FIT_IMAGE_GENERATOR_H_

#include <map>

#include <mitkImage.h>

#include "mitkModelParameterizerBase.h"
#include "mitkModelFitFunctorBase.h"
#include "mitkParameterFitImageGeneratorBase.h"

#include "MitkModelFitExports.h"

namespace mitk
{

class MITKMODELFIT_EXPORT ROIBasedParameterFitImageGenerator: public ParameterFitImageGeneratorBase
{
public:
  mitkClassMacro(ROIBasedParameterFitImageGenerator, ParameterFitImageGeneratorBase);

  itkNewMacro(Self);

    using ParameterImagePixelType = ScalarType;
    using FunctorValueArrayType = std::vector<ParameterImagePixelType>;

    using FitFunctorType = ModelFitFunctorBase;

    using ParameterizerType = ModelParameterizerBase;

    using ModelBaseType = ParameterFitImageGeneratorBase::ModelBaseType;
    using ParameterNameType = ParameterFitImageGeneratorBase::ParameterNameType;
    using ParameterImageMapType = ParameterFitImageGeneratorBase::ParameterImageMapType;

    using TimeGridType = ModelBaseType::TimeGridType;
    using SignalType = ModelBaseType::ModelResultType;

    itkSetObjectMacro(Mask, Image);
    itkGetConstObjectMacro(Mask, Image);

    itkGetConstReferenceMacro(Signal, SignalType);
    itkSetMacro(Signal, SignalType);

    itkGetConstReferenceMacro(TimeGrid, TimeGridType);
    itkSetMacro(TimeGrid, TimeGridType);

    itkSetObjectMacro(FitFunctor, FitFunctorType);
    itkGetObjectMacro(FitFunctor, FitFunctorType);

    itkSetObjectMacro(ModelParameterizer, ParameterizerType);
    itkGetObjectMacro(ModelParameterizer, ParameterizerType);

    double GetProgress() const override;

    ParameterNamesType GetParameterNames() const override;

    ParameterNamesType GetDerivedParameterNames() const override;

    ParameterNamesType GetCriterionNames() const override;

    ParameterNamesType GetEvaluationParameterNames() const override;

protected:
  ROIBasedParameterFitImageGenerator() : m_Progress(0)
  {
    m_Mask = nullptr;
  };

  ~ROIBasedParameterFitImageGenerator() override = default;

    template <typename TPixel, unsigned int VDim>
    void DoImageGeneration(itk::Image<TPixel, VDim>* image, double value);

    void onFitProgressEvent(::itk::Object* caller, const ::itk::EventObject& eventObject);

    bool HasOutdatedResult() const override;
    void CheckValidInputs() const override;
    void DoFitAndGetResults(ParameterImageMapType& parameterImages, ParameterImageMapType& derivedParameterImages, ParameterImageMapType& criterionImages, ParameterImageMapType& evaluationParameterImages) override;

private:
    Image::Pointer m_Mask;
    SignalType m_Signal;
    TimeGridType m_TimeGrid;

    FitFunctorType::Pointer m_FitFunctor;

    Image::Pointer m_TempResultImage;

    ParameterizerType::Pointer m_ModelParameterizer;

    double m_Progress;
};

}

#endif // __MITK_PARAMETER_FIT_IMAGE_GENERATOR_H_