Merge branch 'master' of git+ssh://repo.or.cz/srv/git/pachi
[pachi.git] / pattern3.c
blobc90a9490f1791d0f8aa2c36dced8bbcc3c23cea4
1 #include <assert.h>
2 #include <math.h>
3 #include <stdio.h>
4 #include <stdlib.h>
6 #include "board.h"
7 #include "debug.h"
8 #include "pattern3.h"
11 static void
12 pattern_record(struct pattern3s *p, char *str, hash3_t pat, int fixed_color)
14 p->hash[pat] = fixed_color ? fixed_color : 3;
15 //fprintf(stderr, "[%s] %04x %d\n", str, pat, fixed_color);
18 static int
19 pat_vmirror(hash3_t pat)
21 /* V mirror pattern; reverse order of 3-2-3 chunks */
22 return ((pat & 0xfc00) >> 10) | (pat & 0x03c0) | ((pat & 0x003f) << 10);
25 static int
26 pat_hmirror(hash3_t pat)
28 /* H mirror pattern; reverse order of 2-bit values within the chunks */
29 #define rev3(p) ((p >> 4) | (p & 0xc) | ((p & 0x3) << 4))
30 #define rev2(p) ((p >> 2) | ((p & 0x3) << 2))
31 return (rev3((pat & 0xfc00) >> 10) << 10)
32 | (rev2((pat & 0x03c0) >> 6) << 6)
33 | rev3((pat & 0x003f));
34 #undef rev3
35 #undef rev2
38 static int
39 pat_90rot(hash3_t pat)
41 /* Rotate by 90 degrees:
42 * 5 6 7 7 4 2
43 * 3 4 -> 6 1
44 * 0 1 2 5 3 0 */
45 /* I'm too lazy to optimize this :) */
46 int vals[8];
47 for (int i = 0; i < 8; i++)
48 vals[i] = (pat >> (i * 2)) & 0x3;
49 int vals2[8];
50 vals2[0] = vals[5]; vals2[1] = vals[3]; vals2[2] = vals[0];
51 vals2[3] = vals[6]; vals2[4] = vals[1];
52 vals2[5] = vals[7]; vals2[6] = vals[4]; vals2[7] = vals[2];
53 int p2 = 0;
54 for (int i = 0; i < 8; i++)
55 p2 |= vals2[i] << (i * 2);
56 return p2;
59 void
60 pattern3_transpose(hash3_t pat, hash3_t (*transp)[8])
62 int i = 0;
63 (*transp)[i++] = pat;
64 (*transp)[i++] = pat_vmirror(pat);
65 (*transp)[i++] = pat_hmirror(pat);
66 (*transp)[i++] = pat_vmirror(pat_hmirror(pat));
67 (*transp)[i++] = pat_90rot(pat);
68 (*transp)[i++] = pat_90rot(pat_vmirror(pat));
69 (*transp)[i++] = pat_90rot(pat_hmirror(pat));
70 (*transp)[i++] = pat_90rot(pat_vmirror(pat_hmirror(pat)));
73 static void
74 pattern_gen(struct pattern3s *p, hash3_t pat, char *src, int srclen, int fixed_color)
76 for (; srclen > 0; src++, srclen--) {
77 if (srclen == 5)
78 continue;
79 int patofs = (srclen > 5 ? srclen - 1 : srclen) - 1;
80 switch (*src) {
81 case '?':
82 *src = '.'; pattern_gen(p, pat, src, srclen, fixed_color);
83 *src = 'X'; pattern_gen(p, pat, src, srclen, fixed_color);
84 *src = 'O'; pattern_gen(p, pat, src, srclen, fixed_color);
85 *src = '#'; pattern_gen(p, pat, src, srclen, fixed_color);
86 *src = '?'; // for future recursions
87 return;
88 case 'x':
89 *src = '.'; pattern_gen(p, pat, src, srclen, fixed_color);
90 *src = 'O'; pattern_gen(p, pat, src, srclen, fixed_color);
91 *src = '#'; pattern_gen(p, pat, src, srclen, fixed_color);
92 *src = 'x'; // for future recursions
93 return;
94 case 'o':
95 *src = '.'; pattern_gen(p, pat, src, srclen, fixed_color);
96 *src = 'X'; pattern_gen(p, pat, src, srclen, fixed_color);
97 *src = '#'; pattern_gen(p, pat, src, srclen, fixed_color);
98 *src = 'o'; // for future recursions
99 return;
100 case '.': /* 0 */ break;
101 case 'X': pat |= S_BLACK << (patofs * 2); break;
102 case 'O': pat |= S_WHITE << (patofs * 2); break;
103 case '#': pat |= S_OFFBOARD << (patofs * 2); break;
107 /* Original pattern, all transpositions and rotations */
108 hash3_t transp[8];
109 pattern3_transpose(pat, &transp);
110 for (int i = 0; i < 8; i++) {
111 /* Original color assignment */
112 pattern_record(p, src - 9, transp[i], fixed_color);
113 /* Reverse color assignment */
114 if (fixed_color)
115 fixed_color = 2 - (fixed_color == 2);
116 pattern_record(p, src - 9, pattern3_reverse(transp[i]), fixed_color);
120 static void
121 patterns_gen(struct pattern3s *p, char src[][11], int src_n)
123 for (int i = 0; i < src_n; i++) {
124 //printf("<%s>\n", src[i]);
125 int fixed_color = 0;
126 switch (src[i][9]) {
127 case 'X': fixed_color = S_BLACK; break;
128 case 'O': fixed_color = S_WHITE; break;
130 //fprintf(stderr, "** %s **\n", src[i]);
131 pattern_gen(p, 0, src[i], 9, fixed_color);
135 static bool
136 patterns_load(char src[][11], int src_n, char *filename)
138 FILE *f = fopen("moggy.patterns", "r");
139 if (!f) return false;
141 int i;
142 for (i = 0; i < src_n; i++) {
143 char line[32];
144 if (!fgets(line, sizeof(line), f))
145 goto error;
146 int l = strlen(line);
147 if (l != 10 + (line[l - 1] == '\n'))
148 goto error;
149 memcpy(src[i], line, 10);
151 fprintf(stderr, "moggy.patterns: %d patterns loaded\n", i);
152 fclose(f);
153 return true;
154 error:
155 fprintf(stderr, "Error loading moggy.patterns.\n");
156 fclose(f);
157 return false;
160 void
161 pattern3s_init(struct pattern3s *p, char src[][11], int src_n)
163 char nsrc[src_n][11];
165 if (!patterns_load(nsrc, src_n, "moggy.patterns")) {
166 /* Use default pattern set. */
167 for (int i = 0; i < src_n; i++)
168 strcpy(nsrc[i], src[i]);
171 patterns_gen(p, nsrc, src_n);