#include "cache.h"

#include "object.h"

#include "blob.h"

#include "tree.h"

#include "commit.h"

#include "tag.h"

#include "alloc.h"

#define BLOCKING 1024

union any_object {

struct object object;

struct blob blob;

struct tree tree;

struct commit commit;

struct tag tag;

};

struct alloc_state {

};

struct alloc_state *allocate_alloc_state(void)

{

return xcalloc(1, sizeof(struct alloc_state));

}

void clear_alloc_state(struct alloc_state *s)

{

while (s->slab_nr > 0) {

s->slab_nr--;

free(s->slabs[s->slab_nr]);

}

FREE_AND_NULL(s->slabs);

}

static inline void *alloc_node(struct alloc_state *s, size_t node_size)

{

void *ret;

if (!s->nr) {

s->nr = BLOCKING;

s->p = xmalloc(BLOCKING * node_size);

ALLOC_GROW(s->slabs, s->slab_nr + 1, s->slab_alloc);

s->slabs[s->slab_nr++] = s->p;

}

s->nr--;

s->count++;

ret = s->p;

s->p = (char *)s->p + node_size;

memset(ret, 0, node_size);

return ret;

}

void *alloc_blob_node(struct repository *r)

{

struct blob *b = alloc_node(r->parsed_objects->blob_state, sizeof(struct blob));

b->object.type = OBJ_BLOB;

return b;

}

void *alloc_tree_node(struct repository *r)

{

struct tree *t = alloc_node(r->parsed_objects->tree_state, sizeof(struct tree));

t->object.type = OBJ_TREE;

return t;

}

void *alloc_tag_node(struct repository *r)

{

struct tag *t = alloc_node(r->parsed_objects->tag_state, sizeof(struct tag));

t->object.type = OBJ_TAG;

return t;

}

void *alloc_object_node(struct repository *r)

{

struct object *obj = alloc_node(r->parsed_objects->object_state, sizeof(union any_object));

obj->type = OBJ_NONE;

return obj;

}

static unsigned int alloc_commit_index(struct repository *r)

{

return r->parsed_objects->commit_count++;

}

void init_commit_node(struct repository *r, struct commit *c)

{

c->object.type = OBJ_COMMIT;

c->index = alloc_commit_index(r);

c->graph_pos = COMMIT_NOT_FROM_GRAPH;

c->generation = GENERATION_NUMBER_INFINITY;

}

void *alloc_commit_node(struct repository *r)

{

struct commit *c = alloc_node(r->parsed_objects->commit_state, sizeof(struct commit));

init_commit_node(r, c);

return c;

}

static void report(const char *name, unsigned int count, size_t size)

{

fprintf(stderr, "%10s: %8u (%"PRIuMAX" kB)\n",

name, count, (uintmax_t) size);

}

#define REPORT(name, type) \

report(#name, r->parsed_objects->name##_state->count, \

r->parsed_objects->name##_state->count * sizeof(type) >> 10)

void alloc_report(struct repository *r)

{

REPORT(blob, struct blob);

REPORT(tree, struct tree);

REPORT(commit, struct commit);

REPORT(tag, struct tag);

REPORT(object, union any_object);

}