69 lines
1.7 KiB
C
69 lines
1.7 KiB
C
/*
|
|
* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
|
|
*
|
|
* This program is free software: you can use, redistribute, and/or modify
|
|
* it under the terms of the GNU Affero General Public License, version 3
|
|
* or later ("AGPL"), as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "indexFstSparse.h"
|
|
|
|
FstSparseSet *sparSetCreate(int32_t sz) {
|
|
FstSparseSet *ss = taosMemoryCalloc(1, sizeof(FstSparseSet));
|
|
if (ss == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
ss->dense = (uint32_t *)taosMemoryCalloc(sz, sizeof(uint32_t));
|
|
ss->sparse = (uint32_t *)taosMemoryCalloc(sz, sizeof(uint32_t));
|
|
ss->size = 0;
|
|
return ss;
|
|
}
|
|
void sparSetDestroy(FstSparseSet *ss) {
|
|
if (ss == NULL) {
|
|
return;
|
|
}
|
|
taosMemoryFree(ss->dense);
|
|
taosMemoryFree(ss->sparse);
|
|
taosMemoryFree(ss);
|
|
}
|
|
uint32_t sparSetLen(FstSparseSet *ss) {
|
|
// Get occupied size
|
|
return ss == NULL ? 0 : ss->size;
|
|
}
|
|
uint32_t sparSetAdd(FstSparseSet *ss, uint32_t ip) {
|
|
if (ss == NULL) {
|
|
return 0;
|
|
}
|
|
uint32_t i = ss->size;
|
|
ss->dense[i] = ip;
|
|
ss->sparse[ip] = i;
|
|
ss->size += 1;
|
|
return i;
|
|
}
|
|
uint32_t sparSetGet(FstSparseSet *ss, uint32_t i) {
|
|
// check later
|
|
return ss->dense[i];
|
|
}
|
|
bool sparSetContains(FstSparseSet *ss, uint32_t ip) {
|
|
uint32_t i = ss->sparse[ip];
|
|
if (i < ss->size && ss->dense[i] == ip) {
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
void sparSetClear(FstSparseSet *ss) {
|
|
if (ss == NULL) {
|
|
return;
|
|
}
|
|
ss->size = 0;
|
|
}
|