X-Git-Url: http://git.code-monkey.de/?a=blobdiff_plain;f=ext%2Fext.c;h=02ddba9454dd2789c3e08b86cb7bd9f5747a492a;hb=2c5f0577496dbb429a327fc85e72e7f4dcf4250c;hp=485bfc6062b9da6106912f5a98b612c5883b0e5e;hpb=802b343606893c51ec568679c13b5e2bcb3c5d31;p=ruby-eet.git diff --git a/ext/ext.c b/ext/ext.c index 485bfc6..02ddba9 100644 --- a/ext/ext.c +++ b/ext/ext.c @@ -1,5 +1,5 @@ /* - * $Id: ext.c 29 2005-04-12 17:07:48Z tilman $ + * $Id: ext.c 35 2005-05-10 18:58:49Z tilman $ * * Copyright (c) 2005 Tilman Sauerbeck (tilman at code-monkey de) * @@ -27,18 +27,26 @@ #include #define CHECK_KEY(key) \ - StringValue((key)); \ -\ - if (rb_funcall (key, rb_intern ("include?"), \ - 1, INT2FIX (0)) == Qtrue) \ + if (rb_funcall (key, id_include, 1, INT2FIX (0)) == Qtrue) \ rb_raise (rb_eArgError, "key must not contain binary zeroes"); #define CHECK_CLOSED(ef) \ if (!*(ef)) \ rb_raise (rb_eIOError, "closed stream"); +#ifdef WORDS_BIGENDIAN +# define BSWAP32(x) \ + ((((x) & 0xff000000) >> 24) | (((x) & 0x00ff0000) >> 8) | \ + (((x) & 0x0000ff00) << 8) | (((x) & 0x000000ff) << 24)) +#else /* !WORDS_BIGENDIAN */ +# define BSWAP16(x) (x) +# define BSWAP32(x) (x) +#endif /* WORDS_BIGENDIAN */ + static VALUE c_close (VALUE self); +static VALUE id_include; + static void c_free (Eet_File **ef) { @@ -99,17 +107,15 @@ c_init (int argc, VALUE *argv, VALUE self) VALUE file = Qnil, mode = Qnil; Eet_File **ef = NULL; Eet_File_Mode m = EET_FILE_MODE_READ; - const char *tmp; + const char *tmp, *cfile; Data_Get_Struct (self, Eet_File *, ef); rb_scan_args (argc, argv, "11", &file, &mode); - StringValue (file); + cfile = StringValuePtr (file); if (!NIL_P (mode)) { - StringValue (mode); - tmp = StringValuePtr (mode); if (!strcmp (tmp, "r+")) m = EET_FILE_MODE_READ_WRITE; @@ -121,7 +127,7 @@ c_init (int argc, VALUE *argv, VALUE self) eet_init (); - *ef = eet_open (StringValuePtr (file), m); + *ef = eet_open (cfile, m); if (!*ef) { switch (m) { case EET_FILE_MODE_READ_WRITE: @@ -133,7 +139,7 @@ c_init (int argc, VALUE *argv, VALUE self) break; } - rb_raise (rb_eRuntimeError, tmp, file); + rb_raise (rb_eRuntimeError, tmp, cfile); } return self; @@ -221,17 +227,16 @@ static VALUE c_delete (VALUE self, VALUE key) { Eet_File **ef = NULL; - char *tmp; + char *ckey; Data_Get_Struct (self, Eet_File *, ef); CHECK_CLOSED (ef); + ckey = StringValuePtr (key); CHECK_KEY (key); - tmp = StringValuePtr (key); - - if (!eet_delete (*ef, tmp)) - rb_raise (rb_eIOError, "cannot delete entry - %s", tmp); + if (!eet_delete (*ef, ckey)) + rb_raise (rb_eIOError, "cannot delete entry - %s", ckey); return self; } @@ -250,16 +255,18 @@ c_read (VALUE self, VALUE key) VALUE ret; Eet_File **ef = NULL; void *data; + char *ckey; int size = 0; Data_Get_Struct (self, Eet_File *, ef); CHECK_CLOSED (ef); + ckey = StringValuePtr (key); CHECK_KEY (key); - data = eet_read (*ef, StringValuePtr (key), &size); + data = eet_read (*ef, ckey, &size); if (!data) - rb_raise (rb_eIOError, "cannot read entry - %s", key); + rb_raise (rb_eIOError, "cannot read entry - %s", ckey); ret = rb_str_new (data, size); @@ -283,6 +290,7 @@ c_write (int argc, VALUE *argv, VALUE self) { VALUE key = Qnil, buf = Qnil, comp = Qnil; Eet_File **ef = NULL; + char *ckey, *cbuf; int n; Data_Get_Struct (self, Eet_File *, ef); @@ -293,11 +301,12 @@ c_write (int argc, VALUE *argv, VALUE self) if (NIL_P (comp)) comp = Qtrue; + ckey = StringValuePtr (key); CHECK_KEY (key); - StringValue (buf); + cbuf = StringValuePtr (buf); - n = eet_write (*ef, StringValuePtr (key), - StringValuePtr (buf), RSTRING (buf)->len, + n = eet_write (*ef, ckey, + cbuf, RSTRING (buf)->len, comp == Qtrue); if (!n) rb_raise (rb_eIOError, "couldn't write to file"); @@ -323,18 +332,20 @@ c_read_image (VALUE self, VALUE key) VALUE ret, comp; Eet_File **ef = NULL; void *data; + char *ckey; int w = 0, h = 0, has_alpha = 0, level = 0, quality = 0, lossy = 0; Data_Get_Struct (self, Eet_File *, ef); CHECK_CLOSED (ef); + ckey = StringValuePtr (key); CHECK_KEY (key); - data = eet_data_image_read (*ef, StringValuePtr (key), &w, &h, + data = eet_data_image_read (*ef, ckey, &w, &h, &has_alpha, &level, &quality, &lossy); if (!data) - rb_raise (rb_eIOError, "cannot read entry - %s", key); + rb_raise (rb_eIOError, "cannot read entry - %s", ckey); comp = rb_hash_new (); rb_hash_aset (comp, ID2SYM (rb_intern ("lossy")), INT2FIX (lossy)); @@ -371,6 +382,7 @@ c_write_image (int argc, VALUE *argv, VALUE self) VALUE key = Qnil, buf = Qnil, w = Qnil, h = Qnil, has_alpha = Qnil; VALUE comp = Qnil, tmp; Eet_File **ef = NULL; + char *ckey, *cbuf; int n, lossy = 0, level = 9, quality = 100; Data_Get_Struct (self, Eet_File *, ef); @@ -382,8 +394,9 @@ c_write_image (int argc, VALUE *argv, VALUE self) if (NIL_P (has_alpha)) has_alpha = Qfalse; + ckey = StringValuePtr (key); CHECK_KEY (key); - StringValue (buf); + cbuf = StringValuePtr (buf); Check_Type (w, T_FIXNUM); Check_Type (h, T_FIXNUM); @@ -406,8 +419,7 @@ c_write_image (int argc, VALUE *argv, VALUE self) if (!RSTRING (buf)->len) return INT2FIX (0); - n = eet_data_image_write (*ef, StringValuePtr (key), - StringValuePtr (buf), + n = eet_data_image_write (*ef, ckey, cbuf, FIX2INT (w), FIX2INT (h), has_alpha == Qtrue, level, quality, lossy); @@ -417,6 +429,79 @@ c_write_image (int argc, VALUE *argv, VALUE self) return INT2FIX (n); } +static VALUE +stream_serialize (VALUE self) +{ + VALUE ret; + struct RArray *stream; + static ID id_serialize; + long i; + + ret = rb_str_new2 (""); + + stream = RARRAY (self); + if (!stream->len) + return ret; + + if (!id_serialize) + id_serialize = rb_intern ("serialize"); + + for (i = 0; i < stream->len; i++) { + VALUE str = rb_funcall (stream->ptr[i], id_serialize, 0, NULL); + + rb_str_append (ret, str); + } + + return ret; +} + +static VALUE +chunk_serialize (VALUE self) +{ + VALUE tmp, ret; + unsigned int size, buf_len; + unsigned char *buf; + struct RString *tag, *data; + static ID id_tag, id_data; + + if (!id_tag) + id_tag = rb_intern ("@tag"); + + if (!id_data) + id_data = rb_intern ("@data"); + + tmp = rb_ivar_get (self, id_tag); + StringValue (tmp); + tag = RSTRING (tmp); + + tmp = rb_ivar_get (self, id_data); + StringValue (tmp); + data = RSTRING (tmp); + + buf_len = 9 + tag->len + data->len; + ret = rb_str_buf_new (buf_len); + + buf = RSTRING (ret)->ptr; + RSTRING (ret)->len = buf_len; + + memcpy (buf, "CHnK", 4); + buf += 4; + + size = tag->len + data->len + 1; + size = BSWAP32 (size); + memcpy (buf, &size, 4); + buf += 4; + + memcpy (buf, tag->ptr, tag->len); + buf += tag->len; + + *buf++ = 0; + + memcpy (buf, data->ptr, data->len); + + return ret; +} + void Init_eet_ext () { @@ -435,4 +520,12 @@ Init_eet_ext () rb_define_method (c, "write", c_write, -1); rb_define_method (c, "read_image", c_read_image, 1); rb_define_method (c, "write_image", c_write_image, -1); + + c = rb_define_class_under (m, "Stream", rb_cArray); + rb_define_method (c, "serialize", stream_serialize, 0); + + c = rb_define_class_under (m, "Chunk", rb_cObject); + rb_define_method (c, "serialize", chunk_serialize, 0); + + id_include = rb_intern ("include?"); }