archive/ruby
1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2025-08-15 13:39:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

complex.c: multiply as rotation

Browse source 
* complex.c (nucomp_mul): calculate as rotation in complex plane
  if matrix calculation resulted in NaN.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@49723 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
nobu 2015-02-24 13:58:52 +00:00
parent 87e3aec84d
commit 3bcb10ad2a
3 changed files with 47 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

35
complex.c
View file

@ -17,6 +17,9 @@
VALUE rb_cComplex;
static VALUE nucomp_abs(VALUE self);
static VALUE nucomp_arg(VALUE self);
static ID id_abs, id_arg, id_convert,
id_denominator, id_eqeq_p, id_expt, id_fdiv,
id_negate, id_numerator, id_quo,
@ -720,6 +723,38 @@ nucomp_mul(VALUE self, VALUE other)
imag = f_add(f_mul(adat->real, bdat->imag),
f_mul(adat->imag, bdat->real));
if ((RB_FLOAT_TYPE_P(real) && isnan(RFLOAT_VALUE(real))) ||
(RB_FLOAT_TYPE_P(imag) && isnan(RFLOAT_VALUE(imag)))) {
VALUE abs = f_mul(nucomp_abs(self), nucomp_abs(other));
VALUE arg = f_add(nucomp_arg(self), nucomp_arg(other));
if (f_zero_p(arg)) {
real = abs;
imag = INT2FIX(0);
}
else if (RB_FLOAT_TYPE_P(arg)) {
double a = RFLOAT_VALUE(arg);
if (a == M_PI) {
real = f_negate(abs);
imag = INT2FIX(0);
}
else if (a == M_PI/2) {
imag = abs;
real = INT2FIX(0);
}
else if (a == M_PI*3/2) {
imag = f_negate(abs);
real = INT2FIX(0);
}
else {
goto polar;
}
}
else {
polar:
return f_complex_polar(CLASS_OF(self), abs, arg);
}
}
return f_complex_new2(CLASS_OF(self), real, imag);
}
if (k_numeric_p(other) && f_real_p(other)) {