GestureAPI.py

#!/usr/bin/python

import math
import numpy as np


class Gesture(object):

    def __init__(self, name):
        self.name = name

    def getName(self):
        return self.name

    def set_palm(self, hand_center, hand_radius):
        self.hand_center = hand_center
        self.hand_radius = hand_radius

    def set_finger_pos(self, finger_pos):
        self.finger_pos = finger_pos
        self.finger_count = len(finger_pos)

    def calc_angles(self):
        self.angle = np.zeros(self.finger_count, dtype=int)
        for i in range(self.finger_count):
            y = self.finger_pos[i][1]
            x = self.finger_pos[i][0]
            self.angle[i] = abs(math.atan2(
                (self.hand_center[1] - y), (x - self.hand_center[0])) * 180 / math.pi)


def DefineGestures():
    dict = {}
    # 1. BEGIN ------------------------------------#
    Zero = Gesture("Zero")
    Zero.set_palm((480, 230), 50)
    Zero.set_finger_pos([])
    Zero.calc_angles()
    dict[Zero.getName()] = Zero
    One = Gesture("One")
    One.set_palm((475, 225), 50)
    One.set_finger_pos([(475, 80)])
    One.calc_angles()
    dict[One.getName()] = One
    Two = Gesture("Two")
    Two.set_palm((475, 225), 45)
    Two.set_finger_pos([(490, 90), (415, 105)])
    Two.calc_angles()
    dict[Two.getName()] = Two
    Three = Gesture("Three")
    Three.set_palm((475, 225), 50)
    Three.set_finger_pos([(345, 200), (345, 200), (345, 200)])
    Three.calc_angles()
    dict[Three.getName()] = Three
    Five = Gesture("Five")
    Five.set_palm((475, 225), 50)
    Five.set_finger_pos(
        [(475, 80), (475, 80), (475, 80), (475, 80), (475, 80)])
    Five.calc_angles()
    dict[Five.getName()] = Five

    return dict


def CompareGestures(src1, src2):
    if(src1.finger_count == src2.finger_count):
        if(src1.finger_count == 0):
            result = src2.getName()
        elif(src1.finger_count == 1):
            result = src2.getName()
        elif(src1.finger_count == 2):
            result = src2.getName()
        elif(src1.finger_count == 3):
            result = src2.getName()
        elif(src1.finger_count == 5):
            result = src2.getName()

        else:
            angle_diff = []
            for i in range(src1.finger_count):
                angle_diff.append(src1.angle[i] - src2.angle[i])
            angle_score = max(angle_diff) - min(angle_diff)
            if(angle_score < 15):
                length_diff = []
                for i in range(src1.finger_count):
                    len1 = np.sqrt((src1.finger_pos[i][
                                   0] - src1.hand_center[0])**2 + (src1.finger_pos[i][1] - src1.hand_center[1])**2)
                    len2 = np.sqrt((src2.finger_pos[i][
                                   0] - src2.hand_center[0])**2 + (src2.finger_pos[i][1] - src2.hand_center[1])**2)
                    length_diff.append(len1 / len2)
                length_score = max(length_diff) - min(length_diff)
                if(length_score < 0.06):
                    result = src2.getName()
                else:
                    result = 0
            else:
                result = 0
    else:
        result = 0
    return result


def DecideGesture(src, GestureDictionary):
    result_list = []
    for k in GestureDictionary.keys():
        src2 = '"' + k + '"'
        result = CompareGestures(src, GestureDictionary[k])
        if(result != 0):
            return result
    return "NONE"