Actual source code: dgedi.c

  1: /*
  2:               This file creating by running f2c
  3:             linpack. this version dated 08/14/78
  4:       cleve moler, university of new mexico, argonne national lab.

  6:       Computes the inverse of a matrix given its factors and pivots
  7:     calculated by PetscLINPACKgefa(). Performed in-place for an n by n
  8:     dense matrix.

 10:        Used by the sparse factorization routines in
 11:      src/mat/impls/baij/seq

 13: */

 15: #include <petscsys.h>
 16: #include <petsc/private/kernels/blockinvert.h>

 18: PETSC_INTERN PetscErrorCode PetscLINPACKgedi(MatScalar *a, PetscInt n, PetscInt *ipvt, MatScalar *work)
 19: {
 20:   PetscInt   i__2, kb, kp1, nm1, i, j, k, l, ll, kn, knp1, jn1;
 21:   MatScalar *aa, *ax, *ay, tmp;
 22:   MatScalar  t;

 24:   PetscFunctionBegin;
 25:   --work;
 26:   --ipvt;
 27:   a -= n + 1;

 29:   /*     compute inverse(u) */

 31:   for (k = 1; k <= n; ++k) {
 32:     kn      = k * n;
 33:     knp1    = kn + k;
 34:     a[knp1] = 1.0 / a[knp1];
 35:     t       = -a[knp1];
 36:     i__2    = k - 1;
 37:     aa      = &a[1 + kn];
 38:     for (ll = 0; ll < i__2; ll++) aa[ll] *= t;
 39:     kp1 = k + 1;
 40:     if (n < kp1) continue;
 41:     ax = aa;
 42:     for (j = kp1; j <= n; ++j) {
 43:       jn1        = j * n;
 44:       t          = a[k + jn1];
 45:       a[k + jn1] = 0.;
 46:       ay         = &a[1 + jn1];
 47:       for (ll = 0; ll < k; ll++) ay[ll] += t * ax[ll];
 48:     }
 49:   }

 51:   /*    form inverse(u)*inverse(l) */

 53:   nm1 = n - 1;
 54:   if (nm1 < 1) PetscFunctionReturn(PETSC_SUCCESS);
 55:   for (kb = 1; kb <= nm1; ++kb) {
 56:     k   = n - kb;
 57:     kn  = k * n;
 58:     kp1 = k + 1;
 59:     aa  = a + kn;
 60:     for (i = kp1; i <= n; ++i) {
 61:       work[i] = aa[i];
 62:       aa[i]   = 0.;
 63:     }
 64:     for (j = kp1; j <= n; ++j) {
 65:       t  = work[j];
 66:       ax = &a[j * n + 1];
 67:       ay = &a[kn + 1];
 68:       for (ll = 0; ll < n; ll++) ay[ll] += t * ax[ll];
 69:     }
 70:     l = ipvt[k];
 71:     if (l != k) {
 72:       ax = &a[kn + 1];
 73:       ay = &a[l * n + 1];
 74:       for (ll = 0; ll < n; ll++) {
 75:         tmp    = ax[ll];
 76:         ax[ll] = ay[ll];
 77:         ay[ll] = tmp;
 78:       }
 79:     }
 80:   }
 81:   PetscFunctionReturn(PETSC_SUCCESS);
 82: }