src/gmxlib/gmx_lapack/sgebd2.c

   1 /*
   2  * This file is part of the GROMACS molecular simulation package.
   3  *
   4  * Copyright (c) 2012, by the GROMACS development team, led by
   5  * David van der Spoel, Berk Hess, Erik Lindahl, and including many
   6  * others, as listed in the AUTHORS file in the top-level source
   7  * directory and at http://www.gromacs.org.
   8  *
   9  * GROMACS is free software; you can redistribute it and/or
  10  * modify it under the terms of the GNU Lesser General Public License
  11  * as published by the Free Software Foundation; either version 2.1
  12  * of the License, or (at your option) any later version.
  13  *
  14  * GROMACS is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  17  * Lesser General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU Lesser General Public
  20  * License along with GROMACS; if not, see
  21  * http://www.gnu.org/licenses, or write to the Free Software Foundation,
  22  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
  23  *
  24  * If you want to redistribute modifications to GROMACS, please
  25  * consider that scientific software is very special. Version
  26  * control is crucial - bugs must be traceable. We will be happy to
  27  * consider code for inclusion in the official distribution, but
  28  * derived work must not be called official GROMACS. Details are found
  29  * in the README & COPYING files - if they are missing, get the
  30  * official version at http://www.gromacs.org.
  31  *
  32  * To help us fund GROMACS development, we humbly ask that you cite
  33  * the research papers on the package. Check out http://www.gromacs.org.
  34  */
  35 #include "gmx_lapack.h"
  36
  37 void
  38 F77_FUNC(sgebd2,SGEBD2)(int *m,
  39         int *n,
  40         float *a,
  41         int *lda,
  42         float *d,
  43         float *e,
  44         float *tauq,
  45         float *taup,
  46         float *work,
  47         int *info)
  48 {
  49
  50   int i,i1,i2,i3;
  51
  52     *info = 0;
  53
  54   if(*m>=*n) {
  55     /* reduce to upper bidiag. form */
  56     for(i=0;i<*n;i++) {
  57       i1 = *m - i;
  58       i2 = ( (i+1) < (*m-1)) ? (i+1) : (*m-1);
  59       i3 = 1;
  60       F77_FUNC(slarfg,SLARFG)(&i1,&(a[i*(*lda)+i]),&(a[i*(*lda)+i2]),&i3,&(tauq[i]));
  61       d[i] = a[i*(*lda)+i];
  62       a[i*(*lda)+i] = 1.0;
  63       i2 = *n - i - 1;
  64       F77_FUNC(slarf,SLARF)("L",&i1,&i2,&(a[i*(*lda)+i]),&i3,&(tauq[i]),&(a[(i+1)*(*lda)+i]),lda,work);
  65       a[i*(*lda)+i] = d[i];
  66
  67       if(i<(*n-1)) {
  68
  69         i1 = *n - i -1;
  70         i2 = ( (i+2) < (*n-1)) ? (i+2) : (*n-1);
  71         F77_FUNC(slarfg,SLARFG)(&i1,&(a[(i+1)*(*lda)+i]),&(a[i2*(*lda)+i]),lda,&(taup[i]));
  72
  73         e[i] = a[(i+1)*(*lda)+i];
  74         a[(i+1)*(*lda)+i] = 1.0;
  75
  76         i1 = *m - i - 1;
  77         i2 = *n - i - 1;
  78         F77_FUNC(slarf,SLARF)("R",&i1,&i2,&(a[(i+1)*(*lda)+i]),lda,&(taup[i]),&(a[(i+1)*(*lda)+i+1]),lda,work);
  79         a[(i+1)*(*lda)+i] = e[i];
  80       } else
  81         taup[i] = 0.0;
  82     }
  83   } else {
  84     /* reduce to lower bidiag. form */
  85     for(i=0;i<*m;i++) {
  86       i1 = *n - i;
  87       i2 = ( (i+1) < (*n-1)) ? (i+1) : (*n-1);
  88       i3 = 1;
  89       F77_FUNC(slarfg,SLARFG)(&i1,&(a[i*(*lda)+i]),&(a[i2*(*lda)+i]),lda,&(taup[i]));
  90       d[i] = a[i*(*lda)+i];
  91       a[i*(*lda)+i] = 1.0;
  92
  93       i2 = *m - i - 1;
  94       i3 = ( (i+1) < (*m-1)) ? (i+1) : (*m-1);
  95       F77_FUNC(slarf,SLARF)("R",&i2,&i1,&(a[i*(*lda)+i]),lda,&(taup[i]),&(a[(i)*(*lda)+i3]),lda,work);
  96       a[i*(*lda)+i] = d[i];
  97
  98       if(i<(*m-1)) {
  99
 100         i1 = *m - i - 1;
 101         i2 = ( (i+2) < (*m-1)) ? (i+2) : (*m-1);
 102         i3 = 1;
 103         F77_FUNC(slarfg,SLARFG)(&i1,&(a[(i)*(*lda)+i+1]),&(a[i*(*lda)+i2]),&i3,&(tauq[i]));
 104
 105         e[i] = a[(i)*(*lda)+i+1];
 106         a[(i)*(*lda)+i+1] = 1.0;
 107
 108         i1 = *m - i - 1;
 109         i2 = *n - i - 1;
 110         i3 = 1;
 111         F77_FUNC(slarf,SLARF)("L",&i1,&i2,&(a[(i)*(*lda)+i+1]),&i3,&(tauq[i]),&(a[(i+1)*(*lda)+i+1]),lda,work);
 112         a[(i)*(*lda)+i+1] = e[i];
 113       } else
 114         tauq[i] = 0.0;
 115     }
 116   }
 117   return;
 118 }