-
Notifications
You must be signed in to change notification settings - Fork 47
/
LS.f
executable file
·178 lines (159 loc) · 6.07 KB
/
LS.f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
!
! Copyright (c) Stanford University, The Regents of the University of
! California, and others.
!
! All Rights Reserved.
!
! See Copyright-SimVascular.txt for additional details.
!
! Permission is hereby granted, free of charge, to any person obtaining
! a copy of this software and associated documentation files (the
! "Software"), to deal in the Software without restriction, including
! without limitation the rights to use, copy, modify, merge, publish,
! distribute, sublicense, and/or sell copies of the Software, and to
! permit persons to whom the Software is furnished to do so, subject
! to the following conditions:
!
! The above copyright notice and this permission notice shall be included
! in all copies or substantial portions of the Software.
!
! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
! IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
! TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
! PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
! OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
! EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
! PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
! PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
! LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
! NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
! SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
!
!--------------------------------------------------------------------
!
! Subroutines related to initializing linear solver arrays and
! function calls to svFSILS and Trilinos solver library
!
!--------------------------------------------------------------------
SUBROUTINE LSALLOC(lEq)
USE COMMOD
IMPLICIT NONE
TYPE(eqType), INTENT(IN) :: lEq
IF (ALLOCATED(R)) DEALLOCATE(R)
ALLOCATE (R(dof,tnNo))
R(:,:) = 0._RKIND
IF (.NOT.lEq%assmTLS) THEN
IF (ALLOCATED(Val)) DEALLOCATE(Val)
ALLOCATE (Val(dof*dof,lhs%nnz))
Val(:,:) = 0._RKIND
END IF
#ifdef WITH_TRILINOS
IF (lEq%useTLS) THEN
IF (ALLOCATED(tls%W)) THEN
DEALLOCATE(tls%W, tls%R)
CALL TRILINOS_LHS_FREE()
END IF
ALLOCATE(tls%W(dof,tnNo), tls%R(dof,tnNo))
CALL TRILINOS_LHS_CREATE(gtnNo, lhs%mynNo, tnNo, lhs%nnz,
2 tls%ltg, ltg, rowPtr, colPtr, dof)
END IF
#endif
RETURN
END SUBROUTINE LSALLOC
!####################################################################
SUBROUTINE LSSOLVE(lEq, incL, res)
USE COMMOD
IMPLICIT NONE
TYPE(eqType), INTENT(INOUT) :: lEq
INTEGER(KIND=IKIND), INTENT(IN) :: incL(nFacesLS)
REAL(KIND=RKIND), INTENT(IN) :: res(nFacesLS)
#ifdef WITH_TRILINOS
INTEGER(KIND=IKIND) a
IF (lEq%useTLS) CALL INIT_DIR_AND_COUPNEU_BC(incL, res)
IF (lEq%assmTLS) THEN
lEq%FSILS%RI%suc = .FALSE.
CALL TRILINOS_SOLVE(tls%R, tls%W, lEq%FSILS%RI%fNorm,
2 lEq%FSILS%RI%iNorm, lEq%FSILS%RI%itr, lEq%FSILS%RI%callD,
3 lEq%FSILS%RI%dB, lEq%FSILS%RI%suc, lEq%ls%LS_type,
4 lEq%FSILS%RI%reltol, lEq%FSILS%RI%mItr, lEq%FSILS%RI%sD,
5 lEq%ls%PREC_Type, lEq%assmTLS)
ELSE IF(lEq%useTLS) THEN
CALL TRILINOS_GLOBAL_SOLVE(Val, R, tls%R, tls%W,
2 lEq%FSILS%RI%fNorm, lEq%FSILS%RI%iNorm, lEq%FSILS%RI%itr,
3 lEq%FSILS%RI%callD, lEq%FSILS%RI%dB, lEq%FSILS%RI%suc,
4 lEq%ls%LS_type, lEq%FSILS%RI%reltol, lEq%FSILS%RI%mItr,
5 lEq%FSILS%RI%sD, lEq%ls%PREC_Type)
ELSE
#endif
CALL FSILS_SOLVE(lhs, lEq%FSILS, dof, R, Val,
2 lEq%ls%PREC_Type, incL=incL, res=res)
#ifdef WITH_TRILINOS
END IF
IF (lEq%useTLS) THEN
DO a=1, tnNo
R(:,a) = tls%R(:,lhs%map(a))
END DO
END IF
#endif
RETURN
END SUBROUTINE LSSOLVE
!--------------------------------------------------------------------
SUBROUTINE INIT_DIR_AND_COUPNEU_BC(incL, res)
USE COMMOD
USE ALLFUN
IMPLICIT NONE
INTEGER(KIND=IKIND), INTENT(IN) :: incL(lhs%nFaces)
REAL(KIND=RKIND), INTENT(IN) :: res(lhs%nFaces)
REAL(KIND=RKIND), ALLOCATABLE :: v(:,:)
INTEGER(KIND=IKIND) a, i, Ac, faIn, faDof
LOGICAL flag, isCoupledBC
IF (lhs%nFaces .NE. 0) THEN
lhs%face%incFlag = .TRUE.
DO faIn=1, lhs%nFaces
IF (incL(faIn) .EQ. 0) lhs%face(faIn)%incFlag = .FALSE.
END DO
DO faIn=1, lhs%nFaces
lhs%face(faIn)%coupledFlag = .FALSE.
IF (.NOT.lhs%face(faIn)%incFlag) CYCLE
flag = lhs%face(faIn)%bGrp .EQ. BC_TYPE_Neu
IF (flag .AND. res(faIn).NE.0._RKIND) THEN
lhs%face(faIn)%res = res(faIn)
lhs%face(faIn)%coupledFlag = .TRUE.
END IF
END DO
END IF
tls%W = 1._RKIND
DO faIn=1, lhs%nFaces
IF (.NOT.lhs%face(faIn)%incFlag) CYCLE
faDof = MIN(lhs%face(faIn)%dof,dof)
IF (lhs%face(faIn)%bGrp .EQ. BC_TYPE_Dir) THEN
DO a=1, lhs%face(faIn)%nNo
Ac = lhs%face(faIn)%glob(a)
DO i=1, faDof
tls%W(i,Ac) = tls%W(i,Ac) * lhs%face(faIn)%val(i,a)
END DO
END DO
END IF
END DO
ALLOCATE(v(dof,tnNo))
v = 0._RKIND
isCoupledBC = .FALSE.
DO faIn=1, lhs%nFaces
IF (lhs%face(faIn)%coupledFlag) THEN
isCoupledBC = .TRUE.
faDof = MIN(lhs%face(faIn)%dof,dof)
DO a=1, lhs%face(faIn)%nNo
Ac = lhs%face(faIn)%glob(a)
DO i=1, faDof
v(i,Ac) = v(i,Ac) +
2 SQRT(ABS(res(faIn)))*lhs%face(faIn)%val(i,a)
END DO
END DO
END IF
END DO
#ifdef WITH_TRILINOS
CALL TRILINOS_BC_CREATE(v, isCoupledBC)
#endif
DEALLOCATE(v)
END SUBROUTINE INIT_DIR_AND_COUPNEU_BC
!####################################################################