Source code for luna_handlers.sdk.sdk_loop.enums

"""
Helpers enums.
"""
from enum import Enum


class MultifacePolicy(Enum):
    """
    Multiple face detection policy enum.
    """

    notAllowed = 0  #: multiple face detection not allowed
    allowed = 1  #: multiple face detection allowed
    getBest = 2  #: get best detection from the image


class Estimators(Enum):
    """
    Available loop estimators
    """

    imageOrientation = "imageOrientation"  #: image orientation

    faceDetection = "faceDetection"  #: face detection
    faceWarp = "faceWarp"  #: face warp
    gazeDirection = "gaze"  #: gaze  direction
    headPose = "headPose"  #: head pose
    eyes = "eyes"  #: eyes state, points
    mouthState = "mouthState"  #: mouth state
    quality = "warpQuality"  #: warp quality
    emotions = "emotions"  #: emotions
    faceDescriptor = "faceDescriptor"  #: face descriptor
    basicAttributes = "basicAttributes"  #: basic attributes (age, gender, ethnicity)
    mask = "mask"  #: mask
    glasses = "glasses"  #: glasses
    livenessV1 = "livenessV1"  #: livenessv1
    imageColorType = "image_color_type"  #: image color type
    headwear = "headwear"  #: headwear
    faceNaturalLight = "face_natural_light"  #: face natural light
    fisheye = "fisheye"  #: fisheye
    eyebrowExpression = "eyebrow_expression"  #: eyebrow expression
    redEyes = "red_eyes"  #: red-eyes
    faceDetectionBackground = "face_detection_background"  #: face detection background

    bodyDetection = "bodyDetection"  #: body detection
    bodyWarp = "bodyWarp"  #: body warp
    bodyDescriptor = "bodyDescriptor"  #: body descriptors
    bodyAttributes = "bodyAttributes"  #: body attributes


class LoopEstimations(Enum):
    """
    SDK Loop estimations enum. Name - estimation target.
    """

    exif = "exif"

    imageOrientation = "image_orientation"  #: image orientation

    faceDetection = "face_detection"  #: face detection
    faceWarp = "face_warp"  #: face warp
    faceLandmarks68 = "face_landmarks68"  #: face landmarks68
    gaze = "gaze"  #: gaze direction
    headPose = "head_pose"  #: head pose
    eyes = "eyes"  #: eyes state, points
    mouthAttributes = "mouth_state"  #: mouth state
    faceWarpQuality = "face_warp_quality"  #: face warp quality
    emotions = "emotions"  #: emotions
    faceDescriptor = "face_descriptor"  #: face descriptor
    basicAttributes = "basic_attributes"  #: basic attributes (age, gender, ethnicity)
    mask = "mask"  #: mask
    glasses = "glasses"  #: glasses
    livenessV1 = "livenessV1"  #: livenessV1
    imageColorType = "image_color_type"  #: image color type
    headwear = "headwear"  #: headwear
    faceNaturalLight = "face_natural_light"  #: face natural light
    fisheye = "fisheye"  #: fisheye
    eyebrowExpression = "eyebrow_expression"  #: eyebrow expression
    redEyes = "red_eyes"  #: red-eyes
    # todo
    # faceDetectionBackground = "face_detection_background"  #: face detection background

    bodyWarp = "body_warp"  #: body warp
    bodyDetection = "body_detection"  #: body detection
    bodyDescriptor = "body_descriptor"  #: body descriptor
    bodyLandmarks17 = "body_landmarks17"  #: body landmarks
    bodyAttributes = "body_attributes"  #: body attributes