1234567891011121314151617181920212223242526272829303132333435363738394041424344 |
- #!/usr/bin/ruby
- #
- ## Algorithm from Wikipedia: https://en.wikipedia.org/wiki/Lanczos_approximation#Simple_implementation
- #
- func gamma(z) {
- var epsilon = 0.0000001
- func withinepsilon(x) {
- abs(x - abs(x)) <= epsilon
- }
- var p = [
- 676.5203681218851, -1259.1392167224028,
- 771.32342877765313, -176.61502916214059,
- 12.507343278686905, -0.13857109526572012,
- 9.9843695780195716e-6, 1.5056327351493116e-7,
- ]
- var result
- const pi = Num.pi
- if (z.real < 0.5) {
- result = (pi / (sin(pi * z) * gamma(1 - z)))
- }
- else {
- z--
- var x = 0.99999999999980993
- p.each_kv { |i, v|
- x += v/(z + i + 1)
- }
- var t = (z + p.len - 0.5)
- result = (sqrt(pi*2) * t**(z+0.5) * exp(-t) * x)
- }
- withinepsilon(result.im) ? result.real : result
- }
- for i in [1/2,4,5,6,30,40,50] {
- say "gamma(%3s) =~ %s"%(i, gamma(i));
- }
|