X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~yarrgweb/git?a=blobdiff_plain;f=yarrg%2Frscommon.h;h=d91765e83b4b8ed0d5426897a69fc784d499dc9a;hb=ba117b2a7d73eae16ad76edf42ebf7e7c93f282a;hp=658bbdd71404bbc2bc2f61dbef9a30591050a860;hpb=3ca67ce14212ba4421029d7b8db90f03f106c67c;p=ypp-sc-tools.web-live.git diff --git a/yarrg/rscommon.h b/yarrg/rscommon.h index 658bbdd..d91765e 100644 --- a/yarrg/rscommon.h +++ b/yarrg/rscommon.h @@ -1,14 +1,101 @@ #ifndef RSCOMMON_H #define RSCOMMON_H -#include - #include -#include -#include -extern struct sqlite *db; +#define DEBUG_FLAG_LIST \ + DF(sql) \ + DF(sql2) \ + DF(value) \ + DF(search) \ + DF(check) \ + DF(lp) + +#define debug stdout +#define DEBUG_DEV "/dev/stdout" + +#include "common.h" + + +#define SQL_MUST( call ) ({ \ + /* `call' is an expression returning result, using const char *sqe; \ + * chk1 and chk2 are blocks using sqe and int sqr; */ \ + const char *sql_must_call_string= #call; \ + int sqr; \ + if (DEBUGP(sql2)) fprintf(stderr,"SQL %s", sql_must_call_string); \ + sqr= (call); \ + if (DEBUGP(sql2)) fprintf(stderr," = %d\n", sqr); \ + if (sqr) sql_fatal("(unknown)", sqr, sql_must_call_string); \ + }) \ + +void sql_fatal(const char *stmt_what, int sqr, const char *act_what) NORET; + +#define SQL_STEP(ssh) (sql_step((ssh), #ssh, __FILE__, __LINE__)) +int sql_step(sqlite3_stmt *ssh, const char *ssh_string, + const char *file, int line); + +#define SQL_DISTINCT_DECL(cols, nintcols) \ + int cols[nintcols]; \ + cols[0]= -1; +#define SQL_DISTINCT_STEP(ssh, cols, nkeycols) \ + (sql_step_distinct((ssh), #ssh, __FILE__, __LINE__, \ + (cols), sizeof((cols))/sizeof((cols)[0]), nkeycols)) +int sql_step_distinct(sqlite3_stmt *ssh, const char *ssh_string, + const char *file, int line, + int *cols, int ncols, int nkeycols); + /* These work if we're making a query whose columns consist of: + * - keys: integer column(s) on which the results are sorted by the query + * - consequences: zero or more integer cols strictly dependent on the keys + * - extra: zero or more further (possibly non-integer) columns + * + * Call SQL_DISTINCT_DECL, passing intcols = the total number of keys and + * consequences; it will declare int cols[intfields]; + * + * Then each SQL_DISTINCT_STEP is like SQL_STEP only you have to + * pass the number of key columns and it only returns rows with + * distinct keys. Rows with all-identical keys are asserted to + * have identical consequences. After each call to + * SQL_DISTINCT_STEP the keys and consequences will be stored in + * cols. + */ + +int sql_single_int(const char *stmt); + +#define SQL_PREPARE(ss,stmt) ((ss)= sql_prepare((stmt),#ss)) +sqlite3_stmt *sql_prepare(const char *stmt, const char *what); + +#define SQL_BIND(ss,index,value) (sql_bind((ss),(index),(value),#ss,#value)) +void sql_bind(sqlite3_stmt *ss, int index, int value, + const char *ss_what, const char *val_what); + +#define MAX_ROUTELEN 20 + +extern sqlite3 *db; + +void setup_sql(void); + + +typedef struct { + double distance_loss_factor; + struct TradesBlock *trades; + double route_tail_value; +} IslandPair; + +IslandPair *ipair_get(int si, int di); + +double value_route(int nislands, const int *islands, int exclude_arbitrage); +void setup_value(void); + +void setup_search(void); +void search(int start_isle); + +extern double max_mass, max_volu, max_capi; +extern double distance_loss_factor_per_league; +extern int max_dist; + +#define LOSS_FACTOR_PER_DELAY_SLOT (1-1e-8) + +extern int islandtablesz; -void value_route(int nislands, const int *islands); #endif /*RSCOMMON_H*/