bloom.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdbool.h>
#include <stdint.h>  
#include <assert.h>

#include "hash.h"
#include "hash_f.h"
#include "bloom.h"

void add_hash_functions(bloom_filter *bf) {
    for (int i = 0; i < bf->k; i++) {
        bf->hashfunc[i] = hashf[i];
    }
}

bloom_filter *bloom_init(int size, int k) {
    bloom_filter *bf = malloc(sizeof(bloom_filter));
    assert(bf != NULL);
    bf->size = size;
    bf->bits = malloc(size * sizeof(int));
    assert(bf->bits != NULL);
    assert(k <= 12);
    bf->hashfunc = malloc(k * sizeof(hash_func_t));
    assert(bf->hashfunc != NULL);
    bf->k = k;
    add_hash_functions(bf);
    return bf;
}

void bloom_add(bloom_filter *bf, const char *item) {
    for (int i = 0; i < bf->k; i++) {
        int hash_val = bf->hashfunc[i](item, bf->size);
        bf->bits[hash_val] = 1;
    }
}

bool bloom_check(bloom_filter *bf, const char *item) {
    for (int i = 0; i < bf->k; i++) {
        int hash_val = bf->hashfunc[i](item, bf->size);
        if (bf->bits[hash_val] == 0) {
            return false;
        }
    }
    return true;    
}

void bloom_free(bloom_filter *bf) {
    free(bf->bits);
    free(bf->hashfunc);
    free(bf);
}