From 7516add571888d32d3d33364a35cd8148ed3bff6 Mon Sep 17 00:00:00 2001
From: JP Appel <jeanpierre.appel01@gmail.com>
Date: Sat, 20 Apr 2024 23:49:21 -0400
Subject: add zoom feature to grids and shifted code around

---
 src/grids.c | 78 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 78 insertions(+)

(limited to 'src/grids.c')

diff --git a/src/grids.c b/src/grids.c
index a6d35df..89347db 100644
--- a/src/grids.c
+++ b/src/grids.c
@@ -1,3 +1,4 @@
+#include <complex.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
@@ -26,6 +27,7 @@ grid_t* create_grid(const size_t x, const size_t y, double complex lower_left, d
     *grid = (grid_t){
         .x = x,
         .y = y,
+        .size = x*y,
         .lower_left = lower_left,
         .upper_right = upper_right,
         .data = data
@@ -119,6 +121,23 @@ double complex grid_to_complex(const grid_t* grid, const size_t index) {
     return x + y * I;
 }
 
+/*
+ * Zoom a grid in or out based on its current center
+ *
+ * Resets all grid values to 0
+ */
+void zoom_grid(grid_t* grid, const double magnification){
+    set_grid(grid, 0);
+    const double complex upper_right = grid->upper_right;
+    const double complex lower_left = grid->lower_left;
+
+    const double complex center = (lower_left + upper_right) / 2.0;
+    const double complex offset = (upper_right - lower_left) / magnification;
+
+    grid->lower_left = center - offset;
+    grid->upper_right = center + offset;
+}
+
 /*
  * Writes a grid to a file in the .grid format
  *
@@ -156,6 +175,65 @@ int write_grid(FILE* restrict file, const grid_t *grid){
     return 0;
 }
 
+/*
+ * Prints info about a grid to stdout
+ */
+void print_grid_info(const grid_t* grid){
+    if(!grid){
+        printf("Grid is NULL!\n");
+        return;
+    }
+
+    printf("x\t%zu\n", grid->x);
+    printf("y\t%zu\n", grid->y);
+    printf("size\t%zu\n", grid->size);
+    printf("lower_left\t%f + %fI\n", creal(grid->lower_left), cimag(grid->lower_left));
+    printf("upper_right\t%f + %fI\n", creal(grid->upper_right), cimag(grid->upper_right));
+
+    printf("Data is %s NULL\n", grid->data ? "not" : "");
+}
+
+/*
+ * Attempts an ASCII print of the grid
+ */
+void print_grid(const grid_t* grid, const size_t iterations){
+    const size_t size = grid->size;
+    const size_t x_res = grid->x;
+    const size_t* data = grid->data;
+
+    // char* output_buffer = malloc(size);
+    // if(!output_buffer){
+    //     fprintf(stderr, "Failed to allocate output buffer for %zu points\n");
+    //     return;
+    // }
+
+    const char point_types[] = { ' ', '.', '*', '%', '#'};
+    size_t bin_width = iterations/3;
+    size_t last_bin = iterations - bin_width;
+    char point;
+
+    //TODO: set values in output buffer rather than multiple printf calls
+    for(size_t i = 0; i < size; i++){
+        const size_t value = data[i];
+        if(value == iterations){
+            point = point_types[4];
+        }
+        else if(value == 0){
+            point = point_types[0];
+        }
+        else if(value <= bin_width){
+            point = point_types[1];
+        }
+        else if(value >= last_bin){
+            point = point_types[3];
+        }
+        else {
+            point = point_types[2];
+        }
+        printf("%c%s", point, (i % x_res == x_res - 1) ? "\n" : "");
+    }
+}
+
 /*
  * Creates a grid from a .grid file, reading the amount of data as specified by the file
  * For more details on the .grid format see write_grid
-- 
cgit v1.2.3