Add function to save map

2 files changed, 43 insertions, 0 deletions

diff --git a/map.c b/map.c
index 569ab46..5dc2782 100644
--- a/map.c
+++ b/map.c
@@ -211,6 +211,31 @@ Map file_plaintext_map(char *filename, size_t *width, size_t *height, Position *
     return map;
 }
 
+void map_to_file_plaintext(char *filename, Map map, size_t width, size_t height, Position start, Position end) {
+    FILE *file = fopen(filename, "w");
+    if (file == NULL) {
+        message(height, "Couldn't open %s for writing", filename);
+        return;
+    }
+
+    /* Size and chars */
+    fprintf(file, "%zux%zu\n.#@x\n", width, height);
+    for (size_t row = 0; row < height; row++) {
+        for (size_t col = 0; col < width; col++) {
+            if (row == start.y && col == start.x) { putc('@', file); continue; }
+            if (row == end.y && col == end.x) { putc('x', file); continue; }
+            switch (map[row][col]) {
+                case EMPTY: putc('.', file); break;
+                case WALL: putc('#', file); break;
+            }
+        }
+        putc('\n', file);
+    }
+
+    message(height, "Saved the map to %s", filename);
+    return;
+}
+
 /* TODO: so many fucking arguments lmao. Break it down into several functions? */
 /* TODO: draw the start and goal with no background, allowing us to see the frontier/path/whatever's underneath. I think attr_get() will be useful */
 /* TODO: only draw a portion of the map (for bigger ones) */
@@ -455,6 +480,22 @@ void map_editor(Map *map, size_t *width, size_t *height, Position *start, Positi
                               break;
                       }
                       break;
+            case 's':
+                      curs_set(2);   /* Show the cursor */
+                      echo();        /* Echo characters */
+
+                      message(*height, FILENAME_PROMPT);
+                      char filename[FILENAME_BUF_SIZE] = "map";
+                      mvgetnstr(*height + map_offset_y + 1, map_offset_x - 2 + sizeof(FILENAME_PROMPT), filename, FILENAME_BUF_SIZE - 1);
+
+                      curs_set(0);   /* Hide the cursor */
+                      noecho();      /* Don't echo characters */
+
+                      map_to_file_plaintext(filename, *map, *width, *height, *start, *goal);
+
+                      getch();
+                      break;
+
             /* TODO: keybindings to resize the map */
         }
         draw_map(*map, *width, *height, *start, *goal, NULL, NULL, NULL, NULL);
diff --git a/map.h b/map.h
index e9c5bc3..b2b3ff4 100644
--- a/map.h
+++ b/map.h
@@ -46,6 +46,8 @@ Map rbt_maze_map(size_t width, size_t height, unsigned int seed);
  * ....#.#...
  * ..@....... */
 Map file_plaintext_map(char *filename, size_t *width, size_t *height, Position *start_pos, Position *end_pos);
+/* The reverse of above */
+void map_to_file_plaintext(char *filename, Map map, size_t width, size_t height, Position start, Position end);
 
 /* Draw the map. Bet you didn't expect that.
  * path could be NULL to draw a map with no path. So can cursor, frontier and visited */