nli-gravity

GravityBallCode/GravityBallCode.pde

@@ -1,38 +1,54 @@
 //declare variables
-float x, y, velX, velY, diam;
+int count= 30;
+float []x = new float [count];
+float []y = new float [count];
+float []velX = new float [count];
+float []velY = new float [count];
+float []diam = new float [count];
+float gravity;
+
 
 void setup() {
   //set size of canvas
   size(800, 600);
-
-  //initialize variables
-  x = width/2;
-  y = height/2;
-  diam = 80;
-  velX = random(-5, 5);
-  velY = random(-5, 5);
+  for (int i = 0; i < count; i++) {
+    //initialize variables
+    x[i] = width/2;
+    y[i] = height/2;
+    diam[i] = 80;
+    velX[i] = random(-5, 5);
+    velY[i] = random(-5, 5);
+    gravity= .5;
+  }
 }
 
 void draw() {
   //draw background to cover previous frame
   background(0);
+  for (int i=0; i<count; i++) {
+    //add gravity
+    velY[i]+= gravity;
 
-  //draw ball
-  ellipse(x, y, diam, diam);
+    //draw ball
+    ellipse(x[i], y[i], diam[i], diam[i]);
+    fill(random(100), random(100), random(100));
+    noStroke();
 
-  //add velocity to position
-  x += velX;
-  y += velY;
+    //add velocity to position
+    x[i] += velX[i];
+    y[i] += velY[i];
 
-  //bounce ball if it hits walls
-  if (x + diam/2 >= width) {
-    velX = -abs(velX);    //if the ball hits the right wall, assign x velocity the negative version of itself
-  } else if (x - diam/2 <= 0) {
-    velX = abs(velX);     //if the ball hits the left wall, assign x velocity the positive version of itself
-  }
-  if (y + diam/2 >= height) {
-    velY = -abs(velY);
-  } else if (y - diam/2 <= 0) {
-    velY = abs(velY);
+    //bounce ball if it hits walls
+    if (x[i] + diam[i]/2 >= width) {
+      velX[i] = -abs(velX[i]);    //if the ball hits the right wall, assign x velocity the negative version of itself
+    } else if (x[i] - diam[i]/2 <= 0) {
+      velX[i] = abs(velX[i]);     //if the ball hits the left wall, assign x velocity the positive version of itself
+    }
+    if (y[i] + diam[i]/2 >= height) {
+      velY[i] = -abs(velY[i]);
+      y[i]=height-diam[i]/2;
+    } else if (y[i] - diam[i]/2 <= 0) {
+      velY[i] = abs(velY[i]);
+    }
   }
 }