From 6fb00e8a1ff95de610143e7076209c58c19f5f59 Mon Sep 17 00:00:00 2001
From: nagai <nagai@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Sun, 17 Oct 2004 14:05:19 +0000
Subject: * ext/tk/lib/tk/timer.rb: TkTimer#start and restart accept a block

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@7053 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 ext/tk/lib/tk/timer.rb | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

(limited to 'ext/tk/lib/tk/timer.rb')

diff --git a/ext/tk/lib/tk/timer.rb b/ext/tk/lib/tk/timer.rb
index 27c40e57e4..f3b6465f78 100644
--- a/ext/tk/lib/tk/timer.rb
+++ b/ext/tk/lib/tk/timer.rb
@@ -321,20 +321,25 @@ class TkTimer
     self
   end
 
-  def set_start_proc(sleep, init_proc=nil, *init_args)
+  def set_start_proc(sleep=nil, init_proc=nil, *init_args, &b)
+    # set parameters for 'restart'
+    sleep = @init_sleep unless sleep
+
     if !sleep == 'idle' && !sleep.kind_of?(Integer)
       fail ArguemntError, "expect Integer or 'idle' for 1st argument"
     end
+
     @init_sleep = sleep
     @init_proc = init_proc
     @init_args = init_args
 
+    @init_proc = b if !@init_proc && b
     @init_proc = proc{|*args| } if @init_sleep > 0 && !@init_proc
 
     self
   end
 
-  def start(*init_args)
+  def start(*init_args, &b)
     return nil if @running
 
     Tk_CBTBL[@id] = self
@@ -357,6 +362,7 @@ class TkTimer
     @init_proc = init_args.shift if argc > 1
     @init_args = init_args if argc > 2
 
+    @init_proc = b if !@init_proc && b
     @init_proc = proc{|*args| } if @init_sleep > 0 && !@init_proc
 
     @current_sleep = @init_sleep
@@ -391,12 +397,12 @@ class TkTimer
     self
   end
 
-  def restart(*restart_args)
+  def restart(*restart_args, &b)
     cancel if @running
-    if restart_args == []
+    if restart_args == [] && !b
       start(@init_sleep, @init_proc, *@init_args)
     else
-      start(*restart_args)
+      start(*restart_args, &b)
     end
   end
 
-- 
cgit v1.2.3