motiejus/turbonss
1
Fork 0
Code Releases Activity

included option -k to specify the number of keys to use

Browse Source
This commit is contained in:
fc_botelho 2005-01-05 19:48:23 +00:00
parent b63e04ee09
commit 30d7a654f0
2 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/cmph.c
View File

@ -80,7 +80,7 @@ mphf_t *mph_create(mph_t *mph)
break; break;
case MPH_BMZ: /* included -- Fabiano */ case MPH_BMZ: /* included -- Fabiano */
DEBUGP("Creating bmz hash\n"); DEBUGP("Creating bmz hash\n");
mphf = bmz_mph_create(mph, 1.10); mphf = bmz_mph_create(mph, 1.15);
break; break;
default: default:
assert(0); assert(0);

24
src/main.c
View File

@ -12,12 +12,12 @@
void usage(const char *prg) void usage(const char *prg)
{ {
fprintf(stderr, "usage: %s [-v] [-h] [-V] [-g [-s seed] ] [-m file.mph] [-a algorithm] keysfile\n", prg); fprintf(stderr, "usage: %s [-v] [-h] [-V] [-k] [-g [-s seed] ] [-m file.mph] [-a algorithm] keysfile\n", prg);
} }
void usage_long(const char *prg) void usage_long(const char *prg)
{ {
uint32 i; uint32 i;
fprintf(stderr, "usage: %s [-v] [-h] [-V] [-g [-s seed] ] [-m file.mph] [-a algorithm] keysfile\n", prg); fprintf(stderr, "usage: %s [-v] [-h] [-V] [-k] [-g [-s seed] ] [-m file.mph] [-a algorithm] keysfile\n", prg);
fprintf(stderr, "Minimum perfect hashing tool\n\n"); fprintf(stderr, "Minimum perfect hashing tool\n\n");
fprintf(stderr, " -h\t print this help message\n"); fprintf(stderr, " -h\t print this help message\n");
fprintf(stderr, " -a\t algorithm - valid values are\n"); fprintf(stderr, " -a\t algorithm - valid values are\n");
@ -26,6 +26,7 @@ void usage_long(const char *prg)
for (i = 0; i < HASH_COUNT; ++i) fprintf(stderr, " \t * %s\n", hash_names[i]); for (i = 0; i < HASH_COUNT; ++i) fprintf(stderr, " \t * %s\n", hash_names[i]);
fprintf(stderr, " -V\t print version number and exit\n"); fprintf(stderr, " -V\t print version number and exit\n");
fprintf(stderr, " -v\t increase verbosity (may be used multiple times)\n"); fprintf(stderr, " -v\t increase verbosity (may be used multiple times)\n");
fprintf(stderr, " -k\t number of keys\n");
fprintf(stderr, " -g\t generation mode\n"); fprintf(stderr, " -g\t generation mode\n");
fprintf(stderr, " -s\t random seed\n"); fprintf(stderr, " -s\t random seed\n");
fprintf(stderr, " -m\t minimum perfect hash function file \n"); fprintf(stderr, " -m\t minimum perfect hash function file \n");
@ -91,6 +92,7 @@ int main(int argc, char **argv)
FILE *mphf_fd = stdout; FILE *mphf_fd = stdout;
const char *keys_file = NULL; const char *keys_file = NULL;
FILE *keys_fd; FILE *keys_fd;
uint32 nkeys = UINT_MAX;
uint32 seed = UINT_MAX; uint32 seed = UINT_MAX;
CMPH_HASH *hashes = NULL; CMPH_HASH *hashes = NULL;
uint32 nhashes = 0; uint32 nhashes = 0;
@ -103,7 +105,7 @@ int main(int argc, char **argv)
while (1) while (1)
{ {
char c = getopt(argc, argv, "hVva:f:gm:s:"); char c = getopt(argc, argv, "hVvk:a:f:gm:s:");
if (c == -1) break; if (c == -1) break;
switch (c) switch (c)
{ {
@ -120,6 +122,16 @@ int main(int argc, char **argv)
case 'g': case 'g':
generate = 1; generate = 1;
break; break;
case 'k':
{
char *endptr;
nkeys = strtoul(optarg, &endptr, 10);
if(*endptr != 0) {
fprintf(stderr, "Invalid number of keys %s\n", optarg);
exit(1);
}
}
break;
case 'm': case 'm':
mphf_file = strdup(optarg); mphf_file = strdup(optarg);
break; break;
@ -195,7 +207,7 @@ int main(int argc, char **argv)
memcpy(mphf_file + strlen(keys_file), ".mph\0", 5); memcpy(mphf_file + strlen(keys_file), ".mph\0", 5);
} }
keys_fd = fopen(keys_file, "r"); keys_fd = fopen64(keys_file, "r");
if (keys_fd == NULL) if (keys_fd == NULL)
{ {
fprintf(stderr, "Unable to open file %s: %s\n", keys_file, strerror(errno)); fprintf(stderr, "Unable to open file %s: %s\n", keys_file, strerror(errno));
@ -203,7 +215,9 @@ int main(int argc, char **argv)
} }
source.data = (void *)keys_fd; source.data = (void *)keys_fd;
source.nkeys = count_keys(keys_fd); if (seed == UINT_MAX) seed = time(NULL);
if(nkeys == UINT_MAX) source.nkeys = count_keys(keys_fd);
else source.nkeys = nkeys;
source.read = key_read; source.read = key_read;
source.dispose = key_dispose; source.dispose = key_dispose;
source.rewind = key_rewind; source.rewind = key_rewind;