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:
parent
1ea6c0aa9b
commit
38b9eb1674
1 changed files with 9 additions and 7 deletions
|
@ -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
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue