#include <math.h></math.h>

#include <stdlib.h></stdlib.h>

#include <string.h></string.h>

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

#define TILESIZE     512

#define ROWS           8

#define COLS          12

const double cellRatio = 5/6.0;

Animation colTiles[128];

Animation rowTiles[128];

int colTilesCount = 0;

int rowTilesCount = 0;

Framebuffer framebuffer;

Animation fbAnim;

void generate() {

  saveState();

  target(framebuffer);

  background(COLOR_TRANSPARENT);

  clear();

  zoom(TILESIZE*cellRatio);

  double k = 0.1;

  fill(colorByRGBA(1, 1, 1, 1));

  noStroke();

  saveState();

  translate(1, 0.5);

  if (rowTilesCount > 0)

    for(int y = 0; y < ROWS; ++y)

      for(int x = 0; x < COLS-1; ++x) {

        saveState();

        translate(x, y);

        zoom(1/cellRatio);

        rectTextured(rowTiles[rand()%rowTilesCount], -0.5, -0.5, 1, 1);

        restoreState();

      }

  restoreState();

  saveState();

  translate(0.5, 1);

  if (rowTilesCount > 0)

    for(int x = 0; x < COLS; ++x)

      for(int y = 0; y < ROWS-1; ++y) {

        saveState();

        translate(x, y);

        zoom(1/cellRatio);

        rectTextured(colTiles[rand()%colTilesCount], -0.5, -0.5, 1, 1);

        restoreState();

      }

  restoreState();

  viewportSave("data/output/generated-scheme.png");

  restoreState();

}

Animation subImage(int width, int height, unsigned char *pixels, int x, int y, int w, int h) {

  if (x < 0 || y < 0 || x + w > width || y + h > height || !pixels)

    return NULL;

  unsigned char *px = calloc(4, w*h);

  for(int i = 0; i < h; ++i)

    memcpy(px + i*w*4, pixels + ((y+i)*width + x)*4, w*4);

  Animation anim = createAnimationFromImage(w, h, px, FALSE);

  free(px);

  return anim;

}

void init() {

  int w = 0;

  int h = 0;

  unsigned char *px = NULL;

  imageLoad("data/scheme.png", &w, &h, &px);

  for(int i = 0; i < 4; ++i)

    if ((rowTiles[rowTilesCount] = subImage(w, h, px, i*TILESIZE, 0*TILESIZE, TILESIZE, TILESIZE)))

      ++rowTilesCount;

  for(int i = 0; i < 4; ++i)

    if ((colTiles[colTilesCount] = subImage(w, h, px, i*TILESIZE, 1*TILESIZE, TILESIZE, TILESIZE)))

      ++colTilesCount;

  free(px);

  w = ceil(TILESIZE*COLS*cellRatio - 0.001);

  h = ceil(TILESIZE*ROWS*cellRatio - 0.001);

  framebuffer = createFramebuffer(w, h);

  fbAnim = createAnimationFromFramebuffer(framebuffer);

  generate();

}

void draw() {

  background(COLOR_WHITE);

  double w = windowGetWidth();

  double h = windowGetHeight();

  if (keyWentDown("space")) generate();

  saveState();

  translate(w/2, h/2);

  zoom(w/COLS);

  fill(COLOR_WHITE);

  noStroke();

  rectTextured(fbAnim, -COLS/2.0, -ROWS/2.0, COLS, ROWS);

  restoreState();

}

int main() {

  windowSetVariableFrameRate();

  windowSetResizable(TRUE);

  windowSetInit(&init);

  windowSetDraw(&draw);

  windowRun();

}