#puts and #print that works for multiple arguments too

vidarh · Jul 17, 2013 · 4511cea · 4511cea
1 parent 6901567
commit 4511cea
Showing 1 changed file with 25 additions and 12 deletions.
diff --git a/lib/core/object.rb b/lib/core/object.rb
@@ -40,19 +40,32 @@ def puts *str
       %s(puts "")
       return
     end
-
-    %s(assign raw (index str 0))
-    raw = raw.to_s.__get_raw
-    %s(if raw
-         (puts raw)
-         (puts "")
-         )
+
+    na = na - 2
+    i = 0
+    while i < na
+      %s(assign raw (index str (callm i __get_raw)))
+      raw = raw.to_s.__get_raw
+      %s(if raw (puts raw))
+      i = i + 1
+    end
   end
 
-  def print str
-    raw = str.to_s.__get_raw
-    %s(if raw
-         (printf "%s" raw)
-         )
+  def print *str
+    %s(assign na (__get_fixnum numargs))
+
+    if na == 2
+      %s(puts "")
+      return
+    end
+
+    na = na - 2
+    i = 0
+    while i < na
+      %s(assign raw (index str (callm i __get_raw)))
+      raw = raw.to_s.__get_raw
+      %s(if raw (printf "%s" raw))
+      i = i + 1
+    end
   end
 end