예제 #1
0
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
예제 #3
0
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