class Agent {

  // constructor to set initial agent values

  constructor() {

    this.x = random(17, width - 17);

    this.y = random(17, height - 17);

    this.vx = random([-2, -1, 1, 2]);

    this.vy = random([-2, -1, 1, 2]);

    this.radius = random(8, 16);

    this.diam = 2 * this.radius;

  }

​

  // update agent position

  update() {

    // update position

    this.updateRandom();

​

    // check boundary

    this.wrapBoundary();

  }

​

  // if agent gets to the edges, bounce back

  bounceBoundary() {

    if (this.x + this.radius >= width || this.x - this.radius <= 0) {

      this.vx *= -1;

    }

    if (this.y + this.radius >= height || this.y - this.radius <= 0) {

      this.vy *= -1;

    }

  }

​

  // if agent gets to the edges, wrap around to the opposite edge

  wrapBoundary() {

    if (this.x > width) {

      this.x = this.x % width;

    }

    if (this.x < 0) {

      this.x = this.x + width;

    }

    if (this.y > height) {

      this.y = this.y % height;

    }

    if (this.y < 0) {

      this.y = this.y + height;

    }

  }

​

  // if agent gets to the edges, reset it's position, velocity, etc

  resetBoundary() {

    if (this.x > width || this.x < 0 || this.y > height || this.y < 0) {

      this.x = random(17, width - 17);

      this.y = random(17, height - 17);

      this.vx = random([-2, -1, 1, 2]);

      this.vy = random([-2, -1, 1, 2]);

      this.radius = random(8, 16);

      this.diam = 2 * this.radius;

    }

  }

​

  // use velocity values to update position

  updateByVelocity() {

    this.x += this.vx;

    this.y += this.vy;

  }

​

  // use random velocity values to update position

  updateRandom() {

    this.vx = random([-3, -2, -1, 1, 2, 3]);

    this.vy = random([-3, -2, -1, 1, 2, 3]);

    this.x += this.vx;

    this.y += this.vy;

  }

​

  distComp(agentA, agentB) {

    let distA = dist(this.x, this.y, agentA.x, agentA.y);

    let distB = dist(this.x, this.y, agentB.x, agentB.y);

    return distA - distB;

  }

​

  // move away from nearest agent

  updateNearest() {

    let sortedByDist = agents.toSorted(this.distComp.bind(this));

    let closestAgent = sortedByDist[1];

    this.vx = map(closestAgent.x - this.x, -width, width, 4, -4);

    this.vy = map(closestAgent.y - this.y, -height, height, 4, -4);

    this.x += this.vx;

    this.y += this.vy;

  }

​

  // draw agent

  drawAgent() {

    ellipse(this.x, this.y, this.diam);

  }

​

  // create drawing

  draw() {}

​

  // draw black point at x, y

  drawPoint() {

    point(this.x, this.y);

  }

​

  // draw a line between each agent and the agent furthest away from it

  drawFurthest() {

    let sortedByDist = agents.toSorted(this.distComp.bind(this));

    let furthestAgent = sortedByDist[sortedByDist.length - 1];

    line(this.x, this.y, furthestAgent.x, furthestAgent.y);

  }

​

  // draw a line between each agent and its nearest agent

  drawNearest() {

    let sortedByDist = agents.toSorted(this.distComp.bind(this));

    let nearestAgent = sortedByDist[1];

    line(this.x, this.y, nearestAgent.x, nearestAgent.y);

  }

​

  // draw ellipse between agents when they overlap

  drawOverlap() {

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

      let otherAgent = agents[i];

      if (this != otherAgent) {

        let tDist = dist(this.x, this.y, otherAgent.x, otherAgent.y);

        if (tDist < this.radius + otherAgent.radius) {

          let cx = (this.x + otherAgent.x) / 2;

          let cy = (this.y + otherAgent.y) / 2;

          ellipse(cx, cy, tDist);

        }

      }

    }

  }

}

​

// max number of agents

let maxAgents = 16;

​

// array for keeping track of agents

let agents = [];

​

// keep track of current mode

let AGENT_MODE = 0;

let DRAWING_MODE = 1;

​

let currentMode;

​

function setup() {

  createCanvas(windowWidth, windowHeight);

​

  // set initial state

  currentMode = AGENT_MODE;

​

  // create agents and store them in array

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

    agents.push(new Agent());

  }

}

​

function draw() {

  // update agents

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

    agents[i].update();

  }

​

  // depending on the mode:

  if (currentMode == AGENT_MODE) {

    background(220, 20, 120);

​

    // draw agents

    noStroke();

    fill(255);

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

      agents[i].drawAgent();

    }

  } else if (currentMode == DRAWING_MODE) {

    // create drawing

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

      agents[i].draw();

    }

  }

}

​

function mouseClicked() {

  // cycle through modes

  if (currentMode == AGENT_MODE) {

    currentMode = DRAWING_MODE;

    background(255);

  } else if (currentMode == DRAWING_MODE) {

    currentMode = AGENT_MODE;

  }

}

​

function keyReleased() {

  // if drawing:

  if (currentMode == DRAWING_MODE) {

    // s: save drawing

    if (key == "s" || key == "S") {

      saveCanvas("my-drawing", "jpg");

    }

​

    // r: reset

    if (key == "r" || key == "R") {

      background(255);

    }

  }

}

​