/*
* Copyright (c) 2008, 2009, 2010, 2011, 2012
* Jonathan Schleifer <js@webkeks.org>
*
* All rights reserved.
*
* This file is part of ObjFW. It may be distributed under the terms of the
* Q Public License 1.0, which can be found in the file LICENSE.QPL included in
* the packaging of this file.
*
* Alternatively, it may be distributed under the terms of the GNU General
* Public License, either version 2 or 3, which can be found in the file
* LICENSE.GPLv2 or LICENSE.GPLv3 respectively included in the packaging of this
* file.
*/
#include "config.h"
#include <stdio.h>
#include <stdlib.h>
#import "runtime.h"
#import "runtime-private.h"
static struct objc_sparsearray_level2 *empty_level2 = NULL;
static struct objc_sparsearray_level3 *empty_level3 = NULL;
static void
init(void)
{
size_t i;
empty_level2 = malloc(sizeof(struct objc_sparsearray_level2));
empty_level3 = malloc(sizeof(struct objc_sparsearray_level3));
if (empty_level2 == NULL || empty_level3 == NULL)
ERROR("Not enough memory to allocate sparse array!");
empty_level2->empty = YES;
empty_level3->empty = YES;
for (i = 0; i < 256; i++) {
empty_level2->buckets[i] = empty_level3;
empty_level3->buckets[i] = NULL;
}
}
struct objc_sparsearray*
objc_sparsearray_new(void)
{
struct objc_sparsearray *s;
size_t i;
if (empty_level2 == NULL || empty_level3 == NULL)
init();
if ((s = malloc(sizeof(struct objc_sparsearray))) == NULL)
ERROR("Not enough memory to allocate sparse array!");
for (i = 0; i < 256; i++)
s->buckets[i] = empty_level2;
return s;
}
struct objc_sparsearray*
objc_sparsearray_copy(struct objc_sparsearray *src)
{
struct objc_sparsearray *dst;
size_t i, j, k;
uint32_t idx;
dst = objc_sparsearray_new();
for (i = 0; i < 256; i++) {
if (src->buckets[i]->empty)
continue;
for (j = 0; j < 256; j++) {
if (src->buckets[i]->buckets[j]->empty)
continue;
for (k = 0; k < 256; k++) {
const void *obj;
obj = src->buckets[i]->buckets[j]->buckets[k];
if (obj == NULL)
continue;
idx = (i << 16) | (j << 8) | k;
objc_sparsearray_set(dst, idx, obj);
}
}
}
return dst;
}
void
objc_sparsearray_set(struct objc_sparsearray *s, uint32_t idx, const void *obj)
{
uint8_t i = idx >> 16;
uint8_t j = idx >> 8;
uint8_t k = idx;
if (s->buckets[i]->empty) {
struct objc_sparsearray_level2 *t;
size_t l;
t = malloc(sizeof(struct objc_sparsearray_level2));
if (t == NULL)
ERROR("Not enough memory to insert into sparse array!");
t->empty = NO;
for (l = 0; l < 256; l++)
t->buckets[l] = empty_level3;
s->buckets[i] = t;
}
if (s->buckets[i]->buckets[j]->empty) {
struct objc_sparsearray_level3 *t;
size_t l;
t = malloc(sizeof(struct objc_sparsearray_level3));
if (t == NULL)
ERROR("Not enough memory to insert into sparse array!");
t->empty = NO;
for (l = 0; l < 256; l++)
t->buckets[l] = NULL;
s->buckets[i]->buckets[j] = t;
}
s->buckets[i]->buckets[j]->buckets[k] = obj;
}
void
objc_sparsearray_free(struct objc_sparsearray *s)
{
size_t i, j;
for (i = 0; i < 256; i++) {
if (s->buckets[i]->empty)
continue;
for (j = 0; j < 256; j++)
if (!s->buckets[i]->buckets[j]->empty)
free(s->buckets[i]->buckets[j]);
free(s->buckets[i]);
}
free(s);
}
void
objc_sparsearray_cleanup(void)
{
if (empty_level2 != NULL)
free(empty_level2);
if (empty_level3 != NULL)
free(empty_level3);
empty_level2 = NULL;
empty_level3 = NULL;
}