voronoi.py 文件源码

def testVoronoi():

    size,polygons,poly_site,river_list,city_sites,road_list,regions = setup(main_island_shape_seed=None)

    pygame.init()
    pygame.font.init()
    city_font = pygame.font.SysFont('cardinal', 20) #cardinal
    region_font = pygame.font.SysFont('cardinal', 30)

    GAMEOVER = False
    clock = pygame.time.Clock()

    screen = pygame.display.set_mode(size,pygame.RESIZABLE)
    screen.fill((255,255,255))
    pygame.display.set_caption('Voronoi')
    pygame.draw.rect(screen,(255,0,0),[384,320,32,32],0)
    curr = Floating(384,320)

    poly = polyDict = False
    count = 0
    first = False
    pre_render(polygons,poly_site)
    render_rivers(river_list)
    render_roads(road_list)
    bg = v_fx(screen)
    while not GAMEOVER:
        # --- Main event loop for event handling
        action = False
        for e in pygame.event.get(): # User did something
            action = True
            if e.type == pygame.QUIT: # If user clicked close
                GAMEOVER = True
            elif e.type == pygame.MOUSEBUTTONUP:
                x,y = pygame.mouse.get_pos()
                found = False
                """
                for p in polygons:
                    if p.inside_polygon(x,y) and not found:
                        pts = [i.get_cords() for i in p.vertices]
                        pygame.draw.polygon(screen,(40,40,40),pts)
                        found = True
                    else:
                        p.draw()
                """
        if action:
            for p in polygons:
                p.draw()
            render_rivers(river_list)
            render_roads(road_list)
            render_cities(city_sites,city_font)
            render_regions(regions,region_font)

        pygame.display.update()
        #print(pygame.image.tostring(screen,"RGB"))
        # --- Limit to 5 frames per second
        clock.tick(5)

        if not polyDict:
            polyDict = not polyDict

    pygame.quit()