Class: Zstd::StreamingDecompress

Inherits:
Object
  • Object
show all
Defined in:
ext/zstdruby/streaming_decompress.c

Instance Method Summary collapse

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
 
# File 'ext/zstdruby/streaming_decompress.c', line 70

static VALUE
rb_streaming_decompress_initialize(int argc, VALUE *argv, VALUE obj)
{
  VALUE kwargs;
  rb_scan_args(argc, argv, "00:", &kwargs);

  struct streaming_decompress_t* sd;
  TypedData_Get_Struct(obj, struct streaming_decompress_t, &streaming_decompress_type, sd);
  size_t const buffOutSize = ZSTD_DStreamOutSize();

  ZSTD_DCtx* dctx = ZSTD_createDCtx();
  if (dctx == NULL) {
    rb_raise(rb_eRuntimeError, "%s", "ZSTD_createDCtx error");
  }
  set_decompress_params(dctx, kwargs);

  sd->dctx = dctx;
  sd->buf = rb_str_new(NULL, buffOutSize);
  sd->buf_size = buffOutSize;

  return obj;
}

Instance Method Details

#decompress(src) ⇒ Object 



93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
 
# File 'ext/zstdruby/streaming_decompress.c', line 93

static VALUE
rb_streaming_decompress_decompress(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_decompress_t* sd;
  TypedData_Get_Struct(obj, struct streaming_decompress_t, &streaming_decompress_type, sd);
  VALUE result = rb_str_new(0, 0);

  while (input.pos < input.size) {
    const char* output_data = RSTRING_PTR(sd->buf);
    ZSTD_outBuffer output = { (void*)output_data, sd->buf_size, 0 };
    size_t const ret = zstd_stream_decompress(sd->dctx, &output, &input, false);

    if (ZSTD_isError(ret)) {
      rb_raise(rb_eRuntimeError, "decompress error error code: %s", ZSTD_getErrorName(ret));
    }
    if (output.pos > 0) {
        rb_str_cat(result, output.dst, output.pos);
    }
    if (ret == 0 && output.pos == 0) {
        break;
    }
  }
  return result;
}

#decompress_with_pos(src) ⇒ Object 



123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
 
# File 'ext/zstdruby/streaming_decompress.c', line 123

static VALUE
rb_streaming_decompress_decompress_with_pos(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_decompress_t* sd;
  TypedData_Get_Struct(obj, struct streaming_decompress_t, &streaming_decompress_type, sd);
  const char* output_data = RSTRING_PTR(sd->buf);
  VALUE result = rb_str_new(0, 0);
  ZSTD_outBuffer output = { (void*)output_data, sd->buf_size, 0 };
  size_t const ret = zstd_stream_decompress(sd->dctx, &output, &input, false);
  if (ZSTD_isError(ret)) {
    rb_raise(rb_eRuntimeError, "decompress error error code: %s", ZSTD_getErrorName(ret));
  }
  rb_str_cat(result, output.dst, output.pos);
  return rb_ary_new_from_args(2, result, ULONG2NUM(input.pos));
}