yarrg/rssearch.c

   1 /**/
   2
   3 #include "rscommon.h"
   4
   5 DEBUG_DEFINE_DEBUGF(search);
   6
   7 typedef struct Neighbour {
   8   struct Neighbour *next;
   9   int islandid;
  10   int dist;
  11 } Neighbour;
  12
  13 static Neighbour **neighbours; /* neighbours[islandid]->islandid etc. */
  14 static sqlite3_stmt *ss_neigh;
  15
  16 static int ports[MAX_ROUTELEN];
  17
  18 static Neighbour *get_neighbours(int isle) {
  19   Neighbour **np= &neighbours[isle];
  20   Neighbour *head= *np;
  21   if (head) return head;
  22
  23   SQL_BIND(ss_neigh, 1, isle);
  24   while (SQL_STEP(ss_neigh)) {
  25     Neighbour *add= mmalloc(sizeof(*add));
  26     add->islandid= sqlite3_column_int(ss_neigh, 0);
  27     add->dist= sqlite3_column_int(ss_neigh, 1);
  28     add->next= head;
  29     head= add;
  30   }
  31   SQL_MUST( sqlite3_reset(ss_neigh) );
  32
  33   *np= head;
  34   return head;
  35 }
  36
  37 static double best_absolute, best_perleague;
  38
  39 static double process_route(int nports, int totaldist,
  40                             double overestimate_excepting_tail) {
  41   int i;
  42   int leagues_divisor= totaldist + nports;
  43
  44   debugf("========== ROUTE");
  45   for (i=0; i<nports; i++)
  46     debugf(" %d",ports[i]);
  47   debugf("\n");
  48
  49   if (nports>=2) {
  50     int pair[2], i;
  51     pair[1]= ports[nports-1];
  52     double guess_absolute= overestimate_excepting_tail;
  53
  54     for (i=0; i<nports; i++) {
  55       pair[0]= ports[i];
  56       IslandPair *ip= ipair_get(ports[nports-2], ports[nports-1]);
  57       if (ip->route_tail_value < 0)
  58         ip->route_tail_value= value_route(2, pair, pair[0]!=pair[1]);
  59       guess_absolute += ip->route_tail_value;
  60     }
  61     double guess_perleague= guess_absolute / leagues_divisor;
  62
  63     if (guess_absolute <= best_absolute && guess_perleague <= best_perleague) {
  64       debugf(" ELIM %f %f\n", guess_absolute, guess_perleague);
  65       return guess_absolute;
  66     }
  67     debugf(" COMPUTE %f %f\n", guess_absolute, guess_perleague);
  68   }
  69
  70   double absolute= value_route(nports, ports, 0);
  71   double perleague= absolute / leagues_divisor;
  72
  73   if (absolute <= best_absolute && perleague <= best_perleague)
  74     return absolute;
  75
  76   debugf(" SOMEHOW BEST\n");
  77
  78 #define CHK(absperl)                                    \
  79   fprintf(stderr,#absperl " %15f", absperl);            \
  80   if (absperl < best_##absperl) fputs("   ",stderr);    \
  81   else { best_##absperl= absperl; fputs("** ",stderr); }
  82
  83   CHK(absolute)
  84   CHK(perleague)
  85
  86   fputs(" route",stderr);
  87
  88   for (i=0; i<nports; i++)
  89     fprintf(stderr," %d",ports[i]);
  90   putc('\n',stderr);
  91
  92   return absolute;
  93 }
  94
  95 static void recurse(int last_isle,
  96                     int nports, /* excluding last_isle */
  97                     int totaldist /* including last_isle */,
  98                     double last_estimate) {
  99   ports[nports++]= last_isle;
 100   double estimate= process_route(nports, totaldist, last_estimate);
 101   if (nports >= MAX_ROUTELEN) return;
 102
 103   Neighbour *add;
 104   for (add= get_neighbours(last_isle); add; add=add->next) {
 105     int newdist= totaldist + add->dist;
 106     if (newdist > max_dist) continue;
 107
 108     recurse(add->islandid, nports, newdist, estimate);
 109   }
 110 }
 111
 112 void search(int start_isle) {
 113   recurse(start_isle,0,0,1e6);
 114 }
 115
 116 void setup_search(void) {
 117   neighbours= mcalloc(sizeof(*neighbours) * islandtablesz);
 118
 119   SQL_PREPARE(ss_neigh,
 120               "SELECT biid, dist FROM routes WHERE aiid=?");
 121 }