1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00

mult & div instead of * & /.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@4459 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
shigek 2003-08-29 13:33:37 +00:00
parent 1ea6c0aa9b
commit 38b9eb1674

View file

@ -2,7 +2,8 @@
# ludcmp.rb # ludcmp.rb
# #
module LUSolve module LUSolve
def ludecomp(a,n,zero=0.0,one=1.0) def ludecomp(a,n,zero=0,one=1)
prec = BigDecimal.limit(nil)
ps = [] ps = []
scales = [] scales = []
for i in 0...n do # pick up largest(abs. val.) element in each row. for i in 0...n do # pick up largest(abs. val.) element in each row.
@ -14,7 +15,7 @@ module LUSolve
nrmrow = biggst if biggst>nrmrow nrmrow = biggst if biggst>nrmrow
end end
if nrmrow>zero then if nrmrow>zero then
scales <<= one/nrmrow scales <<= one.div(nrmrow,prec)
else else
raise "Singular matrix" raise "Singular matrix"
end end
@ -38,11 +39,11 @@ module LUSolve
pivot = a[ps[k]*n+k] pivot = a[ps[k]*n+k]
for i in (k+1)...n do for i in (k+1)...n do
psin = ps[i]*n psin = ps[i]*n
a[psin+k] = mult = a[psin+k]/pivot a[psin+k] = mult = a[psin+k].div(pivot,prec)
if mult!=zero then if mult!=zero then
pskn = ps[k]*n pskn = ps[k]*n
for j in (k+1)...n do for j in (k+1)...n do
a[psin+j] -= mult*a[pskn+j] a[psin+j] -= mult.mult(a[pskn+j],prec)
end end
end end
end end
@ -52,13 +53,14 @@ module LUSolve
end end
def lusolve(a,b,ps,zero=0.0) def lusolve(a,b,ps,zero=0.0)
prec = BigDecimal.limit(nil)
n = ps.size n = ps.size
x = [] x = []
for i in 0...n do for i in 0...n do
dot = zero dot = zero
psin = ps[i]*n psin = ps[i]*n
for j in 0...i do for j in 0...i do
dot = a[psin+j]*x[j] + dot dot = a[psin+j].mult(x[j],prec) + dot
end end
x <<= b[ps[i]] - dot x <<= b[ps[i]] - dot
end end
@ -66,9 +68,9 @@ module LUSolve
dot = zero dot = zero
psin = ps[i]*n psin = ps[i]*n
for j in (i+1)...n do for j in (i+1)...n do
dot = a[psin+j]*x[j] + dot dot = a[psin+j].mult(x[j],prec) + dot
end end
x[i] = (x[i]-dot)/a[psin+i] x[i] = (x[i]-dot).div(a[psin+i],prec)
end end
x x
end end