/* * Copyright (c) 2008 - 2009 * Jonathan Schleifer * * All rights reserved. * * This file is part of libobjfw. It may be distributed under the terms of the * Q Public License 1.0, which can be found in the file LICENSE included in * the packaging of this file. */ #include "config.h" #include #include #include #include #import "OFDataArray.h" #import "OFExceptions.h" #import "OFMacros.h" #ifdef _WIN32 #include #endif static int lastpagebyte = 0; @implementation OFDataArray + dataArrayWithItemSize: (size_t)is { return [[[self alloc] initWithItemSize: is] autorelease]; } - initWithItemSize: (size_t)is { Class c; self = [super init]; if (is == 0) { c = isa; [super dealloc]; @throw [OFInvalidArgumentException newWithClass: c andSelector: _cmd]; } data = NULL; itemsize = is; return self; } - (size_t)count { return count; } - (size_t)itemsize { return itemsize; } - (void*)data { return data; } - (void*)itemAtIndex: (size_t)index { if (index >= count) @throw [OFOutOfRangeException newWithClass: isa]; return data + index * itemsize; } - (void*)lastItem { return data + (count - 1) * itemsize; } - addItem: (void*)item { if (SIZE_MAX - count < 1) @throw [OFOutOfRangeException newWithClass: isa]; data = [self resizeMem: data toNItems: count + 1 withSize: itemsize]; memcpy(data + count++ * itemsize, item, itemsize); return self; } - addNItems: (size_t)nitems fromCArray: (void*)carray { if (nitems > SIZE_MAX - count) @throw [OFOutOfRangeException newWithClass: isa]; data = [self resizeMem: data toNItems: count + nitems withSize: itemsize]; memcpy(data + count * itemsize, carray, nitems * itemsize); count += nitems; return self; } - removeNItems: (size_t)nitems { if (nitems > count) @throw [OFOutOfRangeException newWithClass: isa]; data = [self resizeMem: data toNItems: count - nitems withSize: itemsize]; count -= nitems; return self; } - (id)copy { OFDataArray *new = [[OFDataArray alloc] initWithItemSize: itemsize]; [new addNItems: count fromCArray: data]; return new; } - (BOOL)isEqual: (id)obj { if (![obj isKindOf: [OFDataArray class]]) return NO; if ([obj count] != count || [obj itemsize] != itemsize) return NO; if (memcmp([obj data], data, count * itemsize)) return NO; return YES; } - (int)compare: (id)obj { int ret; if (![obj isKindOf: [OFDataArray class]]) @throw [OFInvalidArgumentException newWithClass: isa andSelector: _cmd]; if ([obj itemsize] != itemsize) @throw [OFInvalidArgumentException newWithClass: isa andSelector: _cmd]; if ([obj count] == count) return memcmp(data, [obj data], count * itemsize); if (count > [obj count]) { if ((ret = memcmp(data, [obj data], [obj count] * itemsize))) return ret; return *(char*)[self itemAtIndex: [obj count]]; } else { if ((ret = memcmp(data, [obj data], count * itemsize))) return ret; return *(char*)[obj itemAtIndex: count] * -1; } } - (uint32_t)hash { uint32_t hash; size_t i; OF_HASH_INIT(hash); for (i = 0; i < count * itemsize; i++) OF_HASH_ADD(hash, ((char*)data)[i]); OF_HASH_FINALIZE(hash); return hash; } @end @implementation OFBigDataArray - initWithItemSize: (size_t)is { self = [super initWithItemSize: is]; if (lastpagebyte == 0) { #ifndef _WIN32 if ((lastpagebyte = sysconf(_SC_PAGESIZE)) == -1) lastpagebyte = 4096; lastpagebyte--; #else SYSTEM_INFO si; GetSystemInfo(&si); lastpagebyte = si.dwPageSize - 1; #endif } return self; } - addItem: (void*)item { size_t nsize; if (SIZE_MAX - count < 1 || count + 1 > SIZE_MAX / itemsize) @throw [OFOutOfRangeException newWithClass: isa]; nsize = ((count + 1) * itemsize + lastpagebyte) & ~lastpagebyte; if (size != nsize) data = [self resizeMem: data toSize: nsize]; memcpy(data + count++ * itemsize, item, itemsize); size = nsize; return self; } - addNItems: (size_t)nitems fromCArray: (void*)carray { size_t nsize; if (nitems > SIZE_MAX - count || count + nitems > SIZE_MAX / itemsize) @throw [OFOutOfRangeException newWithClass: isa]; nsize = ((count + nitems) * itemsize + lastpagebyte) & ~lastpagebyte; if (size != nsize) data = [self resizeMem: data toSize: nsize]; memcpy(data + count * itemsize, carray, nitems * itemsize); count += nitems; size = nsize; return self; } - removeNItems: (size_t)nitems { size_t nsize; if (nitems > count) @throw [OFOutOfRangeException newWithClass: isa]; nsize = ((count - nitems) * itemsize + lastpagebyte) & ~lastpagebyte; if (size != nsize) data = [self resizeMem: data toSize: nsize]; count -= nitems; size = nsize; return self; } - (id)copy { OFDataArray *new = [[OFBigDataArray alloc] initWithItemSize: itemsize]; [new addNItems: count fromCArray: data]; return new; } @end