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(VectorIndex index,void *item,void *data) {
  62     Knockout *kod = (Knockout*) data;
  63     VectorIndex 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(VectorIndex 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(VectorIndex 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         VectorIndex 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,VectorIndex *index,tuple *query) {
 141     return HashVector_next( &r->content, index, query );
 142 }
 143