tarray.h 3.1 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 24 25
/*
 * 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"

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

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

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

/**
 *
 * @param pArray
 * @param pData
 * @return
 */
void* taosArrayPush(SArray* pArray, void* pData);

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

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

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

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

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

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

/**
96 97 98 99 100 101 102 103 104 105 106
 * 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 已提交
107
void taosArrayCopy(SArray* pDst, const SArray* pSrc);
108

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

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

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

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

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

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

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

/**
 * search the array
 * @param pArray
 * @param key
 */
160
char* taosArraySearchString(const SArray* pArray, const char* key, __compar_fn_t comparFn);
weixin_48148422's avatar
weixin_48148422 已提交
161

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

#endif  // TDENGINE_TAOSARRAY_H