from cs1lib import *
from ball import Ball

WINDOW_HEIGHT = 400
FRAMERATE = 40
TIMESTEP = 1.0 / FRAMERATE
PIXELS_PER_METER = 10.0


def main():
    set_clear_color(1, 1, 1)
    clear()

    if is_mouse_pressed():
        x = mouse_x() / PIXELS_PER_METER
        y = (WINDOW_HEIGHT - mouse_y()) / PIXELS_PER_METER

        ball = Ball(x, y, 0, 0, 1.0)
        ball_list.append(ball)

    for ball in ball_list:
        ball.draw(PIXELS_PER_METER, WINDOW_HEIGHT)

        ball.update_position(TIMESTEP)
        ball.update_velocity(TIMESTEP)


ball_list = []

start_graphics(main, 600, framerate=FRAMERATE)