Browse Source

* util.c : use bare number instead sample

master
Alex 'AdUser' Z 10 years ago
parent
commit
490be1c7bc
  1. 35
      src/util.c

35
src/util.c

@ -41,16 +41,20 @@ void usage(int exitcode) {
exit(exitcode); exit(exitcode);
} }
int search_similar(imdb_t *db, imdb_rec_t *sample, float tresh) int search_similar(imdb_t *db, uint64_t number, float tresh)
{ {
int ret = 0, i = 0; int ret = 0, i = 0;
imdb_rec_t sample;
imdb_match_t *matches = NULL; imdb_match_t *matches = NULL;
imdb_search_t search; imdb_search_t search;
memset(&search, 0x0, sizeof(search));
memset(&sample, 0x0, sizeof(imdb_rec_t));
memset(&search, 0x0, sizeof(imdb_search_t));
search.tresh_ratio = tresh; search.tresh_ratio = tresh;
search.tresh_bitmap = tresh; search.tresh_bitmap = tresh;
ret = imdb_search(db, sample, &search, &matches); sample.num = number;
ret = imdb_search(db, &sample, &search, &matches);
if (ret == -1) { if (ret == -1) {
printf("%s\n", db->errstr); printf("%s\n", db->errstr);
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
@ -96,24 +100,27 @@ int db_usage_map(imdb_t *db, unsigned short int cols)
return 0; return 0;
} }
int rec_bitmap(imdb_t *db, imdb_rec_t *sample) int rec_bitmap(imdb_t *db, uint64_t number)
{ {
imdb_rec_t rec;
uint16_t row; uint16_t row;
uint8_t i, j; uint8_t i, j;
char c; char c;
assert(db != NULL); assert(db != NULL);
assert(sample != NULL); memset(&rec, 0x0, sizeof(imdb_rec_t));
if (imdb_read_rec(db, sample) < 1) rec.num = number;
if (imdb_read_rec(db, &rec) < 1)
return -1; return -1;
if (!sample->data[0]) { if (!rec.data[0]) {
puts("Sample not exists"); puts("Sample not exists");
return 0; return 0;
} }
for (i = 0; i < 16; i++) { for (i = 0; i < 16; i++) {
row = *(((uint16_t *) (&sample->data[REC_OFF_BM])) + i); row = *(((uint16_t *) (&rec.data[REC_OFF_BM])) + i);
for (j = 0; j < 16; j++) { for (j = 0; j < 16; j++) {
c = (row & 1) == 1 ? CHAR_USED : CHAR_NONE; c = (row & 1) == 1 ? CHAR_USED : CHAR_NONE;
putchar(c); putchar(c);
@ -188,13 +195,11 @@ int main(int argc, char **argv)
float tresh = 0.10; float tresh = 0.10;
unsigned short int cols = 64, map = 0; unsigned short int cols = 64, map = 0;
imdb_t db; imdb_t db;
imdb_rec_t sample;
uint64_t a = 0, b = 0; uint64_t a = 0, b = 0;
char *c = NULL; char *c = NULL;
char opt = '\0'; char opt = '\0';
memset(&db, 0x0, sizeof(imdb_t)); memset(&db, 0x0, sizeof(imdb_t));
memset(&sample, 0x0, sizeof(imdb_rec_t));
if (argc < 3) if (argc < 3)
usage(EXIT_FAILURE); usage(EXIT_FAILURE);
@ -213,7 +218,7 @@ int main(int argc, char **argv)
break; break;
case 'B' : case 'B' :
mode = bitmap; mode = bitmap;
sample.num = atoll(optarg); a = atoll(optarg);
break; break;
case 'D' : case 'D' :
map = 1; map = 1;
@ -226,7 +231,7 @@ int main(int argc, char **argv)
break; break;
case 'S' : case 'S' :
mode = search; mode = search;
sample.num = atoll(optarg); a = atoll(optarg);
break; break;
case 'U' : case 'U' :
mode = usage_map; mode = usage_map;
@ -240,7 +245,7 @@ int main(int argc, char **argv)
} }
} }
if ((mode == search || mode == bitmap) && sample.num <= 0) if ((mode == search || mode == bitmap) && a <= 0)
usage(EXIT_FAILURE); usage(EXIT_FAILURE);
if (mode == diff && (a <= 0 || b <= 0)) if (mode == diff && (a <= 0 || b <= 0))
@ -260,10 +265,10 @@ int main(int argc, char **argv)
} }
if (mode == search) if (mode == search)
search_similar(&db, &sample, tresh); search_similar(&db, a, tresh);
if (mode == bitmap) if (mode == bitmap)
rec_bitmap(&db, &sample); rec_bitmap(&db, a);
if (mode == usage_map) if (mode == usage_map)
db_usage_map(&db, cols); db_usage_map(&db, cols);

Loading…
Cancel
Save