let grid = [];

let piece; 

​

function setup() {

  createCanvas(300, 400);

  piece = new Piece();

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

    grid[i] = [];

    for (let j=0; j<rows; j++) {

      grid[i][j] = color(0, 0, 255);

    }

  }

  piece.newFallingPiece();

}

​

function draw() {

  background(220);

  translate(-size*cols/2 + width/2, -size*rows/2 + height/2);

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

    for (let j=0; j<rows; j++) {

      fill(grid[i][j]);

      rect(i*size, j*size, size, size);

    }

  }

  piece.drawPiece();

}

​

function keyPressed() {

  if (key == 's') {

    piece.newFallingPiece();

  } 

  if (keyCode === DOWN_ARROW) {

    piece.moveFallingPiece(0, 1);

  }

  if (keyCode === UP_ARROW) {

    piece.moveFallingPiece(0, -1);

  }

  if (keyCode === RIGHT_ARROW) {

    piece.moveFallingPiece(1, 0);

​

  }

  if (keyCode === LEFT_ARROW) {

    piece.moveFallingPiece(-1, 0);  

  }

}