guess.c

   1 /*
   2  * guess.c: Mastermind clone.
   3  */
   4
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7 #include <string.h>
   8 #include <assert.h>
   9 #include <ctype.h>
  10 #include <math.h>
  11
  12 #include "puzzles.h"
  13
  14 #define FLASH_FRAME 0.5F
  15
  16 enum {
  17     COL_BACKGROUND,
  18     COL_HIGHLIGHT, COL_LOWLIGHT, COL_FRAME, COL_FLASH, COL_HOLD,
  19     COL_EMPTY, /* must be COL_1 - 1 */
  20     COL_1, COL_2, COL_3, COL_4, COL_5, COL_6, COL_7, COL_8, COL_9, COL_10,
  21     COL_CORRECTPLACE, COL_CORRECTCOLOUR,
  22     NCOLOURS
  23 };
  24
  25 struct game_params {
  26     int ncolours, npegs, nguesses;
  27 };
  28
  29 #define FEEDBACK_CORRECTPLACE  1
  30 #define FEEDBACK_CORRECTCOLOUR 2
  31
  32 typedef struct pegrow {
  33     int npegs;
  34     int *pegs;          /* 0 is 'empty' */
  35     int *feedback;      /* may well be unused */
  36 } *pegrow;
  37
  38 struct game_state {
  39     game_params params;
  40     pegrow *guesses;  /* length params->nguesses */
  41     pegrow solution;
  42     int next_go; /* from 0 to nguesses-1;
  43                     if next_go == nguesses then they've lost. */
  44     int solved;
  45 };
  46
  47 static game_params *default_params(void)
  48 {
  49     game_params *ret = snew(game_params);
  50
  51     /* AFAIK this is the canonical Mastermind ruleset. */
  52     ret->ncolours = 6;
  53     ret->npegs = 4;
  54     ret->nguesses = 10;
  55
  56     return ret;
  57 }
  58
  59 static int game_fetch_preset(int i, char **name, game_params **params)
  60 {
  61     return FALSE;
  62 }
  63
  64 static void free_params(game_params *params)
  65 {
  66     sfree(params);
  67 }
  68
  69 static game_params *dup_params(game_params *params)
  70 {
  71     game_params *ret = snew(game_params);
  72     *ret = *params;                    /* structure copy */
  73     return ret;
  74 }
  75
  76 static void decode_params(game_params *params, char const *string)
  77 {
  78     char const *p = string;
  79     game_params *defs = default_params();
  80
  81     *params = *defs; free_params(defs);
  82
  83     while (*p) {
  84         switch (*p++) {
  85         case 'c':
  86             params->ncolours = atoi(p);
  87             while (*p && isdigit((unsigned char)*p)) p++;
  88             break;
  89
  90         case 'p':
  91             params->npegs = atoi(p);
  92             while (*p && isdigit((unsigned char)*p)) p++;
  93             break;
  94
  95         case 'g':
  96             params->nguesses = atoi(p);
  97             while (*p && isdigit((unsigned char)*p)) p++;
  98             break;
  99
 100         default:
 101             ;
 102         }
 103     }
 104 }
 105
 106 static char *encode_params(game_params *params, int full)
 107 {
 108     char data[256];
 109
 110     sprintf(data, "c%dp%dg%d", params->ncolours, params->npegs, params->nguesses);
 111
 112     return dupstr(data);
 113 }
 114
 115 static config_item *game_configure(game_params *params)
 116 {
 117     config_item *ret;
 118     char buf[80];
 119
 120     ret = snewn(4, config_item);
 121
 122     ret[0].name = "No. of colours";
 123     ret[0].type = C_STRING;
 124     sprintf(buf, "%d", params->ncolours);
 125     ret[0].sval = dupstr(buf);
 126     ret[0].ival = 0;
 127
 128     ret[1].name = "No. of pegs per row";
 129     ret[1].type = C_STRING;
 130     sprintf(buf, "%d", params->npegs);
 131     ret[1].sval = dupstr(buf);
 132     ret[1].ival = 0;
 133
 134     ret[2].name = "No. of guesses";
 135     ret[2].type = C_STRING;
 136     sprintf(buf, "%d", params->nguesses);
 137     ret[2].sval = dupstr(buf);
 138     ret[2].ival = 0;
 139
 140     ret[3].name = NULL;
 141     ret[3].type = C_END;
 142     ret[3].sval = NULL;
 143     ret[3].ival = 0;
 144
 145     return ret;
 146 }
 147
 148 static game_params *custom_params(config_item *cfg)
 149 {
 150     game_params *ret = snew(game_params);
 151
 152     ret->ncolours = atoi(cfg[0].sval);
 153     ret->npegs = atoi(cfg[1].sval);
 154     ret->nguesses = atoi(cfg[2].sval);
 155
 156     return ret;
 157 }
 158
 159 static char *validate_params(game_params *params)
 160 {
 161     if (params->ncolours < 2 || params->npegs < 2)
 162         return "Trivial solutions are uninteresting";
 163     /* NB as well as the no. of colours we define, max(ncolours) must
 164      * also fit in an unsigned char; see new_game_desc. */
 165     if (params->ncolours > 10)
 166         return "Too many colours";
 167     if (params->nguesses < 1)
 168         return "Must have at least one guess";
 169     return NULL;
 170 }
 171
 172 static pegrow new_pegrow(int npegs)
 173 {
 174     pegrow pegs = snew(struct pegrow);
 175
 176     pegs->npegs = npegs;
 177     pegs->pegs = snewn(pegs->npegs, int);
 178     memset(pegs->pegs, 0, pegs->npegs * sizeof(int));
 179     pegs->feedback = snewn(pegs->npegs, int);
 180     memset(pegs->feedback, 0, pegs->npegs * sizeof(int));
 181
 182     return pegs;
 183 }
 184
 185 static pegrow dup_pegrow(pegrow pegs)
 186 {
 187     pegrow newpegs = snew(struct pegrow);
 188
 189     newpegs->npegs = pegs->npegs;
 190     newpegs->pegs = snewn(newpegs->npegs, int);
 191     memcpy(newpegs->pegs, pegs->pegs, newpegs->npegs * sizeof(int));
 192     newpegs->feedback = snewn(newpegs->npegs, int);
 193     memcpy(newpegs->feedback, pegs->feedback, newpegs->npegs * sizeof(int));
 194
 195     return newpegs;
 196 }
 197
 198 static void invalidate_pegrow(pegrow pegs)
 199 {
 200     memset(pegs->pegs, -1, pegs->npegs * sizeof(int));
 201     memset(pegs->feedback, -1, pegs->npegs * sizeof(int));
 202 }
 203
 204 static void free_pegrow(pegrow pegs)
 205 {
 206     sfree(pegs->pegs);
 207     sfree(pegs->feedback);
 208     sfree(pegs);
 209 }
 210
 211 static char *new_game_desc(game_params *params, random_state *rs,
 212                            game_aux_info **aux, int interactive)
 213 {
 214     unsigned char *bmp = snewn(params->npegs, unsigned char);
 215     char *ret;
 216     int i;
 217
 218     for (i = 0; i < params->npegs; i++)
 219         bmp[i] = (unsigned char)(random_upto(rs, params->ncolours)+1);
 220     obfuscate_bitmap(bmp, params->npegs*8, FALSE);
 221
 222     ret = bin2hex(bmp, params->npegs);
 223     sfree(bmp);
 224     return ret;
 225 }
 226
 227 static void game_free_aux_info(game_aux_info *aux)
 228 {
 229     assert(!"Shouldn't happen");
 230 }
 231
 232 static char *validate_desc(game_params *params, char *desc)
 233 {
 234     /* desc is just an (obfuscated) bitmap of the solution; all we
 235      * care is that it's the correct length. */
 236     if (strlen(desc) != params->npegs * 2)
 237         return "Game description is wrong length";
 238     return NULL;
 239 }
 240
 241 static game_state *new_game(midend_data *me, game_params *params, char *desc)
 242 {
 243     game_state *state = snew(game_state);
 244     unsigned char *bmp;
 245     int i;
 246
 247     state->params = *params;
 248     state->guesses = snewn(params->nguesses, pegrow);
 249     for (i = 0; i < params->nguesses; i++)
 250         state->guesses[i] = new_pegrow(params->npegs);
 251     state->solution = new_pegrow(params->npegs);
 252
 253     bmp = hex2bin(desc, params->npegs);
 254     obfuscate_bitmap(bmp, params->npegs*8, TRUE);
 255     for (i = 0; i < params->npegs; i++)
 256         state->solution->pegs[i] = (int)bmp[i];
 257     sfree(bmp);
 258
 259     state->next_go = state->solved = 0;
 260
 261     return state;
 262 }
 263
 264 static game_state *dup_game(game_state *state)
 265 {
 266     game_state *ret = snew(game_state);
 267     int i;
 268
 269     *ret = *state;
 270     ret->guesses = snewn(state->params.nguesses, pegrow);
 271     for (i = 0; i < state->params.nguesses; i++)
 272         ret->guesses[i] = dup_pegrow(state->guesses[i]);
 273     ret->solution = dup_pegrow(state->solution);
 274
 275     return ret;
 276 }
 277
 278 static void free_game(game_state *state)
 279 {
 280     int i;
 281
 282     free_pegrow(state->solution);
 283     for (i = 0; i < state->params.nguesses; i++)
 284         free_pegrow(state->guesses[i]);
 285     sfree(state->guesses);
 286
 287     sfree(state);
 288 }
 289
 290 static game_state *solve_game(game_state *state, game_state *currstate,
 291                               game_aux_info *aux, char **error)
 292 {
 293     game_state *ret = dup_game(currstate);
 294     ret->solved = 1;
 295     return ret;
 296 }
 297
 298 static char *game_text_format(game_state *state)
 299 {
 300     return NULL;
 301 }
 302
 303 struct game_ui {
 304     pegrow curr_pegs; /* half-finished current move */
 305     int *holds;
 306     int colour_cur;   /* position of up-down colour picker cursor */
 307     int peg_cur;      /* position of left-right peg picker cursor */
 308     int display_cur, markable;
 309
 310     int drag_col, drag_x, drag_y; /* x and y are *center* of peg! */
 311 };
 312
 313 static game_ui *new_ui(game_state *state)
 314 {
 315     game_ui *ui = snew(struct game_ui);
 316     memset(ui, 0, sizeof(struct game_ui));
 317     ui->curr_pegs = new_pegrow(state->params.npegs);
 318     ui->holds = snewn(state->params.npegs, int);
 319     memset(ui->holds, 0, sizeof(int)*state->params.npegs);
 320     return ui;
 321 }
 322
 323 static void free_ui(game_ui *ui)
 324 {
 325     free_pegrow(ui->curr_pegs);
 326     sfree(ui->holds);
 327     sfree(ui);
 328 }
 329
 330 static void game_changed_state(game_ui *ui, game_state *oldstate,
 331                                game_state *newstate)
 332 {
 333     int i;
 334
 335     /* just clear the row-in-progress when we have an undo/redo. */
 336     for (i = 0; i < ui->curr_pegs->npegs; i++)
 337         ui->curr_pegs->pegs[i] = 0;
 338 }
 339
 340 #define PEGSZ   (ds->pegsz)
 341 #define PEGOFF  (ds->pegsz + ds->gapsz)
 342 #define HINTSZ  (ds->hintsz)
 343 #define HINTOFF (ds->hintsz + ds->gapsz)
 344
 345 #define PEGRAD  (ds->pegrad)
 346 #define HINTRAD (ds->hintrad)
 347
 348 #define COL_OX          (ds->colx)
 349 #define COL_OY          (ds->coly)
 350 #define COL_X(c)        (COL_OX)
 351 #define COL_Y(c)        (COL_OY + (c)*PEGOFF)
 352 #define COL_W           PEGOFF
 353 #define COL_H           (ds->colours->npegs*PEGOFF)
 354
 355 #define GUESS_OX        (ds->guessx)
 356 #define GUESS_OY        (ds->guessy)
 357 #define GUESS_X(g,p)    (GUESS_OX + (p)*PEGOFF)
 358 #define GUESS_Y(g,p)    (GUESS_OY + (g)*PEGOFF)
 359 #define GUESS_W         (ds->solution->npegs*PEGOFF)
 360 #define GUESS_H         (ds->nguesses*PEGOFF)
 361
 362 #define HINT_OX         (GUESS_OX + GUESS_W + ds->gapsz)
 363 #define HINT_OY         (GUESS_OY + (PEGSZ - HINTOFF - HINTSZ) / 2)
 364 #define HINT_X(g)       HINT_OX
 365 #define HINT_Y(g)       (HINT_OY + (g)*PEGOFF)
 366 #define HINT_W          (ds->hintw*HINTOFF)
 367 #define HINT_H          GUESS_H
 368
 369 #define SOLN_OX         GUESS_OX
 370 #define SOLN_OY         (GUESS_OY + GUESS_H + ds->gapsz + 2)
 371 #define SOLN_W          GUESS_W
 372 #define SOLN_H          PEGOFF
 373
 374 struct game_drawstate {
 375     int nguesses;
 376     pegrow *guesses;    /* same size as state->guesses */
 377     pegrow solution;    /* only displayed if state->solved */
 378     pegrow colours;     /* length ncolours, not npegs */
 379
 380     int *holds;
 381
 382     int pegsz, hintsz, gapsz; /* peg size (diameter), etc. */
 383     int pegrad, hintrad;      /* radius of peg, hint */
 384     int border;
 385     int colx, coly;     /* origin of colours vertical bar */
 386     int guessx, guessy; /* origin of guesses */
 387     int solnx, solny;   /* origin of solution */
 388     int hintw;          /* no. of hint tiles we're wide per row */
 389     int w, h, started, solved;
 390
 391     int colour_cur, peg_cur, display_cur; /* as in game_ui. */
 392     int next_go;
 393
 394     blitter *blit_peg;
 395     int drag_col, blit_ox, blit_oy;
 396 };
 397
 398 static void set_peg(game_ui *ui, int peg, int col)
 399 {
 400     int i;
 401
 402     ui->curr_pegs->pegs[peg] = col;
 403
 404     /* set to 'markable' if all of our pegs are filled. */
 405     for (i = 0; i < ui->curr_pegs->npegs; i++) {
 406         if (ui->curr_pegs->pegs[i] == 0) return;
 407     }
 408     debug(("UI is markable."));
 409     ui->markable = 1;
 410 }
 411
 412 static int mark_pegs(pegrow guess, pegrow solution, int ncols)
 413 {
 414     int nc_place = 0, nc_colour = 0, i, j;
 415
 416     assert(guess && solution && (guess->npegs == solution->npegs));
 417
 418     for (i = 0; i < guess->npegs; i++) {
 419         if (guess->pegs[i] == solution->pegs[i]) nc_place++;
 420     }
 421
 422     /* slight bit of cleverness: we have the following formula, from
 423      * http://mathworld.wolfram.com/Mastermind.html that gives:
 424      *
 425      * nc_colour = sum(colours, min(#solution, #guess)) - nc_place
 426      *
 427      * I think this is due to Knuth.
 428      */
 429     for (i = 1; i <= ncols; i++) {
 430         int n_guess = 0, n_solution = 0;
 431         for (j = 0; j < guess->npegs; j++) {
 432             if (guess->pegs[j] == i) n_guess++;
 433             if (solution->pegs[j] == i) n_solution++;
 434         }
 435         nc_colour += min(n_guess, n_solution);
 436     }
 437     nc_colour -= nc_place;
 438
 439     debug(("mark_pegs, %d pegs, %d right place, %d right colour",
 440            guess->npegs, nc_place, nc_colour));
 441     assert((nc_colour + nc_place) <= guess->npegs);
 442
 443     memset(guess->feedback, 0, guess->npegs*sizeof(int));
 444     for (i = 0, j = 0; i < nc_place; i++)
 445         guess->feedback[j++] = FEEDBACK_CORRECTPLACE;
 446     for (i = 0; i < nc_colour; i++)
 447         guess->feedback[j++] = FEEDBACK_CORRECTCOLOUR;
 448
 449     return nc_place;
 450 }
 451
 452 static game_state *mark_move(game_state *from, game_ui *ui)
 453 {
 454     int i, ncleared = 0, nc_place;
 455     game_state *to = dup_game(from);
 456
 457     for (i = 0; i < to->solution->npegs; i++) {
 458         to->guesses[from->next_go]->pegs[i] = ui->curr_pegs->pegs[i];
 459     }
 460     nc_place = mark_pegs(to->guesses[from->next_go], to->solution, to->params.ncolours);
 461
 462     if (nc_place == to->solution->npegs) {
 463         to->solved = 1; /* win! */
 464     } else {
 465         to->next_go = from->next_go + 1;
 466         if (to->next_go >= to->params.nguesses)
 467             to->solved = 1; /* 'lose' so we show the pegs. */
 468     }
 469
 470     for (i = 0; i < to->solution->npegs; i++) {
 471         if (!ui->holds[i] || to->solved) {
 472             ui->curr_pegs->pegs[i] = 0;
 473             ncleared++;
 474         }
 475         if (to->solved) ui->holds[i] = 0;
 476     }
 477     if (ncleared) {
 478         ui->markable = 0;
 479         if (ui->peg_cur == to->solution->npegs)
 480             ui->peg_cur--;
 481     }
 482
 483     return to;
 484 }
 485
 486 static game_state *make_move(game_state *from, game_ui *ui, game_drawstate *ds,
 487                              int x, int y, int button)
 488 {
 489     int over_col = 0;           /* one-indexed */
 490     int over_guess = -1;        /* zero-indexed */
 491     int over_past_guess_y = -1; /* zero-indexed */
 492     int over_past_guess_x = -1; /* zero-indexed */
 493     int over_hint = 0;          /* zero or one */
 494     game_state *ret = NULL;
 495
 496     int guess_ox = GUESS_X(from->next_go, 0);
 497     int guess_oy = GUESS_Y(from->next_go, 0);
 498
 499     if (from->solved) return NULL;
 500
 501     if (x >= COL_OX && x <= (COL_OX + COL_W) &&
 502         y >= COL_OY && y <= (COL_OY + COL_H)) {
 503         over_col = ((y - COL_OY) / PEGOFF) + 1;
 504     } else if (x >= guess_ox &&
 505                y >= guess_oy && y <= (guess_oy + GUESS_H)) {
 506         if (x <= (guess_ox + GUESS_W)) {
 507             over_guess = (x - guess_ox) / PEGOFF;
 508         } else {
 509             over_hint = 1;
 510         }
 511     } else if (x >= guess_ox &&
 512                y >= GUESS_OY && y < guess_oy) {
 513         over_past_guess_y = (y - GUESS_OY) / PEGOFF;
 514         over_past_guess_x = (x - guess_ox) / PEGOFF;
 515     }
 516     debug(("make_move: over_col %d, over_guess %d, over_hint %d,"
 517            " over_past_guess %d", over_col, over_guess, over_hint,
 518            over_past_guess));
 519
 520     assert(ds->blit_peg);
 521
 522     /* mouse input */
 523     if (button == LEFT_BUTTON) {
 524         if (over_col > 0) {
 525             ui->drag_col = over_col;
 526             debug(("Start dragging from colours"));
 527         } else if (over_guess > -1) {
 528             int col = ui->curr_pegs->pegs[over_guess];
 529             if (col) {
 530                 ui->drag_col = col;
 531                 debug(("Start dragging from a guess"));
 532             }
 533         } else if (over_past_guess_y > -1) {
 534             int col =
 535                 from->guesses[over_past_guess_y]->pegs[over_past_guess_x];
 536             if (col) {
 537                 ui->drag_col = col;
 538                 debug(("Start dragging from a past guess"));
 539             }
 540         }
 541         if (ui->drag_col) {
 542             ui->drag_x = x;
 543             ui->drag_y = y;
 544             debug(("Start dragging, col = %d, (%d,%d)",
 545                    ui->drag_col, ui->drag_x, ui->drag_y));
 546             ret = from;
 547         }
 548     } else if (button == LEFT_DRAG && ui->drag_col) {
 549         ui->drag_x = x;
 550         ui->drag_y = y;
 551         debug(("Keep dragging, (%d,%d)", ui->drag_x, ui->drag_y));
 552         ret = from;
 553     } else if (button == LEFT_RELEASE && ui->drag_col) {
 554         if (over_guess > -1) {
 555             debug(("Dropping colour %d onto guess peg %d",
 556                    ui->drag_col, over_guess));
 557             set_peg(ui, over_guess, ui->drag_col);
 558         }
 559         ui->drag_col = 0;
 560         debug(("Stop dragging."));
 561         ret = from;
 562     } else if (button == RIGHT_BUTTON) {
 563         if (over_guess > -1) {
 564             /* we use ths feedback in the game_ui to signify
 565              * 'carry this peg to the next guess as well'. */
 566             ui->holds[over_guess] = 1 - ui->holds[over_guess];
 567             ret = from;
 568         }
 569     } else if (button == LEFT_RELEASE && over_hint && ui->markable) {
 570         /* NB this won't trigger if on the end of a drag; that's on
 571          * purpose, in case you drop by mistake... */
 572         ret = mark_move(from, ui);
 573     }
 574
 575     /* keyboard input */
 576     if (button == CURSOR_UP || button == CURSOR_DOWN) {
 577         ui->display_cur = 1;
 578         if (button == CURSOR_DOWN && (ui->colour_cur+1) < from->params.ncolours)
 579             ui->colour_cur++;
 580         if (button == CURSOR_UP && ui->colour_cur > 0)
 581             ui->colour_cur--;
 582         ret = from;
 583     } else if (button == CURSOR_LEFT || button == CURSOR_RIGHT) {
 584         int maxcur = from->params.npegs;
 585         if (ui->markable) maxcur++;
 586
 587         ui->display_cur = 1;
 588         if (button == CURSOR_RIGHT && (ui->peg_cur+1) < maxcur)
 589             ui->peg_cur++;
 590         if (button == CURSOR_LEFT && ui->peg_cur > 0)
 591             ui->peg_cur--;
 592         ret = from;
 593     } else if (button == CURSOR_SELECT || button == ' ' || button == '\r' ||
 594                button == '\n') {
 595         if (ui->peg_cur == from->params.npegs) {
 596             ret = mark_move(from, ui);
 597         } else {
 598             set_peg(ui, ui->peg_cur, ui->colour_cur+1);
 599             ret = from;
 600         }
 601     }
 602     return ret;
 603 }
 604
 605 /* ----------------------------------------------------------------------
 606  * Drawing routines.
 607  */
 608
 609 #define PEG_PREFER_SZ 32
 610
 611 /* next three are multipliers for pegsz. It will look much nicer if
 612  * (2*PEG_HINT) + PEG_GAP = 1.0 as the hints are formatted like that. */
 613 #define PEG_GAP   0.10
 614 #define PEG_HINT  0.35
 615
 616 #define BORDER    0.5
 617
 618 static void game_size(game_params *params, game_drawstate *ds,
 619                       int *x, int *y, int expand)
 620 {
 621     double hmul, vmul_c, vmul_g, vmul, szx, szy;
 622     int sz, colh, guessh;
 623
 624     hmul = BORDER   * 2.0 +               /* border */
 625            1.0      * 2.0 +               /* vertical colour bar */
 626            1.0      * params->npegs +     /* guess pegs */
 627            PEG_GAP  * params->npegs +     /* guess gaps */
 628            PEG_HINT * ds->hintw +         /* hint pegs */
 629            PEG_GAP  * (ds->hintw - 1);    /* hint gaps */
 630
 631     vmul_c = BORDER  * 2.0 +                    /* border */
 632              1.0     * params->ncolours +       /* colour pegs */
 633              PEG_GAP * (params->ncolours - 1);  /* colour gaps */
 634
 635     vmul_g = BORDER  * 2.0 +                    /* border */
 636              1.0     * (params->nguesses + 1) + /* guesses plus solution */
 637              PEG_GAP * (params->nguesses + 1);  /* gaps plus gap above soln */
 638
 639     vmul = max(vmul_c, vmul_g);
 640
 641     szx = *x / hmul;
 642     szy = *y / vmul;
 643     sz = max(min((int)szx, (int)szy), 1);
 644     if (expand)
 645         ds->pegsz = sz;
 646     else
 647         ds->pegsz = min(sz, PEG_PREFER_SZ);
 648
 649     ds->hintsz = (int)((double)ds->pegsz * PEG_HINT);
 650     ds->gapsz  = (int)((double)ds->pegsz * PEG_GAP);
 651     ds->border = (int)((double)ds->pegsz * BORDER);
 652
 653     ds->pegrad  = (ds->pegsz -1)/2; /* radius of peg to fit in pegsz (which is 2r+1) */
 654     ds->hintrad = (ds->hintsz-1)/2;
 655
 656     *x = (int)ceil((double)ds->pegsz * hmul);
 657     *y = (int)ceil((double)ds->pegsz * vmul);
 658     ds->w = *x; ds->h = *y;
 659
 660     colh = ((ds->pegsz + ds->gapsz) * params->ncolours) - ds->gapsz;
 661     guessh = ((ds->pegsz + ds->gapsz) * params->nguesses);      /* guesses */
 662     guessh += ds->gapsz + ds->pegsz;                            /* solution */
 663
 664     ds->colx = ds->border;
 665     ds->coly = (*y - colh) / 2;
 666
 667     ds->guessx = ds->solnx = ds->border + ds->pegsz * 2;     /* border + colours */
 668     ds->guessy = (*y - guessh) / 2;
 669     ds->solny = ds->guessy + ((ds->pegsz + ds->gapsz) * params->nguesses) + ds->gapsz;
 670
 671     if (ds->pegsz > 0) {
 672         if (ds->blit_peg) blitter_free(ds->blit_peg);
 673         ds->blit_peg = blitter_new(ds->pegsz, ds->pegsz);
 674     }
 675 }
 676
 677 static float *game_colours(frontend *fe, game_state *state, int *ncolours)
 678 {
 679     float *ret = snewn(3 * NCOLOURS, float);
 680
 681     frontend_default_colour(fe, &ret[COL_BACKGROUND * 3]);
 682
 683     /* red */
 684     ret[COL_1 * 3 + 0] = 1.0F;
 685     ret[COL_1 * 3 + 1] = 0.0F;
 686     ret[COL_1 * 3 + 2] = 0.0F;
 687
 688     /* yellow (toned down a bit due to pale grey background) */
 689     ret[COL_2 * 3 + 0] = 0.7F;
 690     ret[COL_2 * 3 + 1] = 0.7F;
 691     ret[COL_2 * 3 + 2] = 0.0F;
 692
 693     /* green (also toned down) */
 694     ret[COL_3 * 3 + 0] = 0.0F;
 695     ret[COL_3 * 3 + 1] = 0.5F;
 696     ret[COL_3 * 3 + 2] = 0.0F;
 697
 698     /* blue */
 699     ret[COL_4 * 3 + 0] = 0.0F;
 700     ret[COL_4 * 3 + 1] = 0.0F;
 701     ret[COL_4 * 3 + 2] = 1.0F;
 702
 703     /* orange */
 704     ret[COL_5 * 3 + 0] = 1.0F;
 705     ret[COL_5 * 3 + 1] = 0.5F;
 706     ret[COL_5 * 3 + 2] = 0.0F;
 707
 708     /* purple */
 709     ret[COL_6 * 3 + 0] = 0.5F;
 710     ret[COL_6 * 3 + 1] = 0.0F;
 711     ret[COL_6 * 3 + 2] = 0.7F;
 712
 713     /* brown */
 714     ret[COL_7 * 3 + 0] = 0.4F;
 715     ret[COL_7 * 3 + 1] = 0.2F;
 716     ret[COL_7 * 3 + 2] = 0.2F;
 717
 718     /* light blue */
 719     ret[COL_8 * 3 + 0] = 0.4F;
 720     ret[COL_8 * 3 + 1] = 0.7F;
 721     ret[COL_8 * 3 + 2] = 1.0F;
 722
 723     /* light green */
 724     ret[COL_9 * 3 + 0] = 0.5F;
 725     ret[COL_9 * 3 + 1] = 0.8F;
 726     ret[COL_9 * 3 + 2] = 0.5F;
 727
 728     /* pink */
 729     ret[COL_10 * 3 + 0] = 1.0F;
 730     ret[COL_10 * 3 + 1] = 0.6F;
 731     ret[COL_10 * 3 + 2] = 1.0F;
 732
 733     ret[COL_FRAME * 3 + 0] = 0.0F;
 734     ret[COL_FRAME * 3 + 1] = 0.0F;
 735     ret[COL_FRAME * 3 + 2] = 0.0F;
 736
 737     ret[COL_HIGHLIGHT * 3 + 0] = 1.0F;
 738     ret[COL_HIGHLIGHT * 3 + 1] = 1.0F;
 739     ret[COL_HIGHLIGHT * 3 + 2] = 1.0F;
 740
 741     ret[COL_LOWLIGHT * 3 + 0] = ret[COL_BACKGROUND * 3 + 0] * 2.0 / 3.0;
 742     ret[COL_LOWLIGHT * 3 + 1] = ret[COL_BACKGROUND * 3 + 1] * 2.0 / 3.0;
 743     ret[COL_LOWLIGHT * 3 + 2] = ret[COL_BACKGROUND * 3 + 2] * 2.0 / 3.0;
 744
 745     ret[COL_FLASH * 3 + 0] = 0.5F;
 746     ret[COL_FLASH * 3 + 1] = 1.0F;
 747     ret[COL_FLASH * 3 + 2] = 1.0F;
 748
 749     ret[COL_HOLD * 3 + 0] = 1.0F;
 750     ret[COL_HOLD * 3 + 1] = 0.5F;
 751     ret[COL_HOLD * 3 + 2] = 0.5F;
 752
 753     ret[COL_EMPTY * 3 + 0] = ret[COL_BACKGROUND * 3 + 0] * 2.0 / 3.0;
 754     ret[COL_EMPTY * 3 + 1] = ret[COL_BACKGROUND * 3 + 1] * 2.0 / 3.0;
 755     ret[COL_EMPTY * 3 + 2] = ret[COL_BACKGROUND * 3 + 2] * 2.0 / 3.0;
 756
 757     ret[COL_CORRECTPLACE*3 + 0] = 0.0F;
 758     ret[COL_CORRECTPLACE*3 + 1] = 0.0F;
 759     ret[COL_CORRECTPLACE*3 + 2] = 0.0F;
 760
 761     ret[COL_CORRECTCOLOUR*3 + 0] = 1.0F;
 762     ret[COL_CORRECTCOLOUR*3 + 1] = 1.0F;
 763     ret[COL_CORRECTCOLOUR*3 + 2] = 1.0F;
 764
 765     *ncolours = NCOLOURS;
 766     return ret;
 767 }
 768
 769 static game_drawstate *game_new_drawstate(game_state *state)
 770 {
 771     struct game_drawstate *ds = snew(struct game_drawstate);
 772     int i;
 773
 774     memset(ds, 0, sizeof(struct game_drawstate));
 775
 776     ds->guesses = snewn(state->params.nguesses, pegrow);
 777     ds->nguesses = state->params.nguesses;
 778     for (i = 0; i < state->params.nguesses; i++) {
 779         ds->guesses[i] = new_pegrow(state->params.npegs);
 780         invalidate_pegrow(ds->guesses[i]);
 781     }
 782     ds->solution = new_pegrow(state->params.npegs);
 783     invalidate_pegrow(ds->solution);
 784     ds->colours = new_pegrow(state->params.ncolours);
 785     invalidate_pegrow(ds->colours);
 786
 787     ds->hintw = (state->params.npegs+1)/2; /* must round up */
 788
 789     ds->holds = snewn(state->params.npegs, int);
 790     memset(ds->holds, 0, state->params.npegs*sizeof(int));
 791
 792     ds->blit_peg = NULL;
 793
 794     return ds;
 795 }
 796
 797 static void game_free_drawstate(game_drawstate *ds)
 798 {
 799     int i;
 800
 801     if (ds->blit_peg) blitter_free(ds->blit_peg);
 802     free_pegrow(ds->colours);
 803     free_pegrow(ds->solution);
 804     for (i = 0; i < ds->nguesses; i++)
 805         free_pegrow(ds->guesses[i]);
 806     sfree(ds->holds);
 807     sfree(ds->guesses);
 808     sfree(ds);
 809 }
 810
 811 static void draw_peg(frontend *fe, game_drawstate *ds, int cx, int cy, int col)
 812 {
 813     if (PEGRAD > 0) {
 814         draw_circle(fe, cx+PEGRAD, cy+PEGRAD, PEGRAD, 1, COL_EMPTY + col);
 815         draw_circle(fe, cx+PEGRAD, cy+PEGRAD, PEGRAD, 0, COL_EMPTY + col);
 816     } else
 817         draw_rect(fe, cx, cy, PEGSZ, PEGSZ, COL_EMPTY + col);
 818     draw_update(fe, cx, cy, PEGSZ, PEGSZ);
 819 }
 820
 821 static void guess_redraw(frontend *fe, game_drawstate *ds, int guess,
 822                          pegrow src, int force)
 823 {
 824     pegrow dest;
 825     int rowx, rowy, i, scol;
 826
 827     if (guess == -1) {
 828         dest = ds->solution;
 829         rowx = SOLN_OX;
 830         rowy = SOLN_OY;
 831     } else {
 832         dest = ds->guesses[guess];
 833         rowx = GUESS_X(guess,0);
 834         rowy = GUESS_Y(guess,0);
 835     }
 836     if (src) assert(src->npegs == dest->npegs);
 837
 838     for (i = 0; i < dest->npegs; i++) {
 839         scol = src ? src->pegs[i] : 0;
 840         if ((dest->pegs[i] != scol) || force)
 841             draw_peg(fe, ds, rowx + PEGOFF * i, rowy, scol);
 842         dest->pegs[i] = scol;
 843     }
 844 }
 845
 846 static void hint_redraw(frontend *fe, game_drawstate *ds, int guess,
 847                         pegrow src, int force, int emptycol)
 848 {
 849     pegrow dest = ds->guesses[guess];
 850     int rowx, rowy, i, scol, col, hintlen;
 851
 852     if (src) assert(src->npegs == dest->npegs);
 853
 854     hintlen = (dest->npegs + 1)/2;
 855
 856     for (i = 0; i < dest->npegs; i++) {
 857         scol = src ? src->feedback[i] : 0;
 858         col = (scol == FEEDBACK_CORRECTPLACE) ? COL_CORRECTPLACE :
 859               (scol == FEEDBACK_CORRECTCOLOUR) ? COL_CORRECTCOLOUR : emptycol;
 860         if ((scol != dest->feedback[i]) || force) {
 861             rowx = HINT_X(guess);
 862             rowy = HINT_Y(guess);
 863             if (i < hintlen) {
 864                 rowx += HINTOFF * i;
 865             } else {
 866                 rowx += HINTOFF * (i - hintlen);
 867                 rowy += HINTOFF;
 868             }
 869             if (HINTRAD > 0) {
 870                 draw_circle(fe, rowx+HINTRAD, rowy+HINTRAD, HINTRAD, 1, col);
 871                 draw_circle(fe, rowx+HINTRAD, rowy+HINTRAD, HINTRAD, 0, col);
 872             } else {
 873                 draw_rect(fe, rowx, rowy, HINTSZ, HINTSZ, col);
 874             }
 875             draw_update(fe, rowx, rowy, HINTSZ, HINTSZ);
 876         }
 877         dest->feedback[i] = scol;
 878     }
 879 }
 880
 881 static void hold_redraw(frontend *fe, game_drawstate *ds, int guess, int *src, int force)
 882 {
 883     int shold, col, ox, oy, i;
 884
 885     if (guess >= ds->nguesses)
 886         return;
 887
 888     for (i = 0; i < ds->solution->npegs; i++) {
 889         shold = src ? src[i] : 0;
 890         col = shold ? COL_HOLD : COL_BACKGROUND;
 891         if ((shold != ds->holds[i]) || force) {
 892             ox = GUESS_X(guess, i);
 893             oy = GUESS_Y(guess, i) + PEGSZ + ds->gapsz/2;
 894
 895             draw_rect(fe, ox, oy, PEGSZ, 2, col);
 896             draw_update(fe, ox, oy, PEGSZ, 2);
 897         }
 898         if (src) ds->holds[i] = shold;
 899     }
 900 }
 901
 902 static void currmove_redraw(frontend *fe, game_drawstate *ds, int guess, int col)
 903 {
 904     int ox = GUESS_X(guess, 0), oy = GUESS_Y(guess, 0), off = PEGSZ/4;
 905
 906     draw_rect(fe, ox-off-1, oy, 2, PEGSZ, col);
 907     draw_update(fe, ox-off-1, oy, 2, PEGSZ);
 908 }
 909
 910 static void cur_redraw(frontend *fe, game_drawstate *ds,
 911                        int x, int y, int erase)
 912 {
 913     int cgap = ds->gapsz / 2;
 914     int x1, y1, x2, y2, hi, lo;
 915
 916     x1 = x-cgap; x2 = x+PEGSZ+cgap;
 917     y1 = y-cgap; y2 = y+PEGSZ+cgap;
 918     hi = erase ? COL_BACKGROUND : COL_HIGHLIGHT;
 919     lo = erase ? COL_BACKGROUND : COL_LOWLIGHT;
 920
 921     draw_line(fe, x1, y1, x2, y1, hi);
 922     draw_line(fe, x2, y1, x2, y2, lo);
 923     draw_line(fe, x2, y2, x1, y2, lo);
 924     draw_line(fe, x1, y2, x1, y1, hi);
 925
 926     draw_update(fe, x1, y1, x2, y2);
 927 }
 928
 929 static void game_redraw(frontend *fe, game_drawstate *ds, game_state *oldstate,
 930                         game_state *state, int dir, game_ui *ui,
 931                         float animtime, float flashtime)
 932 {
 933     int i, cur_erase = 0, cur_draw = 0, new_move, last_go;
 934
 935     new_move = (state->next_go != ds->next_go) || !ds->started;
 936     last_go = (state->next_go == state->params.nguesses-1);
 937
 938     if (!ds->started) {
 939       draw_rect(fe, 0, 0, ds->w, ds->h, COL_BACKGROUND);
 940       draw_rect(fe, SOLN_OX, SOLN_OY - ds->gapsz - 1, SOLN_W, 2, COL_FRAME);
 941       draw_update(fe, 0, 0, ds->w, ds->h);
 942     }
 943
 944     if (ds->drag_col != 0) {
 945         debug(("Loading from blitter."));
 946         blitter_load(fe, ds->blit_peg, ds->blit_ox, ds->blit_oy);
 947         draw_update(fe, ds->blit_ox, ds->blit_oy, PEGSZ, PEGSZ);
 948     }
 949
 950     /* draw the colours */
 951     for (i = 0; i < state->params.ncolours; i++) {
 952         if (ds->colours->pegs[i] != i+1) {
 953             draw_peg(fe, ds, COL_X(i), COL_Y(i), i+1);
 954             ds->colours->pegs[i] = i+1;
 955         }
 956     }
 957
 958     /* draw the guesses (so far) and the hints */
 959     for (i = 0; i < state->params.nguesses; i++) {
 960         if (state->next_go > i || state->solved) {
 961             /* this info is stored in the game_state already */
 962             guess_redraw(fe, ds, i, state->guesses[i], 0);
 963             hint_redraw(fe, ds, i, state->guesses[i],
 964                         i == (state->next_go-1) ? 1 : 0, COL_EMPTY);
 965         } else if (state->next_go == i) {
 966             /* this is the one we're on; the (incomplete) guess is
 967              * stored in the game_ui. */
 968             guess_redraw(fe, ds, i, ui->curr_pegs, 0);
 969             hint_redraw(fe, ds, i, NULL, 1, ui->markable ? COL_FLASH : COL_EMPTY);
 970         } else {
 971             /* we've not got here yet; it's blank. */
 972             guess_redraw(fe, ds, i, NULL, 0);
 973             hint_redraw(fe, ds, i, NULL, 0, COL_EMPTY);
 974         }
 975     }
 976
 977     /* draw the 'hold' markers */
 978     if (state->solved) {
 979         hold_redraw(fe, ds, state->next_go, NULL, 1);
 980     } else if (new_move) {
 981         hold_redraw(fe, ds, ds->next_go, NULL, 1);
 982         hold_redraw(fe, ds, state->next_go, last_go ? NULL : ui->holds, 1);
 983     } else
 984         hold_redraw(fe, ds, state->next_go, last_go ? NULL : ui->holds, 0);
 985
 986     /* draw the 'current move' and 'able to mark' sign. */
 987     if (new_move)
 988         currmove_redraw(fe, ds, ds->next_go, COL_BACKGROUND);
 989     if (!state->solved)
 990         currmove_redraw(fe, ds, state->next_go, COL_HOLD);
 991
 992     /* draw the solution (or the big rectangle) */
 993     if ((state->solved != ds->solved) || !ds->started) {
 994         draw_rect(fe, SOLN_OX, SOLN_OY, SOLN_W, SOLN_H,
 995                   state->solved ? COL_BACKGROUND : COL_EMPTY);
 996         draw_update(fe, SOLN_OX, SOLN_OY, SOLN_W, SOLN_H);
 997     }
 998     if (state->solved)
 999         guess_redraw(fe, ds, -1, state->solution, !ds->solved);
1000     ds->solved = state->solved;
1001
1002     if (ui->display_cur && !ds->display_cur)
1003         cur_draw = 1;
1004     else if (!ui->display_cur && ds->display_cur)
1005         cur_erase = 1;
1006     else if (ui->display_cur) {
1007         if ((state->next_go != ds->next_go) ||
1008             (ui->peg_cur != ds->peg_cur) ||
1009             (ui->colour_cur != ds->colour_cur)) {
1010             cur_erase = 1;
1011             cur_draw = 1;
1012         }
1013     }
1014     if (cur_erase) {
1015         cur_redraw(fe, ds, COL_X(ds->colour_cur), COL_Y(ds->colour_cur), 1);
1016         cur_redraw(fe, ds,
1017                    GUESS_X(ds->next_go, ds->peg_cur), GUESS_Y(ds->next_go, ds->peg_cur), 1);
1018     }
1019     if (cur_draw) {
1020         cur_redraw(fe, ds, COL_X(ui->colour_cur), COL_Y(ui->colour_cur), 0);
1021         cur_redraw(fe, ds,
1022                    GUESS_X(state->next_go, ui->peg_cur), GUESS_Y(state->next_go, ui->peg_cur), 0);
1023     }
1024     ds->display_cur = ui->display_cur;
1025     ds->peg_cur = ui->peg_cur;
1026     ds->colour_cur = ui->colour_cur;
1027     ds->next_go = state->next_go;
1028
1029     /* if ui->drag_col != 0, save the screen to the blitter,
1030      * draw the peg where we saved, and set ds->drag_* == ui->drag_*. */
1031     if (ui->drag_col != 0) {
1032         int ox = ui->drag_x - (PEGSZ/2);
1033         int oy = ui->drag_y - (PEGSZ/2);
1034         debug(("Saving to blitter at (%d,%d)", ox, oy));
1035         blitter_save(fe, ds->blit_peg, ox, oy);
1036         draw_peg(fe, ds, ox, oy, ui->drag_col);
1037
1038         ds->blit_ox = ox; ds->blit_oy = oy;
1039     }
1040     ds->drag_col = ui->drag_col;
1041
1042     ds->started = 1;
1043 }
1044
1045 static float game_anim_length(game_state *oldstate, game_state *newstate,
1046                               int dir, game_ui *ui)
1047 {
1048     return 0.0F;
1049 }
1050
1051 static float game_flash_length(game_state *oldstate, game_state *newstate,
1052                                int dir, game_ui *ui)
1053 {
1054     return 0.0F;
1055 }
1056
1057 static int game_wants_statusbar(void)
1058 {
1059     return FALSE;
1060 }
1061
1062 static int game_timing_state(game_state *state)
1063 {
1064     return TRUE;
1065 }
1066
1067 #ifdef COMBINED
1068 #define thegame guess
1069 #endif
1070
1071 const struct game thegame = {
1072     "Guess", NULL,
1073     default_params,
1074     game_fetch_preset,
1075     decode_params,
1076     encode_params,
1077     free_params,
1078     dup_params,
1079     TRUE, game_configure, custom_params,
1080     validate_params,
1081     new_game_desc,
1082     game_free_aux_info,
1083     validate_desc,
1084     new_game,
1085     dup_game,
1086     free_game,
1087     TRUE, solve_game,
1088     FALSE, game_text_format,
1089     new_ui,
1090     free_ui,
1091     game_changed_state,
1092     make_move,
1093     game_size,
1094     game_colours,
1095     game_new_drawstate,
1096     game_free_drawstate,
1097     game_redraw,
1098     game_anim_length,
1099     game_flash_length,
1100     game_wants_statusbar,
1101     FALSE, game_timing_state,
1102     0,                                 /* mouse_priorities */
1103 };
1104
1105 /* vim: set shiftwidth=4 tabstop=8: */