let ants;

let numAnts;

let numNodes;

let paused;

let totalDone;

​

let ticks;

​

function euclidean(p1, p2) {

  return Math.sqrt(Math.pow(p1.x - p2.x, 2) + Math.pow(p1.y - p2.y, 2));

}

​

function getRandomInt(min, max) {

  min = Math.ceil(min);

  max = Math.floor(max);

  return Math.floor(Math.random() * (max - min) + min); //The maximum is exclusive and the minimum is inclusive

}

​

function keyPressed() {

  if (key == "p") {

    paused = !paused;

    console.log(`paused [${paused}]`);

  }

  if (key == "R") {

    ants = [];

    console.log("forced continuing to colors");

  }

  if (key == "Q") {

    console.log(`ants: ${ants}`);

  }

}

class ant {

  constructor(x, y, trailColor) {

    this.position = createVector(x, y);

    this.trailColor = trailColor;

    this.notVisited = Array.from({ length: numNodes }, (item, index) => index);

    this.velocity = createVector(0, 0);

    this.target = null; // don't randomly jump around

  }

​

  update() {

    if (this.notVisited.length > 0) {

      let closest = this.target;

​

      if (closest == null) {

        if (random() > 0.5) {

          // if (random() > 0.1) {

          closest = this.notVisited.reduce((a, b) =>

            euclidean(this.position, nodes[a]) <

            euclidean(this.position, nodes[b])

              ? a

              : b

          );

        } else {

          closest = this.notVisited[getRandomInt(0, this.notVisited.length)];

        }

      }

​

      // https://stackoverflow.com/questions/56306097/get-nearest-point-from-another-one-in-javascript

      // let dx = nodes[closest].x - this.position.x;

      // let dy = nodes[closest].y - this.position.y;

      let dx = abs(nodes[closest].x - this.position.x);

      let dy = abs(nodes[closest].y - this.position.y);

      let angle = atan2(dy, dx);

​

      this.velocity.x = random(1, 5) * cos(angle);

      this.velocity.y = random(1, 5) * sin(angle);

​

      this.position.x += this.velocity.x;

      this.position.y += this.velocity.y;

​

      // if (this.position.x > width || this.position.x < 0) this.velocity.x *= -1;

      // if (this.position.y > height || this.position.y < 0) this.velocity.y *= -1;

      // if (this.position.x > width || this.position.x < 0) this.position.x = random(width);

      // if (this.position.y > height || this.position.y < 0) this.velocity.y = random(height);

​

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

        let _i = getRandomInt(0,this.notVisited.length);

        this.position.x = nodes[this.notVisited[_i]].x;

        this.position.y = nodes[this.notVisited[_i]].y;

        this.velocity.x = random(-4,4);

        this.velocity.y = random(-4,4);

      }

      // l, x and y must satisfy (x - center_x)^2 + (y - center_y)^2 < radius^2.

      if (

        Math.pow(this.position.x - nodes[closest].x, 2) +

          Math.pow(this.position.y - nodes[closest].y, 2) <

        25

      ) {

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

          if (this.notVisited[i] == closest) {

            this.notVisited.splice(i, 1);

            this.target = null;

            break;

          }

        }

      }

​

      // if (random() > 0.59) {

      //   this.velocity.x *= random(-2.0, 2.0);

      //   this.velocity.y *= random(-2.0, 2.0);

      // }

​

      if (random() > 0.8) {

        this.target = null;

      }

      let _v = [-2, -1, 1, 2];

      if (this.velocity.x < 1 && this.velocity.x > -1) this.velocity.x = _v[getRandomInt(0,_v.length)];

      if (this.velocity.y < 1 && this.velocity.y > -1) this.velocity.y = _v[getRandomInt(0,_v.length)];

    }

  }

​

  draw() {

    noStroke();

    fill(this.trailColor);

    circle(this.position.x, this.position.y, 0.5);

  }

​

  done() {

    let retval = false;

    if (this.notVisited.length == 0) retval = true;

    return retval;

  }

}

​

function setup() {

  numNodes = 500;

  // numAnts = 100;

  numAnts = 1000;

  // numAnts = 20000;

  nodes = [];

  ants = [];

  totalDone = 0;

  ticks = 0;

​

  createCanvas(7200, 7200);

  // createCanvas(1000, 1000);

  // createCanvas(400, 400);

​

  background(0);

  noStroke();

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

    nodes.push(createVector(random(width), random(height)));

    fill(color(180));

    circle(nodes[i].x, nodes[i].y, 10);

  }

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

    let _start = [

      [0, 0],

      [width, 0],

      [0, height],

      [width, height],

      [width / 2, height / 2],

    ];

    let _r = getRandomInt(0, _start.length);

    ants.push(

      new ant(

        // _start[_r][0],

        // _start[_r][1],

        //0,0,

        random(width),

        random(height),

        color(random(255))

        // color(random(255), random(255), random(255))

      )

    ); //color("#FF00FF")));

  }

}

​

function draw() {

  if (!paused) {

    // background(0);

    // nodes.forEach((n) => {

    //   fill(180);

    //   circle(n.x, n.y, 10);

    // });

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

      // ants.forEach((a) => {

      let a = ants[i];

      a.update();

      a.draw();

​

      if (a.done()) ants.splice(i, 1);

​

      // });

    }

    // background(220);

​

    if (ants.length == 0) {

      ticks++;

​

      if (ticks == 1) {

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

          let _start = [

            [0, 0],

            [width, 0],

            [0, height],

            [width, height],

            [width / 2, height / 2],

          ];

          let _r = getRandomInt(0, _start.length);

          ants.push(

            new ant(

              // width/2,height/2,

              // _start[_r][0],

              // _start[_r][1],

              //0,0,

              random(width),

              random(height),

              color(0,random(255),0)

              // color(random(255))

              // color(random(255), random(255), random(255))

            )

          ); //color("#FF00FF")));

        }

      } else if (ticks == 2) {

        console.log("done");

        noLoop();

      }

    }

  }

}

​