mas01cr@401
|
1 #include "audioDB.h"
|
mas01cr@401
|
2 extern "C" {
|
mas01cr@401
|
3 #include "audioDB_API.h"
|
mas01cr@401
|
4 #include "audioDB-internals.h"
|
mas01cr@401
|
5 }
|
mas01cr@401
|
6
|
mas01cr@401
|
7 static int audiodb_l2norm_existing(adb_t *adb) {
|
mas01cr@401
|
8 double *data_buffer, *l2norm_buffer;
|
mas01cr@401
|
9 adb_header_t *header = adb->header;
|
mas01cr@401
|
10 size_t data_buffer_size = ALIGN_PAGE_UP(header->length);
|
mas01cr@401
|
11 size_t nvectors = header->length / (sizeof(double) * header->dim);
|
mas01cr@401
|
12 /* FIXME: this map of the vector data will lose if we ever turn the
|
mas01cr@401
|
13 * l2norm flag on when we have already inserted a large number of
|
mas01cr@401
|
14 * vectors, as the mmap() will fail. "Don't do that, then" is one
|
mas01cr@401
|
15 * possible answer. */
|
mas01cr@401
|
16 mmap_or_goto_error(double *, data_buffer, header->dataOffset, data_buffer_size);
|
mas01cr@401
|
17 l2norm_buffer = (double *) malloc(nvectors * sizeof(double));
|
mas01cr@401
|
18 if(!l2norm_buffer) {
|
mas01cr@401
|
19 goto error;
|
mas01cr@401
|
20 }
|
mas01cr@426
|
21 audiodb_l2norm_buffer(data_buffer, header->dim, nvectors, l2norm_buffer);
|
mas01cr@401
|
22 if(lseek(adb->fd, adb->header->l2normTableOffset, SEEK_SET) == (off_t) -1) {
|
mas01cr@401
|
23 goto error;
|
mas01cr@401
|
24 }
|
mas01cr@410
|
25 write_or_goto_error(adb->fd, l2norm_buffer, nvectors * sizeof(double));
|
mas01cr@401
|
26
|
mas01cr@401
|
27 munmap(data_buffer, data_buffer_size);
|
mas01cr@401
|
28 free(l2norm_buffer);
|
mas01cr@401
|
29
|
mas01cr@401
|
30 return 0;
|
mas01cr@401
|
31
|
mas01cr@401
|
32 error:
|
mas01cr@401
|
33 maybe_munmap(data_buffer, data_buffer_size);
|
mas01cr@401
|
34 if(l2norm_buffer) {
|
mas01cr@401
|
35 free(l2norm_buffer);
|
mas01cr@401
|
36 }
|
mas01cr@401
|
37 return 1;
|
mas01cr@401
|
38 }
|
mas01cr@401
|
39
|
mas01cr@401
|
40 int audiodb_l2norm(adb_t *adb) {
|
mas01cr@401
|
41 adb_header_t *header = adb->header;
|
mas01cr@403
|
42 if(!(adb->flags & O_RDWR)) {
|
mas01cr@403
|
43 return 1;
|
mas01cr@403
|
44 }
|
mas01cr@401
|
45 if(header->flags & O2_FLAG_L2NORM) {
|
mas01cr@401
|
46 /* non-error code for forthcoming backwards-compatibility
|
mas01cr@401
|
47 * reasons */
|
mas01cr@401
|
48 return 0;
|
mas01cr@401
|
49 }
|
mas01cr@401
|
50 if((!(header->flags & O2_FLAG_LARGE_ADB)) && (header->length > 0)) {
|
mas01cr@401
|
51 if(audiodb_l2norm_existing(adb)) {
|
mas01cr@401
|
52 goto error;
|
mas01cr@401
|
53 }
|
mas01cr@401
|
54 }
|
mas01cr@401
|
55 adb->header->flags |= O2_FLAG_L2NORM;
|
mas01cr@401
|
56 return audiodb_sync_header(adb);
|
mas01cr@401
|
57
|
mas01cr@401
|
58 error:
|
mas01cr@401
|
59 return 1;
|
mas01cr@401
|
60 }
|