more
This commit is contained in:
Hongze Cheng
parent
4fc5fe028c
commit
a074622115
|
|
@ -0,0 +1,281 @@
|
||||||
|
/*
|
||||||
|
* 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 _TD_UTIL_ENCODE_H_
|
||||||
|
#define _TD_UTIL_ENCODE_H_
|
||||||
|
|
||||||
|
#include "os.h"
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
td_endian_t endian;
|
||||||
|
uint8_t* data;
|
||||||
|
int64_t size;
|
||||||
|
int64_t pos;
|
||||||
|
} SEncoder;
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
td_endian_t endian;
|
||||||
|
const uint8_t* data;
|
||||||
|
int64_t size;
|
||||||
|
int16_t pos;
|
||||||
|
} SDecoder;
|
||||||
|
|
||||||
|
#define tPut(TYPE, BUF, VAL) ((TYPE*)(BUF))[0] = (VAL)
|
||||||
|
#define tGet(TYPE, BUF, VAL) (VAL) = ((TYPE*)(BUF))[0]
|
||||||
|
|
||||||
|
#define tRPut16(PDEST, PSRC) \
|
||||||
|
((uint8_t*)(PDEST))[0] = ((uint8_t*)(PSRC))[1]; \
|
||||||
|
((uint8_t*)(PDEST))[1] = ((uint8_t*)(PSRC))[0];
|
||||||
|
|
||||||
|
#define tRPut32(PDEST, PSRC) \
|
||||||
|
((uint8_t*)(PDEST))[0] = ((uint8_t*)(PSRC))[3]; \
|
||||||
|
((uint8_t*)(PDEST))[1] = ((uint8_t*)(PSRC))[2]; \
|
||||||
|
((uint8_t*)(PDEST))[2] = ((uint8_t*)(PSRC))[1]; \
|
||||||
|
((uint8_t*)(PDEST))[3] = ((uint8_t*)(PSRC))[0];
|
||||||
|
|
||||||
|
#define tRPut64(PDEST, PSRC) \
|
||||||
|
((uint8_t*)(PDEST))[0] = ((uint8_t*)(PSRC))[7]; \
|
||||||
|
((uint8_t*)(PDEST))[1] = ((uint8_t*)(PSRC))[6]; \
|
||||||
|
((uint8_t*)(PDEST))[2] = ((uint8_t*)(PSRC))[5]; \
|
||||||
|
((uint8_t*)(PDEST))[3] = ((uint8_t*)(PSRC))[4]; \
|
||||||
|
((uint8_t*)(PDEST))[4] = ((uint8_t*)(PSRC))[3]; \
|
||||||
|
((uint8_t*)(PDEST))[5] = ((uint8_t*)(PSRC))[2]; \
|
||||||
|
((uint8_t*)(PDEST))[6] = ((uint8_t*)(PSRC))[1]; \
|
||||||
|
((uint8_t*)(PDEST))[7] = ((uint8_t*)(PSRC))[0];
|
||||||
|
|
||||||
|
#define tRGet16 tRPut16
|
||||||
|
#define tRGet32 tRPut32
|
||||||
|
#define tRGet64 tRPut64
|
||||||
|
|
||||||
|
#define TD_CODER_CURRENT(CODER) ((CODER)->data + (CODER)->pos)
|
||||||
|
#define TD_CODER_MOVE_POS(CODER, MOVE) ((CODER)->pos += (MOVE))
|
||||||
|
#define TD_CHECK_CODER_CAPACITY_FAILED(CODER, EXPSIZE) (((CODER)->size - (CODER)->pos) < (EXPSIZE))
|
||||||
|
|
||||||
|
/* ------------------------ FOR ENCODER ------------------------ */
|
||||||
|
static FORCE_INLINE void tInitEncoder(SEncoder* pEncoder, td_endian_t endian, uint8_t* data, int64_t size) {
|
||||||
|
pEncoder->endian = endian;
|
||||||
|
pEncoder->data = data;
|
||||||
|
pEncoder->size = (data) ? size : 0;
|
||||||
|
pEncoder->pos = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 8
|
||||||
|
static FORCE_INLINE int tEncodeU8(SEncoder* pEncoder, uint8_t val) {
|
||||||
|
if (pEncoder->data) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(val))) return -1;
|
||||||
|
tPut(uint8_t, TD_CODER_CURRENT(pEncoder), val);
|
||||||
|
}
|
||||||
|
TD_CODER_MOVE_POS(pEncoder, sizeof(val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static FORCE_INLINE int tEncodeI8(SEncoder* pEncoder, int8_t val) {
|
||||||
|
if (pEncoder->data) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(val))) return -1;
|
||||||
|
tPut(int8_t, TD_CODER_CURRENT(pEncoder), val);
|
||||||
|
}
|
||||||
|
TD_CODER_MOVE_POS(pEncoder, sizeof(val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 16
|
||||||
|
static FORCE_INLINE int tEncodeU16(SEncoder* pEncoder, uint16_t val) {
|
||||||
|
if (pEncoder->data) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(val))) return -1;
|
||||||
|
if (TD_RT_ENDIAN() == pEncoder->endian) {
|
||||||
|
tPut(uint16_t, TD_CODER_CURRENT(pEncoder), val);
|
||||||
|
} else {
|
||||||
|
tRPut16(TD_CODER_CURRENT(pEncoder), &val);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
TD_CODER_MOVE_POS(pEncoder, sizeof(val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static FORCE_INLINE void tEncodeI16(SEncoder* pEncoder, int16_t val) {
|
||||||
|
if (pEncoder->data) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(val))) return -1;
|
||||||
|
if (TD_RT_ENDIAN() == pEncoder->endian) {
|
||||||
|
tPut(int16_t, TD_CODER_CURRENT(pEncoder), val);
|
||||||
|
} else {
|
||||||
|
tRPut16(TD_CODER_CURRENT(pEncoder), &val);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
TD_CODER_MOVE_POS(pEncoder, sizeof(val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 32
|
||||||
|
static FORCE_INLINE void tEncodeU32(SEncoder* pEncoder, uint32_t val) {
|
||||||
|
if (pEncoder->data) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(val))) return -1;
|
||||||
|
if (TD_RT_ENDIAN() == pEncoder->endian) {
|
||||||
|
tPut(uint32_t, TD_CODER_CURRENT(pEncoder), val);
|
||||||
|
} else {
|
||||||
|
tRPut32(TD_CODER_CURRENT(pEncoder), &val);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
TD_CODER_MOVE_POS(pEncoder, sizeof(val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static FORCE_INLINE void tEncodeI32(SEncoder* pEncoder, int32_t val) {
|
||||||
|
if (pEncoder->data) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(val))) return -1;
|
||||||
|
if (TD_RT_ENDIAN() == pEncoder->endian) {
|
||||||
|
tPut(int32_t, TD_CODER_CURRENT(pEncoder), val);
|
||||||
|
} else {
|
||||||
|
tRPut32(TD_CODER_CURRENT(pEncoder), &val);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
TD_CODER_MOVE_POS(pEncoder, sizeof(val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 64
|
||||||
|
static FORCE_INLINE void tEncodeU64(SEncoder* pEncoder, uint64_t val) {
|
||||||
|
if (pEncoder->data) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(val))) return -1;
|
||||||
|
if (TD_RT_ENDIAN() == pEncoder->endian) {
|
||||||
|
tPut(uint64_t, TD_CODER_CURRENT(pEncoder), val);
|
||||||
|
} else {
|
||||||
|
tRPut64(TD_CODER_CURRENT(pEncoder), &val);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
TD_CODER_MOVE_POS(pEncoder, sizeof(val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static FORCE_INLINE void tEncodeI32(SEncoder* pEncoder, int32_t val) {
|
||||||
|
if (pEncoder->data) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(val))) return -1;
|
||||||
|
if (TD_RT_ENDIAN() == pEncoder->endian) {
|
||||||
|
tPut(int32_t, TD_CODER_CURRENT(pEncoder), val);
|
||||||
|
} else {
|
||||||
|
tRPut64(TD_CODER_CURRENT(pEncoder), &val);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
TD_CODER_MOVE_POS(pEncoder, sizeof(val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ------------------------ FOR DECODER ------------------------ */
|
||||||
|
static FORCE_INLINE void tInitDecoder(SDecoder* pDecoder, td_endian_t endian, const uint8_t* data, int64_t size) {
|
||||||
|
ASSERT(!TD_IS_NULL(data));
|
||||||
|
pDecoder->endian = endian;
|
||||||
|
pDecoder->data = data;
|
||||||
|
pDecoder->size = size;
|
||||||
|
pDecoder->pos = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 8
|
||||||
|
static FORCER_INLINE int tDecodeU8(SDecoder* pDecoder, uint8_t* val) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(*val))) return -1;
|
||||||
|
tGet(uint8_t, TD_CODER_CURRENT(pDecoder), *val);
|
||||||
|
TD_CODER_MOVE_POS(pDecoder, sizeof(*val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static FORCER_INLINE int tDecodeI8(SDecoder* pDecoder, int8_t* val) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(*val))) return -1;
|
||||||
|
tGet(int8_t, TD_CODER_CURRENT(pDecoder), *val);
|
||||||
|
TD_CODER_MOVE_POS(pDecoder, sizeof(*val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 16
|
||||||
|
static FORCER_INLINE int tDecodeU16(SDecoder* pDecoder, uint16_t* val) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(*val))) return -1;
|
||||||
|
if (TD_RT_ENDIAN() == pDecoder->endian) {
|
||||||
|
tGet(uint16_t, TD_CODER_CURRENT(pDecoder), *val);
|
||||||
|
} else {
|
||||||
|
tRGet16(val, TD_CODER_CURRENT(pDecoder));
|
||||||
|
}
|
||||||
|
|
||||||
|
TD_CODER_MOVE_POS(pDecoder, sizeof(*val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static FORCER_INLINE int tDecodeI16(SDecoder* pDecoder, int16_t* val) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(*val))) return -1;
|
||||||
|
if (TD_RT_ENDIAN() == pDecoder->endian) {
|
||||||
|
tGet(int16_t, TD_CODER_CURRENT(pDecoder), *val);
|
||||||
|
} else {
|
||||||
|
tRGet16(val, TD_CODER_CURRENT(pDecoder));
|
||||||
|
}
|
||||||
|
|
||||||
|
TD_CODER_MOVE_POS(pDecoder, sizeof(*val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 32
|
||||||
|
static FORCER_INLINE int tDecodeU32(SDecoder* pDecoder, uint32_t* val) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(*val))) return -1;
|
||||||
|
if (TD_RT_ENDIAN() == pDecoder->endian) {
|
||||||
|
tGet(uint32_t, TD_CODER_CURRENT(pDecoder), *val);
|
||||||
|
} else {
|
||||||
|
tRGet32(val, TD_CODER_CURRENT(pDecoder));
|
||||||
|
}
|
||||||
|
|
||||||
|
TD_CODER_MOVE_POS(pDecoder, sizeof(*val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static FORCER_INLINE int tDecodeI32(SDecoder* pDecoder, int32_t* val) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(*val))) return -1;
|
||||||
|
if (TD_RT_ENDIAN() == pDecoder->endian) {
|
||||||
|
tGet(int32_t, TD_CODER_CURRENT(pDecoder), *val);
|
||||||
|
} else {
|
||||||
|
tRGet32(val, TD_CODER_CURRENT(pDecoder));
|
||||||
|
}
|
||||||
|
|
||||||
|
TD_CODER_MOVE_POS(pDecoder, sizeof(*val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 64
|
||||||
|
static FORCER_INLINE int tDecodeU64(SDecoder* pDecoder, uint64_t* val) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(*val))) return -1;
|
||||||
|
if (TD_RT_ENDIAN() == pDecoder->endian) {
|
||||||
|
tGet(uint64_t, TD_CODER_CURRENT(pDecoder), *val);
|
||||||
|
} else {
|
||||||
|
tRGet64(val, TD_CODER_CURRENT(pDecoder));
|
||||||
|
}
|
||||||
|
|
||||||
|
TD_CODER_MOVE_POS(pDecoder, sizeof(*val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static FORCER_INLINE int tDecodeI64(SDecoder* pDecoder, int64_t* val) {
|
||||||
|
if (TD_CHECK_CODER_CAPACITY_FAILED(pEncoder, sizeof(*val))) return -1;
|
||||||
|
if (TD_RT_ENDIAN() == pDecoder->endian) {
|
||||||
|
tGet(int64_t, TD_CODER_CURRENT(pDecoder), *val);
|
||||||
|
} else {
|
||||||
|
tRGet64(val, TD_CODER_CURRENT(pDecoder));
|
||||||
|
}
|
||||||
|
|
||||||
|
TD_CODER_MOVE_POS(pDecoder, sizeof(*val));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif /*_TD_UTIL_ENCODE_H_*/
|
||||||
Loading…
Reference in New Issue