Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-20 11:30:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

support abs() for complex objects

Browse Source
This commit is contained in:
BLUELOVETH 2024-07-21 08:29:53 +00:00
parent 066554f241
commit 981e6e6e88
5 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
include/pocketpy/str.h
View File

@ -224,6 +224,7 @@ extern const StrName __package__;
extern const StrName __path__;
extern const StrName __class__;
extern const StrName __missing__;
extern const StrName __abs__;
extern const StrName pk_id_add;
extern const StrName pk_id_set;

8
python/cmath.py
View File

@ -77,8 +77,8 @@ class complex:
def __pow__(self, other: int | float):
if type(other) in (int, float):
return complex(self.__abs__() ** other * math.cos(other * phase(self)),
self.__abs__() ** other * math.sin(other * phase(self)))
return complex(abs(self) ** other * math.cos(other * phase(self)),
abs(self) ** other * math.sin(other * phase(self)))
return NotImplemented
def __abs__(self) -> float:
@ -97,7 +97,7 @@ def phase(z: complex):
return math.atan2(z.imag, z.real)
def polar(z: complex):
return z.__abs__(), phase(z)
return abs(z), phase(z)
def rect(r: float, phi: float):
return r * math.cos(phi) + r * math.sin(phi) * 1j
@ -108,7 +108,7 @@ def exp(z: complex):
return math.exp(z.real) * rect(1, z.imag)
def log(z: complex, base=2.718281828459045):
return math.log(z.__abs__(), base) + phase(z) * 1j
return math.log(abs(z), base) + phase(z) * 1j
def log10(z: complex):
return log(z, 10)

2
src/_generated.cpp
View File

File diff suppressed because one or more lines are too long

3
src/pocketpy.cpp
View File

@ -158,8 +158,7 @@ void __init_builtins(VM* _vm) {
_vm->bind_func(_vm->builtins, "abs", 1, [](VM* vm, ArgsView args) {
if(is_int(args[0])) return VAR(std::abs(_CAST(i64, args[0])));
if(is_float(args[0])) return VAR(std::abs(_CAST(f64, args[0])));
vm->TypeError("bad operand type for abs()");
return vm->None;
return vm->call_method(args[0], __abs__);
});
_vm->bind(_vm->builtins, "max(*args, key=None)", [](VM* vm, ArgsView args){

1
src/str.cpp
View File

@ -592,6 +592,7 @@ const StrName __package__ = StrName::get("__package__");
const StrName __path__ = StrName::get("__path__");
const StrName __class__ = StrName::get("__class__");
const StrName __missing__ = StrName::get("__missing__");
const StrName __abs__ = StrName::get("__abs__");
const StrName pk_id_add = StrName::get("add");
const StrName pk_id_set = StrName::get("set");