tarray.h 3.2 KB
Newer Older
H
more  
hzcheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
/*
 * 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/>.
 */

#ifndef TDENGINE_TAOSARRAY_H
#define TDENGINE_TAOSARRAY_H

#ifdef __cplusplus
extern "C" {
#endif

#include "os.h"
H
refact  
Hongze Cheng 已提交
24
#include "talgo.h"
H
more  
hzcheng 已提交
25 26

#define TARRAY_MIN_SIZE 8
H
Haojun Liao 已提交
27
#define TARRAY_GET_ELEM(array, index) ((void*)((char*)((array)->pData) + (index) * (array)->elemSize))
H
more  
hzcheng 已提交
28 29 30 31 32

typedef struct SArray {
  size_t size;
  size_t capacity;
  size_t elemSize;
H
Haojun Liao 已提交
33
  void*  pData;
H
more  
hzcheng 已提交
34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
} SArray;

/**
 *
 * @param size
 * @param elemSize
 * @return
 */
void* taosArrayInit(size_t size, size_t elemSize);

/**
 *
 * @param pArray
 * @param pData
 * @return
 */
H
Hongze Cheng 已提交
50
void* taosArrayPush(SArray* pArray, const void* pData);
H
more  
hzcheng 已提交
51 52 53 54 55

/**
 *
 * @param pArray
 */
weixin_48148422's avatar
weixin_48148422 已提交
56
void* taosArrayPop(SArray* pArray);
H
more  
hzcheng 已提交
57 58

/**
H
hjxilinx 已提交
59
 * get the data from array
H
more  
hzcheng 已提交
60 61 62 63
 * @param pArray
 * @param index
 * @return
 */
64
void* taosArrayGet(const SArray* pArray, size_t index);
H
more  
hzcheng 已提交
65 66

/**
H
hjxilinx 已提交
67 68 69 70 71
 * get the pointer data from the array
 * @param pArray
 * @param index
 * @return
 */
72
void* taosArrayGetP(const SArray* pArray, size_t index);
H
hjxilinx 已提交
73

H
Haojun Liao 已提交
74 75 76 77 78 79 80
/**
 * get the last element in the array list
 * @param pArray
 * @return
 */
void* taosArrayGetLast(const SArray* pArray);

H
hjxilinx 已提交
81 82
/**
 * return the size of array
H
more  
hzcheng 已提交
83 84 85
 * @param pArray
 * @return
 */
H
hjxilinx 已提交
86
size_t taosArrayGetSize(const SArray* pArray);
H
more  
hzcheng 已提交
87 88

/**
H
hjxilinx 已提交
89
 * insert data into array
H
more  
hzcheng 已提交
90 91 92 93
 * @param pArray
 * @param index
 * @param pData
 */
H
hjxilinx 已提交
94
void* taosArrayInsert(SArray* pArray, size_t index, void* pData);
H
more  
hzcheng 已提交
95 96

/**
97 98 99 100 101 102 103 104 105 106 107
 * remove data entry of the given index
 * @param pArray
 * @param index
 */
void taosArrayRemove(SArray* pArray, size_t index);

/**
 * copy the whole array from source to destination
 * @param pDst
 * @param pSrc
 */
H
hjxilinx 已提交
108
void taosArrayCopy(SArray* pDst, const SArray* pSrc);
109

H
hjxilinx 已提交
110 111 112 113
/**
 * clone a new array
 * @param pSrc
 */
H
hjxilinx 已提交
114
SArray* taosArrayClone(const SArray* pSrc);
H
hjxilinx 已提交
115

weixin_48148422's avatar
weixin_48148422 已提交
116 117 118 119 120 121
/**
 * clear the array (remove all element)
 * @param pArray
 */
void taosArrayClear(SArray* pArray);

122 123
/**
 * destroy array list
H
more  
hzcheng 已提交
124 125
 * @param pArray
 */
H
hjxilinx 已提交
126
void taosArrayDestroy(SArray* pArray);
H
more  
hzcheng 已提交
127

H
Haojun Liao 已提交
128 129 130 131 132 133 134
/**
 *
 * @param pArray
 * @param fp
 */
void taosArrayDestroyEx(SArray* pArray, void (*fp)(void*));

weixin_48148422's avatar
weixin_48148422 已提交
135 136 137 138 139 140 141
/**
 * sort the array
 * @param pArray
 * @param compar
 */
void taosArraySort(SArray* pArray, int (*compar)(const void*, const void*));

weixin_48148422's avatar
weixin_48148422 已提交
142 143 144 145
/**
 * sort string array
 * @param pArray
 */
146
void taosArraySortString(SArray* pArray, __compar_fn_t comparFn);
weixin_48148422's avatar
weixin_48148422 已提交
147

weixin_48148422's avatar
weixin_48148422 已提交
148 149 150 151 152 153
/**
 * search the array
 * @param pArray
 * @param compar
 * @param key
 */
H
refact  
Hongze Cheng 已提交
154
void* taosArraySearch(const SArray* pArray, const void* key, __compar_fn_t comparFn, int flags);
weixin_48148422's avatar
weixin_48148422 已提交
155 156 157 158 159 160

/**
 * search the array
 * @param pArray
 * @param key
 */
H
refact  
Hongze Cheng 已提交
161
char* taosArraySearchString(const SArray* pArray, const char* key, __compar_fn_t comparFn, int flags);
weixin_48148422's avatar
weixin_48148422 已提交
162

H
more  
hzcheng 已提交
163 164 165 166 167
#ifdef __cplusplus
}
#endif

#endif  // TDENGINE_TAOSARRAY_H