Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-22 12:30:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

optimize hashtable access

Browse Source
This commit is contained in:
方而静 2024-06-14 12:04:39 +08:00
parent ed2e95b3f4
commit a47b52f086
1 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/objects/dict.c
View File

@ -18,9 +18,10 @@ inline static int pkpy_Dict__idx_size(const pkpy_Dict* self) {
} }
inline static unsigned int pkpy_Dict__idx_null(const pkpy_Dict* self) { inline static unsigned int pkpy_Dict__idx_null(const pkpy_Dict* self) {
if(self->_htcap < 255) return 255; // if(self->_htcap < 255) return 255;
if(self->_htcap < 65535) return 65535; // if(self->_htcap < 65535) return 65535;
return 4294967295u; // 2^32 - 1 // return 4294967295u; // 2^32 - 1
return (1u << ((pkpy_Dict__idx_size(self) * 8) & 31)) - 1u;
} }
inline static int pkpy_Dict__ht_byte_size(const pkpy_Dict* self) { return self->_htcap * pkpy_Dict__idx_size(self); } inline static int pkpy_Dict__ht_byte_size(const pkpy_Dict* self) { return self->_htcap * pkpy_Dict__idx_size(self); }
@ -49,23 +50,20 @@ pkpy_Dict pkpy_Dict__copy(const pkpy_Dict* self) {
} }
static unsigned int pkpy_Dict__htget(const pkpy_Dict* self, int h) { static unsigned int pkpy_Dict__htget(const pkpy_Dict* self, int h) {
int sz = pkpy_Dict__idx_size(self); const int *p = (int*)(((char*)self->_hashtable) + h * pkpy_Dict__idx_size(self));
switch(sz) { return (*p) & pkpy_Dict__idx_null(self);
case 1: return ((uint8_t*)self->_hashtable)[h];
case 2: return ((uint16_t*)self->_hashtable)[h];
case 4: return ((uint32_t*)self->_hashtable)[h];
default: PK_UNREACHABLE();
}
} }
static void pkpy_Dict__htset(pkpy_Dict* self, int h, int v) { static void pkpy_Dict__htset(pkpy_Dict* self, int h, unsigned int v) {
int sz = pkpy_Dict__idx_size(self); int sz = pkpy_Dict__idx_size(self);
switch(sz) { // switch(sz) {
case 1: ((uint8_t*)self->_hashtable)[h] = v; break; // case 1: ((uint8_t*)self->_hashtable)[h] = v; break;
case 2: ((uint16_t*)self->_hashtable)[h] = v; break; // case 2: ((uint16_t*)self->_hashtable)[h] = v; break;
case 4: ((uint32_t*)self->_hashtable)[h] = v; break; // case 4: ((uint32_t*)self->_hashtable)[h] = v; break;
default: PK_UNREACHABLE(); // default: PK_UNREACHABLE();
} // }
int *p = ((char*)self->_hashtable) + h * pkpy_Dict__idx_size(self);
*p = v | (*p & ~pkpy_Dict__idx_null(self));
} }
static int pkpy_Dict__probe0(const pkpy_Dict* self, void* vm, pkpy_Var key, int64_t hash) { static int pkpy_Dict__probe0(const pkpy_Dict* self, void* vm, pkpy_Var key, int64_t hash) {