eyetracking/awesomeClient.py at master · ncheney/eyetracking · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
#! /usr/bin/env python

from socket import *
from time import sleep
import re, sys, random
import subprocess

from LoggedProcess import LoggedProcess
from helper import getLocalIP


def simpleListenLoop(proc, tcpCliSock):
    # Eye-tracker API specific
    tcpCliSock.send(str.encode('<SET ID="ENABLE_SEND_POG_FIX" STATE="1" />\r\n"'))
    tcpCliSock.send(str.encode('<SET ID="ENABLE_SEND_DATA" STATE="1" />\r\n"'))

    # Loop forever
    RE_STRING = '<REC FPOGX="([-0-9\.]+)" FPOGY="([-0-9\.]+)" FPOGS="[0-9\.]+" FPOGD="[0-9\.]+" FPOGID="[0-9]+" FPOGV="([01])"/>'
    ii = 0
    while True:
        ii += 1
        data = tcpCliSock.recv(1024)
        st = bytes.decode(data)
        #print 'got something', st

        results = re.search(RE_STRING, st)

        if results is None:
            print 'Could not parse "%s", skipping' % st
            continue
        #print 'results is', results

        rawXX       = float(results.group(1))
        rawYY       = float(results.group(2))
        validEye = int(results.group(3))

        xx = rawXX
        yy = 1-rawYY

        idxY = int(yy * 3)
        xx = (xx-.1) * (1.0/.8)  # rescale hack....
        idxX = int(xx * 5)
        validIdx = idxY in [0, 1, 2] and idxX in [0, 1, 2, 3, 4]
        idx = idxY * 5 + idxX if validIdx else -1

#        print '%d (%.3f, %.3f) %d (%d, %d) %d' % (validEye, xx, yy, validIdx, idxX, idxY, idx)

        sendPositionEvery = 5
        if ii % sendPositionEvery == 0:
            # Send bin
            #proc.write('%d\n' % idx)
            # Send x, y (in real x, y order)
            proc.write('%f %f\n' % (rawXX, rawYY))
            print('%f %f\n' % (rawXX, rawYY))

        #print proc._quit, dir(proc._process)
        #print
        #print
        if(proc._quit):
            print 'QUITTING'
            break
    tcpCliSock.close()


def main():
#    args = ('./v2Hyperneat', '-O', 'run', '-R', '102', '-I', 'ShapesExperiment.dat', '-XG', '5000', '-X', 'einstein.xml')

# with random seed set
#    args = ('./v2Hyperneat', '-O', 'run', '-R', '102', '-I', 'ShapesExperiment.dat')
#without random seed set

    #args = ('./Hypercube_NEAT', '-O', 'run', '-R', str(random.randint(0,1000)), '-I' ,'ShapesExperiment.dat')#, '-K', '1')
    args = ['./Hypercube_NEAT'] + sys.argv[1:]

    print args

    proc = LoggedProcess(args         = args,                         logStdOut    = 'stdout.log',                         logStdErr    = 'stderr.log')
    #subprocess.Popen(args,shell=1)

    print 'Started Hyperneat proc!'
    #for ii in range(10):
    #    print 'hi %d' % ii
    #    proc.write('hi %d\n' % ii)
    #    sleep(1)

#    if len(sys.argv) > 1:
 #       HOST = sys.argv[1]
  #  else:
        #HOST = '128.253.224.134'
        #HOST = '128.84.62.237'
        #HOST = '10.211.55.3'
        #HOST = '192.168.0.115'
        #HOST = getLocalIP()
    HOST = '128.253.224.94'

    PORT = 4242
    BUFSIZ = 1024
    ADDR = (HOST, PORT)

    print 'Attempting to connect to %s...' % repr(ADDR),
    sys.stdout.flush()
    tcpCliSock = socket(AF_INET, SOCK_STREAM)
    tcpCliSock.connect(ADDR)
    print 'connected!'

    print 'Starting loop'
    simpleListenLoop(proc, tcpCliSock)


if __name__ == '__main__':
    main()