Added Puzzle export as PNG image
[wordblah.git] / wordblox.c
index 6ad88e1..223dde7 100644 (file)
@@ -8,7 +8,33 @@
 #include "wordblox.h"
 #include "constantstrings.h"
 
-void set_clue_across_word (Puzzle *p)
+/* export the puzzle to a png file */
+void export_puzzle (Puzzle *p)
+{
+       if (p->grid_frozen == false)
+       {
+               printf (UNFROZEN_GRID);
+               char ch = getchar ();
+               return;
+       }
+       char fname[256];
+       printf (INPUT_FILE);
+       fgets (fname, 256, stdin);
+       char* filename = strtok (fname, "\n");
+
+       printf (INPUT_EXPORT_ANSWERS);
+       char ans[3];
+       fgets (ans, 3, stdin);
+       bool solution;
+       solution = (toupper (ans[0]) == 'Y') ? true : false;
+               
+       export_grid_image (p, filename, solution);
+       printf (FILE_SAVED);
+       char ch = getchar ();
+}
+
+/* set clue for a word - only for frozen grid */
+void set_clue_word (Puzzle *p, enum ORIENTATION orient)
 {
        print_puzzle (p);
        if (p->grid_frozen == false)
@@ -25,9 +51,14 @@ void set_clue_across_word (Puzzle *p)
        fgets (clue, MAX_CLUE_LENGTH, stdin);
        char* cl = strtok (clue, "\n");
        
-       bool res = set_across_clue (p, cl, index);
+       bool res;
+       res = set_clue (p, cl, index, orient);
+
        if (res == false)
+       {
                printf (NO_WORD_INDEX);
+               char ch = getchar ();
+       }
 }
 
 /* clear a cell in the grid */
@@ -115,7 +146,7 @@ void add_across_word (Puzzle *p)
        char wd[MAX_PUZZLE_SIZE];
        int row, col;   
        printf (INPUT_WORD);
-       fgets (wd, p->grid_size, stdin);
+       fgets (wd, MAX_PUZZLE_SIZE, stdin);
        char *word = strtok (wd, "\n");
        if (! is_valid_word (word))
        {
@@ -147,13 +178,25 @@ void add_across_word (Puzzle *p)
        print_puzzle (p);
        char ch = getchar ();
 }
+/* confirm exit */
+bool confirm_exit ()
+{
+       printf (INPUT_CONFIRM_EXIT);
+       char res[3];
+       fgets (res, 3, stdin);
+       if (toupper(res[0]) == 'Y')
+               return true;
+       else
+               return false;
+} 
 
+/* main loop for the puzzle editor */
 void puzzle_editor_loop (Puzzle *p, const char *filename) 
 {
        bool loop = true;
        while (loop) 
        {
-               print_menu (WHITE, RED, PUZZLE_MENU_TITLE, PUZZLE_EDIT_MENU, 9, 50);
+               print_menu (WHITE, RED, PUZZLE_MENU_TITLE, PUZZLE_EDIT_MENU, 11, 50);
                printf (INPUT_CHOICE);
                int ch = get_num ();
                switch (ch)
@@ -175,16 +218,18 @@ void puzzle_editor_loop (Puzzle *p, const char *filename)
                                        print_puzzle (p);
                                        ch = getchar ();
                                        break;
-                       case 7: set_clue_across_word (p);
-                                       print_puzzle (p);
-                                       ch = getchar ();
+                       case 7: set_clue_word (p, ACROSS);
+                                       break;
+                       case 8: set_clue_word (p, DOWN);
                                        break;
-                       case 8: save_puzzle (p, filename);
+                       case 9: save_puzzle (p, filename);
                                        printf ("%s\n",FILE_SAVED);
                                        ch = getchar ();
                                        break;
-                       case 9: loop = false;
-                                       break;
+                       case 10: export_puzzle (p);
+                                        break;
+                       case 11: loop = !confirm_exit ();
+                                        break;
                }
        }
 }
@@ -245,6 +290,27 @@ int main_loop ()
 }
 
 int main (int argc, char* argv[]) 
-{      
+{
+       if (argc >= 2)
+       {
+               Puzzle p;
+               switch (argc)
+               {
+                       case 2 : p = load_puzzle (argv[1]);
+                                        puzzle_editor_loop (&p, argv[1]);
+                                        break;
+                       case 4 : if (strcmp (argv[2], "new") == 0)
+                                        {
+                                               int grid_size = atoi (argv[3]);
+                                               init_puzzle (&p, grid_size);
+                                               puzzle_editor_loop (&p, argv[1]);
+                                               break;
+                                        }
+                       default: fprintf (stderr, USAGE_LINE_1, argv[0]);
+                                        fprintf (stderr, USAGE_LINE_2);
+                                        fprintf (stderr, USAGE_LINE_3);
+                                        exit (3);
+               }
+       }
        return (main_loop ());
 }