chiark / gitweb /
74a2f4504feb0840444a3da69cbd166333010aa5
[ypp-sc-tools.db-live.git] / yarrg / rssearch.c
1 /**/
2
3 #include "rscommon.h"
4
5 DEBUG_DEFINE_DEBUGF(search);
6 DEBUG_DEFINE_SOME_DEBUGF(filter,fildebugf);
7
8 typedef struct Neighbour {
9   struct Neighbour *next;
10   int islandid;
11   int dist;
12 } Neighbour;
13
14 static Neighbour **neighbours; /* neighbours[islandid]->islandid etc. */
15 static sqlite3_stmt *ss_neigh;
16
17 static int ports[MAX_ROUTELEN];
18 static int final_isle;
19
20 static Neighbour *get_neighbours(int isle) {
21   Neighbour **np= &neighbours[isle];
22   Neighbour *head= *np;
23   if (head) return head;
24
25   SQL_BIND(ss_neigh, 1, isle);
26   while (SQL_STEP(ss_neigh)) {
27     Neighbour *add;
28     NEW(add);
29     add->islandid= sqlite3_column_int(ss_neigh, 0);
30     add->dist= sqlite3_column_int(ss_neigh, 1);
31     add->next= head;
32     head= add;
33   }
34   SQL_MUST( sqlite3_reset(ss_neigh) );
35
36   *np= head;
37   return head;
38 }
39
40
41 static PotentialResult ***buckets_base[GRANUS];
42
43
44 static double process_route(int nports, int totaldist,
45                             double overestimate_excepting_tail) {
46   int i, ap;
47   int leagues_divisor= totaldist + nports;
48
49   ctr_routes_considered++;
50
51   int wrong_final= final_isle && ports[nports-1] != final_isle;
52
53   debugf("========== ROUTE");
54   for (i=0; i<nports; i++)
55     debugf(" %d",ports[i]);
56   debugf("\n");
57
58   double guess[AP]={0,0};
59   if (nports>=2) {
60     int pair[2], i;
61     pair[1]= ports[nports-1];
62     guess[A]= overestimate_excepting_tail;
63
64     for (i=0; i<nports; i++) {
65       pair[0]= ports[i];
66       IslandPair *ip= ipair_get_maybe(pair[0], pair[1]);
67       if (!ip) continue;
68       if (ip->route_tail_value < 0) {
69         ctr_subroute_tails_valued++;
70         ip->route_tail_value= value_route(2, pair, pair[0]!=pair[1]);
71       }
72       guess[A] += ip->route_tail_value;
73     }
74     guess[P]= guess[A] / leagues_divisor;
75
76     if (wrong_final) {
77       ctr_routes_wrongfinalelim++;
78       debugf(" WFELIM\n");
79       return guess[A];
80     }
81
82     if (guess[A] <= highscores[GRANUS-1][A][0].value &&
83         guess[P] <= highscores[GRANUS-1][P][0].value) {
84       ctr_routes_quickelim++;
85       debugf(" QELIM %f %f\n", guess[A], guess[P]);
86       return guess[A];
87     }
88   }
89
90   int finisle= ports[nports-1];
91   int finarch= isle2arch(finisle);
92
93   int midisle= ports[nports/2];
94   int midarch= route2midarch(ports,nports);
95
96   PotentialResult *buckets[GRANUS];
97   int granui;
98   for (granui=mingranu; granui<GRANUS; granui++) {
99     PotentialResult **buckets_fin;
100     int mid, fin;
101     switch (granui) {
102     case 0: fin=finisle; mid=midisle; break;
103     case 1: fin=finisle; mid=midarch; break;
104     case 2: fin=finarch; mid=midarch; break;
105     default: abort();
106     }
107     buckets_fin= ONDEMAND(buckets_base[granui][fin], granusz_mid[granui]);
108     buckets[granui]= ONDEMAND(buckets_fin[mid], 1);
109   }
110
111   if (nports>=2) {
112     if (guess[A] <= buckets[mingranu]->value[A] &&
113         guess[P] <= buckets[mingranu]->value[P]) {
114       ctr_routes_bucketelim++;
115       debugf(" ELIM %f %f\n", guess[A], guess[P]);
116       return guess[A];
117     }
118     debugf(" COMPUTE %f %f\n", guess[A], guess[P]);
119   }
120
121   ctr_routes_valued++;
122
123   double value[AP];
124   value[A]= value_route(nports, ports, 0);
125   value[P]= value[A] / leagues_divisor;
126
127   if (wrong_final) {
128     ctr_routes_wrongfinal++;
129     return value[0];
130   }
131
132   for (granui=mingranu; granui<GRANUS; granui++) {
133     PotentialResult *bucket= buckets[granui];
134
135     if (value[A] <= bucket->value[A] &&
136         value[P] <= bucket->value[P])
137       continue;
138
139     debugf(" SOMEHOW %d BEST\n",granui);
140
141     fildebugf("final %d:%3d mid %d ",finarch,finisle,midarch);
142
143     for (ap=0; ap<AP; ap++) {
144       HighScoreEntry *scores= highscores[granui][ap];
145       int *nscores= &nhighscores[granui][ap];
146
147       fildebugf("ap=%d %15f", ap, value[ap]);
148       if (value[ap] < bucket->value[ap]) {
149         debugf("      ");
150       } else {
151         int pos;
152         ctr_newbests_granu[granui*AP+ap]++;
153         bucket->value[ap]= value[ap];
154         memcpy(bucket->ports[ap], ports, sizeof(*ports) * nports);
155         if (nports < MAX_ROUTELEN-1) bucket->ports[ap][nports]= -1;
156         fildebugf("** ");
157         for (pos=0; pos < *nscores; pos++)
158           if (scores[pos].pr == bucket) goto found;
159         /* not found */
160         pos= -1;
161       found:
162         for (;;) {
163           pos++;
164           if (pos >= *nscores-1) break; /* new top */
165           if (scores[pos].value >= value[ap]) break; /* found spot */
166           if (pos>0)
167             scores[pos-1]= scores[pos];
168         }
169         pos--;
170         if (pos>0) {
171           scores[pos].value= value[ap];
172           scores[pos].pr= bucket;
173         }
174         fildebugf("@%2d", pos);
175       }
176     }
177   }
178
179   fildebugf(" route");
180
181   for (i=0; i<nports; i++)
182     fildebugf(" %d",ports[i]);
183   fildebugf("\n");
184
185   return value[0];
186 }
187
188 static void recurse(int last_isle,
189                     int nports /* excluding last_isle */,
190                     int totaldist /* including last_isle */,
191                     double last_estimate) {
192   ports[nports++]= last_isle;
193   double estimate= process_route(nports, totaldist, last_estimate);
194   if (nports >= MAX_ROUTELEN) return;
195
196   Neighbour *add;
197   for (add= get_neighbours(last_isle); add; add=add->next) {
198     int newdist= totaldist + add->dist;
199     if (newdist > max_dist) continue;
200
201     recurse(add->islandid, nports, newdist, estimate);
202   }
203 }
204
205 void search(int start_isle, int final_isle_spec,
206             PotentialResult ****buckets_base_io[GRANUS]) {
207   int granui;
208   for (granui=0; granui<GRANUS; granui++)
209     buckets_base[granui]=
210       ONDEMAND(*buckets_base_io[granui], granusz_fin[granui]);
211
212   final_isle= final_isle_spec <= 0 ? 0 : final_isle_spec;
213   recurse(start_isle,0,0,1e6);
214 }
215
216 int nhighscores[GRANUS][AP];
217 HighScoreEntry *highscores[GRANUS][AP];
218 int mingranu, granusz_fin[GRANUS], granusz_mid[GRANUS];
219
220 int narches;
221 char **archnames;
222 int *islandid2arch;
223
224 void setup_search(void) {
225   MCALLOC(neighbours, islandtablesz);
226
227   SQL_PREPARE(ss_neigh,
228               "SELECT biid, dist FROM routes WHERE aiid=?");
229
230   int max_narches=
231     sql_single_int(" SELECT count(*) FROM (\n"
232                    "  SELECT DISTINCT archipelago\n"
233                    "   FROM islands\n"
234                    "  )");
235   MCALLOC(archnames, max_narches);
236   MCALLOC_INITEACH(islandid2arch, islandtablesz, *this=-1);
237
238   sqlite3_stmt *archs;
239   SQL_PREPARE(archs,
240               " SELECT islandid, archipelago\n"
241               "  FROM islands\n"
242               "  ORDER BY archipelago");
243   while (SQL_STEP(archs)) {
244     int isle= sqlite3_column_int(archs,0);
245     const char *archname= (const char*)sqlite3_column_text(archs,1);
246     int arch;
247     for (arch=0; arch<narches; arch++)
248       if (!strcmp(archnames[arch], archname)) goto found;
249     assert(narches < max_narches);
250     arch= narches++;
251     archnames[arch]= masprintf("%s",archname);
252   found:
253     islandid2arch[isle]= arch;
254   }
255   sqlite3_finalize(archs);
256
257   granusz_fin[0]=                granusz_mid[0]= islandtablesz;
258   granusz_fin[1]= islandtablesz; granusz_mid[1]= narches;
259   granusz_fin[2]=                granusz_mid[2]= narches;
260 }