patternprob.c

   1 #define DEBUG
   2 #include <assert.h>
   3 #include <ctype.h>
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6
   7 #include "board.h"
   8 #include "debug.h"
   9 #include "pattern.h"
  10 #include "patternsp.h"
  11 #include "patternprob.h"
  12
  13
  14 /* We try to avoid needlessly reloading probability dictionary
  15  * since it may take rather long time. */
  16 static struct pattern_pdict *cached_dict;
  17
  18 struct pattern_pdict *
  19 pattern_pdict_init(char *filename, struct pattern_config *pc)
  20 {
  21         if (cached_dict) {
  22                 cached_dict->pc = pc;
  23                 return cached_dict;
  24         }
  25
  26         if (!filename)
  27                 filename = "patterns.prob";
  28         FILE *f = fopen(filename, "r");
  29         if (!f) {
  30                 if (DEBUGL(1))
  31                         fprintf(stderr, "No pattern probtable, will not use learned patterns.\n");
  32                 return NULL;
  33         }
  34
  35         struct pattern_pdict *dict = calloc2(1, sizeof(*dict));
  36         dict->pc = pc;
  37         dict->table = calloc2(pc->spat_dict->nspatials + 1, sizeof(*dict->table));
  38
  39         char *sphcachehit = calloc2(pc->spat_dict->nspatials, 1);
  40         hash_t (*sphcache)[PTH__ROTATIONS] = malloc(pc->spat_dict->nspatials * sizeof(sphcache[0]));
  41
  42         int i = 0;
  43         char sbuf[1024];
  44         while (fgets(sbuf, sizeof(sbuf), f)) {
  45                 struct pattern_prob *pb = calloc2(1, sizeof(*pb));
  46                 int c, o;
  47
  48                 char *buf = sbuf;
  49                 if (buf[0] == '#') continue;
  50                 while (isspace(*buf)) buf++;
  51                 while (!isspace(*buf)) buf++; // we recompute the probability
  52                 while (isspace(*buf)) buf++;
  53                 c = strtol(buf, &buf, 10);
  54                 while (isspace(*buf)) buf++;
  55                 o = strtol(buf, &buf, 10);
  56                 pb->prob = (floating_t) c / o;
  57                 while (isspace(*buf)) buf++;
  58                 str2pattern(buf, &pb->p);
  59
  60                 uint32_t spi = pattern2spatial(dict, &pb->p);
  61                 pb->next = dict->table[spi];
  62                 dict->table[spi] = pb;
  63
  64                 /* We rehash spatials in the order of loaded patterns. This way
  65                  * we make sure that the most popular patterns will be hashed
  66                  * last and therefore take priority. */
  67                 if (!sphcachehit[spi]) {
  68                         sphcachehit[spi] = 1;
  69                         for (int r = 0; r < PTH__ROTATIONS; r++)
  70                                 sphcache[spi][r] = spatial_hash(r, &pc->spat_dict->spatials[spi]);
  71                 }
  72                 for (int r = 0; r < PTH__ROTATIONS; r++)
  73                         spatial_dict_addh(pc->spat_dict, sphcache[spi][r], spi);
  74
  75                 i++;
  76         }
  77
  78         free(sphcache);
  79         free(sphcachehit);
  80         if (DEBUGL(3))
  81                 spatial_dict_hashstats(pc->spat_dict);
  82
  83         fclose(f);
  84         if (DEBUGL(1))
  85                 fprintf(stderr, "Loaded %d pattern-probability pairs.\n", i);
  86         cached_dict = dict;
  87         return dict;
  88 }
  89
  90 floating_t
  91 pattern_rate_moves(struct pattern_setup *pat,
  92                    struct board *b, enum stone color,
  93                    struct pattern *pats, floating_t *probs)
  94 {
  95         floating_t total = 0;
  96         for (int f = 0; f < b->flen; f++) {
  97                 probs[f] = NAN;
  98
  99                 struct move mo = { .coord = b->f[f], .color = color };
 100                 if (is_pass(mo.coord))
 101                         continue;
 102                 if (!board_is_valid_move(b, &mo))
 103                         continue;
 104
 105                 pattern_match(&pat->pc, pat->ps, &pats[f], b, &mo);
 106                 floating_t prob = pattern_prob(pat->pd, &pats[f]);
 107                 if (!isnan(prob)) {
 108                         probs[f] = prob;
 109                         total += prob;
 110                 }
 111         }
 112         return total;
 113 }
 114