HexDecimal / python-tdl

def render_bsp(first, key, mouse):

    global bsp, bsp_generate, bsp_refresh, bsp_map

    global bsp_random_room, bsp_room_walls, bsp_depth, bsp_min_room_size

    if bsp_generate or bsp_refresh:

        # dungeon generation

        if bsp is None:

            # create the bsp

            bsp = libtcod.bsp_new_with_size(0, 0, SAMPLE_SCREEN_WIDTH,

                                            SAMPLE_SCREEN_HEIGHT)

        else:

            # restore the nodes size

            libtcod.bsp_resize(bsp, 0, 0, SAMPLE_SCREEN_WIDTH,

                               SAMPLE_SCREEN_HEIGHT)

        bsp_map = list()

        for x in range(SAMPLE_SCREEN_WIDTH):

            bsp_map.append([False] * SAMPLE_SCREEN_HEIGHT)

        if bsp_generate:

            # build a new random bsp tree

            libtcod.bsp_remove_sons(bsp)

            if bsp_room_walls:

                libtcod.bsp_split_recursive(bsp, 0, bsp_depth,

                                            bsp_min_room_size + 1,

                                            bsp_min_room_size + 1, 1.5, 1.5)

            else:

                libtcod.bsp_split_recursive(bsp, 0, bsp_depth,

                                            bsp_min_room_size,

                                            bsp_min_room_size, 1.5, 1.5)

        # create the dungeon from the bsp

        libtcod.bsp_traverse_inverted_level_order(bsp, traverse_node)

        bsp_generate = False

        bsp_refresh = False

    libtcod.console_clear(sample_console)

    libtcod.console_set_default_foreground(sample_console, libtcod.white)

    rooms = 'OFF'

            if left.y + left.h - 1 < right.y or right.y + right.h - 1 < left.y:

                # no overlapping zone. we need a Z shaped corridor

                y1 = libtcod.random_get_int(None, left.y, left.y + left.h - 1)

                y2 = libtcod.random_get_int(None,

                                            right.y, right.y + right.h - 1)

                x = libtcod.random_get_int(None, left.x + left.w, right.x)

                hline_left(bsp_map, x - 1, y1)

                vline(bsp_map, x, y1, y2)

                hline_right(bsp_map, x + 1, y2)

            else:

                # straight horizontal corridor

                miny = max(left.y, right.y)

                maxy = min(left.y + left.h - 1, right.y + right.h - 1)

                y = libtcod.random_get_int(None, miny, maxy)

                hline_left(bsp_map, right.x - 1, y)

                hline_right(bsp_map, right.x, y)

    return True

bsp = None

bsp_generate = True

bsp_refresh = False

class BSPSample(Sample):

    def __init__(self):

        self.name = 'Bsp toolkit'

    def on_draw(self, delta_time):

        global bsp, bsp_generate, bsp_refresh, bsp_map

        global bsp_random_room, bsp_room_walls, bsp_depth, bsp_min_room_size

        if bsp_generate or bsp_refresh:

            # dungeon generation

            if bsp is None:

                # create the bsp

                bsp = libtcod.bsp_new_with_size(0, 0, SAMPLE_SCREEN_WIDTH,

                                                SAMPLE_SCREEN_HEIGHT)