Source code for aicssegmentation.workflow.segmenter_function

from enum import Enum
from dataclasses import dataclass
from typing import Dict, List, Union


class WidgetType(Enum):
    SLIDER = "slider"
    DROPDOWN = "drop-down"

    @staticmethod
    def from_str(value: str):
        if value is not None:
            value = value.lower()
        if value == WidgetType.SLIDER.value:
            return WidgetType.SLIDER
        if value == WidgetType.DROPDOWN.value:
            return WidgetType.DROPDOWN
        raise NotImplementedError()


@dataclass
class FunctionParameter:
    """
    Represents an input parameter to a segmenter function
    """

    name: str
    widget_type: WidgetType
    data_type: str
    min_value: Union[int, float] = None
    max_value: Union[int, float] = None
    increment: Union[int, float] = None
    options: List[str] = None


@dataclass
class SegmenterFunction:
    """
    Represents an aicssegmentation function.
    Functions are the smallest executable entity in a workflow and directly map
    to a python callable
    """

    name: str
    display_name: str
    function: str
    module: str
    parameters: Dict[str, List[FunctionParameter]] = None