visualization.py

import pygame


class Visualization:
    def __init__(self):
        pygame.init()

    def draw(self, draw_info, algorithm_name, ascending) -> None:

        draw_info.window.fill(draw_info.BACKGROUND_COLOR)

        algorithm_title = draw_info.LARGE_FONT.render(
            f"{algorithm_name} - {"Ascending" if ascending else "Descending"}",
            1,
            draw_info.GREEN,
        )
        draw_info.window.blit(
            algorithm_title, (draw_info.width / 2 - algorithm_title.get_width() / 2, 65)
        )

        sorting = draw_info.SMALL_FONT.render(
            "S - Selection | X - Radix | M - Merge | H - Heap | C - Counting | B - Bubble | I - Insertion | K - Comb | L - Shell",
            1,
            draw_info.WHITE,
        )
        draw_info.window.blit(
            sorting, (draw_info.width / 2 - sorting.get_width() / 2, 35)
        )

        controls = draw_info.SMALL_FONT.render(
            "R - Reset | SPACE - Start | A - Ascending | D - Descending",
            1,
            draw_info.WHITE,
        )
        draw_info.window.blit(
            controls, (draw_info.width / 2 - controls.get_width() / 2, 5)
        )

        self.draw_list(draw_info)

        pygame.display.update()

    def draw_list(self, draw_info, color_positions={}, clear_bg=False) -> None:

        lst = draw_info.list

        if clear_bg:
            clear_rect = (
                draw_info.SIDE_PADDING // 2,
                draw_info.TOP_PADDING,
                draw_info.width - draw_info.SIDE_PADDING,
                draw_info.height - draw_info.TOP_PADDING,
            )

            pygame.draw.rect(draw_info.window, draw_info.BACKGROUND_COLOR, clear_rect)

        for i, value in enumerate(lst):
            X = draw_info.start_x + i * draw_info.block_width
            Y = (
                draw_info.height
                - (value - draw_info.min_value) * draw_info.block_height
            )

            colors = draw_info.GRADIENT[i % len(draw_info.GRADIENT)]

            if i in color_positions:
                colors = color_positions[i]

            pygame.draw.rect(
                draw_info.window,
                colors,
                (X, Y, draw_info.block_width, draw_info.height),
            )

        if clear_bg:
            pygame.display.update()