sustaining_gazes/python_scripts/testing_head_pose.py

38 lines
777 B
Python

import time
def main():
import zmq
port = "5000"
context = zmq.Context()
socket = context.socket(zmq.SUB)
print "Collecting head pose updates..."
socket.connect ("tcp://localhost:%s" % port)
topic_filter = "HeadPose:"
socket.setsockopt(zmq.SUBSCRIBE, topic_filter)
while True:
head_pose = socket.recv()
head_pose = head_pose[9:].split(',')
X = float(head_pose[0])
Y = float(head_pose[1])
Z = float(head_pose[2])
pitch = float(head_pose[3])
yaw = float(head_pose[4])
roll = float(head_pose[5])
print 'X: %.1f, Y: %.1f, Z:%.1f, pitch: %.1f, yaw: %.1f, roll: %.1f' % (X, Y, Z, pitch, yaw, roll)
time.sleep(0.01)
if __name__ == '__main__':
main()