src/gmxlib/gmx_lapack/dgetrs.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_blas.h"
  36 #include "gmx_lapack.h"
  37
  38 void
  39 F77_FUNC(dgetrs,DGETRS)(const char *trans,
  40         int *n,
  41         int *nrhs,
  42         double *a,
  43         int *lda,
  44         int *ipiv,
  45         double *b,
  46         int *ldb,
  47         int *info)
  48 {
  49     int a_dim1, a_offset, b_dim1, b_offset;
  50     int notran;
  51     int c__1 = 1;
  52     int c_n1 = -1;
  53     double one = 1.0;
  54
  55     a_dim1 = *lda;
  56     a_offset = 1 + a_dim1;
  57     a -= a_offset;
  58     --ipiv;
  59     b_dim1 = *ldb;
  60     b_offset = 1 + b_dim1;
  61     b -= b_offset;
  62
  63     *info = 0;
  64     notran = (*trans=='N' || *trans=='n');
  65
  66     if (*n <= 0 || *nrhs <= 0)
  67         return;
  68
  69     if (notran) {
  70         F77_FUNC(dlaswp,DLASWP)(nrhs, &b[b_offset], ldb, &c__1, n, &ipiv[1], &c__1);
  71         F77_FUNC(dtrsm,DTRSM)("Left", "Lower", "No transpose", "Unit", n, nrhs, &one,
  72                &a[a_offset], lda, &b[b_offset], ldb);
  73
  74         F77_FUNC(dtrsm,DTRSM)("Left", "Upper", "No transpose", "Non-unit", n, nrhs, &one,
  75                &a[a_offset], lda, &b[b_offset], ldb);
  76     } else {
  77         F77_FUNC(dtrsm,DTRSM)("Left", "Upper", "Transpose", "Non-unit", n, nrhs, &one,
  78                &a[a_offset], lda, &b[b_offset], ldb);
  79         F77_FUNC(dtrsm,DTRSM)("Left", "Lower", "Transpose", "Unit", n, nrhs, &one,
  80                &a[a_offset], lda, &b[b_offset], ldb);
  81
  82         F77_FUNC(dlaswp,DLASWP)(nrhs, &b[b_offset], ldb, &c__1, n, &ipiv[1], &c_n1);
  83     }
  84
  85     return;
  86
  87 }