def loop(): global tstart global config global sock global tc print "connecting..." #sock = ctx.socket(zmq.REQ) #sock.connect("tcp://" + host_address + ":5556") print "... connected @" + host_address + ":" + "5556" tc = TDW_Client(host_address, initial_command='request_create_environment', description="test script", #selected_build='test_none.x86_64', # or skip to select from UI #queue_port_num="23402", get_obj_data=True, send_scene_info=True ) tc.load_config(config) tc.load_profile({'screen_width': 256, 'screen_height': 256}) sock = tc.run() #Receive answer #print "receiving initial answer..." #sock.recv() #for i in range(3): # msg = sock.recv() # print i #print "...received\n" #print "sending scene switch..." #handle_message(sock, write=True, outdir=path, prefix=str(0) + '_' + str(0)) #sock.send_json({"msg_type" : "CLIENT_JOIN", "get_obj_data" : True, "send_scene_info" : True}) #sock.send_json({"msg_type" : "CLIENT_JOIN_WITH_CONFIG", "config" : config, "get_obj_data" : True, "send_scene_info" : True}) #sock.send_json({'n': 4, 'msg': {"msg_type": "SCENE_SWITCH", "config": config, 'send_scene_info': True}}) #msg = {'n': 4, 'msg': {"msg_type": "CLIENT_INPUT", "get_obj_data": True, "actions": []}} #sock.send_json(msg) #print "...switch sent" ''' while True: print "waiting on messages..." msg1 = sock.recv() msg2 = sock.recv() msg3 = sock.recv() msg4 = sock.recv() print "...messages received\n\nsending input..." print msg1 img1 = Image.open(StringIO(msg2)).convert('RGB') img2 = Image.open(StringIO(msg3)).convert('RGB') img3 = Image.open(StringIO(msg4)).convert('RGB') img1.show() img2.show() img3.show() time.sleep(10) sock.send_json({"msg_type" : "CLIENT_INPUT", "vel": [0.0, 0.0, 0.0], "ang_vel" : [0.0, 0.0, 0.0], "teleport_random" : True, "get_obj_data" : True, "send_scene_info" : True}) print "...input sent" ''' #teleport=0 bn = 0 while True: print "waiting on messages" #msg = [sock.recv() for _ in range(4)] print "messages received" make_new_batch(bn) bn = bn + 1 end = time.time() FPS = number_of_frames / (end - tstart) print('--------TIME----------') print(end - tstart) print(number_of_frames) print('--------FPS-----------') print(FPS) print('----------------------')