mirror of
https://github.com/darlinghq/cctools-port.git
synced 2024-11-23 04:09:48 +00:00
267 lines
6.6 KiB
C
267 lines
6.6 KiB
C
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include <assert.h>
|
|
|
|
#include "sarray2.h"
|
|
#include "visibility.h"
|
|
|
|
static void *EmptyArrayData[256];
|
|
static SparseArray EmptyArray = { 0xff, 0, 0, (void**)&EmptyArrayData};
|
|
static void *EmptyArrayData8[256] = { [0 ... 255] = &EmptyArray };
|
|
static SparseArray EmptyArray8 = { 0xff00, 8, 0, (void**)&EmptyArrayData8};
|
|
static void *EmptyArrayData16[256] = { [0 ... 255] = &EmptyArray8 };
|
|
static SparseArray EmptyArray16 = { 0xff0000, 16, 0, (void**)&EmptyArrayData16};
|
|
static void *EmptyArrayData24[256] = { [0 ... 255] = &EmptyArray16 };
|
|
static SparseArray EmptyArray24 = { 0xff0000, 24, 0, (void**)&EmptyArrayData24};
|
|
|
|
#define MAX_INDEX(sarray) (sarray->mask >> sarray->shift)
|
|
#define DATA_SIZE(sarray) ((sarray->mask >> sarray->shift) + 1)
|
|
|
|
// Tweak this value to trade speed for memory usage. Bigger values use more
|
|
// memory, but give faster lookups.
|
|
#define base_shift 8
|
|
#define base_mask ((1<<base_shift) - 1)
|
|
|
|
void *EmptyChildForShift(uint32_t shift)
|
|
{
|
|
switch(shift)
|
|
{
|
|
default: UNREACHABLE("Broken sparse array");
|
|
case 8:
|
|
return &EmptyArray;
|
|
case 16:
|
|
return &EmptyArray8;
|
|
case 24:
|
|
return &EmptyArray16;
|
|
case 32:
|
|
return &EmptyArray24;
|
|
}
|
|
}
|
|
|
|
static void init_pointers(SparseArray * sarray)
|
|
{
|
|
sarray->data = calloc(DATA_SIZE(sarray), sizeof(void*));
|
|
if(sarray->shift != 0)
|
|
{
|
|
void *data = EmptyChildForShift(sarray->shift);
|
|
for(unsigned i=0 ; i<=MAX_INDEX(sarray) ; i++)
|
|
{
|
|
sarray->data[i] = data;
|
|
}
|
|
}
|
|
}
|
|
|
|
PRIVATE SparseArray * SparseArrayNewWithDepth(uint32_t depth)
|
|
{
|
|
SparseArray * sarray = calloc(1, sizeof(SparseArray));
|
|
sarray->refCount = 1;
|
|
sarray->shift = depth-base_shift;
|
|
sarray->mask = base_mask << sarray->shift;
|
|
init_pointers(sarray);
|
|
return sarray;
|
|
}
|
|
|
|
PRIVATE SparseArray *SparseArrayNew()
|
|
{
|
|
return SparseArrayNewWithDepth(32);
|
|
}
|
|
PRIVATE SparseArray *SparseArrayExpandingArray(SparseArray *sarray, uint32_t new_depth)
|
|
{
|
|
if (new_depth == sarray->shift)
|
|
{
|
|
return sarray;
|
|
}
|
|
assert(new_depth > sarray->shift);
|
|
// Expanding a child sarray has undefined results.
|
|
assert(sarray->refCount == 1);
|
|
SparseArray *new = calloc(1, sizeof(SparseArray));
|
|
new->refCount = 1;
|
|
new->shift = sarray->shift;
|
|
new->mask = sarray->mask;
|
|
void **newData = malloc(DATA_SIZE(sarray) * sizeof(void*));
|
|
void *data = EmptyChildForShift(new->shift + 8);
|
|
for(unsigned i=1 ; i<=MAX_INDEX(sarray) ; i++)
|
|
{
|
|
newData[i] = data;
|
|
}
|
|
new->data = sarray->data;
|
|
newData[0] = new;
|
|
sarray->data = newData;
|
|
// Now, any lookup in sarray for any value less than its capacity will have
|
|
// all non-zero values shifted away, resulting in 0. All lookups will
|
|
// therefore go to the new sarray.
|
|
sarray->shift += base_shift;
|
|
// Finally, set the mask to the correct value. Now all lookups should work.
|
|
sarray->mask <<= base_shift;
|
|
return sarray;
|
|
}
|
|
|
|
static void *SparseArrayFind(SparseArray * sarray, uint32_t * index)
|
|
{
|
|
uint32_t j = MASK_INDEX((*index));
|
|
uint32_t max = MAX_INDEX(sarray);
|
|
if (sarray->shift == 0)
|
|
{
|
|
while (j<=max)
|
|
{
|
|
if (sarray->data[j] != SARRAY_EMPTY)
|
|
{
|
|
return sarray->data[j];
|
|
}
|
|
(*index)++;
|
|
j++;
|
|
}
|
|
}
|
|
else while (j<max)
|
|
{
|
|
// If the shift is not 0, then we need to recursively look at child
|
|
// nodes.
|
|
uint32_t zeromask = ~(sarray->mask >> base_shift);
|
|
while (j<max)
|
|
{
|
|
//Look in child nodes
|
|
SparseArray *child = sarray->data[j];
|
|
// Skip over known-empty children
|
|
if ((&EmptyArray == child) ||
|
|
(&EmptyArray8 == child) ||
|
|
(&EmptyArray16 == child) ||
|
|
(&EmptyArray24 == child))
|
|
{
|
|
//Add 2^n to index so j is still correct
|
|
(*index) += 1<<sarray->shift;
|
|
//Zero off the next component of the index so we don't miss any.
|
|
*index &= zeromask;
|
|
}
|
|
else
|
|
{
|
|
// The recursive call will set index to the correct value for
|
|
// the next index, but won't update j
|
|
void * ret = SparseArrayFind(child, index);
|
|
if (ret != SARRAY_EMPTY)
|
|
{
|
|
return ret;
|
|
}
|
|
}
|
|
//Go to the next child
|
|
j++;
|
|
}
|
|
}
|
|
return SARRAY_EMPTY;
|
|
}
|
|
|
|
PRIVATE void *SparseArrayNext(SparseArray * sarray, uint32_t * idx)
|
|
{
|
|
(*idx)++;
|
|
return SparseArrayFind(sarray, idx);
|
|
}
|
|
|
|
PRIVATE void SparseArrayInsert(SparseArray * sarray, uint32_t index, void *value)
|
|
{
|
|
if (sarray->shift > 0)
|
|
{
|
|
uint32_t i = MASK_INDEX(index);
|
|
SparseArray *child = sarray->data[i];
|
|
if ((&EmptyArray == child) ||
|
|
(&EmptyArray8 == child) ||
|
|
(&EmptyArray16 == child) ||
|
|
(&EmptyArray24 == child))
|
|
{
|
|
// Insert missing nodes
|
|
SparseArray * newsarray = calloc(1, sizeof(SparseArray));
|
|
newsarray->refCount = 1;
|
|
if (base_shift >= sarray->shift)
|
|
{
|
|
newsarray->shift = 0;
|
|
}
|
|
else
|
|
{
|
|
newsarray->shift = sarray->shift - base_shift;
|
|
}
|
|
newsarray->mask = sarray->mask >> base_shift;
|
|
init_pointers(newsarray);
|
|
sarray->data[i] = newsarray;
|
|
child = newsarray;
|
|
}
|
|
else if (child->refCount > 1)
|
|
{
|
|
// Copy the copy-on-write part of the tree
|
|
sarray->data[i] = SparseArrayCopy(child);
|
|
SparseArrayDestroy(child);
|
|
child = sarray->data[i];
|
|
}
|
|
SparseArrayInsert(child, index, value);
|
|
}
|
|
else
|
|
{
|
|
sarray->data[index & sarray->mask] = value;
|
|
}
|
|
}
|
|
|
|
PRIVATE SparseArray *SparseArrayCopy(SparseArray * sarray)
|
|
{
|
|
SparseArray *copy = calloc(1, sizeof(SparseArray));
|
|
copy->refCount = 1;
|
|
copy->shift = sarray->shift;
|
|
copy->mask = sarray->mask;
|
|
copy->data = malloc(sizeof(void*) * DATA_SIZE(sarray));
|
|
memcpy(copy->data, sarray->data, sizeof(void*) * DATA_SIZE(sarray));
|
|
// If the sarray has children, increase their refcounts and link them
|
|
if (sarray->shift > 0)
|
|
{
|
|
for (unsigned int i = 0 ; i<=MAX_INDEX(sarray); i++)
|
|
{
|
|
SparseArray *child = copy->data[i];
|
|
__sync_fetch_and_add(&child->refCount, 1);
|
|
// Non-lazy copy. Uncomment if debugging
|
|
// copy->data[i] = SparseArrayCopy(copy->data[i]);
|
|
}
|
|
}
|
|
return copy;
|
|
}
|
|
|
|
PRIVATE void SparseArrayDestroy(SparseArray * sarray)
|
|
{
|
|
// Don't really delete this sarray if its ref count is > 0
|
|
if (sarray == &EmptyArray ||
|
|
sarray == &EmptyArray8 ||
|
|
sarray == &EmptyArray16 ||
|
|
(__sync_sub_and_fetch(&sarray->refCount, 1) > 0))
|
|
{
|
|
return;
|
|
}
|
|
|
|
if(sarray->shift > 0)
|
|
{
|
|
uint32_t max = (sarray->mask >> sarray->shift) + 1;
|
|
for(uint32_t i=0 ; i<max ; i++)
|
|
{
|
|
SparseArrayDestroy((SparseArray*)sarray->data[i]);
|
|
}
|
|
}
|
|
free(sarray->data);
|
|
free(sarray);
|
|
}
|
|
|
|
PRIVATE int SparseArraySize(SparseArray *sarray)
|
|
{
|
|
int size = 0;
|
|
if (sarray->shift == 0)
|
|
{
|
|
return 256*sizeof(void*) + sizeof(SparseArray);
|
|
}
|
|
size += 256*sizeof(void*) + sizeof(SparseArray);
|
|
for(unsigned i=0 ; i<=MAX_INDEX(sarray) ; i++)
|
|
{
|
|
SparseArray *child = sarray->data[i];
|
|
if (child == &EmptyArray ||
|
|
child == &EmptyArray8 ||
|
|
child == &EmptyArray16)
|
|
{
|
|
continue;
|
|
}
|
|
size += SparseArraySize(child);
|
|
}
|
|
return size;
|
|
}
|