cpython/Modules/_zstd/clinic/decompressor.c.h at main · python/cpython · GitHub

Repository navigation

History

213 lines (190 loc) · 7.01 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

/*[clinic input]

preserve

[clinic start generated code]*/

#if defined(Py_BUILD_CORE) && !defined(Py_BUILD_CORE_MODULE)

# include "pycore_gc.h" // PyGC_Head

# include "pycore_runtime.h" // _Py_ID()

#endif

#include "pycore_abstract.h" // _PyNumber_Index()

#include "pycore_modsupport.h" // _PyArg_UnpackKeywords()

PyDoc_STRVAR(_zstd_ZstdDecompressor_new__doc__,

"ZstdDecompressor(zstd_dict=None, options=None)\n"

"--\n"

"\n"

"Create a decompressor object for decompressing data incrementally.\n"

"\n"

" zstd_dict\n"

" A ZstdDict object, a pre-trained Zstandard dictionary.\n"

" options\n"

" A dict object that contains advanced decompression parameters.\n"

"\n"

"Thread-safe at method level. For one-shot decompression, use the\n"

"decompress() function instead.");

static PyObject *

_zstd_ZstdDecompressor_new_impl(PyTypeObject *type, PyObject *zstd_dict,

PyObject *options);

static PyObject *

_zstd_ZstdDecompressor_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)

{

PyObject *return_value = NULL;

#if defined(Py_BUILD_CORE) && !defined(Py_BUILD_CORE_MODULE)

#define NUM_KEYWORDS 2

static struct {

PyGC_Head _this_is_not_used;

PyObject_VAR_HEAD

Py_hash_t ob_hash;

PyObject *ob_item[NUM_KEYWORDS];

} _kwtuple = {

.ob_base = PyVarObject_HEAD_INIT(&PyTuple_Type, NUM_KEYWORDS)

.ob_hash = -1,

.ob_item = { &_Py_ID(zstd_dict), &_Py_ID(options), },

};

#undef NUM_KEYWORDS

#define KWTUPLE (&_kwtuple.ob_base.ob_base)

#else // !Py_BUILD_CORE

# define KWTUPLE NULL

#endif // !Py_BUILD_CORE

static const char * const _keywords[] = {"zstd_dict", "options", NULL};

static _PyArg_Parser _parser = {

.keywords = _keywords,

.fname = "ZstdDecompressor",

.kwtuple = KWTUPLE,

};

#undef KWTUPLE

PyObject *argsbuf[2];

PyObject * const *fastargs;

Py_ssize_t nargs = PyTuple_GET_SIZE(args);

Py_ssize_t noptargs = nargs + (kwargs ? PyDict_GET_SIZE(kwargs) : 0) - 0;

PyObject *zstd_dict = Py_None;

PyObject *options = Py_None;

fastargs = _PyArg_UnpackKeywords(_PyTuple_CAST(args)->ob_item, nargs, kwargs, NULL, &_parser,

/*minpos*/ 0, /*maxpos*/ 2, /*minkw*/ 0, /*varpos*/ 0, argsbuf);

if (!fastargs) {

goto exit;

}

if (!noptargs) {

goto skip_optional_pos;

}

if (fastargs[0]) {

zstd_dict = fastargs[0];

if (!--noptargs) {

goto skip_optional_pos;

}

options = fastargs[1];

skip_optional_pos:

return_value = _zstd_ZstdDecompressor_new_impl(type, zstd_dict, options);

exit:

return return_value;

}

PyDoc_STRVAR(_zstd_ZstdDecompressor_unused_data__doc__,

"A bytes object of un-consumed input data.\n"

"\n"

"When ZstdDecompressor object stops after a frame is\n"

"decompressed, unused input data after the frame. Otherwise this\n"

"will be b\'\'.");

static PyObject *

_zstd_ZstdDecompressor_unused_data_get_impl(ZstdDecompressor *self);

static PyObject *

_zstd_ZstdDecompressor_unused_data_get(PyObject *self, void *Py_UNUSED(context))

{

return _zstd_ZstdDecompressor_unused_data_get_impl((ZstdDecompressor *)self);

}

PyDoc_STRVAR(_zstd_ZstdDecompressor_decompress__doc__,

"decompress($self, /, data, max_length=-1)\n"

"--\n"

"\n"

"Decompress *data*, returning uncompressed bytes if possible, or b\'\' otherwise.\n"

"\n"

" data\n"

" A bytes-like object, Zstandard data to be decompressed.\n"

" max_length\n"

" Maximum size of returned data. When it is negative, the size of\n"

" output buffer is unlimited. When it is nonnegative, returns at\n"

" most max_length bytes of decompressed data.\n"

"\n"

"If *max_length* is nonnegative, returns at most *max_length* bytes\n"

"of decompressed data. If this limit is reached and further output\n"

"can be produced, *self.needs_input* will be set to ``False``. In\n"

"this case, the next call to *decompress()* may provide *data* as b\'\'\n"

"to obtain more of the output.\n"

"\n"

"If all of the input data was decompressed and returned (either\n"

"because this was less than *max_length* bytes, or because\n"

"*max_length* was negative), *self.needs_input* will be set to True.\n"

"\n"

"Attempting to decompress data after the end of a frame is reached\n"

"raises an EOFError. Any data found after the end of the frame is\n"

"ignored and saved in the self.unused_data attribute.");

#define _ZSTD_ZSTDDECOMPRESSOR_DECOMPRESS_METHODDEF \

{"decompress", _PyCFunction_CAST(_zstd_ZstdDecompressor_decompress), METH_FASTCALL|METH_KEYWORDS, _zstd_ZstdDecompressor_decompress__doc__},

static PyObject *

_zstd_ZstdDecompressor_decompress_impl(ZstdDecompressor *self,

Py_buffer *data,

Py_ssize_t max_length);

static PyObject *

_zstd_ZstdDecompressor_decompress(PyObject *self, PyObject *const *args, Py_ssize_t nargs, PyObject *kwnames)

{

PyObject *return_value = NULL;

#if defined(Py_BUILD_CORE) && !defined(Py_BUILD_CORE_MODULE)

#define NUM_KEYWORDS 2

static struct {

PyGC_Head _this_is_not_used;

PyObject_VAR_HEAD

Py_hash_t ob_hash;

PyObject *ob_item[NUM_KEYWORDS];

} _kwtuple = {

.ob_base = PyVarObject_HEAD_INIT(&PyTuple_Type, NUM_KEYWORDS)

.ob_hash = -1,

.ob_item = { &_Py_ID(data), &_Py_ID(max_length), },

};

#undef NUM_KEYWORDS

#define KWTUPLE (&_kwtuple.ob_base.ob_base)

#else // !Py_BUILD_CORE

# define KWTUPLE NULL

#endif // !Py_BUILD_CORE

static const char * const _keywords[] = {"data", "max_length", NULL};

static _PyArg_Parser _parser = {

.keywords = _keywords,

.fname = "decompress",

.kwtuple = KWTUPLE,

};

#undef KWTUPLE

PyObject *argsbuf[2];

Py_ssize_t noptargs = nargs + (kwnames ? PyTuple_GET_SIZE(kwnames) : 0) - 1;

Py_buffer data = {NULL, NULL};

Py_ssize_t max_length = -1;

args = _PyArg_UnpackKeywords(args, nargs, NULL, kwnames, &_parser,

/*minpos*/ 1, /*maxpos*/ 2, /*minkw*/ 0, /*varpos*/ 0, argsbuf);

if (!args) {

goto exit;

}

if (PyObject_GetBuffer(args[0], &data, PyBUF_SIMPLE) != 0) {

goto exit;

}

if (!noptargs) {

goto skip_optional_pos;

}

{

Py_ssize_t ival = -1;

PyObject *iobj = _PyNumber_Index(args[1]);

if (iobj != NULL) {

ival = PyLong_AsSsize_t(iobj);

Py_DECREF(iobj);

}

if (ival == -1 && PyErr_Occurred()) {

goto exit;

}

max_length = ival;

}

skip_optional_pos:

return_value = _zstd_ZstdDecompressor_decompress_impl((ZstdDecompressor *)self, &data, max_length);

exit:

/* Cleanup for data */

if (data.obj) {

PyBuffer_Release(&data);

}

return return_value;

}

#define _ZSTD_ZSTDDECOMPRESSOR_UNUSED_DATA_GETSETDEF {"unused_data", (getter)_zstd_ZstdDecompressor_unused_data_get, (setter)NULL, _zstd_ZstdDecompressor_unused_data__doc__},

/*[clinic end generated code: output=91b84afb07b1188c input=a9049054013a1b77]*/

cpython/Modules/_zstd/clinic/decompressor.c.h at main · python/cpython · GitHub

Repository navigation

Expand file tree

Breadcrumbs

decompressor.c.h

Latest commit

History

Breadcrumbs

decompressor.c.h

File metadata and controls