from pyg import *
import sqlite3
from random import *

con = sqlite3.connect('ball.db');
cur = con.cursor();

#generate 10000 balls
#cur.execute("DELETE FROM ball");
#for i in range(10000):
#  cur.executemany("INSERT INTO ball VALUES(?,?,?,?,?)", [(150+i,randint(10,500),randint(10,250),random(),random())]);

physiQuery = """
UPDATE ball SET x = x+vx, y = y+vy,
vx = CASE WHEN x+vx>1024 OR x+vx<0 THEN -vx ELSE vx END,
vy = CASE WHEN y+vy>512 OR y+vy<0 THEN -vy ELSE vy END
""";

readQuery = "SELECT x,y FROM ball LIMIT 1000";

if __name__ == "__main__":
  while main():
    cur.execute(physiQuery);
    cur.execute(readQuery);
    lst = [];
    for data in cur.fetchall():
      x,y = data;
      circle(x,y);
  con.commit();
  cur.close();