module CMath include Math alias exp! exp alias log! log alias log10! log10 alias sqrt! sqrt alias sin! sin alias cos! cos alias tan! tan alias sinh! sinh alias cosh! cosh alias tanh! tanh alias asin! asin alias acos! acos alias atan! atan alias atan2! atan2 alias asinh! asinh alias acosh! acosh alias atanh! atanh def exp(z) if z.scalar? exp!(z) else Complex(exp!(z.real) * cos!(z.image), exp!(z.real) * sin!(z.image)) end end def log(*args) z, b = args if z.scalar? and z >= 0 and (b.nil? or b >= 0) log!(*args) else r, theta = z.polar a = Complex(log!(r.abs), theta) if b a /= log(b) end a end end def log10(z) if z.scalar? log10!(z) else log(z) / log!(10) end end def sqrt(z) if z.scalar? if z >= 0 sqrt!(z) else Complex(0,sqrt!(-z)) end else if z.image < 0 sqrt(z.conjugate).conjugate else r = z.abs x = z.real Complex(sqrt!((r + x) / 2), sqrt!((r - x) / 2)) end end end def sin(z) if z.scalar? sin!(z) else Complex(sin!(z.real) * cosh!(z.image), cos!(z.real) * sinh!(z.image)) end end def cos(z) if z.scalar? cos!(z) else Complex(cos!(z.real) * cosh!(z.image), -sin!(z.real) * sinh!(z.image)) end end def tan(z) if z.scalar? tan!(z) else sin(z)/cos(z) end end def sinh(z) if z.scalar? sinh!(z) else Complex(sinh!(z.real) * cos!(z.image), cosh!(z.real) * sin!(z.image)) end end def cosh(z) if z.scalar? cosh!(z) else Complex(cosh!(z.real) * cos!(z.image), sinh!(z.real) * sin!(z.image)) end end def tanh(z) if z.scalar? tanh!(z) else sinh(z) / cosh(z) end end def asin(z) if z.scalar? and z >= -1 and z <= 1 asin!(z) else -1.0.im * log(1.0.im * z + sqrt(1.0 - z * z)) end end def acos(z) if z.scalar? and z >= -1 and z <= 1 acos!(z) else -1.0.im * log(z + 1.0.im * sqrt(1.0 - z * z)) end end def atan(z) if z.scalar? atan!(z) else 1.0.im * log((1.0.im + z) / (1.0.im - z)) / 2.0 end end def atan2(y,x) if y.scalar? and x.scalar? atan2!(y,x) else -1.0.im * log((x + 1.0.im * y) / sqrt(x * x + y * y)) end end def acosh(z) if z.scalar? and z >= 1 acosh!(z) else log(z + sqrt(z * z - 1.0)) end end def asinh(z) if z.scalar? asinh!(z) else log(z + sqrt(1.0 + z * z)) end end def atanh(z) if z.scalar? and z >= -1 and z <= 1 atanh!(z) else log((1.0 + z) / (1.0 - z)) / 2.0 end end module_function :exp! module_function :exp module_function :log! module_function :log module_function :log10! module_function :log10 module_function :sqrt! module_function :sqrt module_function :sin! module_function :sin module_function :cos! module_function :cos module_function :tan! module_function :tan module_function :sinh! module_function :sinh module_function :cosh! module_function :cosh module_function :tanh! module_function :tanh module_function :asin! module_function :asin module_function :acos! module_function :acos module_function :atan! module_function :atan module_function :atan2! module_function :atan2 module_function :asinh! module_function :asinh module_function :acosh! module_function :acosh module_function :atanh! module_function :atanh end