LJnano/clitools/generators/trckr.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# -*- mode: Python -*-

'''

A Face tracker 
v0.1.0

Get all points fom redis /trckr/frame/WSclientID points

LICENCE : CC

by cocoa and Sam Neurohack

'''

from __future__ import print_function
import time
import argparse
import sys
import redis
import ast

name="generator::trckr"


def debug(*args, **kwargs):
    if( verbose == False ):
        return
    print(*args, file=sys.stderr, **kwargs)


argsparser = argparse.ArgumentParser(description="Face tracking generator")
argsparser.add_argument("-f","--fps",help="Frame Per Second",default=30,type=int)
argsparser.add_argument("-v","--verbose",action="store_true",help="Verbose output")
argsparser.add_argument("-i","--id",help="Trckr client ID",default="0",type=str)
argsparser.add_argument("-s","--server",help="redis server IP (127.0.0.1 by default)", type=str)
args = argsparser.parse_args()

fps=args.fps
verbose=args.verbose
idclient = args.id

if args.server:
    redisIP = args.server
else:
    redisIP = "127.0.0.1"



optimal_looptime = 1 / fps
debug(name+" optimal looptime "+str(optimal_looptime))

color = 65280


def rgb2int(rgb):
    return int('0x%02x%02x%02x' % tuple(rgb),0)

# Useful variables init.
white = rgb2int((255,255,255))
red = rgb2int((255,0,0))
blue = rgb2int((0,0,255))
green = rgb2int((0,255,0))

#
# Redis functions
#

r = redis.StrictRedis(host=redisIP , port=6379, db=0)

# read from redis key
def fromKey(keyname):

   return r.get(keyname)

# Write to redis key 
def toKey(keyname,keyvalue):
   return r.set(keyname,keyvalue)

#
# Trckr faces
# 

TrckrPts = [[159.39, 137.68], [155.12, 159.31], [155.56, 180.13], [159.81, 201.6], [170.48, 220.51], [187.46, 234.81], [208.4, 244.68], [229.46, 248.21], [246.44, 244.91], [259.69, 234.83], [270.95, 221.51], [278.54, 204.66], [283.53, 185.63], [286.27, 165.79], [284.72, 144.84], [280.06, 125.01], [274.35, 118.7], [260.71, 117.23], [249.52, 118.86], [182.04, 121.5], [193.63, 114.79], [210.24, 114.77], [222.35, 117.57], [190.6, 137.49], [203.59, 132.42], [214.75, 137.58], [203.04, 140.46], [203.32, 136.53], [272.45, 141.57], [263.33, 135.42], [250.31, 138.89], [262.15, 143.27], [261.99, 139.37], [235.82, 131.74], [221.87, 156.09], [213.66, 165.88], [219.28, 173.53], [236.3, 175.25], [249.02, 174.4], [254.22, 167.81], [248.83, 157.39], [237.94, 147.51], [227.01, 168.39], [245.68, 170.02], [204.94, 197.32], [217.56, 192.77], [228.27, 190.55], [234.66, 192.19], [240.47, 191.09], [247.96, 193.87], [254.52, 199.19], [249.35, 204.25], [242.74, 207.16], [233.2, 207.87], [222.13, 206.52], [212.44, 203.09], [220.34, 198.74], [233.31, 200.04], [244.0, 199.6], [244.27, 197.8], [233.81, 197.44], [220.88, 196.99], [239.57, 162.69], [196.52, 133.86], [210.2, 133.98], [209.43, 139.41], [196.59, 139.47], [268.99, 137.59], [256.36, 136.02], [255.95, 141.5], [267.9, 142.85]]
toKey('/trckr/frame/0',str(TrckrPts))

# get absolute face position points 
def getPART(TrckrPts, pose_points):

    dots = []
    #debug(pose_points)
    #debug(TrckrPts)
    for dot in pose_points:
      dots.append((TrckrPts[dot][0], TrckrPts[dot][1],0))
    #debug(dots)
    return dots


# Face keypoints
def face(TrckrPts):
    pose_points = [0,1,2,3,4,5,6,7,8,9,10,11,12,13,14]
    return getPART(TrckrPts, pose_points)

def browL(TrckrPts):
    pose_points = [15,16,17,18]
    return getPART(TrckrPts, pose_points)

def browR(TrckrPts):
    pose_points = [22,21,20,19]
    return getPART(TrckrPts, pose_points)

def eyeR(TrckrPts):
    pose_points = [25,64,24,63,23,66,26,65,25]
    return getPART(TrckrPts, pose_points)

def eyeL(TrckrPts):
    pose_points = [28,67,29,68,30,69,31,28]
    return getPART(TrckrPts, pose_points)

def pupR(TrckrPts):
    pose_points = [27]
    return getPART(TrckrPts, pose_points)

def pupL(TrckrPts):
    pose_points = [32]
    return getPART(TrckrPts, pose_points)


def nose1(TrckrPts):
    pose_points = [62,41,33]
    return getPART(TrckrPts, pose_points)

def nose2(TrckrPts):
    pose_points = [40,39,38,43,37,42,36,35,34]
    return getPART(TrckrPts, pose_points)

def mouth(TrckrPts):
    pose_points = [50,49,48,47,46,45,44,55,54,53,52,51,50]
    return getPART(TrckrPts, pose_points)

def mouthfull(TrckrPts):
    pose_points = [50,49,48,47,46,45,44,55,54,53,52,51,50,59,60,61,44,56,57,58,50]
    return getPART(TrckrPts, pose_points)


while True:

  start = time.time()
  shape =[]
  points = ast.literal_eval(fromKey('/trckr/frame/'+idclient).decode('ascii'))
  shape.append(browL(points))
  shape.append(eyeL(points))
  shape.append(browR(points))
  shape.append(eyeR(points))
  shape.append(pupL(points))
  shape.append(pupR(points))
  shape.append(nose1(points))
  shape.append(nose2(points))
  shape.append(mouthfull(points))
  line = str(shape)
  line = line.replace("(",'[')
  line = line.replace(")",']')
  line = "[{}]".format(line)
  print(line, flush=True);
  #debug(shape)
  #print(shape, flush=True);

  looptime = time.time() - start
  if( looptime < optimal_looptime ):
    time.sleep( optimal_looptime - looptime)
    debug(name+" micro sleep:"+str( optimal_looptime - looptime))