mitkTransferFunctionInitializer.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#ifndef MITK_TRANSFER_FUNCTION_MODE_CREATOR_H_HEADER_INCLUDED
#define MITK_TRANSFER_FUNCTION_MODE_CREATOR_H_HEADER_INCLUDED

#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);
    virtual ~TransferFunctionInitializer();

  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 /* MITK_TRANSFER_FUNCTION_MODE_CREATOR_H_HEADER_INCLUDED */