Class: Zstd::StreamingCompress
- Inherits:
-
Object
- Object
- Zstd::StreamingCompress
- Defined in:
- ext/zstdruby/streaming_compress.c
Constant Summary collapse
- CONTINUE =
INT2FIX(ZSTD_e_continue)
- FLUSH =
INT2FIX(ZSTD_e_flush)
- END =
INT2FIX(ZSTD_e_end)
Instance Method Summary collapse
-
#<< ⇒ Object
Same as IO.
- #compress(src) ⇒ Object
- #finish ⇒ Object
- #flush ⇒ Object
- #initialize(*args) ⇒ Object constructor
-
#print ⇒ Object
Same as IO.
-
#printf ⇒ Object
Same as IO.
-
#puts ⇒ Object
Same as IO.
- #write(*args) ⇒ Object
Constructor Details
#initialize(*args) ⇒ Object
70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
# File 'ext/zstdruby/streaming_compress.c', line 70
static VALUE
rb_streaming_compress_initialize(int argc, VALUE *argv, VALUE obj)
{
VALUE kwargs;
VALUE compression_level_value;
rb_scan_args(argc, argv, "01:", &compression_level_value, &kwargs);
struct streaming_compress_t* sc;
TypedData_Get_Struct(obj, struct streaming_compress_t, &streaming_compress_type, sc);
size_t const buffOutSize = ZSTD_CStreamOutSize();
ZSTD_CCtx* ctx = ZSTD_createCCtx();
if (ctx == NULL) {
rb_raise(rb_eRuntimeError, "%s", "ZSTD_createCCtx error");
}
set_compress_params(ctx, compression_level_value, kwargs);
sc->ctx = ctx;
sc->buf = rb_str_new(NULL, buffOutSize);
sc->buf_size = buffOutSize;
return obj;
}
|
Instance Method Details
#<< ⇒ Object
Same as IO.
#compress(src) ⇒ Object
118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 |
# File 'ext/zstdruby/streaming_compress.c', line 118
static VALUE
rb_streaming_compress_compress(VALUE obj, VALUE src)
{
StringValue(src);
const char* input_data = RSTRING_PTR(src);
size_t input_size = RSTRING_LEN(src);
ZSTD_inBuffer input = { input_data, input_size, 0 };
struct streaming_compress_t* sc;
TypedData_Get_Struct(obj, struct streaming_compress_t, &streaming_compress_type, sc);
const char* output_data = RSTRING_PTR(sc->buf);
VALUE result = rb_str_new(0, 0);
while (input.pos < input.size) {
ZSTD_outBuffer output = { (void*)output_data, sc->buf_size, 0 };
size_t const ret = zstd_stream_compress(sc->ctx, &output, &input, ZSTD_e_continue, false);
if (ZSTD_isError(ret)) {
rb_raise(rb_eRuntimeError, "compress error error code: %s", ZSTD_getErrorName(ret));
}
rb_str_cat(result, output.dst, output.pos);
}
return result;
}
|
#finish ⇒ Object
200 201 202 203 204 205 206 207 |
# File 'ext/zstdruby/streaming_compress.c', line 200
static VALUE
rb_streaming_compress_finish(VALUE obj)
{
struct streaming_compress_t* sc;
TypedData_Get_Struct(obj, struct streaming_compress_t, &streaming_compress_type, sc);
VALUE result = no_compress(sc, ZSTD_e_end);
return result;
}
|
#flush ⇒ Object
191 192 193 194 195 196 197 198 |
# File 'ext/zstdruby/streaming_compress.c', line 191
static VALUE
rb_streaming_compress_flush(VALUE obj)
{
struct streaming_compress_t* sc;
TypedData_Get_Struct(obj, struct streaming_compress_t, &streaming_compress_type, sc);
VALUE result = no_compress(sc, ZSTD_e_flush);
return result;
}
|
#print ⇒ Object
Same as IO.
#printf ⇒ Object
Same as IO.
#puts ⇒ Object
Same as IO.
#write(*args) ⇒ Object
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 |
# File 'ext/zstdruby/streaming_compress.c', line 142
static VALUE
rb_streaming_compress_write(int argc, VALUE *argv, VALUE obj)
{
size_t total = 0;
VALUE result = rb_str_new(0, 0);
struct streaming_compress_t* sc;
TypedData_Get_Struct(obj, struct streaming_compress_t, &streaming_compress_type, sc);
const char* output_data = RSTRING_PTR(sc->buf);
while (argc-- > 0) {
VALUE str = *argv++;
StringValue(str);
const char* input_data = RSTRING_PTR(str);
size_t input_size = RSTRING_LEN(str);
ZSTD_inBuffer input = { input_data, input_size, 0 };
while (input.pos < input.size) {
ZSTD_outBuffer output = { (void*)output_data, sc->buf_size, 0 };
size_t const ret = zstd_stream_compress(sc->ctx, &output, &input, ZSTD_e_continue, false);
if (ZSTD_isError(ret)) {
rb_raise(rb_eRuntimeError, "compress error error code: %s", ZSTD_getErrorName(ret));
}
total += RSTRING_LEN(str);
}
}
return SIZET2NUM(total);
}
|