lua-sci/complex.lua at master · stepelu/lua-sci · GitHub

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
--------------------------------------------------------------------------------
-- Complex numbers.
--
-- Copyright (C) 2011-2016 Stefano Peluchetti. All rights reserved.
--------------------------------------------------------------------------------

localffi=require'ffi'

localsqrt=math.sqrt

localcomplex=ffi.typeof("complex")

localcomplex_mt= {
__add=function(x, y)
x, y=complex(x), complex(y)
returncomplex(x.re+y.re, x.im+y.im)
end,
__sub=function(x, y)
x, y=complex(x), complex(y)
returncomplex(x.re-y.re, x.im-y.im)
end,
__mul=function(x, y)
x, y=complex(x), complex(y)
returncomplex(x.re*y.re-x.im*y.im, x.re*y.im+x.im*y.re)
end,
__div=function(x, y)
x, y=complex(x), complex(y)
locald=y.re^2+y.im^2
returncomplex((x.re*y.re+x.im*y.im)/d, (x.im*y.re-x.re*y.im)/d)
end,
}

ffi.metatype(complex, complex_mt)

localfunctioncabs(x)
returnsqrt(x.re^2+x.im^2)
end

return {
new=complex,

abs=cabs,
}