add controller

This commit is contained in:
Jing Hua
2023-10-24 02:26:24 +08:00
parent 14a4f5c008
commit a7144824d2
6 changed files with 61 additions and 13 deletions

17
controller.py Normal file
View File

@@ -0,0 +1,17 @@
import numpy as np
from PIL import Image
from simulator import EmulatorSimulator
class Controller:
def __init__(self, vm_name: str, username: str, password: str, host: str) -> None:
self.simulator = EmulatorSimulator(vm_name=vm_name, username=username,
password=password, host=host)
def get_state(self) -> np.ndarray:
image_path = self.simulator.get_screenshot()
with Image.open(image_path) as img:
return np.array(img)
def step(self) -> None:
self.simulator.keyboard_type("hello word")