1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
"""
Wrapper functions to more user-friendly calling of certain math functions
whose output data-type is different than the input data-type in certain domains of the input.
"""
__all__ = ['sqrt', 'log', 'log2', 'logn','log10', 'power', 'arccos',
'arcsin', 'arctanh']
import numeric as nx
from numeric import *
from type_check import isreal, asscalar
__all__.extend([key for key in dir(nx.umath)
if key[0] != '_' and key not in __all__])
_ln2 = log(2.0)
def _tocomplex(arr):
if isinstance(arr.dtype, (nx.single, nx.byte, nx.short, nx.ubyte,
nx.ushort)):
return arr.astype(nx.csingle)
else:
return arr.astype(nx.cdouble)
def _fix_real_lt_zero(x):
x = asarray(x)
if any(isreal(x) & (x<0)):
x = _tocomplex(x)
return asscalar(x)
def _fix_real_abs_gt_1(x):
x = asarray(x)
if any(isreal(x) & (abs(x)>1)):
x = _tocomplex(x)
return x
def sqrt(x):
x = _fix_real_lt_zero(x)
return nx.sqrt(x)
def log(x):
x = _fix_real_lt_zero(x)
return nx.log(x)
def log10(x):
x = _fix_real_lt_zero(x)
return nx.log10(x)
def logn(n, x):
""" Take log base n of x.
"""
x = _fix_real_lt_zero(x)
n = _fix_real_lt_zero(n)
return log(x)/log(n)
def log2(x):
""" Take log base 2 of x.
"""
x = _fix_real_lt_zero(x)
return log(x)/_ln2
def power(x, p):
x = _fix_real_lt_zero(x)
return nx.power(x, p)
def arccos(x):
x = _fix_real_abs_gt_1(x)
return arccos(x)
def arcsin(x):
x = _fix_real_abs_gt_1(x)
return arcsin(x)
def arctanh(x):
x = _fix_real_abs_gt_1(x)
return arctanh(x)
|
