fix boundaries

[lifeboot.git] / prototype.c

diff --git a/prototype.c b/prototype.c
index 405654e5741109820dc367ecc75be18917c47dc6..bc414d5f23a2932ff38547affb00b7ec20ba8917 100644 (file)
--- a/prototype.c
+++ b/prototype.c
@@ -1,5 +1,7 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <stdio.h>
 #include <stdlib.h>

+#include <unistd.h>
+#include <string.h>

 #include <time.h>
 
 #define COLS 80
 #include <time.h>
 
 #define COLS 80


@@ -27,7 +29,7 @@ static void print_grid(char grid[COLS * ROWS]) {
 }
 
 static void init_grid(char grid[COLS * ROWS]) {
 }
 
 static void init_grid(char grid[COLS * ROWS]) {

-  for (int i = 0; i < ROWS * COLS; i++) {
+  for (int i = 0; i < COLS * ROWS; i++) {

     if (rand() % INIT_RATIO)
       grid[i] = DEAD;
     else
     if (rand() % INIT_RATIO)
       grid[i] = DEAD;
     else


@@ -35,14 +37,66 @@ static void init_grid(char grid[COLS * ROWS]) {
   }
 }
 
   }
 }
 

+static char get_state(char grid[COLS * ROWS], int row, int col) {
+  if (row < 0 || row >= ROWS || col < 0 || col >= COLS)
+    return DEAD;
+
+  return grid[row * COLS + col];
+}
+
+static int alive_neighbours(char grid[COLS * ROWS], int idx) {
+  int row = idx / COLS;
+  int col = idx % COLS;
+  int neighbours = 0;
+
+  for (int i = -1; i <= 1; i++)
+    for (int j = -1; j <= 1; j++)
+      if (get_state(grid, row + i, col + j) == ALIVE)
+        neighbours++;
+
+  if (get_state(grid, row, col) == ALIVE)
+    neighbours--;
+
+  return neighbours;
+}
+
+static char next_state(char grid[COLS * ROWS], int idx) {
+  int n = alive_neighbours(grid, idx);
+
+  if (get_state(grid, idx / COLS, idx % COLS) == ALIVE) {
+    if (n < 2 || n > 3)
+      return DEAD;
+
+    return ALIVE;
+  }
+
+  if (n == 3)
+    return ALIVE;
+
+  return DEAD;
+}
+
+static void update_grid(char grid[COLS * ROWS]) {
+  char new[COLS * ROWS];
+
+  for (int i = 0; i < COLS * ROWS; i++)
+    new[i] = next_state(grid, i);
+
+  memcpy(grid, new, COLS * ROWS);
+}
+

 int main () {
   char grid[COLS * ROWS];
 
   srand(time(0));
   init_grid(grid);
 
 int main () {
   char grid[COLS * ROWS];
 
   srand(time(0));
   init_grid(grid);
 

-  clearscr();
-  print_grid(grid);
+  for (;;) {
+    clearscr();
+    print_grid(grid);
+    update_grid(grid);
+    getchar();
+  }

 
   return 0;
 }
 
   return 0;
 }