93 lines
2.2 KiB
C
93 lines
2.2 KiB
C
#include "../hashmap/myhashmap.h"
|
|
#include <assert.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#define MAX_STR_LEN 64
|
|
|
|
/* My custom data type stored as value */
|
|
struct my_custom_type {
|
|
int age;
|
|
char favourite_brand[MAX_STR_LEN];
|
|
};
|
|
|
|
/* Let's hash our name */
|
|
static unsigned int my_hash_func(const void *key) {
|
|
char *name = (char *)key;
|
|
size_t len = strlen(name);
|
|
|
|
unsigned int hash = 0;
|
|
for (size_t i = 0; i < len; ++i) {
|
|
hash += (int)name[i];
|
|
}
|
|
|
|
return hash % 2069;
|
|
}
|
|
|
|
/* Let's write our compare function */
|
|
bool my_equal_fun(const void *key_a, const void *key_b) {
|
|
char *name_a = (char *)key_a;
|
|
char *name_b = (char *)key_b;
|
|
|
|
if (strcmp(name_a, name_b) == 0) {
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/* And our last two functions, the free key and value called inside mcl_hm_remove() */
|
|
void my_free_key(void *key) {
|
|
free(key);
|
|
}
|
|
|
|
void my_free_value(void *value) {
|
|
struct my_custom_type *mct = (struct my_custom_type *)value;
|
|
|
|
free(mct);
|
|
}
|
|
|
|
void test_hm1(void) {
|
|
/* Allocate a new hashmap */
|
|
/* Pass your custom hash, equal and free functions */
|
|
/* This hashmap will contain names as keys and a custom type as value */
|
|
size_t key_size = sizeof(char) * MAX_STR_LEN;
|
|
size_t value_size = sizeof(int) + sizeof(char) * MAX_STR_LEN;
|
|
hashmap_s *map =
|
|
hm_new(my_hash_func, my_equal_fun, my_free_key, my_free_value, key_size, value_size);
|
|
assert(map != NULL);
|
|
|
|
/* Make a new value */
|
|
struct my_custom_type p1 = {
|
|
.age = 21,
|
|
};
|
|
strncpy(p1.favourite_brand, "Ferrari", sizeof(p1.favourite_brand));
|
|
|
|
/* Insert a new pair */
|
|
assert(hm_set(map, "John", &p1));
|
|
|
|
/* Retrieve the data */
|
|
/* Remember to free the value from the get function */
|
|
bucket_s *john = hm_get(map, "John");
|
|
assert(john != NULL);
|
|
|
|
char *name = (char *)john->key;
|
|
struct my_custom_type *john_v = (struct my_custom_type *)john->value;
|
|
int age = john_v->age;
|
|
char *fav_brand = john_v->favourite_brand;
|
|
|
|
assert(strcmp(name, "John") == 0);
|
|
assert(age == 21);
|
|
assert(strcmp(fav_brand, "Ferrari") == 0);
|
|
|
|
/* Free the bucket */
|
|
hm_free_bucket(john);
|
|
|
|
/* Remove a key from hash map */
|
|
assert(hm_remove(map, "John"));
|
|
|
|
/* Deallocate */
|
|
hm_free(map);
|
|
}
|