Add method roots() to Numbers

leto · Mar 21, 2009 · e5ae093 · e5ae093
1 parent fb7d44d
commit e5ae093
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/builtins/any-num.pir b/src/builtins/any-num.pir
@@ -184,6 +184,15 @@ error.
 
 =cut
 
+.sub 'roots' :method :multi(I)
+    .param num n
+    .local pmc x
+    .local pmc result
+     x = self
+     result = 'roots'(x,n)
+    .return (result)
+.end
+
 .sub 'unpolar' :method
     .param num angle
     .local num mag

diff --git a/src/builtins/math.pir b/src/builtins/math.pir
@@ -36,7 +36,9 @@ Returns the $N roots of a Complex number $z.
 #    $P5[1] = 1
 
   if n > 0 goto positive
-    .return ('NaN')
+    roots = 1
+    roots[0] = 'NaN'
+    goto done
   positive:
   if n > 1 goto general
     roots[0] = x