def get_stop(): fo = open("stop.txt", "rw+") data = fo.read().split(',') list_of_gridpoints, mapd, rmapd = gp.grid(cv2.imread("cropped.png"), 9) d = 10000 dat = resource() dat.shape = "stop" dat.real_point = (int(data[0]), int(data[1])) for gridp in list_of_gridpoints: dist = distance(dat.real_point, gridp) if dist < d: d = dist dat.nearest_grid = gridp return dat
def get_start(): fo = open("start.txt", "r") data = fo.read().split(',') list_of_gridpoints, mapd, rmapd = gp.grid(cv2.imread("arena.jpg"), 9) d = 10000 dat = resource() dat.shape = "start" dat.real_point = (int(data[0]), int(data[1])) for gridp in list_of_gridpoints: dist = distance(dat.real_point, gridp) if dist < d: d = dist dat.nearest_grid = gridp fo.close() return dat
def get_resources(): fo = open("resources.txt", "r") data = fo.read() x = ast.literal_eval(data) list_of_obj_res = [] for i in x: list_of_shit = i.split(',') elem = resource() elem.shape = list_of_shit[0] elem.real_point = (int(list_of_shit[1]), int(list_of_shit[2])) list_of_obj_res.append(elem) list_of_gridpoints, mapd, rmapd = gp.grid(cv2.imread("cropped.png"), 9) for res in list_of_obj_res: d = 10000 for gridp in list_of_gridpoints: dist = distance(res.real_point, gridp) if dist < d: d = dist res.nearest_grid = gridp return list_of_obj_res