var segments = [];

var USER = 0;

var RUN = 1;

var STATE = RUN;

var runButton;

var iterations = 10;

​

function setup() {

  createCanvas(640, 480);

  for (var i = 0; i < 100; i++) {

    balls[i] = new Circle(i * 0.1 + 55, 0, 10);

  }

  segments[0] = new Segment(50, 100, 250, 150);

  segments[1] = new Segment(600, 300, 400, 450);

  runButton = createButton("Design / Play!");

  runButton.mousePressed(changeState);

}

​

function changeState() {

  if (STATE == USER) {

    STATE = RUN;

  } else {

    STATE = USER;

  }

}

​

function mousePressed() {

  balls.push(new Circle(mouseX, mouseY, 10));

}

​

function draw() {

  background(0);

  for (var segm of segments) {

    // segm.update(segm.a.x, segm.a.y, mouseX, mouseY);

    segm.show();

  }

  if (balls[0]) {

    for (var ball of balls) {

      for (var segment of segments) {

        ball.addSegment(segment);

      }

      var g = createVector(0, 0.25);

      ball.applyForce(g);

      for (var i = 0; i < iterations; i++) {

        ball.update();

      }

      ball.show();

    }

  }

}