mas01cr@498: extern "C" {
mas01cr@498: #include "audioDB_API.h"
mas01cr@509: }
mas01cr@498: #include "audioDB-internals.h"
mas01cr@498: 
mas01cr@498: static int audiodb_l2norm_existing(adb_t *adb) {
mas01mc@519:   double *data_buffer = 0, *l2norm_buffer = 0;
mas01cr@498:   adb_header_t *header = adb->header;
mas01cr@509:   size_t data_buffer_size = align_page_up(header->length);
mas01cr@498:   size_t nvectors = header->length / (sizeof(double) * header->dim);
mas01cr@498:   /* FIXME: this map of the vector data will lose if we ever turn the
mas01cr@498:    * l2norm flag on when we have already inserted a large number of
mas01cr@498:    * vectors, as the mmap() will fail.  "Don't do that, then" is one
mas01cr@498:    * possible answer. */
mas01cr@498:   mmap_or_goto_error(double *, data_buffer, header->dataOffset, data_buffer_size);
mas01cr@498:   l2norm_buffer = (double *) malloc(nvectors * sizeof(double));
mas01cr@498:   if(!l2norm_buffer) {
mas01cr@498:     goto error;
mas01cr@498:   }
mas01cr@498:   audiodb_l2norm_buffer(data_buffer, header->dim, nvectors, l2norm_buffer);
mas01cr@498:   if(lseek(adb->fd, adb->header->l2normTableOffset, SEEK_SET) == (off_t) -1) {
mas01cr@498:     goto error;
mas01cr@498:   }
mas01cr@498:   write_or_goto_error(adb->fd, l2norm_buffer, nvectors * sizeof(double));
mas01cr@498: 
mas01cr@498:   munmap(data_buffer, data_buffer_size);
mas01cr@498:   free(l2norm_buffer);
mas01cr@498: 
mas01cr@498:   return 0;
mas01cr@498: 
mas01cr@498:  error:
mas01cr@498:   maybe_munmap(data_buffer, data_buffer_size);
mas01cr@498:   if(l2norm_buffer) {
mas01cr@498:     free(l2norm_buffer);
mas01cr@498:   }
mas01cr@498:   return 1;
mas01cr@498: }
mas01cr@498: 
mas01cr@498: int audiodb_l2norm(adb_t *adb) {
mas01cr@498:   adb_header_t *header = adb->header;
mas01cr@498:   if(!(adb->flags & O_RDWR)) {
mas01cr@498:     return 1;
mas01cr@498:   }
mas01cr@509:   if(header->flags & ADB_HEADER_FLAG_L2NORM) {
mas01cr@498:     /* non-error code for forthcoming backwards-compatibility
mas01cr@498:      * reasons */
mas01cr@498:     return 0;
mas01cr@498:   }
mas01cr@509:   if((!(header->flags & ADB_HEADER_FLAG_REFERENCES)) && (header->length > 0)) {
mas01cr@498:     if(audiodb_l2norm_existing(adb)) {
mas01cr@498:       goto error;
mas01cr@498:     }
mas01cr@498:   }
mas01cr@509:   adb->header->flags |= ADB_HEADER_FLAG_L2NORM;
mas01cr@498:   return audiodb_sync_header(adb);
mas01cr@498: 
mas01cr@498:  error:
mas01cr@498:   return 1;
mas01cr@498: }