#include <helianthus.h></helianthus.h>

#define SIZE 5

#define LAMPSIZE 100

int lamps[SIZE][SIZE];

void toggle(int i, int j) {

  if (i < 0 || j < 0 || i >= SIZE || j >= SIZE) return;

  if (i > 0     ) lamps[i-1][j] = !lamps[i-1][j];

  if (i < SIZE-1) lamps[i+1][j] = !lamps[i+1][j];

  if (j > 0     ) lamps[i][j-1] = !lamps[i][j-1];

  if (j < SIZE-1) lamps[i][j+1] = !lamps[i][j+1];

  lamps[i][j] = !lamps[i][j];

}

void toggleRandom() {

  toggle(randomNumber(0, SIZE-1), randomNumber(0, SIZE-1));

}

void init() {

  background(colorByName("black"));

  for(int i = 0; i < SIZE*SIZE; ++i)

    toggleRandom();

}

void draw() {

  saveState();

  zoom(LAMPSIZE);

  strokeWidth(0.1);

  stroke( colorByRGB(0.25, 0.25, 0.25) );

  for(int i = 0; i < SIZE; ++i) {

    for(int j = 0; j < SIZE; ++j) {

      fill(colorByName( lamps[i][j] ? "yellow" : "black" ));

      ellipse(j + 0.2, i + 0.2, 0.6, 0.6);

    }

  }

  if (mouseWentDown("left"))

    toggle((int)mouseTransformedY(), (int)mouseTransformedX());

  if (keyWentDown("space"))

    toggleRandom();

  restoreState();

}

int main() {

  windowSetTitle("Lamps");

  windowSetSize(SIZE*LAMPSIZE, SIZE*LAMPSIZE);

  windowSetVariableFrameRate();

  windowSetInit(&init);

  windowSetDraw(&draw);

  windowRun();

  return 0;

}