Snake2 - OpenProcessing

mySketch
xxxxxxxxxx
 
// Inspired by the work of Prashant Gupta (https://p5js.org/examples/interaction-snake-game.html)
​
let numSegments = 10;
let direction = 'right';
​
let xStart = 0; 
let yStart = 250; 
let diff = 10;
​
let xCor = [];
let yCor = [];
​
let xFruit = 0;
let yFruit = 0;
let scoreElem;
​
let titleElem;
let fruitSound;
​
var isRunning = false;
​
function preload() {
  fruitSound = loadSound('B.mp3');
}
​
function setup() {
  scoreElem = createDiv('Score = 0');
  scoreElem.position(20, 20);
  scoreElem.id = 'score';
  scoreElem.style('color', 'black');
  
  createCanvas(windowWidth, windowHeight);
  colorMode(HSL, 100);
  rainbow = createImage(width, height)
  rainbow.loadPixels();
  for (let i = 0; i < width; i++) {
    for (let j = 0; j < height; j++) {
      let h = map(i, 0, width, 0, 100)
      rainbow.set(i, j, color(h, 100, 70));
    }
  }
  rainbow.updatePixels();
​
  frameRate(15);
  stroke(0); 
  strokeWeight(10);
}
​
function draw() {
  background(220);
  image(rainbow, 0, 0);
  if (isRunning == false) titleScreen();
  if (isRunning){
  for (let i = 0; i < numSegments - 1; i++) {
    line(xCor[i], yCor[i], xCor[i + 1], yCor[i + 1]);
  }
  updateSnakeCoordinates();
  checkGameStatus();
  checkForFruit();
}
}
​
​
function updateSnakeCoordinates() {
  for (let i = 0; i < numSegments - 1; i++) {
    xCor[i] = xCor[i + 1];
    yCor[i] = yCor[i + 1];
  }
  if (direction == 'right') {
      xCor[numSegments - 1] = xCor[numSegments - 2] + diff;
      yCor[numSegments - 1] = yCor[numSegments - 2];
  }
    else if (direction =='up') {
      xCor[numSegments - 1] = xCor[numSegments - 2];
      yCor[numSegments - 1] = yCor[numSegments - 2] - diff;
  }
    else if (direction =='left') {
      xCor[numSegments - 1] = xCor[numSegments - 2] - diff;
      yCor[numSegments - 1] = yCor[numSegments - 2];
      
    }
    else if (direction =='down') {
      xCor[numSegments - 1] = xCor[numSegments - 2];
      yCor[numSegments - 1] = yCor[numSegments - 2] + diff;
  }
}
​
​
function checkGameStatus() {
  if (
    xCor[xCor.length - 1] > width ||
    xCor[xCor.length - 1] < 0 ||
    yCor[yCor.length - 1] > height ||
    yCor[yCor.length - 1] < 0 ||
    checkSnakeCollision()
  ) {
    noLoop();
    const scoreVal = parseInt(scoreElem.html().substring(8));
    textAlign(CENTER);
    endScreen();
    scoreElem.html('Your score was : ' + scoreVal);
    
  }
}
​
function checkSnakeCollision() {
  const snakeHeadX = xCor[xCor.length - 1];
  const snakeHeadY = yCor[yCor.length - 1];
  for (let i = 0; i < xCor.length - 1; i++) {
    if (xCor[i] === snakeHeadX && yCor[i] === snakeHeadY) {
      return true;
    }
  }
}
​
​
function checkForFruit() {
  point(xFruit, yFruit);
  if (xCor[xCor.length - 1] === xFruit && yCor[yCor.length - 1] === yFruit) {
    const prevScore = parseInt(scoreElem.html().substring(8));
    scoreElem.html('Score = ' + (prevScore + 1));
    xCor.unshift(xCor[0]);
    yCor.unshift(yCor[0]);
    numSegments++;
    updateFruitCoordinates();
  }
}
​
function updateFruitCoordinates() {
​
  xFruit = floor(random(10, (width - 100) / 10)) * 10;
  yFruit = floor(random(10, (height - 100) / 10)) * 10;
  fruitSound.play();
}
​
function keyPressed() {
  if (keyCode === LEFT_ARROW){
    if (direction !== 'right') {
        direction = 'left';
      }
  }
  else if (keyCode === RIGHT_ARROW){
    if (direction !== 'left') {
        direction = 'right';
      }
  }
  else if (keyCode === UP_ARROW){
    if (direction !== 'down') {
        direction = 'up';
      }
  }
  else if (keyCode === DOWN_ARROW){
    if (direction !== 'up') {
        direction = 'down';
      }
  }
}
​
​
function titleScreen(){
  textSize(150);
  textAlign(CENTER);
  text('CLICK TO START', width/2, height/2);
  
}
​
function endScreen(){
  textSize(150);
  textAlign(CENTER);
  text('GAME OVER', width/2, height/2);
}
​
function mousePressed(){
    updateFruitCoordinates();
​
  for (let i = 0; i < numSegments; i++) {
    xCor.push(xStart + i * diff);
    yCor.push(yStart);
  }
  isRunning = true;
}
​