main.c

   1 #include "main.h"
   2
   3 #define DBG 0
   4
   5 bool move_possible_north(board *b);
   6 bool move_possible_south(board *b);
   7 bool move_possible_east(board *b);
   8 bool move_possible_west(board *b);
   9
  10 void move_north(board *b);
  11 void move_south(board *b);
  12 void move_east(board *b);
  13 void move_west(board *b);
  14 void merge(board *b, const int d);
  15 void merge_north(board *b);
  16 void merge_south(board *b);
  17 void merge_east(board *b);
  18 void merge_west(board *b);
  19
  20 void place_new_piece(board *b);
  21 void game_loop(board *b);
  22
  23 board* new_board() {
  24         board *tmp = malloc(sizeof(board));
  25         init_board(tmp);
  26         return tmp;
  27 }
  28
  29 void init_board(board *b) {
  30         for(int i = 0; i < 4; ++i) {
  31                 for(int j = 0; j < 4; ++j)
  32                         b->x[i][j] = 0;
  33         }
  34         b->points = 0;
  35         b->num_p = 0;
  36         srand((uint)time(NULL));
  37         place_new_piece(b);
  38         // second piece will be placed in game_loop
  39 }
  40
  41 void game_start() {
  42         printf("\e[?1049h\e[?25l]");
  43         struct termios oldtio, newtio;
  44         tcgetattr(STDIN_FILENO, &oldtio);
  45         newtio = oldtio;
  46         newtio.c_lflag &= ~(ICANON | ECHO);
  47         tcsetattr(STDIN_FILENO, TCSANOW, &newtio);
  48         board *b = new_board();
  49         game_loop(b);
  50         printf("\e[?1049l\e[?25h");
  51         tcsetattr(STDIN_FILENO, TCSANOW, &oldtio);
  52         print_score(b);
  53         free_board(b);
  54 }
  55
  56 enum direction get_input()
  57 {
  58         switch(fgetc(stdin)) {
  59                 case 'a':
  60                         return west;
  61                 case 's':
  62                         return south;
  63                 case 'w':
  64                         return north;
  65                 case 'd':
  66                         return east;
  67                 case 'q':
  68                         return quit;
  69                 default:
  70                         return get_input();
  71         }
  72 }
  73
  74 void game_loop(board *b) {
  75         while(move_possible_any(b)) {
  76                 place_new_piece(b);
  77                 print_board(b);
  78                 while(true) {
  79                         enum direction d = get_input();
  80
  81                         if (d == quit) {
  82                                 return;
  83                         } else if(move_possible(b, d)) {
  84                                 make_move(b, d);
  85                                 break;
  86                         }
  87                 }
  88         }
  89 }
  90
  91 void place_new_piece(board *b) {
  92         int prob = rand() % 100;
  93         int possibs[32];
  94         uint val = 0;
  95         int cnt = 0;
  96         int pair = 0;
  97         int rx = 0, ry = 0;
  98         for(int i = 0; i < 4; ++i) {
  99                 for(int j = 0; j < 4; ++j) {
 100                         if(b->x[i][j] == 0) {
 101                                 possibs[2*cnt] = i;
 102                                 possibs[2*cnt+1] = j;
 103                                 cnt++;
 104                         }
 105                 }
 106         }
 107         assert(cnt > 0);
 108         pair = rand() % cnt;
 109         rx = possibs[2*pair];
 110         ry = possibs[2*pair+1];
 111         assert(b->x[rx][ry] == 0);
 112         if(prob > 9)
 113                 val = 2;
 114         else
 115                 val = 4;
 116         b->x[rx][ry] = val;
 117         b->num_p++;
 118 }
 119
 120 void free_board(board *b) {
 121         free(b);
 122 }
 123
 124 void make_move(board *b, const int d) {
 125         bool ok = (d == south) ||
 126                 (d == east) ||
 127                 (d == north) ||
 128                 (d == west);
 129         assert(ok);
 130         /*
 131          * checked before already
 132          * if(!move_possible(b, d))
 133          *      return;
 134         */
 135         move(b, d);
 136         merge(b, d);
 137         move(b, d);
 138 }
 139
 140 bool move_possible_any(board *b) {
 141         if(b->num_p < 16)
 142                 return true;
 143         if(move_possible_south(b))
 144                 return true;
 145         if(move_possible_east(b))
 146                 return true;
 147         if(move_possible_north(b))
 148                 return true;
 149         if(move_possible_west(b))
 150                 return true;
 151         return false;
 152 }
 153
 154 bool move_possible(board *b, const int d) {
 155         bool ok = (d == south) ||
 156                 (d == east) ||
 157                 (d == north) ||
 158                 (d == west);
 159         assert(ok);
 160         switch(d) {
 161                 case north:
 162                         return move_possible_north(b);
 163                 case south:
 164                         return move_possible_south(b);
 165                 case east:
 166                         return move_possible_east(b);
 167                 case west:
 168                         return move_possible_west(b);
 169                 default: // will never execute anyway
 170                         return false;
 171         }
 172 }
 173
 174 void move(board *b, const int d) {
 175         bool ok = (d == south) ||
 176                 (d == east) ||
 177                 (d == north) ||
 178                 (d == west);
 179         assert(ok);
 180         switch(d) {
 181                 case north:
 182                         move_north(b);
 183                         break;
 184                 case south:
 185                         move_south(b);
 186                         break;
 187                 case east:
 188                          move_east(b);
 189                         break;
 190                 case west:
 191                         move_west(b);
 192                         break;
 193                 default: // will never execute anyway
 194                         return;
 195         }
 196 }
 197
 198 void merge(board *b, const int d) {
 199         bool ok = (d == south) ||
 200                 (d == east) ||
 201                 (d == north) ||
 202                 (d == west);
 203         assert(ok);
 204         switch(d) {
 205                 case north:
 206                         merge_north(b);
 207                         break;
 208                 case south:
 209                         merge_south(b);
 210                         break;
 211                 case east:
 212                         merge_east(b);
 213                         break;
 214                 case west:
 215                         merge_west(b);
 216                         break;
 217                 default: // will never execute anyway
 218                         return;
 219         }
 220 }
 221
 222 bool move_possible_south(board *b) {
 223         for(int i = 0; i < 4; ++i) {
 224                 for(int j = 0; j < 3; ++j) {
 225                         if(b->x[i][j] != 0) {
 226                                 if(b->x[i][j+1] == 0)
 227                                         return true;
 228                                 if(b->x[i][j] == b->x[i][j+1])
 229                                         return true;
 230                         }
 231                 }
 232         }
 233         return false;
 234 }
 235
 236 bool move_possible_north(board *b) {
 237         for(int i = 0; i < 4; ++i) {
 238                 for(int j = 1; j < 4; ++j) {
 239                         if(b->x[i][j] != 0) {
 240                                 if(b->x[i][j-1] == 0)
 241                                         return true;
 242                                 if(b->x[i][j] == b->x[i][j-1])
 243                                         return true;
 244                         }
 245                 }
 246         }
 247         return false;
 248 }
 249
 250 bool move_possible_east(board *b) {
 251         for(int i = 0; i < 3; ++i) {
 252                 for(int j = 0; j < 4; ++j) {
 253                         if(b->x[i][j] != 0) {
 254                                 if(b->x[i+1][j] == 0)
 255                                         return true;
 256                                 if(b->x[i+1][j] == b->x[i][j])
 257                                         return true;
 258                         }
 259                 }
 260         }
 261         return false;
 262 }
 263
 264 bool move_possible_west(board *b) {
 265         for(int i = 1; i < 4; ++i) {
 266                 for(int j = 0; j < 4; ++j) {
 267                         if(b->x[i][j] != 0) {
 268                                 if(b->x[i-1][j] == 0)
 269                                         return true;
 270                                 if(b->x[i-1][j] == b->x[i][j])
 271                                         return true;
 272                         }
 273                 }
 274         }
 275         return false;
 276 }
 277
 278 void move_north(board *b) {
 279         int k;
 280         for(int i = 0; i < 4; ++i) {
 281                 for(int j = 1; j < 4; ++j) {
 282                         if(b->x[i][j] != 0) {
 283                                 k = j;
 284                                 while(b->x[i][k-1] == 0) {
 285                                         b->x[i][k-1] = b->x[i][k];
 286                                         b->x[i][k] = 0;
 287                                         k--;
 288                                         if(k < 1)
 289                                                 break;
 290                                 }
 291                         }
 292                 }
 293         }
 294 }
 295
 296 void move_south(board *b) {
 297         int k;
 298         for(int i = 0; i < 4; ++i) {
 299                 for(int j = 2; j >= 0; --j) {
 300                         if(b->x[i][j] != 0) {
 301                                 k = j;
 302                                 while(b->x[i][k+1] == 0) {
 303                                         b->x[i][k+1] = b->x[i][k];
 304                                         b->x[i][k] = 0;
 305                                         k++;
 306                                         if(k > 2)
 307                                                 break;
 308                                 }
 309                         }
 310                 }
 311         }
 312 }
 313
 314 void move_east(board *b) {
 315         int k;
 316         for(int i = 2; i >= 0; --i) {
 317                 for(int j = 0; j < 4; ++j) {
 318                         if(b->x[i][j] != 0) {
 319                                 k = i;
 320                                 while(b->x[k+1][j] == 0) {
 321                                         b->x[k+1][j] = b->x[k][j];
 322                                         b->x[k][j] = 0;
 323                                         k++;
 324                                         if(k > 2)
 325                                                 break;
 326                                 }
 327                         }
 328                 }
 329         }
 330 }
 331
 332 void move_west(board *b) {
 333         int k;
 334         for(int i = 1; i < 4; ++i) {
 335                 for(int j = 0; j < 4; ++j) {
 336                         if(b->x[i][j] != 0) {
 337                                 k = i;
 338                                 while(b->x[k-1][j] == 0) {
 339                                         b->x[k-1][j] = b->x[k][j];
 340                                         b->x[k][j] = 0;
 341                                         k--;
 342                                         if(k < 1)
 343                                                 break;
 344                                 }
 345                         }
 346                 }
 347         }
 348 }
 349
 350 void merge_north(board *b) {
 351         for(int i = 0; i < 4; ++i) {
 352                 for(int j = 1; j < 4; ++j) {
 353                         if(b->x[i][j] != 0 && b->x[i][j] == b->x[i][j-1]) {
 354                                 b->x[i][j-1] *= 2;
 355                                 b->x[i][j] = 0;
 356                                 b->num_p--;
 357                                 b->points += b->x[i][j-1];
 358                         }
 359                 }
 360         }
 361 }
 362
 363 void merge_south(board *b) {
 364         for(int i = 0; i < 4; ++i) {
 365                 for(int j = 2; j >= 0; --j) {
 366                         if(b->x[i][j] != 0 && b->x[i][j] == b->x[i][j+1]) {
 367                                 b->x[i][j+1] *= 2;
 368                                 b->x[i][j] = 0;
 369                                 b->num_p--;
 370                                 b->points += b->x[i][j+1];
 371                         }
 372                 }
 373         }
 374 }
 375
 376 void merge_east(board *b) {
 377         for(int i = 2; i >= 0; --i) {
 378                 for(int j = 0; j < 4; ++j) {
 379                         if(b->x[i][j] != 0 && b->x[i+1][j] == b->x[i][j]) {
 380                                 b->x[i+1][j] *= 2;
 381                                 b->points += b->x[i+1][j];
 382                                 b->num_p--;
 383                                 b->x[i][j] = 0;
 384                         }
 385                 }
 386         }
 387 }
 388
 389 void merge_west(board *b) {
 390         for(int i = 1; i < 4; ++i) {
 391                 for(int j = 0; j < 4; ++j) {
 392                         if(b->x[i][j] != 0 && b->x[i-1][j] == b->x[i][j]) {
 393                                 b->x[i-1][j] *= 2;
 394                                 b->points += b->x[i-1][j];
 395                                 b->num_p--;
 396                                 b->x[i][j] = 0;
 397                         }
 398                 }
 399         }
 400 }
 401
 402 void print_sep(const char *left, const char *right, const char *cross, const char *line)
 403 {
 404         printf("%s", left);
 405         for(int i = 0; i < 4; i++) {
 406                 for(int j = 0; j < 4; j++)
 407                         printf("%s", line);
 408                 if(i == 3)
 409                         printf("%s", right);
 410                 else
 411                         printf("%s", cross);
 412         }
 413         printf("\n");
 414 }
 415
 416 void print_board_line(board *b, int l) {
 417         printf("\u2503");
 418
 419         for(int i = 0; i < 4; i++) {
 420                 uint n = b->x[i][l];
 421
 422                 if(n == 0)
 423                         printf("    ");
 424                 else
 425                         printf("%4u", n);
 426
 427                 if(i == 3)
 428                         printf("\u2503");
 429                 else
 430                         printf("\u2502");
 431         }
 432
 433         printf("\n");
 434
 435         if(l == 3)
 436                 print_sep("\u2517", "\u251B", "\u2537", "\u2501");
 437         else
 438                 print_sep("\u2520", "\u2528", "\u253C", "\u2500");
 439 }
 440
 441 void print_board(board *b) {
 442         printf("\e[2J\e[0;0H");
 443         printf("\e[1mScore: \e[0m%u\n", b->points);
 444         print_sep("\u250F", "\u2513", "\u252F", "\u2501");
 445         for(int i = 0; i < 4; ++i) {
 446                 print_board_line(b, i);
 447         }
 448 }
 449
 450 void print_score(board *b) {
 451         printf("\e[1m\e[91mGame Over\e[0m\n\e[1mScore: \e[0m%u\n", b->points);
 452 }
 453
 454 void merge_test1() {
 455         board *b = new_board();
 456         b->x[2][0] = 2;
 457         b->x[3][0] = 2;
 458         merge_east(b);
 459         assert(b->x[3][0] == 4);
 460 }
 461
 462 void merge_test2() {
 463         board *b = new_board();
 464         b->x[2][0] = 2;
 465         b->x[3][0] = 2;
 466         print_board(b);
 467         move_west(b);
 468         print_board(b);
 469         merge_west(b);
 470         print_board(b);
 471         assert(b->x[0][0] == 4);
 472 }
 473
 474 void merge_test3() {
 475         board *b = new_board();
 476         b->x[0][0] = 2;
 477         b->x[0][1] = 2;
 478         print_board(b);
 479         merge_north(b);
 480         print_board(b);
 481         assert(b->x[0][0] == 4);
 482 }
 483
 484 void merge_test4() {
 485         board *b = new_board();
 486         b->x[0][2] = 2;
 487         b->x[0][3] = 2;
 488         print_board(b);
 489         merge_south(b);
 490         print_board(b);
 491         assert(b->x[0][3] == 4);
 492 }
 493
 494 void move_merge_test1() {
 495         board *b = new_board();
 496         b->x[2][0] = 4;
 497         b->x[3][0] = 2;
 498         b->x[3][2] = 2;
 499         print_board(b);
 500         move_south(b);
 501         print_board(b);
 502         merge_south(b);
 503         print_board(b);
 504         assert(b->x[3][3] == 4);
 505 }
 506
 507 void move_north_test() {
 508         board *b = new_board();
 509         b->x[2][3] = 8;
 510         b->x[2][2] = 0;
 511         b->x[2][1] = 0;
 512         b->x[2][0] = 0;
 513         b->x[0][1] = 2;
 514         b->x[0][0] = 0;
 515         b->x[3][0] = 4;
 516         b->x[3][1] = 2;
 517         b->x[3][2] = 3;
 518         b->x[3][3] = 16;
 519         print_board(b);
 520         move_north(b);
 521         merge_north(b);
 522         move_north(b);
 523         print_board(b);
 524         assert(b->x[2][0] == 8);
 525         assert(b->x[0][0] == 2);
 526 }
 527
 528 void move_merge_test2() {
 529         board *b = new_board();
 530         b->x[3][0] = 2;
 531         b->x[3][1] = 2;
 532         b->x[3][2] = 4;
 533         b->x[3][3] = 32;
 534         print_board(b);
 535         move_south(b);
 536         merge_south(b);
 537         move_south(b);
 538         print_board(b);
 539         assert(b->x[3][1] == 4);
 540 }
 541
 542 void move_merge_test3() {
 543         board *b = new_board();
 544         b->x[0][0] = 0;
 545         b->x[0][1] = 0;
 546         b->x[0][2] = 0;
 547         b->x[0][3] = 2;
 548         b->x[1][0] = 0;
 549         b->x[1][1] = 0;
 550         b->x[1][2] = 0;
 551         b->x[1][3] = 8;
 552         b->x[2][0] = 0;
 553         b->x[2][1] = 0;
 554         b->x[2][2] = 0;
 555         b->x[2][3] = 4;
 556         b->x[3][0] = 2;
 557         b->x[3][1] = 2;
 558         b->x[3][2] = 4;
 559         b->x[3][3] = 16;
 560         print_board(b);
 561         move_south(b);
 562         merge_south(b);
 563         move_south(b);
 564         print_board(b);
 565         assert(b->x[3][1] == 4);
 566 }
 567
 568 void run_tests() {
 569         /*
 570         merge_test1();
 571         merge_test2();
 572         merge_test3();
 573         merge_test4();
 574         move_merge_test1();
 575         move_north_test();
 576         */
 577         move_merge_test2();
 578         move_merge_test3();
 579 }
 580
 581 int main() {
 582 #if DBG
 583         run_tests();
 584 #else
 585         game_start();
 586 #endif
 587 }