green = [0,255,0],

    blue = [0,0,255],

    yellow = [255,255,0],

    black = [0,0,0],

        tolerance = 0.001,

    zeroTolerance = 0.000001;

​

function setup() {

  createCanvas(600, 600);

    pixelDensity(1);

  noLoop();

}

​

function draw() {

  background([32,32,32]);

​

    let startTime = millis();

    let halfWidth = width / 2;

    let halfHeight = height / 2;

    loadPixels();

    for(let y = 0; y < height; y++) {

        for(let x = 0; x < width; x++) {

            let color = getColor((x - halfWidth) / halfWidth, (y - halfHeight) / halfHeight);

            let xy = (x + y * width) * 4;

            pixels[xy] = color[0];

            pixels[xy + 1] = color[1];

            pixels[xy + 2] = color[2];

            pixels[xy + 3] = 255;

        }

    }

    updatePixels();

    let totalTime = millis() - startTime;

    text(totalTime, 32, 32);

}

​

function newton(x,y) {

  let bigCoeff = .25 / (Math.pow(x * x * x - 3 * x * y * y, 2) + Math.pow(3 * x * x * y - y * y * y, 2));

  return [.75 * x + bigCoeff * (x * x * x - 3 * x * y * y), .75 * y - bigCoeff * (3 * x * x * y - y * y * y)];

}

​

​

function getColor(x,y) {

    while (true) {

    if ((x-1)*(x-1) + y*y < tolerance) {

      return green;

    }

    else if (x*x + (y-1)*(y-1) < tolerance) {

      return red;

    }

    else if ((x+1)*(x+1) + y*y < tolerance) {

      return blue;

    }

    else if (x*x + (y+1)*(y+1) < tolerance) {

      return yellow;

    }

    else if (x*x + y*y < zeroTolerance) {

      return black;

    }

    else {

      [x,y] = newton(x,y);

    }

  }

}