Skip to content

Commit

Permalink
Drotmg added (not tested!)
Browse files Browse the repository at this point in the history
  • Loading branch information
michalderkacz committed Nov 27, 2011
1 parent a95b423 commit a923e5d
Show file tree
Hide file tree
Showing 2 changed files with 106 additions and 0 deletions.
1 change: 1 addition & 0 deletions dgemv.go
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
package blas

// Performs: y = alpha * A * x + beta * y or y = alpha * A^T * x + beta * y
/*func Dgemv(order Order, TransA Transpose, N, M, KL, KU int,
alpha float64, A []float64, lda int, X []float64, incX int,
beta float64, Y []float64, incY int) {
Expand Down
105 changes: 105 additions & 0 deletions drotmg.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,105 @@
package blas

import "math"

type DrotmgParam struct {
flag, h11, h21, h12, h22 float64
}

// Compute a modified Givens transformation
func Drotmg(d1, d2, x1, y1 float64) (p DrotmgParam, rd1, rd2, rx1 float64) {
var p1, p2, q1, q2, u float64

gam := 4096.0
gamsq := 16777216.0
rgamsq := 5.9604645e-8

rd1 = d1
rd2 = d2
rx1 = x1

if d1 < 0 {
p.flag = -1
} else {
p2 = rd2 * y1
if p2 == 0 {
p.flag = -2
return
}
p1 = rd1 * x1
q2 = p2 * y1
q1 = p1 * x1
if math.Abs(q1) > math.Abs(q2) {
p.h21 = -y1 / x1
p.h12 = p2 / p1
u = 1 - p.h12 * p.h21
if u > 0 {
p.flag = 0
rd1 /= u
rd2 /= u
rx1 /= u
}
} else {
if q2 < 0 {
p.flag = -1
rd1 = 0
rd2 = 0
rx1 = 0
} else {
p.flag = 1
p.h11 = p1 / p2
p.h22 = x1 / y1
u = 1 + p.h11 + p.h22
rd1, rd2 = rd2 / u, rd1 / u
rx1 = y1 / u
}
}
if rd1 != 0 {
for rd1 <= rgamsq || rd1 >= gamsq {
if p.flag == 0 {
p.h11 = 1
p.h22 = 1
p.flag = -1
} else {
p.h21 = -1
p.h12 = 1
p.flag = -1
}
if rd1 <= rgamsq {
rd1 *= gam * gam
rx1 /= gam
p.h11 /= gam
p.h12 /= gam
} else {
rd1 /= gam * gam
rx1 *= gam
p.h11 *= gam
p.h12 *= gam
}
}
}
if rd2 != 0 {
for math.Abs(rd2) <= rgamsq || math.Abs(rd2) >= gamsq {
if p.flag == 0 {
p.h11 = 1
p.h22 = 1
p.flag = -1
} else {
p.h21 = -1
p.h12 = 1
p.flag = -1
}
if math.Abs(rd2) <= rgamsq {
rd2 *= gam * gam
p.h21 /= gam
p.h22 /= gam
} else {
rd2 /= gam * gam
p.h21 *= gam
p.h22 *= gam
}
}
}
}
return
}

0 comments on commit a923e5d

Please sign in to comment.