X-Git-Url: https://git.rrq.au/?a=blobdiff_plain;f=vector%2Ftupleitem.c;h=ce5c71386476be7ed1449ef96a6c9f9b27240e2c;hb=fff7d7c2ac83984bd2fc7ec3c72dc455c7c72807;hp=eff77cedb56e0146c80571aeffc504856d197c20;hpb=6f54a8281e4e5d6bc05e6b4eadc3327d5e48614a;p=rrq%2Frrqmisc.git

diff --git a/vector/tupleitem.c b/vector/tupleitem.c
index eff77ce..ce5c713 100644
--- a/vector/tupleitem.c
+++ b/vector/tupleitem.c
@@ -1,9 +1,9 @@
+#include <stdarg.h>
 #include <stdlib.h>
+#include <string.h>
 #include <tupleitem.h>
 
-typedef void *tuple[];
-
-#define COLUMN def->schema[i]
+#define COLUMN def->columns
 
 unsigned long tupleitem_hashcode(itemkeyfun *this,void *key) {
     tupleschema *def = (tupleschema *) this;
@@ -11,9 +11,9 @@ unsigned long tupleitem_hashcode(itemkeyfun *this,void *key) {
     int i = 0;
     unsigned long value = 0;
     for ( ; i < def->arity; i++ ) {
-	if ( COLUMN ) {
+	if ( COLUMN[i] ) {
 	    value <<= 3;
-	    value += COLUMN->hashcode( COLUMN, (*kp)[i] );
+	    value += COLUMN[i]->hashcode( COLUMN[i], (*kp)[i] );
 	}
     }
     return value;
@@ -30,8 +30,8 @@ int tupleitem_haskey(itemkeyfun *this,void *item,void *key) {
     int i = 0;
     int haskey = 1;
     for ( ; i < def->arity; i++ ) {
-	if ( COLUMN ) {
-	    haskey &= COLUMN->haskey( COLUMN, (*tp)[i], (*kp)[i] );
+	if ( COLUMN[i] ) {
+	    haskey &= COLUMN[i]->haskey( COLUMN[i], (*tp)[i], (*kp)[i] );
 	}
     }
     return haskey;
@@ -48,14 +48,14 @@ void *tupleitem_itemkey(itemkeyfun *this,void *item) {
     int i, j;
     int keylen = 0;
     for ( i = 0 ; i < def->arity; i++ ) {
-	if ( COLUMN ) {
+	if ( COLUMN[i] ) {
 	    keylen++;
 	}
     }
     void **parts = calloc( keylen, sizeof( void* ) );
     for ( i = 0, j = 0; i < def->arity; i++ ) {
-	if ( COLUMN ) {
-	    parts[j++] = COLUMN->itemkey( COLUMN, (*tp)[i] );
+	if ( COLUMN[i] ) {
+	    parts[j++] = COLUMN[i]->itemkey( COLUMN[i], (*tp)[i] );
 	}
     }
     return (void*) parts;
@@ -70,10 +70,54 @@ void tupleitem_releasekey(itemkeyfun *this,void *key) {
     tuple *kp = (tuple*) key;
     int i,j;
     for ( i = 0, j = 0; i < def->arity; i++ ) {
-	if ( COLUMN ) {
-	    COLUMN->releasekey( COLUMN, (*kp)[j++] );
+	if ( COLUMN[i] ) {
+	    COLUMN[i]->releasekey( COLUMN[i], (*kp)[j++] );
 	}
     }
     free( key );
 }
 
+// Allocate
+tuple *tuple_create(int arity,...) {
+    va_list ap;
+    int i;
+    tuple *t = (tuple *)malloc( arity * sizeof( void* ) );
+    for ( i = 0; i < arity; i++ ) {
+	(*t)[i] = va_arg( ap, void* );
+    }
+    return t;
+}
+
+tupleschema *tupleschema_create(int arity,tuple *columns) {
+    tupleschema *ts = (tupleschema*) malloc( sizeof( tupleschema ) );
+    (*ts) = (tupleschema) {
+	.functions = {
+	    .hashcode = tupleitem_hashcode,
+	    .haskey = tupleitem_haskey,
+	    .itemkey = tupleitem_itemkey,
+	    .releasekey = tupleitem_releasekey
+	},
+	.arity = arity,
+	.columns = (itemkeyfun**) columns
+    };
+    return ts;
+}
+
+#define COPYA(T,P,N) (T*) memcpy( malloc( N * sizeof(T) ), P, N * sizeof( T ) )
+#define COPY(T,P) COPYA(T,P,1)
+
+// Duplicate a tupleschema with optionally some columns reset.
+tupleschema *tupleschema_mask(tupleschema *schema,...) {
+    tupleschema *masked = COPY(tupleschema,schema);
+    masked->columns = COPYA( itemkeyfun*, schema->columns, schema->arity );
+    va_list ap;
+    int i;
+    for ( ;; ) {
+	i = va_arg( ap, int );
+	if ( i < 0 || i >= schema->arity ) {
+	    break;
+	}
+	masked->columns[i] = 0;
+    };
+    return masked;
+}