00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037 #ifndef TEMPLATE_LAPACK_ORGQR_HEADER
00038 #define TEMPLATE_LAPACK_ORGQR_HEADER
00039
00040 #include "template_lapack_common.h"
00041
00042 template<class Treal>
00043 int template_lapack_orgqr(
00044 const integer *m,
00045 const integer *n,
00046 const integer *k,
00047 Treal * a,
00048 const integer *lda,
00049 const Treal *tau,
00050 Treal *work,
00051 const integer *lwork,
00052 integer *info
00053 )
00054 {
00055
00056
00057
00058
00059
00060
00061
00062
00063
00064
00065
00066
00067
00068
00069
00070
00071
00072
00073
00074
00075
00076
00077
00078
00079
00080
00081
00082
00083
00084
00085
00086
00087
00088
00089
00090
00091
00092
00093
00094
00095
00096
00097
00098
00099
00100
00101
00102
00103
00104
00105
00106
00107
00108
00109
00110
00111
00112
00113
00114
00115
00116
00117
00118
00119
00120
00121
00122
00123 integer c__1 = 1;
00124 integer c_n1 = -1;
00125 integer c__3 = 3;
00126 integer c__2 = 2;
00127
00128
00129 integer a_dim1, a_offset, i__1, i__2, i__3;
00130
00131 integer i__, j, l, nbmin, iinfo;
00132 integer ib, nb, ki, kk;
00133 integer nx;
00134 integer ldwork, lwkopt;
00135 logical lquery;
00136 integer iws;
00137 #define a_ref(a_1,a_2) a[(a_2)*a_dim1 + a_1]
00138
00139
00140 a_dim1 = *lda;
00141 a_offset = 1 + a_dim1 * 1;
00142 a -= a_offset;
00143 --tau;
00144 --work;
00145
00146
00147 ki = 0;
00148
00149 *info = 0;
00150 nb = template_lapack_ilaenv(&c__1, "DORGQR", " ", m, n, k, &c_n1, (ftnlen)6, (ftnlen)1);
00151 lwkopt = maxMACRO(1,*n) * nb;
00152 work[1] = (Treal) lwkopt;
00153 lquery = *lwork == -1;
00154 if (*m < 0) {
00155 *info = -1;
00156 } else if (*n < 0 || *n > *m) {
00157 *info = -2;
00158 } else if (*k < 0 || *k > *n) {
00159 *info = -3;
00160 } else if (*lda < maxMACRO(1,*m)) {
00161 *info = -5;
00162 } else if (*lwork < maxMACRO(1,*n) && ! lquery) {
00163 *info = -8;
00164 }
00165 if (*info != 0) {
00166 i__1 = -(*info);
00167 template_blas_erbla("ORGQR ", &i__1);
00168 return 0;
00169 } else if (lquery) {
00170 return 0;
00171 }
00172
00173
00174
00175 if (*n <= 0) {
00176 work[1] = 1.;
00177 return 0;
00178 }
00179
00180 nbmin = 2;
00181 nx = 0;
00182 iws = *n;
00183 if (nb > 1 && nb < *k) {
00184
00185
00186
00187
00188 i__1 = 0, i__2 = template_lapack_ilaenv(&c__3, "DORGQR", " ", m, n, k, &c_n1, (
00189 ftnlen)6, (ftnlen)1);
00190 nx = maxMACRO(i__1,i__2);
00191 if (nx < *k) {
00192
00193
00194
00195 ldwork = *n;
00196 iws = ldwork * nb;
00197 if (*lwork < iws) {
00198
00199
00200
00201
00202 nb = *lwork / ldwork;
00203
00204 i__1 = 2, i__2 = template_lapack_ilaenv(&c__2, "DORGQR", " ", m, n, k, &c_n1,
00205 (ftnlen)6, (ftnlen)1);
00206 nbmin = maxMACRO(i__1,i__2);
00207 }
00208 }
00209 }
00210
00211 if (nb >= nbmin && nb < *k && nx < *k) {
00212
00213
00214
00215
00216 ki = (*k - nx - 1) / nb * nb;
00217
00218 i__1 = *k, i__2 = ki + nb;
00219 kk = minMACRO(i__1,i__2);
00220
00221
00222
00223 i__1 = *n;
00224 for (j = kk + 1; j <= i__1; ++j) {
00225 i__2 = kk;
00226 for (i__ = 1; i__ <= i__2; ++i__) {
00227 a_ref(i__, j) = 0.;
00228
00229 }
00230
00231 }
00232 } else {
00233 kk = 0;
00234 }
00235
00236
00237
00238 if (kk < *n) {
00239 i__1 = *m - kk;
00240 i__2 = *n - kk;
00241 i__3 = *k - kk;
00242 template_lapack_org2r(&i__1, &i__2, &i__3, &a_ref(kk + 1, kk + 1), lda, &tau[kk + 1]
00243 , &work[1], &iinfo);
00244 }
00245
00246 if (kk > 0) {
00247
00248
00249
00250 i__1 = -nb;
00251 for (i__ = ki + 1; i__1 < 0 ? i__ >= 1 : i__ <= 1; i__ += i__1) {
00252
00253 i__2 = nb, i__3 = *k - i__ + 1;
00254 ib = minMACRO(i__2,i__3);
00255 if (i__ + ib <= *n) {
00256
00257
00258
00259
00260 i__2 = *m - i__ + 1;
00261 template_lapack_larft("Forward", "Columnwise", &i__2, &ib, &a_ref(i__, i__),
00262 lda, &tau[i__], &work[1], &ldwork);
00263
00264
00265
00266 i__2 = *m - i__ + 1;
00267 i__3 = *n - i__ - ib + 1;
00268 template_lapack_larfb("Left", "No transpose", "Forward", "Columnwise", &
00269 i__2, &i__3, &ib, &a_ref(i__, i__), lda, &work[1], &
00270 ldwork, &a_ref(i__, i__ + ib), lda, &work[ib + 1], &
00271 ldwork);
00272 }
00273
00274
00275
00276 i__2 = *m - i__ + 1;
00277 template_lapack_org2r(&i__2, &ib, &ib, &a_ref(i__, i__), lda, &tau[i__], &work[
00278 1], &iinfo);
00279
00280
00281
00282 i__2 = i__ + ib - 1;
00283 for (j = i__; j <= i__2; ++j) {
00284 i__3 = i__ - 1;
00285 for (l = 1; l <= i__3; ++l) {
00286 a_ref(l, j) = 0.;
00287
00288 }
00289
00290 }
00291
00292 }
00293 }
00294
00295 work[1] = (Treal) iws;
00296 return 0;
00297
00298
00299
00300 }
00301
00302 #undef a_ref
00303
00304
00305 #endif