mirror of
https://github.com/johndoe6345789/typthon.git
synced 2026-04-25 14:15:29 +00:00
b198f511d2
Massive automated renaming of all Py_/PyObject/etc. prefixes to Ty_/TyObject/etc. This includes: - All public API types (TyObject, TyTypeObject, etc.) - All public API functions (Ty_Initialize, Ty_BuildValue, etc.) - All internal API (_Ty_ prefixes) - Reference counting macros (Ty_INCREF, Ty_DECREF, etc.) - Type flags (Ty_TPFLAGS_*) - Debug flags (Ty_DEBUG, Ty_TRACE_REFS, etc.) - All object type APIs (TyList_, TyDict_, TyUnicode_, etc.) This changes over 60,000 occurrences across 1000+ files. Co-authored-by: johndoe6345789 <224850594+johndoe6345789@users.noreply.github.com>
109 lines
2.8 KiB
C
109 lines
2.8 KiB
C
/* Low level interface to the Zstandard algorthm & the zstd library. */
|
|
|
|
#ifndef ZSTD_BUFFER_H
|
|
#define ZSTD_BUFFER_H
|
|
|
|
#include "pycore_blocks_output_buffer.h"
|
|
|
|
#include <zstd.h> // ZSTD_outBuffer
|
|
|
|
/* Blocks output buffer wrapper code */
|
|
|
|
/* Initialize the buffer, and grow the buffer.
|
|
Return 0 on success
|
|
Return -1 on failure */
|
|
static inline int
|
|
_OutputBuffer_InitAndGrow(_BlocksOutputBuffer *buffer, ZSTD_outBuffer *ob,
|
|
Ty_ssize_t max_length)
|
|
{
|
|
/* Ensure .list was set to NULL */
|
|
assert(buffer->list == NULL);
|
|
|
|
Ty_ssize_t res = _BlocksOutputBuffer_InitAndGrow(buffer, max_length,
|
|
&ob->dst);
|
|
if (res < 0) {
|
|
return -1;
|
|
}
|
|
ob->size = (size_t) res;
|
|
ob->pos = 0;
|
|
return 0;
|
|
}
|
|
|
|
/* Initialize the buffer, with an initial size.
|
|
init_size: the initial size.
|
|
Return 0 on success
|
|
Return -1 on failure */
|
|
static inline int
|
|
_OutputBuffer_InitWithSize(_BlocksOutputBuffer *buffer, ZSTD_outBuffer *ob,
|
|
Ty_ssize_t max_length, Ty_ssize_t init_size)
|
|
{
|
|
Ty_ssize_t block_size;
|
|
|
|
/* Ensure .list was set to NULL */
|
|
assert(buffer->list == NULL);
|
|
|
|
/* Get block size */
|
|
if (0 <= max_length && max_length < init_size) {
|
|
block_size = max_length;
|
|
}
|
|
else {
|
|
block_size = init_size;
|
|
}
|
|
|
|
Ty_ssize_t res = _BlocksOutputBuffer_InitWithSize(buffer, block_size,
|
|
&ob->dst);
|
|
if (res < 0) {
|
|
return -1;
|
|
}
|
|
// Set max_length, InitWithSize doesn't do this
|
|
buffer->max_length = max_length;
|
|
ob->size = (size_t) res;
|
|
ob->pos = 0;
|
|
return 0;
|
|
}
|
|
|
|
/* Grow the buffer.
|
|
Return 0 on success
|
|
Return -1 on failure */
|
|
static inline int
|
|
_OutputBuffer_Grow(_BlocksOutputBuffer *buffer, ZSTD_outBuffer *ob)
|
|
{
|
|
assert(ob->pos == ob->size);
|
|
Ty_ssize_t res = _BlocksOutputBuffer_Grow(buffer, &ob->dst, 0);
|
|
if (res < 0) {
|
|
return -1;
|
|
}
|
|
ob->size = (size_t) res;
|
|
ob->pos = 0;
|
|
return 0;
|
|
}
|
|
|
|
/* Finish the buffer.
|
|
Return a bytes object on success
|
|
Return NULL on failure */
|
|
static inline TyObject *
|
|
_OutputBuffer_Finish(_BlocksOutputBuffer *buffer, ZSTD_outBuffer *ob)
|
|
{
|
|
return _BlocksOutputBuffer_Finish(buffer, ob->size - ob->pos);
|
|
}
|
|
|
|
/* Clean up the buffer */
|
|
static inline void
|
|
_OutputBuffer_OnError(_BlocksOutputBuffer *buffer)
|
|
{
|
|
_BlocksOutputBuffer_OnError(buffer);
|
|
}
|
|
|
|
/* Whether the output data has reached max_length.
|
|
The avail_out must be 0, please check it before calling. */
|
|
static inline int
|
|
_OutputBuffer_ReachedMaxLength(_BlocksOutputBuffer *buffer, ZSTD_outBuffer *ob)
|
|
{
|
|
/* Ensure (data size == allocated size) */
|
|
assert(ob->pos == ob->size);
|
|
|
|
return buffer->allocated == buffer->max_length;
|
|
}
|
|
|
|
#endif // !ZSTD_BUFFER_H
|