marcandre	2010-04-08 02:36:15 +0900 (Thu, 08 Apr 2010)

  New Revision: 27252

  http://svn.ruby-lang.org/cgi-bin/viewvc.cgi?view=rev&revision=27252

  Log:
    * array.c (rb_ary_permutation): Remove limitation for lengthy permutations
      [ruby-core:29240]
    
    * test/ruby/test_array.rb: ditto

  Modified files:
    trunk/ChangeLog
    trunk/array.c
    trunk/test/ruby/test_array.rb

Index: array.c
===================================================================
--- array.c	(revision 27251)
+++ array.c	(revision 27252)
@@ -3958,26 +3958,6 @@
     return ary;
 }
 
-static long
-combi_len(long n, long k)
-{
-    long i, val = 1;
-
-    if (k*2 > n) k = n-k;
-    if (k == 0) return 1;
-    if (k < 0) return 0;
-    val = 1;
-    for (i=1; i <= k; i++,n--) {
-	long m = val;
-	val *= n;
-	if (val < m) {
-	    rb_raise(rb_eRangeError, "too big for combination");
-	}
-	val /= i;
-    }
-    return val;
-}
-
 /*
  *  call-seq:
  *     ary.combination(n) { |c| block }    -> ary
@@ -4024,14 +4004,13 @@
     else {
 	volatile VALUE t0 = tmpbuf(n+1, sizeof(long));
 	long *stack = (long*)RSTRING_PTR(t0);
-	long nlen = combi_len(len, n);
 	volatile VALUE cc = tmpary(n);
 	VALUE *chosen = RARRAY_PTR(cc);
 	long lev = 0;
 
 	MEMZERO(stack, long, n);
 	stack[0] = -1;
-	for (i = 0; i < nlen; i++) {
+	for (;;) {
 	    chosen[lev] = RARRAY_PTR(ary)[stack[lev+1]];
 	    for (lev++; lev < n; lev++) {
 		chosen[lev] = RARRAY_PTR(ary)[stack[lev+1] = stack[lev]+1];
@@ -4041,9 +4020,11 @@
 		rb_raise(rb_eRuntimeError, "combination reentered");
 	    }
 	    do {
+		if (lev == 0) goto done;
 		stack[lev--]++;
-	    } while (lev && (stack[lev+1]+n == len+lev+1));
+	    } while (stack[lev+1]+n == len+lev+1);
 	}
+    done:
 	tmpbuf_discard(t0);
 	tmpary_discard(cc);
     }
Index: ChangeLog
===================================================================
--- ChangeLog	(revision 27251)
+++ ChangeLog	(revision 27252)
@@ -1,3 +1,10 @@
+Wed Apr  8 02:33:55 2010  Marc-Andre Lafortune  <ruby-core@m...>
+
+	* array.c (rb_ary_permutation): Remove limitation for lengthy permutations
+	  [ruby-core:29240]
+
+	* test/ruby/test_array.rb: ditto
+
 Wed Apr  7 23:33:55 2010  KOSAKI Motohiro  <kosaki.motohiro@g...>
 
 	* misc/ruby-mode.el (ruby-mode-map): binded C-c C-c and C-c C-c C-u
Index: test/ruby/test_array.rb
===================================================================
--- test/ruby/test_array.rb	(revision 27251)
+++ test/ruby/test_array.rb	(revision 27252)
@@ -1842,8 +1842,8 @@
   end
 
   def test_combination2
-    assert_raise(RangeError) do
-      (0..100).to_a.combination(50) {}
+    assert_nothing_raised do
+      (0..100).to_a.combination(50) { break }
     end
   end
 

--
ML: ruby-changes@q...
Info: http://www.atdot.net/~ko1/quickml/