# -*- coding: utf-8 -*-

TIME = "Time"

HEAD_X = "Head_x"
HEAD_Y = "Head_y"
HEAD_Z = "Head_z"
HEAD_POS = [
    HEAD_X,
    HEAD_Y,
    HEAD_Z
]

HEAD_ROTW = "Head_rotw"
HEAD_ROTX = "Head_rotx"
HEAD_ROTY = "Head_roty"
HEAD_ROTZ = "Head_rotz"
HEAD_ROT = [
    HEAD_ROTW,
    HEAD_ROTX,
    HEAD_ROTY,
    HEAD_ROTZ
]

HEAD = HEAD_POS + HEAD_ROT

CONTROLLING_HAND_X = "ControllingHand_x"
CONTROLLING_HAND_Y = "ControllingHand_y"
CONTROLLING_HAND_Z = "ControllingHand_z"
CONTROLLING_HAND_POS = [
    CONTROLLING_HAND_X,
    CONTROLLING_HAND_Y,
    CONTROLLING_HAND_Z
]

CONTROLLING_HAND_ROTW = "ControllingHand_rotw"
CONTROLLING_HAND_ROTX = "ControllingHand_rotx"
CONTROLLING_HAND_ROTY = "ControllingHand_roty"
CONTROLLING_HAND_ROTZ = "ControllingHand_rotz"
CONTROLLING_HAND_ROT = [
    CONTROLLING_HAND_ROTW,
    CONTROLLING_HAND_ROTX,
    CONTROLLING_HAND_ROTY,
    CONTROLLING_HAND_ROTZ
]

CONTROLLING_HAND = CONTROLLING_HAND_POS + CONTROLLING_HAND_ROT

SECONDARY_HAND_X = "SecondaryHand_x"
SECONDARY_HAND_Y = "SecondaryHand_y"
SECONDARY_HAND_Z = "SecondaryHand_z"
SECONDARY_HAND_POS = [
    SECONDARY_HAND_X,
    SECONDARY_HAND_Y,
    SECONDARY_HAND_Z
]

SECONDARY_HAND_ROTW = "SecondaryHand_rotw"
SECONDARY_HAND_ROTX = "SecondaryHand_rotx"
SECONDARY_HAND_ROTY = "SecondaryHand_roty"
SECONDARY_HAND_ROTZ = "SecondaryHand_rotz"
SECONDARY_HAND_ROT = [
    SECONDARY_HAND_ROTW,
    SECONDARY_HAND_ROTX,
    SECONDARY_HAND_ROTY,
    SECONDARY_HAND_ROTZ
]

SECONDARY_HAND = SECONDARY_HAND_POS + SECONDARY_HAND_ROT

BODY = HEAD + CONTROLLING_HAND + SECONDARY_HAND

GRIP = "Grip"

ENVIRONMENT = [GRIP]

GESTURE_CLASS = "Gesture"
GESTURE_PREDICTION = "PredictedGesture"

INDEX = TIME
DATA_COLUMNS = HEAD + CONTROLLING_HAND + SECONDARY_HAND + ENVIRONMENT
COLUMNS = DATA_COLUMNS + [GESTURE_CLASS]


# Sword net

SWORD_HILT_X = "SwordHilt_x"
SWORD_HILT_Y = "SwordHilt_y"
SWORD_HILT_Z = "SwordHilt_z"
SWORD_HILT_POS = [
    SWORD_HILT_X,
    SWORD_HILT_Y,
    SWORD_HILT_Z
]

SWORD_TIP_X = "SwordTip_x"
SWORD_TIP_Y = "SwordTip_y"
SWORD_TIP_Z = "SwordTip_z"
SWORD_TIP_POS = [
    SWORD_TIP_X,
    SWORD_TIP_Y,
    SWORD_TIP_Z
]

SWORD_FEATURES = HEAD + SWORD_HILT_POS + SWORD_TIP_POS
