Crude but mostly working implementation of Complex.roots. Straightfor…

…ward implementation of Num.Complex.
rakudo · Jan 12, 2010 · 393c457 · 393c457
1 parent f2d5c76
commit 393c457
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 7 deletions.
diff --git a/src/core/Complex.pm b/src/core/Complex.pm
@@ -162,13 +162,24 @@ class Complex {
     }
 
     multi method roots($n is copy) {
-#        my ($mag, $angle) = @.polar;
-#        return NaN  if $n < 1;
-#        return self if $n == 1;
-#        return NaN  if $!re|$!im ~~  Inf|NaN|-Inf;
-#        $n = $n.Int;
-#        $mag **= 1/$n;
-#        (^$n).map: { $mag.unpolar( ($angle + $_ * 2 * pi) / $n) };
+       # my ($mag, $angle) = @.polar;
+       my $mag = $.abs;
+       my $angle = atan2($.im, $.re);
+       if $n < 1
+       {
+           return NaN;
+       }
+
+       if $n == 1
+       {
+           return self;
+       }
+
+       # return NaN  if $!re|$!im ~~  Inf|NaN|-Inf;
+       $n = $n.Int;
+       $mag **= 1/$n;
+       # (^$n).map: { $mag.unpolar( ($angle + $_ * 2 * pi) / $n) };
+       (0 ... ($n-1)).map: { $mag.unpolar( ($angle + $^x * 2 * 312689/99532) / $n) };
     }
 
     multi method sign() {

diff --git a/src/core/Num.pm b/src/core/Num.pm
@@ -11,6 +11,8 @@ augment class Num {
 
     multi method Num() { self; }
 
+    multi method Complex() { self + 0i; }
+
     multi method abs() {
         pir::abs__Nn(self);
     }