let size;

let fluid;

let iterations = 10;

let t = 0;

​

function setup() {

  createCanvas(400, 400);

  angleMode(DEGREES);

  size = width/N;

  fluid = new Fluid(0.2, 0, 0.001);

}

​

function mouseDragged() {

  fluid.addDensity(floor(mouseX/size), floor(mouseY/size), 100);

}

​

function draw() {

  background(220);

  let cx = int((0.5 * width) / size);

  let cy = int((0.5 * height) / size);

  for (let i = -1; i <= 1; i++) {

    for (let j = -1; j <= 1; j++) {

      fluid.addDensity(cx + i, cy + j, random(50, 150));

    }

  }

​

  for (let i = 0; i < 2; i++) {

    let angle = noise(t) * TWO_PI * 2;

    let v = p5.Vector.fromAngle(angle);

    v.mult(0.2);

    t += 0.01;

   fluid.addVelocity(cx, cy, v.x, v.y);

  }

  fluid.step();

  fluid.renderD();

  fluid.renderV(); 

}