// The Coding Train / Daniel Shiffman

// https://thecodingtrain.com/CodingChallenges/167-prime-spiral.html

// https://youtu.be/a35KWEjRvc0

​

// Prime Spiral: https://editor.p5js.org/codingtrain/sketches/0Ud-XsaYb

// Number Spiral: https://editor.p5js.org/codingtrain/sketches/Zs65bV-Al

// Prime vs Random Spiral: https://editor.p5js.org/codingtrain/sketches/3np1hBlXD

// Shapes and Color: https://editor.p5js.org/codingtrain/sketches/mCvvSKpZ5

// Prime Spiral 3D: https://editor.p5js.org/codingtrain/sketches/-eX078HZ5

​

// State of spiral

let x, y;

let px, py;

let step = 1;

let state = 0;

let numSteps = 0

let turnCounter = 1;

​

// Scale / resolution

let stepSize = 12;

let halfStepSize;

let halfRootThreeStepSize;

let totalSteps;

let lastturn = 0;

let bonus = 0;

​

// Function to test if number is prime

function isPrime(value) {

  if (value == 1) return false;

  for (let i = 2; i <= sqrt(value); i++) {

    if (value % i == 0) {

      return false;

    }

  }

  return true;

}

​

function setup() {

  createCanvas(800, 800);

  halfRootThreeStepSize = sqrt(3) * 0.5 * stepSize;

  halfStepSize = 0.5 * stepSize;

​

  // set up spiral

  const cols = width / stepSize;

  const rows = height / stepSize;

  totalSteps = cols * rows;

  x = width / 2;

  y = height / 2;

  px = x;

  py = y;

  background(0);

}

​

function draw() {

  // If prime draw circle

  if (isPrime(step)) {

    fill(255);

    stroke(255);

    circle(x, y, stepSize * 0.5);

  }

  // Connect current to previous with a line

  line(x, y, px, py);

  px = x;

  py = y;

​

  // Move according to state

  switch (state) {

    case 0:

      x += stepSize;

      bonus = 1;

      break;

    case 1:

      y -= halfRootThreeStepSize;

      x += halfStepSize;

      bonus = 0;

      break;

    case 2:

      y -= halfRootThreeStepSize;

      x -= halfStepSize;

      bonus = 0;

      break;

    case 3:

      x -= stepSize;

      bonus = 0;

      break;

    case 4:

      y += halfRootThreeStepSize;

      x -= halfStepSize;

      bonus = 0;

      break;

    case 5:

      y += halfRootThreeStepSize;

      x += halfStepSize;

      bonus = 0;

      break;

  }

  // Change state

  if ((step-lastturn) >= numSteps+bonus) {

    state = (state + 1) % 6;

    turnCounter++;

    if ((turnCounter >3 )) {

      if (turnCounter % 6 == 3) {

        numSteps++;

      }

    } else {

      numSteps = 1;

      if (step == 1) {

        state++;

        turnCounter++;

      }

    }

    lastturn = step;

  }

  step++;

  // Are we done?

  if (step > totalSteps) {

    noLoop();

  }

​

}

​