From 10abae8a191bd0ea6c20dbe1703463a60e6a898c Mon Sep 17 00:00:00 2001
From: Vladimir Chalupecky <vladimir.chalupecky@gmail.com>
Date: Sun, 5 Feb 2017 20:51:42 +0100
Subject: [PATCH] cgo: add Dorgtr

---
 cgo/lapack.go      | 36 ++++++++++++++++++++++++++++++++++++
 cgo/lapack_test.go |  4 ++++
 2 files changed, 40 insertions(+)

diff --git a/cgo/lapack.go b/cgo/lapack.go
index e1ee6920..4d420cfb 100644
--- a/cgo/lapack.go
+++ b/cgo/lapack.go
@@ -1486,6 +1486,42 @@ func (impl Implementation) Dorgqr(m, n, k int, a []float64, lda int, tau, work [
 	lapacke.Dorgqr(m, n, k, a, lda, tau, work, lwork)
 }
 
+// Dorgtr generates a real orthogonal matrix Q which is defined as the product
+// of n-1 elementary reflectors of order n as returned by Dsytrd.
+//
+// The construction of Q depends on the value of uplo:
+//  Q = H_{n-1} * ... * H_1 * H_0  if uplo == blas.Upper
+//  Q = H_0 * H_1 * ... * H_{n-1}  if uplo == blas.Lower
+// where H_i is constructed from the elementary reflectors as computed by Dsytrd.
+// See the documentation for Dsytrd for more information.
+//
+// tau must have length at least n-1, and Dorgtr will panic otherwise.
+//
+// work is temporary storage, and lwork specifies the usable memory length. At
+// minimum, lwork >= max(1,n-1), and Dorgtr will panic otherwise. The amount of blocking
+// is limited by the usable length.
+// If lwork == -1, instead of computing Dorgtr the optimal work length is stored
+// into work[0].
+//
+// Dorgtr is an internal routine. It is exported for testing purposes.
+func (impl Implementation) Dorgtr(uplo blas.Uplo, n int, a []float64, lda int, tau, work []float64, lwork int) {
+	checkMatrix(n, n, a, lda)
+	if len(tau) < n-1 {
+		panic(badTau)
+	}
+	if len(work) < lwork {
+		panic(badWork)
+	}
+	if lwork < n-1 && lwork != -1 {
+		panic(badWork)
+	}
+	upper := uplo == blas.Upper
+	if !upper && uplo != blas.Lower {
+		panic(badUplo)
+	}
+	lapacke.Dorgtr(uplo, n, a, lda, tau, work, lwork)
+}
+
 // Dormbr applies a multiplicative update to the matrix C based on a
 // decomposition computed by Dgebrd.
 //
diff --git a/cgo/lapack_test.go b/cgo/lapack_test.go
index 28298964..ef60745a 100644
--- a/cgo/lapack_test.go
+++ b/cgo/lapack_test.go
@@ -180,6 +180,10 @@ func TestDorgqr(t *testing.T) {
 	testlapack.DorgqrTest(t, blockedTranslate{impl})
 }
 
+func TestDorgtr(t *testing.T) {
+	testlapack.DorgtrTest(t, impl)
+}
+
 func TestDorgl2(t *testing.T) {
 	testlapack.Dorgl2Test(t, blockedTranslate{impl})
 }