Added Ogg::Vorbis::Comments#merge!.
[ruby-vorbistagger.git] / ext / comments.c
index b37f59146ea3067ff1902137cb554389f61731eb..cdaf576ae7e697571297a7b2a573525a8edf1e02 100644 (file)
  */
 
 #include <ruby.h>
+#include <st.h>
 #include <stdbool.h>
 #include <ctype.h>
 #include <assert.h>
 
 #include "vcedit.h"
 
-typedef struct {
-       vcedit_state *state;
-
-       VALUE items;
-} RbVorbisComments;
-
 static ID id_casecmp, id_replace, id_compare;
 
 void
 comments_init (VALUE self, vcedit_state *state)
 {
-       RbVorbisComments *o;
+       VALUE items;
        vorbis_comment *vc;
        int i;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
+       vc = vcedit_comments (state);
 
-       o->state = state;
-       vcedit_state_ref (state);
-
-       vc = vcedit_comments (o->state);
-
-       o->items = rb_ary_new2 (vc->comments);
+       items = rb_ary_new2 (vc->comments);
+       rb_iv_set (self, "items", items);
 
        for (i = 0; i < vc->comments; i++) {
                VALUE k, v, pair;
@@ -62,28 +53,23 @@ comments_init (VALUE self, vcedit_state *state)
                pair = rb_ary_new3 (2, k, v);
                OBJ_FREEZE (pair);
 
-               rb_ary_store (o->items, i, pair);
+               rb_ary_store (items, i, pair);
        }
-
-       rb_iv_set (self, "@items", o->items);
 }
 
 void
-comments_sync (VALUE self)
+comments_sync (VALUE self, vcedit_state *state)
 {
-       RbVorbisComments *o;
        vorbis_comment *vc;
        struct RArray *items;
        int i;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
-
-       vc = vcedit_comments (o->state);
+       vc = vcedit_comments (state);
 
        vorbis_comment_clear (vc);
        vorbis_comment_init (vc);
 
-       items = RARRAY (o->items);
+       items = RARRAY (rb_iv_get (self, "items"));
 
        for (i = 0; i < items->len; i++) {
                struct RArray *pair = RARRAY (items->ptr[i]);
@@ -94,28 +80,6 @@ comments_sync (VALUE self)
        }
 }
 
-static void
-c_mark (RbVorbisComments *o)
-{
-       rb_gc_mark (o->items);
-}
-
-static void
-c_free (RbVorbisComments *o)
-{
-       vcedit_state_unref (o->state);
-
-       ruby_xfree (o);
-}
-
-static VALUE
-c_alloc (VALUE klass)
-{
-       RbVorbisComments *o;
-
-       return Data_Make_Struct (klass, RbVorbisComments, c_mark, c_free, o);
-}
-
 /*
  * call-seq:
  *  object.inspect -> string
@@ -126,13 +90,10 @@ static VALUE
 c_inspect (VALUE self)
 {
        VALUE ret;
-       RbVorbisComments *o;
        struct RArray *items;
        int i;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
-
-       items = RARRAY (o->items);
+       items = RARRAY (rb_iv_get (self, "items"));
 
        ret = rb_str_buf_new (128);
        rb_str_buf_cat (ret, "{", 1);
@@ -162,11 +123,7 @@ c_inspect (VALUE self)
 static VALUE
 c_clear (VALUE self)
 {
-       RbVorbisComments *o;
-
-       Data_Get_Struct (self, RbVorbisComments, o);
-
-       rb_ary_clear (o->items);
+       rb_ary_clear (rb_iv_get (self, "items"));
 
        return self;
 }
@@ -182,13 +139,10 @@ static VALUE
 c_delete (VALUE self, VALUE key)
 {
        VALUE ret = Qnil;
-       RbVorbisComments *o;
        struct RArray *items;
        int i, pos = -1;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
-
-       items = RARRAY (o->items);
+       items = RARRAY (rb_iv_get (self, "items"));
 
        for (i = 0; i < items->len; i++) {
                struct RArray *pair = RARRAY (items->ptr[i]);
@@ -203,7 +157,7 @@ c_delete (VALUE self, VALUE key)
        }
 
        if (pos != -1)
-               rb_ary_delete_at (o->items, pos);
+               rb_ary_delete_at (rb_iv_get (self, "items"), pos);
 
        return ret;
 }
@@ -218,13 +172,10 @@ static VALUE
 c_keys (VALUE self)
 {
        VALUE ret;
-       RbVorbisComments *o;
        struct RArray *items;
        int i;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
-
-       items = RARRAY (o->items);
+       items = RARRAY (rb_iv_get (self, "items"));
        ret = rb_ary_new2 (items->len);
 
        for (i = 0; i < items->len; i++) {
@@ -246,13 +197,10 @@ static VALUE
 c_values (VALUE self)
 {
        VALUE ret;
-       RbVorbisComments *o;
        struct RArray *items;
        int i;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
-
-       items = RARRAY (o->items);
+       items = RARRAY (rb_iv_get (self, "items"));
        ret = rb_ary_new2 (items->len);
 
        for (i = 0; i < items->len; i++) {
@@ -273,11 +221,11 @@ c_values (VALUE self)
 static VALUE
 c_length (VALUE self)
 {
-       RbVorbisComments *o;
+       struct RArray *items;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
+       items = RARRAY (rb_iv_get (self, "items"));
 
-       return LONG2NUM (RARRAY (o->items)->len);
+       return LONG2NUM (items->len);
 }
 
 /*
@@ -289,11 +237,11 @@ c_length (VALUE self)
 static VALUE
 c_get_empty (VALUE self)
 {
-       RbVorbisComments *o;
+       struct RArray *items;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
+       items = RARRAY (rb_iv_get (self, "items"));
 
-       return !RARRAY(o->items)->len;
+       return items->len ? Qfalse : Qtrue;
 }
 
 /*
@@ -306,13 +254,10 @@ c_get_empty (VALUE self)
 static VALUE
 c_aref (VALUE self, VALUE key)
 {
-       RbVorbisComments *o;
        struct RArray *items;
        int i;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
-
-       items = RARRAY (o->items);
+       items = RARRAY (rb_iv_get (self, "items"));
 
        for (i = 0; i < items->len; i++) {
                struct RArray *pair = RARRAY (items->ptr[i]);
@@ -336,13 +281,10 @@ static VALUE
 c_aset (VALUE self, VALUE key, VALUE value)
 {
        VALUE tmp;
-       RbVorbisComments *o;
        struct RArray *items;
        int i;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
-
-       items = RARRAY (o->items);
+       items = RARRAY (rb_iv_get (self, "items"));
 
        for (i = 0; i < items->len; i++) {
                struct RArray *pair = RARRAY (items->ptr[i]);
@@ -358,7 +300,7 @@ c_aset (VALUE self, VALUE key, VALUE value)
        tmp = rb_ary_new3 (2, rb_str_dup_frozen (key), value);
        OBJ_FREEZE (tmp);
 
-       rb_ary_push (o->items, tmp);
+       rb_ary_push (rb_iv_get (self, "items"), tmp);
 
        return value;
 }
@@ -373,13 +315,10 @@ c_aset (VALUE self, VALUE key, VALUE value)
 static VALUE
 c_has_key (VALUE self, VALUE key)
 {
-       RbVorbisComments *o;
        struct RArray *items;
        int i;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
-
-       items = RARRAY (o->items);
+       items = RARRAY (rb_iv_get (self, "items"));
 
        for (i = 0; i < items->len; i++) {
                struct RArray *pair = RARRAY (items->ptr[i]);
@@ -393,6 +332,31 @@ c_has_key (VALUE self, VALUE key)
        return Qfalse;
 }
 
+static int
+merge_cb (VALUE key, VALUE value, VALUE self)
+{
+       c_aset (self, key, value);
+
+       return ST_CONTINUE;
+}
+
+/*
+ * call-seq:
+ *  object.merge!(hash) -> object
+ *
+ * Adds the key-value pairs from *hash* to *object*, overwriting existing
+ * values if a key already existed in *object*.
+ */
+static VALUE
+c_merge (VALUE self, VALUE hash)
+{
+       Check_Type (hash, T_HASH);
+
+       rb_hash_foreach (hash, merge_cb, self);
+
+       return self;
+}
+
 /*
  * call-seq:
  *  object <=> other -> -1, 0 or 1
@@ -403,18 +367,14 @@ c_has_key (VALUE self, VALUE key)
 static VALUE
 c_compare (VALUE self, VALUE other)
 {
-       RbVorbisComments *o, *o2;
        struct RArray *a, *b;
        int i, j;
 
        if (rb_obj_is_kind_of (other, CLASS_OF (self)) != Qtrue)
                rb_raise (rb_eArgError, "invalid argument");
 
-       Data_Get_Struct (self, RbVorbisComments, o);
-       Data_Get_Struct (other, RbVorbisComments, o2);
-
-       a = RARRAY (o->items);
-       b = RARRAY (o2->items);
+       a = RARRAY (rb_iv_get (self, "items"));
+       b = RARRAY (rb_iv_get (other, "items"));
 
        if (a->len < b->len)
                return -1;
@@ -449,13 +409,10 @@ c_compare (VALUE self, VALUE other)
 static VALUE
 c_each (VALUE self)
 {
-       RbVorbisComments *o;
        struct RArray *items;
        int i;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
-
-       items = RARRAY (o->items);
+       items = RARRAY (rb_iv_get (self, "items"));
 
        for (i = 0; i < items->len; i++) {
                struct RArray *pair = RARRAY (items->ptr[i]);
@@ -477,13 +434,10 @@ c_each (VALUE self)
 static VALUE
 c_each_key (VALUE self)
 {
-       RbVorbisComments *o;
        struct RArray *items;
        int i;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
-
-       items = RARRAY (o->items);
+       items = RARRAY (rb_iv_get (self, "items"));
 
        for (i = 0; i < items->len; i++) {
                struct RArray *pair = RARRAY (items->ptr[i]);
@@ -504,13 +458,10 @@ c_each_key (VALUE self)
 static VALUE
 c_each_value (VALUE self)
 {
-       RbVorbisComments *o;
        struct RArray *items;
        int i;
 
-       Data_Get_Struct (self, RbVorbisComments, o);
-
-       items = RARRAY (o->items);
+       items = RARRAY (rb_iv_get (self, "items"));
 
        for (i = 0; i < items->len; i++) {
                struct RArray *pair = RARRAY (items->ptr[i]);
@@ -528,8 +479,6 @@ Init_Comments (VALUE mVorbis)
 
        c = rb_define_class_under (mVorbis, "Comments", rb_cObject);
 
-       rb_define_alloc_func (c, c_alloc);
-
        rb_define_method (c, "inspect", c_inspect, 0);
        rb_define_method (c, "clear", c_clear, 0);
        rb_define_method (c, "delete", c_delete, 1);
@@ -540,6 +489,7 @@ Init_Comments (VALUE mVorbis)
        rb_define_method (c, "empty?", c_get_empty, 0);
        rb_define_method (c, "keys", c_keys, 0);
        rb_define_method (c, "values", c_values, 0);
+       rb_define_method (c, "merge!", c_merge, 1);
 
        rb_include_module (c, rb_mComparable);
        rb_define_method (c, "<=>", c_compare, 1);