home · contact · privacy
Added memory allocation error checks to windows library.
[plomrogue] / src / main.c
1 /* main.c */
2
3 #include "main.h"
4 #include <stdlib.h> /* for atoi(), exit(), EXIT_FAILURE, calloc() */
5 #include <stdio.h> /* for FILE typedef, F_OK, rename() */
6 #include <ncurses.h> /* for initscr(), noecho(), curs_set(), keypad(), raw() */
7 #include <time.h> /* for time() */
8 #include <unistd.h> /* for unlink(), getopt(), optarg */
9 #include <stdint.h> /* for uint8_t */
10 #include "windows.h" /* for structs WinMeta, Win, init_win(), init_win_meta(),
11                       * draw_all_wins()
12                       */
13 #include "draw_wins.h" /* for draw_keys_win(), draw_map_win(), draw_info_win(),
14                         * draw_log_win()
15                         */
16 #include "keybindings.h" /* for initkeybindings(), get_action_key() */
17 #include "readwrite.h" /* for [read/write]_uint[8/16/32][_bigendian]() */
18 #include "map_objects.h" /* for structs Monster, Item, Player,
19                           * init_map_object_defs(), read_map_objects(),
20                           * build_map_objects()
21                           */
22 #include "map_object_actions.h" /* for player_wait(), move_player() */
23 #include "map.h" /* for struct Map, init_map() */
24 #include "misc.h" /* for update_log(), toggle_window(), find_passable_pos(),
25                    * meta_keys(), save_game()
26                    */
27 #include "yx_uint16.h" /* for dir enum */
28 #include "rrand.h" /* for rrand(), rrand_seed() */
29 #include "rexit.h" /* for exit_game() */
30
31 int main(int argc, char *argv[])
32 {
33     struct World world;
34     char * recordfile_tmp = "record_tmp";
35     char * savefile_tmp   = "savefile_tmp";
36     char * err_x = "A file 'recordfile_tmp' exists, probably from a corrupted "
37                    "previous record saving process. To avoid game record "
38                    "corruption, I won't start until it is removed or renamed.";
39     exit_err(!access(recordfile_tmp, F_OK), &world, err_x);
40     err_x        = "A file 'savefile_tmp' exists, probably from a corrupted "
41                    "previous game saving process. To avoid savegame "
42                    "corruption, I won't start until it is removed or renamed.";
43     exit_err(!access(savefile_tmp,   F_OK), &world, err_x);
44
45     /* Read in startup options (i.e. replay option and replay start turn). */
46     int opt;
47     uint32_t start_turn;
48     world.interactive = 1;
49     while ((opt = getopt(argc, argv, "s::")) != -1)
50     {
51         switch (opt)
52         {
53             case 's':
54                 world.interactive = 0;
55                 start_turn = 0;
56                 if (optarg)
57                     start_turn = atoi(optarg);
58                  break;
59             default:
60                 exit(EXIT_FAILURE);
61         }
62     }
63
64     /* Initialize log, player, monster/item definitions and monsters/items. */
65     world.log = calloc(1, sizeof(char));
66     set_cleanup_flag(CLEANUP_LOG);
67     update_log (&world, " ");
68     struct Player player;
69     player.hitpoints = 5;
70     world.player = &player;
71     world.monster = 0;
72     world.item = 0;
73     init_map_object_defs(&world, "defs");
74
75     /* For interactive mode, try to load world state from savefile. */
76     char * err_o = "Error loading game: "
77                    "Unable to open 'savefile' for reading.";
78     char * err_r = "Error loading game: "
79                    "Trouble reading from opened 'savefile'.";
80     char * err_c = "Error loading game: "
81                    "Unable to close opened 'savefile'.";
82     char * savefile = "savefile";
83     FILE * file;
84     if (1 == world.interactive && 0 == access(savefile, F_OK))
85     {
86         file = fopen(savefile, "r");
87         exit_err(0 == file, &world, err_o);
88         if (   read_uint32_bigendian(file, &world.seed)
89             || read_uint32_bigendian(file, &world.turn)
90             || read_uint16_bigendian(file, &player.pos.y)
91             || read_uint16_bigendian(file, &player.pos.x)
92             || read_uint8(file, &player.hitpoints)
93             || read_map_objects(&world, &world.monster, file)
94             || read_map_objects(&world, &world.item,    file))
95         {
96             exit_err(1, &world, err_r);
97         }
98         exit_err(fclose(file), &world, err_c);
99         player.pos.y--;
100         player.pos.x--;
101     }
102
103     /* For non-interactive mode, try to load world state from record file. */
104     else
105     {
106         err_o = "Error loading record file: "
107                 "Unable to open file 'record' for reading.";
108         err_r = "Error loading record file: "
109                 "Trouble reading from opened file 'record'.";
110         char * recordfile = "record";
111         world.turn = 1;
112         if (0 == world.interactive)
113         {
114             file = fopen(recordfile, "r");
115             exit_err(0 == file, &world, err_o);
116             exit_err(read_uint32_bigendian(file, &world.seed), &world, err_r);
117         }
118
119         /* For interactive-mode in newly started world, generate a start seed
120          * from the current time.
121          */
122         else
123         {
124             world.seed = time(NULL);
125
126             err_x        = "Error recording new seed: "
127                            "A file 'record' already exists, when it shouldn't.";
128             err_o        = "Error recording new seed: "
129                            "Unable to open 'record_tmp' file for writing.";
130             char * err_w = "Error recording new seed: "
131                            "Trouble writing to opened file 'record_tmp'.";
132             err_c        = "Error recording new seed: "
133                            "Unable to close opened file 'record_tmp'.";
134             char * err_m = "Error recording new seed: "
135                            "Unable to rename file 'record_tmp' to 'record'.";
136             exit_err(!access(recordfile, F_OK), &world, err_x);
137             file = fopen(recordfile_tmp, "w");
138             exit_err(0 == file, &world, err_o);
139             exit_err(write_uint32_bigendian(world.seed, file), &world, err_w);
140             exit_err(fclose(file), &world, err_c);
141             exit_err(rename(recordfile_tmp, recordfile), &world, err_m);
142         }
143     }
144
145
146     /* Generate map from seed and, if newly generated world, start positions of
147      * actors.
148      */
149     rrand_seed(world.seed);
150     struct Map map = init_map();
151     world.map = &map;
152     set_cleanup_flag(CLEANUP_MAP);
153     if (1 == world.turn)
154     {
155         player.pos = find_passable_pos(world.map);
156         void * foo;
157         foo = build_map_objects(&world, &world.monster, 1, 1 + rrand() % 27);
158         foo = build_map_objects(&world, foo, 2, 1 + rrand() % 9);
159         build_map_objects(&world, foo, 3, 1 + rrand() % 3);
160         foo = build_map_objects(&world, &world.item, 4, 1 + rrand() % 3);
161         build_map_objects(&world, foo, 5, 1 + rrand() % 3);
162     }
163
164     /* Initialize window system and windows. */
165     WINDOW * screen = initscr();
166     set_cleanup_flag(CLEANUP_NCURSES);
167     noecho();
168     curs_set(0);
169     keypad(screen, TRUE);
170     raw();
171     init_keybindings(&world);
172     set_cleanup_flag(CLEANUP_KEYBINDINGS);
173     struct WinMeta win_meta;
174     init_win_meta(screen, &win_meta);
175     struct Win win_keys = init_win(&win_meta, "Keys",
176                                    0, 29, &world, draw_keys_win);
177     struct Win win_info = init_win(&win_meta, "Info",
178                                    2, 20, &world, draw_info_win);
179     uint16_t height_logwin = win_meta.padframe.size.y
180                              - (2 + win_info.frame.size.y);
181     struct Win win_log = init_win(&win_meta, "Log",
182                                   height_logwin, 20, &world, draw_log_win);
183     uint16_t width_mapwin = win_meta.padframe.size.x - win_keys.frame.size.x
184                              - win_log.frame.size.x - 2;
185     struct Win win_map = init_win(&win_meta, "Map",
186                                   0, width_mapwin, &world, draw_map_win);
187     toggle_window(&win_meta, &win_keys);
188     toggle_window(&win_meta, &win_map);
189     toggle_window(&win_meta, &win_info);
190     toggle_window(&win_meta, &win_log);
191
192     /* Replay mode. */
193     int key;
194     uint8_t quit_called = 0;
195     uint8_t await_actions = 1;
196     if (0 == world.interactive)
197     {
198         int action;
199         while (1)
200         {
201             if (start_turn == world.turn)
202             {
203                 start_turn = 0;
204             }
205             if (0 == start_turn)
206             {
207                 draw_all_wins (&win_meta);
208                key = getch();
209             }
210             if (1 == await_actions
211                 && (world.turn < start_turn
212                     || key == get_action_key(world.keybindings,
213                                              "wait / next turn")) )
214             {
215                 action = getc(file);
216                 if (EOF == action)
217                 {
218                     start_turn = 0;
219                     await_actions = 0;
220                 }
221                 else if (0 == action)
222                 {
223                     player_wait (&world);
224                 }
225                 else if (NORTH == action)
226                 {
227                     move_player(&world, NORTH);
228                 }
229                 else if (EAST  == action)
230                 {
231                     move_player(&world, EAST);
232                 }
233                 else if (SOUTH == action)
234                 {
235                     move_player(&world, SOUTH);
236                 }
237                 else if (WEST == action)
238                 {
239                     move_player(&world, WEST);
240                 }
241             }
242             else
243             {
244                 quit_called = meta_keys(key, &world, &win_meta, &win_keys,
245                                              &win_map, &win_info, &win_log);
246                 if (1 == quit_called)
247                 {
248                     err_c = "Error closing read 'record' file.";
249                     exit_err(fclose(file), &world, err_c);
250                     exit_game(&world);
251                 }
252             }
253         }
254     }
255
256     /* Interactive mode. */
257     else
258     {
259         uint32_t last_turn = 0;
260         while (1)
261         {
262             if (last_turn != world.turn)
263             {
264                 save_game(&world);
265                 last_turn = world.turn;
266             }
267             if (1 == await_actions && 0 == player.hitpoints)
268             {
269                 await_actions = 0;
270             }
271             draw_all_wins (&win_meta);
272             key = getch();
273             if      (1 == await_actions
274                      && key == get_action_key(world.keybindings,
275                                               "player up"))
276             {
277                 move_player(&world, NORTH);
278             }
279             else if (1 == await_actions
280                      && key == get_action_key(world.keybindings,
281                                               "player right"))
282             {
283                 move_player(&world, EAST);
284             }
285             else if (1 == await_actions
286                      && key == get_action_key(world.keybindings,
287                                               "player down"))
288             {
289                 move_player(&world, SOUTH);
290             }
291             else if (1 == await_actions
292                      && key == get_action_key(world.keybindings,
293                                               "player left"))
294             {
295                 move_player(&world, WEST);
296             }
297             else if (1 == await_actions
298                      && key == get_action_key(world.keybindings,
299                                               "wait / next turn"))
300             {
301                 player_wait (&world);
302             }
303             else
304             {
305                 quit_called = meta_keys(key, &world, &win_meta, &win_keys,
306                                              &win_map, &win_info, &win_log);
307                 if (1 == quit_called)
308                 {
309                     exit_game(&world);
310                 }
311             }
312         }
313     }
314 }