def color_change_grid_search()

in roboschool/random_stadium.py [0:0]

• 25 lines of code
• 9 McCabe index (conditional complexity)

def color_change_grid_search(tr, x1, x2, y1, y2, W, H):
    tr.p = []
    tr.material_palette = ["stadium_white", "stadium_dirt", "stadium_grass"]

    kx = float(x2-x1) / W
    ky = float(y2-y1) / H

    for y in tqdm.tqdm(range(H)):
        ry = ky*y + y1
        rx_prev,c_prev = None,-1
        for x in np.arange(0,W,0.1):
            rx = kx*x + x1
            c = color(rx,ry)
            if c != c_prev and rx_prev != None:
                tr.p.append( find_color_change(rx_prev,ry,c_prev, rx,ry,c) )
            c_prev = c
            rx_prev = rx

    for x in tqdm.tqdm(range(W)):
        rx = kx*x + x1
        ry_prev,c_prev = None,-1
        for y in np.arange(0,H,0.1):
            ry = ky*y + y1
            c = color(rx,ry)
            if c != c_prev and ry_prev != None:
                tr.p.append( find_color_change(rx,ry_prev,c_prev, rx,ry,c) )
            c_prev = c
            ry_prev = ry