#ifndef ZSTD_BUFFER_H

#define ZSTD_BUFFER_H

#include "pycore_blocks_output_buffer.h"

#include <zstd.h> // ZSTD_outBuffer

static inline int

_OutputBuffer_InitAndGrow(_BlocksOutputBuffer *buffer, ZSTD_outBuffer *ob,

Py_ssize_t max_length)

{

/* Ensure .writer was set to NULL */

assert(buffer->writer == NULL);

Py_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;

}

static inline int

_OutputBuffer_InitWithSize(_BlocksOutputBuffer *buffer, ZSTD_outBuffer *ob,

Py_ssize_t max_length, Py_ssize_t init_size)

{

Py_ssize_t block_size;

/* Ensure .writer was set to NULL */

assert(buffer->writer == NULL);

/* Get block size */

if (0 <= max_length && max_length < init_size) {

block_size = max_length;

}

else {

block_size = init_size;

}

Py_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;

}

static inline int

_OutputBuffer_Grow(_BlocksOutputBuffer *buffer, ZSTD_outBuffer *ob)

{

assert(ob->pos == ob->size);

Py_ssize_t res = _BlocksOutputBuffer_Grow(buffer, &ob->dst, 0);

if (res < 0) {

return -1;

}

ob->size = (size_t) res;

ob->pos = 0;

return 0;

}

static inline PyObject *

_OutputBuffer_Finish(_BlocksOutputBuffer *buffer, ZSTD_outBuffer *ob)

{

return _BlocksOutputBuffer_Finish(buffer, ob->size - ob->pos);

}

static inline void

_OutputBuffer_OnError(_BlocksOutputBuffer *buffer)

{

_BlocksOutputBuffer_OnError(buffer);

}

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