mitknnInteractiveEnums.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 mitknnInteractiveEnums_h
#define mitknnInteractiveEnums_h
 
#include <mitkColorProperty.h>
#include <MitkPythonSegmentationExports.h>
 
namespace mitk::nnInteractive
{
  enum class InteractionType
  {
    Point,    
    Box,      
    Scribble, 
    Lasso     
  };
 
  MITKPYTHONSEGMENTATION_EXPORT const std::string& GetInteractionTypeAsString(InteractionType interactionType);
 
  MITKPYTHONSEGMENTATION_EXPORT const std::array<InteractionType, 4>& GetAllInteractionTypes();
 
  enum class PromptType
  {
    Positive, 
    Negative  
  };
 
  MITKPYTHONSEGMENTATION_EXPORT const std::string& GetPromptTypeAsString(PromptType promptType);
 
  MITKPYTHONSEGMENTATION_EXPORT const std::array<PromptType, 2>& GetAllPromptTypes();
 
  enum class Backend
  {
    CUDA, 
    CPU   
  };
 
  MITKPYTHONSEGMENTATION_EXPORT const std::string& GetBackendAsString(Backend backend);
 
  MITKPYTHONSEGMENTATION_EXPORT const std::array<Backend, 2>& GetAllBackends();
 
  enum class ColorIntensity
  {
    Muted,  
    Vibrant 
  };
 
  MITKPYTHONSEGMENTATION_EXPORT const Color& GetColor(PromptType promptType, ColorIntensity colorIntensity);
}
 
#endif