var leaves = [];

​

var count = 0;

​

function setup() {

  createCanvas(400, 400);

  var a = createVector(width / 2, height);

  var b = createVector(width / 2, height - 100);

  var root = new Branch(a, b);

​

  tree[0] = root;

  for(let j = 0; j <= 6; j++){

    for (let i = tree.length - 1; i >= 0; i--) {

      if (!tree[i].finished) {

        tree.push(tree[i].branchA());

        tree.push(tree[i].branchB());

      }

      tree[i].finished = true;

    }

    count++;

​

    if (count === 6) {

      for (let i = 0; i < tree.length; i++) {

        if (!tree[i].finished) {

          var leaf = tree[i].end.copy();

          leaves.push(leaf);

        }

      }

    }  

  }

}

​

function draw() {

  background(51);

  for (let i = 0; i < tree.length; i++) {

    tree[i].show();

  }

​

  for (let i = 0; i < leaves.length; i++) {

    fill(255, 0, 100, 100);

    noStroke();

    ellipse(leaves[i].x, leaves[i].y, 8, 8);

    leaves[i].y += random(0, 2);

  }

​

}