fengqi/lv_micropython
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

py: Implement complex division.

Browse Source
This commit is contained in:
Damien George 2014-04-09 20:21:00 +01:00
parent e2835c16f4
commit 5589db88c7
1 changed files with 28 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
py/objcomplex.c
View File

@ -144,22 +144,40 @@ mp_obj_t mp_obj_complex_binary_op(int op, mp_float_t lhs_real, mp_float_t lhs_im
lhs_imag -= rhs_imag; lhs_imag -= rhs_imag;
break; break;
case MP_BINARY_OP_MULTIPLY: case MP_BINARY_OP_MULTIPLY:
case MP_BINARY_OP_INPLACE_MULTIPLY: case MP_BINARY_OP_INPLACE_MULTIPLY: {
{ mp_float_t real;
mp_float_t real = lhs_real * rhs_real - lhs_imag * rhs_imag; multiply:
real = lhs_real * rhs_real - lhs_imag * rhs_imag;
lhs_imag = lhs_real * rhs_imag + lhs_imag * rhs_real; lhs_imag = lhs_real * rhs_imag + lhs_imag * rhs_real;
lhs_real = real; lhs_real = real;
break; break;
} }
/* TODO floor(?) the value
case MP_BINARY_OP_FLOOR_DIVIDE: case MP_BINARY_OP_FLOOR_DIVIDE:
case MP_BINARY_OP_INPLACE_FLOOR_DIVIDE: val = lhs_val / rhs_val; break; case MP_BINARY_OP_INPLACE_FLOOR_DIVIDE:
*/ nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_TypeError, "can't do truncated division of a complex number"));
/* TODO
case MP_BINARY_OP_TRUE_DIVIDE: case MP_BINARY_OP_TRUE_DIVIDE:
case MP_BINARY_OP_INPLACE_TRUE_DIVIDE: val = lhs_val / rhs_val; break; case MP_BINARY_OP_INPLACE_TRUE_DIVIDE:
*/ if (rhs_imag == 0) {
return NULL; // op not supported if (rhs_real == 0) {
nlr_raise(mp_obj_new_exception_msg(&mp_type_ZeroDivisionError, "complex division by zero"));
}
lhs_real /= rhs_real;
lhs_imag /= rhs_real;
} else if (rhs_real == 0) {
mp_float_t real = lhs_imag / rhs_imag;
lhs_imag = -lhs_real / rhs_imag;
lhs_real = real;
} else {
mp_float_t rhs_len_sq = rhs_real*rhs_real + rhs_imag*rhs_imag;
rhs_real /= rhs_len_sq;
rhs_imag /= -rhs_len_sq;
goto multiply;
}
break;
default:
return MP_OBJ_NULL; // op not supported
} }
return mp_obj_new_complex(lhs_real, lhs_imag); return mp_obj_new_complex(lhs_real, lhs_imag);
} }