vis.py

from math import sin, cos, radians, pi
import os, sys, time, errno, pygame, random, platform
import ard

pygame.init()
info            = pygame.display.Info()
clock           = pygame.time.Clock()

fps             = 60

clr_black       = (0, 0, 0)
clr_red         = (255, 0, 0)
clr_green       = (0, 255, 0)
clr_blue        = (0, 0, 255)
clr_white       = (255, 255, 255)
clr_yellow      = (255, 255, 0)
clr_ligthgray   = (250, 250, 250)

screenWidth     = info.current_w
screenHeight    = info.current_h
screenRes       = (screenWidth, screenHeight)

#avail_x         = range(10)
#avail_y         = range(-1, 4)
#avail_force     = range(1000)
#avail_angle     = range(-10, 31)
avail_x         = x
avail_y         = y
avail_force     = force
avail_angle     = angle


screen          = pygame.display.set_mode(screenRes, pygame.FULLSCREEN)
# screen          = pygame.display.set_mode((400, 300))
screen.fill(clr_black)
pygame.display.update()


if __name__ == "__main__":
    run     = True
    pause   = False

    render_event    = pygame.USEREVENT + 1
    pygame.time.set_timer(render_event, 1000)

    animation_top   = screenHeight / 3
    layer_width     = 2
    layer_height    = screenHeight / 2

    first_posX      = (screenWidth * 3 / 8)
    second_posX     = (screenWidth * 5 / 8)
    third_posX      = (screenWidth * 7 / 8)

    needle_width    = first_posX / 2
    needle_height   = 20

    while run :
        clock.tick(fps)

        for event in pygame.event.get():
            if event.type == pygame.QUIT :
                run = not run
            elif event.type == pygame.KEYDOWN :
                if event.key == pygame.K_ESCAPE : run = not run
                if event.key == pygame.K_SPACE  : pause = not pause
            elif event.type == render_event :
                x       = random.choice(avail_x)
                y       = random.choice(avail_y)
                force   = random.choice(avail_force)
                angle   = random.choice(avail_angle)

                step_X  = (third_posX - first_posX) / (len(avail_x) - 1)
                step_Y  = layer_height / (len(avail_y) - 1)
                rads    = radians(-angle)

                start_pos   = [first_posX + (x * step_X) - (needle_width * cos(rads)), ((y - avail_y[0]) * step_Y) + animation_top - (needle_width * sin(rads))]
                end_pos     = [first_posX + (x * step_X), ((y - avail_y[0]) * step_Y) + animation_top]

                screen.fill(clr_black)
                pygame.draw.line(screen, clr_blue, start_pos, end_pos, needle_height)

        # lapisan 1
        pygame.draw.line(screen, clr_green, [first_posX, animation_top], [first_posX, animation_top + layer_height], layer_width)
        # lapisan 2
        pygame.draw.line(screen, clr_yellow, [second_posX, animation_top], [second_posX, animation_top + layer_height], layer_width)
        # lapisan 3
        pygame.draw.line(screen, clr_red, [third_posX, animation_top], [third_posX, animation_top + layer_height], layer_width)

        pygame.display.flip()

    pygame.quit()