René Scharfe <[email protected]> writes:
> Storing integer values in pointers is a trick that seems to have worked
> so far for fast-export. A portable way to avoid that trick without
> requiring more memory would be to use a union.
>
> Or we could roll our own custom hash map, as I mused in an earlier post.
> That would duplicate quite a bit of code; are there reusable pieces
> hidden within that could be extracted into common functions?
Hmm, this together with your follow-up does not look too bad, but it
does introduce quite a lot of code that could be refactored, so I am
not sure if I really like it or not.
>
> ---
> builtin/fast-export.c | 105 ++++++++++++++++++++++++++++++++----------
> 1 file changed, 81 insertions(+), 24 deletions(-)
>
> diff --git a/builtin/fast-export.c b/builtin/fast-export.c
> index 530df12f05..627b0032f3 100644
> --- a/builtin/fast-export.c
> +++ b/builtin/fast-export.c
> @@ -14,7 +14,6 @@
> #include "diffcore.h"
> #include "log-tree.h"
> #include "revision.h"
> -#include "decorate.h"
> #include "string-list.h"
> #include "utf8.h"
> #include "parse-options.h"
> @@ -71,9 +70,65 @@ static int parse_opt_tag_of_filtered_mode(const struct
> option *opt,
> return 0;
> }
>
> -static struct decoration idnums;
> +struct object_mark_entry {
> + const struct object *base;
> + uint32_t mark;
> +};
> +
> +struct object_marks {
> + unsigned int size;
> + unsigned int nr;
> + struct object_mark_entry *entries;
> +};
> +
> +static struct object_marks idnums;
> static uint32_t last_idnum;
>
> +static unsigned int hash_obj(const struct object *obj, unsigned int n)
> +{
> + return sha1hash(obj->oid.hash) % n;
> +}
> +
> +static void set_object_mark(struct object_marks *n, const struct object
> *base,
> + uint32_t mark)
> +{
> + unsigned int size = n->size;
> + struct object_mark_entry *entries = n->entries;
> + unsigned int j = hash_obj(base, size);
> +
> + while (entries[j].base) {
> + if (entries[j].base == base) {
> + entries[j].mark = mark;
> + return;
> + }
> + if (++j >= size)
> + j = 0;
> + }
> + entries[j].base = base;
> + entries[j].mark = mark;
> + n->nr++;
> +}
> +
> +static void grow_object_marks(struct object_marks *n)
> +{
> + unsigned int i;
> + unsigned int old_size = n->size;
> + struct object_mark_entry *old_entries = n->entries;
> +
> + n->size = (old_size + 1000) * 3 / 2;
> + n->entries = xcalloc(n->size, sizeof(n->entries[0]));
> + n->nr = 0;
> +
> + for (i = 0; i < old_size; i++) {
> + const struct object *base = old_entries[i].base;
> + uint32_t mark = old_entries[i].mark;
> +
> + if (mark)
> + set_object_mark(n, base, mark);
> + }
> + free(old_entries);
> +}
> +
> static int has_unshown_parent(struct commit *commit)
> {
> struct commit_list *parent;
> @@ -156,20 +211,13 @@ static void anonymize_path(struct strbuf *out, const
> char *path,
> }
> }
>
> -/* Since intptr_t is C99, we do not use it here */
> -static inline uint32_t *mark_to_ptr(uint32_t mark)
> -{
> - return ((uint32_t *)NULL) + mark;
> -}
> -
> -static inline uint32_t ptr_to_mark(void * mark)
> -{
> - return (uint32_t *)mark - (uint32_t *)NULL;
> -}
> -
> static inline void mark_object(struct object *object, uint32_t mark)
> {
> - add_decoration(&idnums, object, mark_to_ptr(mark));
> + unsigned int nr = idnums.nr + 1;
> +
> + if (nr > idnums.size * 2 / 3)
> + grow_object_marks(&idnums);
> + return set_object_mark(&idnums, object, mark);
> }
>
> static inline void mark_next_object(struct object *object)
> @@ -179,10 +227,21 @@ static inline void mark_next_object(struct object
> *object)
>
> static int get_object_mark(struct object *object)
> {
> - void *decoration = lookup_decoration(&idnums, object);
> - if (!decoration)
> + unsigned int j;
> +
> + /* nothing to lookup */
> + if (!idnums.size)
> return 0;
> - return ptr_to_mark(decoration);
> + j = hash_obj(object, idnums.size);
> + for (;;) {
> + struct object_mark_entry *ref = idnums.entries + j;
> + if (ref->base == object)
> + return ref->mark;
> + if (!ref->base)
> + return 0;
> + if (++j == idnums.size)
> + j = 0;
> + }
> }
>
> static void show_progress(void)
> @@ -897,8 +956,7 @@ static void handle_tags_and_duplicates(void)
> static void export_marks(char *file)
> {
> unsigned int i;
> - uint32_t mark;
> - struct decoration_entry *deco = idnums.entries;
> + struct object_mark_entry *entry = idnums.entries;
> FILE *f;
> int e = 0;
>
> @@ -907,15 +965,14 @@ static void export_marks(char *file)
> die_errno("Unable to open marks file %s for writing.", file);
>
> for (i = 0; i < idnums.size; i++) {
> - if (deco->base && deco->base->type == 1) {
> - mark = ptr_to_mark(deco->decoration);
> - if (fprintf(f, ":%"PRIu32" %s\n", mark,
> - oid_to_hex(&deco->base->oid)) < 0) {
> + if (entry->base && entry->base->type == 1) {
> + if (fprintf(f, ":%"PRIu32" %s\n", entry->mark,
> + oid_to_hex(&entry->base->oid)) < 0) {
> e = 1;
> break;
> }
> }
> - deco++;
> + entry++;
> }
>
> e |= ferror(f);
