Added to_binary_array method

brendon9x · brendon9x · commit d07d96ded24c · 2011-10-06T15:16:11.000+02:00
diff --git a/Rakefile b/Rakefile
@@ -17,7 +17,7 @@ Jeweler::Tasks.new do |gem|
   gem.summary = 'Bitset implementation.'
   gem.description = 'A fast C-based Bitset. It supports the standard set operations as well as operations you may expect on bit arrays. (popcount, for instance)'
   gem.email = "tbmcmullen@gmail.com"
-  gem.authors = ["Tyler McMullen"]
+  gem.authors = ["Tyler McMullen", "Brendon McLean"]
 end
 Jeweler::RubygemsDotOrgTasks.new
 
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-0.1.2
+0.1.3
diff --git a/bitset.gemspec b/bitset.gemspec
@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{bitset}
-  s.version = "0.1.2"
+  s.version = "0.1.3"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
-  s.authors = ["Tyler McMullen"]
-  s.date = %q{2011-05-07}
+  s.authors = ["Tyler McMullen", "Brendon McLean"]
+  s.date = %q{2011-10-06}
   s.description = %q{A fast C-based Bitset. It supports the standard set operations as well as operations you may expect on bit arrays. (popcount, for instance)}
   s.email = %q{tbmcmullen@gmail.com}
   s.extensions = ["ext/bitset/extconf.rb"]
diff --git a/ext/bitset/bitset.c b/ext/bitset/bitset.c
@@ -349,6 +349,18 @@ static VALUE rb_bitset_to_a(VALUE self) {
     return array;
 }
 
+static VALUE rb_bitset_to_binary_array(VALUE self) {
+    Bitset * bs = get_bitset(self);
+    int i;
+
+    VALUE array = rb_ary_new2(bs->len / 2);
+    for(i = 0; i < bs->len; i++) {
+        rb_ary_push(array, INT2NUM(get_bit(bs, i) > 0 ? 1 : 0));
+    }
+
+    return array;
+}
+
 void Init_bitset() {
     cBitset = rb_define_class("Bitset", rb_cObject);
     rb_include_module(cBitset, rb_mEnumerable);
@@ -380,4 +392,5 @@ void Init_bitset() {
     rb_define_method(cBitset, "marshal_dump", rb_bitset_marshall_dump, 0);
     rb_define_method(cBitset, "marshal_load", rb_bitset_marshall_load, 1);
     rb_define_method(cBitset, "to_a", rb_bitset_to_a, 0);
+    rb_define_method(cBitset, "to_binary_array", rb_bitset_to_binary_array, 0);
 }
diff --git a/spec/bitset_spec.rb b/spec/bitset_spec.rb
@@ -278,4 +278,13 @@
       bs.to_a.should == [1, 64, 65]
     end
   end
+
+  describe :to_binary_array do
+    it "can convert to an array of 1s and 0s" do
+      bs = Bitset.new(68)
+      bs.set 1, 64, 65
+
+      bs.to_binary_array.values_at(1, 64, 65, 66).should == [1, 1, 1, 0]
+    end
+  end
 end
