def __init__(self):
self.seed()
self.session = go_vncdriver.VNCSession()
self.session.connect("conn1",
address="localhost:5900",
password="******",
encoding="tight",
subscription=[(0, 1920, 0, 1080)])
self.tcpconnection = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.tcpconnection.connect(('localhost', 5005))
self.action_space = spaces.Discrete(4)
self.observation_space = spaces.Box(low=0,
high=255,
shape=(192, 256, 3),
dtype=np.uint8)
self.last_length = 10
import gc
import go_vncdriver
import time
session = go_vncdriver.VNCSession()
session.start_profile("/tmp/profile.pprof")
session.connect("conn1", address="127.0.0.1:5900", encoding="tight")
for i in range(10000):
# observations, infos, errors = session.step({"conn1": [("KeyEvent", 1, 2)]})
# session.render("conn1")
# print(observations, infos, errors)
# time.sleep(0.016)
observations, infos, errors = session.step({"conn1": [("KeyEvent", 1, 2)]})
# gc.collect()
# print(observations, infos, errors)
# session.connect("conn1", address="127.0.0.1:5900", encoding="tight")
session.end_profile()
# session.close("conn1")
# for i in range(10):
# observations, infos, errors = session.step({"conn2": [("KeyEvent", 1, 2)]})
# print(observations, infos, errors)
# time.sleep(0.016)
# session.render("first")
def __init__(self):
self.vnc_session = go_vncdriver.VNCSession()
self._configure()
import go_vncdriver
import time
from PIL import Image
from lab.envs.classic_control import rendering
# res = go_vncdriver.example(1, 2)
# res = go_vncdriver.connect(["hi", "tester"])
# import ipdb;ipdb.set_trace()
go_vncdriver.setup()
h = go_vncdriver.VNCSession(["localhost:5900"], None)
print(h.remotes)
viewer = rendering.SimpleImageViewer()
while True:
observation, info = h.flip()
if any(i['vnc.updates.n'] for i in info):
for ob in observation:
viewer.imshow(ob)
# Image.fromarray(ob).show()
# print(ob.shape)
time.sleep(1 / 10)
h.close()