sgejsv.f

Go to the documentation of this file.

C     ====================================================================
C                            N U L A P A C K
C                            U U L A P A C K
C                            L L L A P A C K
C                            A A A A P A C K
C                            P P P P P A C K
C                            A A A A A A C K
C                            C C C C C C C K
C                            K K K K K K K K
C
C     This file is part of NULAPACK - NUmerical Linear Algebra PACKage
C
C     Copyright (C) 2025  Saud Zahir
C
C     NULAPACK is free software: you can redistribute it and/or modify
C     it under the terms of the GNU General Public License as published by
C     the Free Software Foundation, either version 3 of the License, or
C     (at your option) any later version.
C
C     NULAPACK is distributed in the hope that it will be useful,
C     but WITHOUT ANY WARRANTY; without even the implied warranty of
C     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
C     GNU General Public License for more details.
C
C     You should have received a copy of the GNU General Public License
C     along with NULAPACK.  If not, see <https://www.gnu.org/licenses/>.
C
C     ====================================================================
C       SGEJSV  -   Jacobi Solver for A * X = B
C     ====================================================================
C       Description:
C       ------------------------------------------------------------------
C         Iterative Jacobi solver for solving linear systems of
C         equations A * X = B, where A is a square N x N matrix in
C         row-major flat array format. Single precision version.
C
C         On input:  X contains initial guess
C         On output: X contains solution
C
C         Convergence is based on maximum absolute difference per iteration.
C     ====================================================================
C       Arguments:
C       ------------------------------------------------------------------
C         N         : INTEGER       -> size of the matrix (N x N)
C         A(*)      : REAL          -> flat array, row-major matrix A
C         B(N)      : REAL          -> right-hand side vector
C         X(N)      : REAL          -> input: initial guess, output: solution
C         MAX_ITER  : INTEGER       -> max number of iterations
C         TOL       : REAL          -> convergence tolerance
C         OMEGA     : REAL          -> relaxation coefficient
C         INFO      : INTEGER       -> return code:
C                                          0 = success
C                                         >0 = did not converge
C                                         <0 = illegal or zero diagonal
C     ====================================================================
      SUBROUTINE sgejsv(N, A, B, X, MAX_ITER, TOL, OMEGA, INFO)
 
C   I m p l i c i t   T y p e s
C   ------------------------------------------------------------------
      IMPLICIT NONE
 
C   D u m m y   A r g u m e n t s
C   ------------------------------------------------------------------
      INTEGER          :: N, MAX_ITER, INFO
      REAL             :: A(*), B(N), X(N), TOL, OMEGA
 
C   L o c a l   V a r i a b l e s
C   ------------------------------------------------------------------
      INTEGER          :: I, J, K, INDEX
      REAL             :: X_NEW(N)
      REAL             :: S, DIFF, MAX_DIFF
 
C   I n i t i a l   S t a t u s
C   ------------------------------------------------------------------
      info = 1   ! Default: did not converge
 
C   M a i n   I t e r a t i o n   L o o p
C   ------------------------------------------------------------------
      DO k = 1, max_iter
         DO i = 1, n
            s = 0.0
 
C           Compute sum: S = sum_{j=1}^{N, j!=i} A(i,j) * X(j)
            DO j = 1, n
               IF (j .NE. i) THEN
                  index = (i - 1) * n + j
                  s = s + a(index) * x(j)
               END IF
            END DO
 
C           Check diagonal element A(i,i)
            index = (i - 1) * n + i
            IF (a(index) .EQ. 0.0) THEN
               info = -i
               RETURN
            END IF
 
C           Jacobi update with relaxation
            x_new(i) = (b(i) - s) / a(index)
            x_new(i) = x(i) + omega * (x_new(i) - x(i))
         END DO
 
C        C o n v e r g e n c e   C h e c k
C        ----------------------------------------------------------------
         max_diff = 0.0
         DO i = 1, n
            diff = abs(x_new(i) - x(i))
            IF (diff .GT. max_diff) max_diff = diff
            x(i) = x_new(i)
         END DO
 
         IF (max_diff .LT. tol) THEN
            info = 0  ! Success
            RETURN
         END IF
      END DO
 
C   N o n - C o n v e r g e n c e   E x i t
C   ------------------------------------------------------------------
      RETURN
      END