Prefer rb_funcall2() over rb_funcall().
[ruby-eet.git] / ext / ext.c
index 4fb3f1e45895a2873f0ed3ee115fa320a6448eef..e99d41ccf618604a1d9b40413abc1c8d79bf45f4 100644 (file)
--- a/ext/ext.c
+++ b/ext/ext.c
@@ -1,7 +1,5 @@
 /*
- * $Id: ext.c 72 2005-07-16 13:15:42Z tilman $
- *
- * Copyright (c) 2005 Tilman Sauerbeck (tilman at code-monkey de)
+ * Copyright (c) 2005-2007 Tilman Sauerbeck (tilman at code-monkey de)
  *
  * Permission is hereby granted, free of charge, to any person obtaining
  * a copy of this software and associated documentation files (the
 #include <st.h>
 #include <locale.h>
 
-#define CHECK_KEY(key) \
+#define CHECK_NO_BIN0(key) \
        if (rb_funcall (key, id_include, 1, INT2FIX (0)) == Qtrue) \
-               rb_raise (rb_eArgError, "key must not contain binary zeroes");
+               rb_raise (rb_eArgError, #key " must not contain binary zeroes");
 
 #define CHECK_CLOSED(ef) \
        if (!*(ef)) \
                rb_raise (rb_eIOError, "closed stream");
 
+#define CHECK_READABLE(ef) \
+       switch (eet_mode_get (*ef)) { \
+               case EET_FILE_MODE_READ: \
+               case EET_FILE_MODE_READ_WRITE: \
+                       break; \
+               default: \
+                       rb_raise (rb_eIOError, "permission denied"); \
+       }
+
 #ifdef WORDS_BIGENDIAN
 # define BSWAP32(x) \
        ((((x) & 0xff000000) >> 24) | (((x) & 0x00ff0000) >> 8) | \
@@ -179,47 +186,61 @@ c_close (VALUE self)
        return self;
 }
 
+static VALUE
+get_keys (Eet_File *ef, const char *glob)
+{
+       VALUE ret;
+       char **keys;
+       int i, count = 0;
+
+       keys = eet_list (ef, glob, &count);
+       ret = rb_ary_new2 (count);
+
+       for (i = 0; i < count; i++)
+               rb_ary_store (ret, i, rb_str_new2 (keys[i]));
+
+       free (keys);
+
+       return ret;
+}
+
 /*
  * call-seq:
- *  ef.list([glob]) -> array
+ *  ef.entries -> array
  *
- * Returns an Array of entries in _ef_ that match the shell glob
- * _glob_ (defaulting to "*").
+ * Returns an Array with the keys of the entries in _ef_.
+ * If the keys cannot be retrieved, an +IOError+ is raised.
  */
 static VALUE
-c_list (int argc, VALUE *argv, VALUE self)
+c_entries (VALUE self)
 {
-       VALUE glob = Qnil, ret;
        Eet_File **ef = NULL;
-       char **entries, *tmp = "*";
-       int i, count = 0;
 
        Data_Get_Struct (self, Eet_File *, ef);
        CHECK_CLOSED (ef);
+       CHECK_READABLE (ef);
 
-       switch (eet_mode_get (*ef)) {
-               case EET_FILE_MODE_READ:
-               case EET_FILE_MODE_READ_WRITE:
-                       break;
-               default:
-                       rb_raise (rb_eIOError, "cannot list entries");
-       }
-
-       rb_scan_args (argc, argv, "01", &glob);
-
-       if (!NIL_P (glob))
-               tmp = StringValuePtr (glob);
-
-       ret = rb_ary_new ();
-
-       entries = eet_list (*ef, tmp, &count);
+       return get_keys (*ef, "*");
+}
 
-       for (i = 0; i < count; i++)
-               rb_ary_push (ret, rb_str_new2 (entries[i]));
+/*
+ * call-seq:
+ *  ef[glob] -> array
+ *
+ * Returns an Array with the keys of entries in _ef_ that match the
+ * shell glob _glob_.
+ * If the keys cannot be retrieved, an +IOError+ is raised.
+ */
+static VALUE
+c_glob (VALUE self, VALUE glob)
+{
+       Eet_File **ef = NULL;
 
-       free (entries);
+       Data_Get_Struct (self, Eet_File *, ef);
+       CHECK_CLOSED (ef);
+       CHECK_READABLE (ef);
 
-       return ret;
+       return get_keys (*ef, StringValuePtr (glob));
 }
 
 /*
@@ -240,7 +261,7 @@ c_delete (VALUE self, VALUE key)
        CHECK_CLOSED (ef);
 
        ckey = StringValuePtr (key);
-       CHECK_KEY (key);
+       CHECK_NO_BIN0 (key);
 
        if (!eet_delete (*ef, ckey))
                rb_raise (rb_eIOError, "cannot delete entry - %s", ckey);
@@ -269,7 +290,7 @@ c_read (VALUE self, VALUE key)
        CHECK_CLOSED (ef);
 
        ckey = StringValuePtr (key);
-       CHECK_KEY (key);
+       CHECK_NO_BIN0 (key);
 
        data = eet_read (*ef, ckey, &size);
        if (!data)
@@ -309,7 +330,7 @@ c_write (int argc, VALUE *argv, VALUE self)
                comp = Qtrue;
 
        ckey = StringValuePtr (key);
-       CHECK_KEY (key);
+       CHECK_NO_BIN0 (key);
        cbuf = StringValuePtr (buf);
 
        n = eet_write (*ef, ckey,
@@ -340,13 +361,14 @@ c_read_image (VALUE self, VALUE key)
        Eet_File **ef = NULL;
        void *data;
        char *ckey;
-       int w = 0, h = 0, has_alpha = 0, level = 0, quality = 0, lossy = 0;
+       unsigned int w = 0, h = 0;
+       int 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);
+       CHECK_NO_BIN0 (key);
 
        data = eet_data_image_read (*ef, ckey, &w, &h,
                                    &has_alpha, &level, &quality,
@@ -402,7 +424,7 @@ c_write_image (int argc, VALUE *argv, VALUE self)
                has_alpha = Qfalse;
 
        ckey = StringValuePtr (key);
-       CHECK_KEY (key);
+       CHECK_NO_BIN0 (key);
        cbuf = StringValuePtr (buf);
        Check_Type (w, T_FIXNUM);
        Check_Type (h, T_FIXNUM);
@@ -444,8 +466,7 @@ chunk_init (VALUE self, VALUE tag, VALUE data)
        StringValue (tag);
        StringValue (data);
 
-       if (rb_funcall (tag, id_include, 1, INT2FIX (0)) == Qtrue)
-               rb_raise (rb_eArgError, "tag must not contain binary zeroes");
+       CHECK_NO_BIN0 (tag);
 
        /* libeet uses a signed 32bit integer to store the
         * chunk size, so make sure we don't overflow it
@@ -480,7 +501,7 @@ chunk_to_s (VALUE self)
        buf_len = 9 + tag->len + data->len;
        ret = rb_str_buf_new (buf_len);
 
-       buf = RSTRING (ret)->ptr;
+       buf = (unsigned char *) RSTRING (ret)->ptr;
        RSTRING (ret)->len = buf_len;
 
        memcpy (buf, "CHnK", 4);
@@ -536,13 +557,13 @@ c_to_eet (VALUE self)
        long i;
 #endif
 
-       props = rb_funcall (self, id_to_eet_properties, 0);
+       props = rb_funcall2 (self, id_to_eet_properties, 0, NULL);
 
        if (rb_obj_is_kind_of (props, rb_cHash) == Qfalse ||
            !RHASH (props)->tbl->num_entries)
                rb_raise (ePropError, "invalid EET properties");
 
-       name = rb_funcall (self, id_to_eet_name, 0);
+       name = rb_funcall2 (self, id_to_eet_name, 0, NULL);
        StringValue (name);
 
        if (!RSTRING (name)->len ||
@@ -554,7 +575,7 @@ c_to_eet (VALUE self)
 #ifdef HAVE_RB_HASH_FOREACH
        rb_hash_foreach (props, for_each_prop, stream);
 #else
-       keys = RARRAY (rb_funcall (props, id_keys, 0));
+       keys = RARRAY (rb_funcall2 (props, id_keys, 0, NULL));
 
        for (i = 0; i < keys->len; i++)
                for_each_prop (keys->ptr[i],
@@ -569,18 +590,18 @@ c_to_eet (VALUE self)
 
        chunk = rb_class_new_instance (2, args, cChunk);
 
-       return rb_funcall (chunk, id_to_s, 0);
+       return rb_funcall2 (chunk, id_to_s, 0, NULL);
 }
 
 static VALUE
 int_to_eet_chunks (int argc, VALUE *argv, VALUE self)
 {
        VALUE tag, type = Qnil, ary, args[2], chunk;
-       char *cfmt = "V";
+       const char *cfmt = "V";
 
        rb_scan_args (argc, argv, "11", &tag, &type);
 
-       ary = rb_ary_new3 (1, self);
+       ary = rb_ary_new4 (1, &self);
 
        if (type == sym_char)
                cfmt = "c";
@@ -593,7 +614,7 @@ int_to_eet_chunks (int argc, VALUE *argv, VALUE self)
        args[1] = rb_funcall (ary, id_pack, 1, rb_str_new2 (cfmt));
        chunk = rb_class_new_instance (2, args, cChunk);
 
-       return rb_ary_new3 (1, chunk);
+       return rb_ary_new4 (1, &chunk);
 }
 
 static VALUE
@@ -623,7 +644,7 @@ float_to_eet_chunks (int argc, VALUE *argv, VALUE self)
        args[1] = rb_str_new (buf, len);
        chunk = rb_class_new_instance (2, args, cChunk);
 
-       return rb_ary_new3 (1, chunk);
+       return rb_ary_new4 (1, &chunk);
 }
 
 void
@@ -638,7 +659,8 @@ Init_eet_ext ()
        rb_define_singleton_method (c, "open", c_open, -1);
        rb_define_method (c, "initialize", c_init, -1);
        rb_define_method (c, "close", c_close, 0);
-       rb_define_method (c, "list", c_list, -1);
+       rb_define_method (c, "entries", c_entries, 0);
+       rb_define_method (c, "[]", c_glob, 1);
        rb_define_method (c, "delete", c_delete, 1);
        rb_define_method (c, "read", c_read, 1);
        rb_define_method (c, "write", c_write, -1);