mitkTransferFunctionInitializer.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 mitkTransferFunctionInitializer_h
#define mitkTransferFunctionInitializer_h
 
#include <vtkColorTransferFunction.h>
#include <vtkPiecewiseFunction.h>
#include <vtkSmartPointer.h>
 
#include "mitkTransferFunction.h"
 
#include <itkObject.h>
#include <itkObjectFactory.h>
 
#include <MitkCoreExports.h>
 
namespace mitk
{
  class MITKCORE_EXPORT TransferFunctionInitializer : public itk::Object
  {
  public:
    mitkClassMacroItkParent(TransferFunctionInitializer, itk::Object);
 
    itkFactorylessNewMacro(Self);
 
    itkCloneMacro(Self)
      mitkNewMacro1Param(TransferFunctionInitializer, TransferFunction::Pointer);
 
    static void GetPresetNames(std::vector<std::string> &presetNames);
 
    void SetTransferFunction(TransferFunction::Pointer transferFunction);
    mitk::TransferFunction::Pointer GetTransferFunction();
    void SetTransferFunctionMode(int mode);
 
    void InitTransferFunctionMode();
 
  protected:
    TransferFunctionInitializer(TransferFunction::Pointer transferFunction = nullptr);
    ~TransferFunctionInitializer() override;
 
  private:
    int m_Mode;
    mitk::TransferFunction::Pointer m_transferFunction;
 
    // Define Transfer Function
    enum TransferFunctionMode
    {
      TF_CT_DEFAULT,
      TF_CT_BLACK_WHITE,
      TF_CT_THORAX_LARGE,
      TF_CT_THORAX_SMALL,
      TF_CT_BONE,
      TF_CT_BONE_GRADIENT,
      TF_CT_CARDIAC,
      TF_MR_GENERIC
    };
 
    // remove all old points
    void RemoveAllPoints();
    void SetModified();
    void SetCtDefaultMode();
    void SetCtBlackWhiteMode();
    void SetCtThoraxLargeMode();
    void SetCtThoraxSmallMode();
    void SetCtBoneMode();
    void SetCtBoneGradientMode();
    void SetCtCardiacMode();
    void SetMrGenericMode();
  };
}
 
#endif