vector/relation.c

   1 #include <stdarg.h>
   2 #include <stdlib.h>
   3 #include <relation.h>
   4
   5 relation *relation_create(tupleschema *schema) {
   6     relation *r = (relation *) malloc( sizeof( relation ) );
   7     (*r) = (relation) {
   8         .content = (hashvector) {
   9             .table = (vector) {
  10                 .variant = nibble_index_levels,
  11                 .size = 16,
  12                 .entries = 0
  13             },
  14             .fill = 0, .holes = 0, .type = (itemkeyfun*) schema
  15         },
  16         .constraints = (vector) {
  17             .variant = single_index_level,
  18             .size = 0,
  19             .entries = 0
  20         },
  21     };
  22     return r;
  23 }
  24
  25 #define COPYA(T,P,N) (T*) memcpy( malloc( N * sizeof(T) ), P, N * sizeof( T ) )
  26 #define COPY(T,P) COPYA(T,P,1)
  27
  28 // Add an indexing hashvector to the relation using the given column
  29 // flags with 1 indicating key column and 0 indicating value column.
  30 int relation_add_constraint(relation *r,...) {
  31     va_list ap;
  32     tupleschema *ts = (tupleschema *) r->content.type;
  33     tuple *columns = (tuple*) calloc( ts->arity, sizeof( void* ) );
  34     int i = 0;
  35     va_start( ap, r );
  36     for ( ; i < ts->arity; i++ ) {
  37         if ( va_arg( ap, int ) ) {
  38             (*columns)[i] = ts->columns[i];
  39         }
  40     }
  41     va_end( ap );
  42     ts = tupleschema_create( ts->arity, columns );
  43     i = (int) r->constraints.size;
  44     vector_append(
  45         &r->constraints,
  46         hashvector_create( nibble_index_levels, (itemkeyfun*) ts ) );
  47     return i;
  48 }
  49
  50 //============== Adding an item =============
  51 // Iteration context for adding or querying a relation
  52 typedef struct {
  53     relation *rel;
  54     hashvector knockouts;
  55     void *item;
  56 } knockout;
  57
  58 // Determine matches to ((knockout*)data)->key in
  59 // (hashvector*)item, optionally using ((knockout*)data)->columns
  60 // for ignoring full matches to the key tuple.
  61 static int knockout_check(vector_index index,void *item,void *data) {
  62     knockout *kod = (knockout*) data;
  63     vector_index i = 0;
  64     for ( ; i < ((hashvector*) item)->table.size; i++ ) {
  65         void *old = hashvector_next( (hashvector*) item, &i, kod->item );
  66         if ( old ) {
  67             hashvector_add( &kod->knockouts, old );
  68         }
  69     }
  70     return 0;
  71 }
  72
  73 // delete the (tuple*)item from the (hashvector*)data
  74 static int knockout_delete(vector_index index,void *item,void *data) {
  75     hashvector_delete( (hashvector*) item, data );
  76     return 0;
  77 }
  78
  79 // add the (tuple*)data to the (hashvector*)item
  80 static int knockout_add(vector_index index,void *item,void *data) {
  81     hashvector_add( (hashvector*)item, data );
  82     return 0;
  83 }
  84
  85 // Find and remove all collisions for a query, unless "add" is
  86 // non-zero in which case the function aborts if there is any match in
  87 // the main content.
  88 static int knockout_clear(knockout *this,relation *r,tuple *item,int add) {
  89     (*this) = (knockout) {
  90         .rel = r,
  91         .knockouts = {
  92             .table = {
  93                 .variant = nibble_index_levels, .size = 16, .entries = 0
  94             },
  95             .fill = 0, .holes = 0, .type = r->content.type,
  96         },
  97         .item = item
  98     };
  99     knockout_check( 0, &r->content, this );
 100     if ( add ) {
 101         if ( this->knockouts.fill > 0 ) {
 102             return 0;
 103         }
 104         // Find all constraint knockouts for addition
 105         vector_iterate( &r->constraints, 0, knockout_check, this );
 106     }
 107     if ( this->knockouts.fill > 0 ) {
 108         // Delete them from all tables
 109         vector_index i;
 110         for ( i = 0; i < this->knockouts.table.size; i++ ) {
 111             void *t = hashvector_next( &this->knockouts, &i, 0 );
 112             if ( t ) {
 113                 hashvector_delete( &r->content, t );
 114                 vector_iterate( &r->constraints, 0, knockout_delete, t );
 115             }
 116         }
 117     }
 118     return 1;
 119 }
 120
 121 // add a tuple to a relation and return a vector of knocked out
 122 // tuples, if any, or 0 otherwise.
 123 vector *relation_add(relation *r,tuple *item) {
 124     knockout data;
 125     if ( knockout_clear( &data, r, item, 1 ) ) {
 126         // Add the new tuple
 127         hashvector_add( &r->content, item );
 128         vector_iterate( &r->constraints, 0, knockout_add, item );
 129         return hashvector_contents( &data.knockouts, single_index_level, 0 );
 130     }
 131     return 0;
 132 }
 133
 134 vector *relation_delete(relation *r,tuple *item) {
 135     knockout data;
 136     (void) knockout_clear( &data, r, item, 0 );
 137     return hashvector_contents( &data.knockouts, single_index_level, 0 );
 138 }
 139
 140 void *relation_next(relation *r,vector_index *index,tuple *query) {
 141     return hashvector_next( &r->content, index, query );
 142 }
 143