let speed = 4; // Speed of the fireworks

​

function setup() {

  createCanvas(600, 600); 

}

​

function draw() {

  background(0);

​

  // Access the current firework object

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

    let firework = fireworks[i];

    firework.update();  

    firework.display(); 

​

    // check if the firework has moved beyond its explosion radius

    if (dist(firework.x, firework.y, firework.startX, firework.startY) > firework.radius) {

      firework.finished = true; 

    }

  }

  // Remove the firework marked as "finished"

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

    if (fireworks[i].finished) {

      fireworks.splice(i, 1); 

    } 

  }

}

​

function mousePressed() {

  let fireworkGroups = int(random(5, 10)); 

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

    let startX = mouseX + random(-200, 200); 

    let startY = mouseY + random(-200, 200);

    let radius = random(5, 100); 

    let count = int(random(8, 16)); 

​

    // Create a new firework object and add it to the array

    fireworks.push(new Firework(startX, startY, count, radius));

  }

}

​

class Firework {

  constructor(startX, startY, count, radius) {

    this.startX = startX;

    this.startY = startY;

    this.x = startX;

    this.y = startY;

    this.color = [random(0, 255), random(0, 255), random(0, 255)];

    this.radius = radius;

    this.speed = speed;

    this.particles = [];

    this.finished = false; // Initialize as "not finished"

​

    // Create the particles of the firework

    // This is calculated using the formula 360 / count so that particles are evenly distributed in a circular manner.

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

      let angle = radians(i * (360 / count));

      this.particles.push({

        x: this.x,

        y: this.y,

        angle: angle

      });

    }

  }

​

  update() {

    for (let particle of this.particles) {

      particle.x += this.speed * cos(particle.angle); 

      particle.y += this.speed * sin(particle.angle); 

    }

  }

​

  display() {

    for (let particle of this.particles) {

      fill(this.color);

      noStroke();

      ellipse(particle.x, particle.y, 5, 5); 

    }

  }

}

​

​

​