Blame view

fvn_sparse/fvn_sparse.f90 24.6 KB
b93026039   daniau   git-svn-id: https...
1
2
3
4
5
6
7
8
  module fvn_sparse
  use fvn_common
  implicit none
  
  ! Sparse solving
  interface fvn_sparse_solve
      module procedure fvn_zl_sparse_solve,fvn_zi_sparse_solve,fvn_dl_sparse_solve,fvn_di_sparse_solve
  end interface fvn_sparse_solve
2b83390c6   wdaniau   1) added fvn_spar...
9
10
11
  interface fvn_sparse_det
      module procedure fvn_zl_sparse_det,fvn_zi_sparse_det,fvn_dl_sparse_det,fvn_di_sparse_det
  end interface fvn_sparse_det
b93026039   daniau   git-svn-id: https...
12
  contains
aee8aaaf1   wdaniau   Added a function ...
13
14
15
  
  function umfpack_return_code(c)
    implicit none
14c923613   wdaniau   Return explicit u...
16
    integer(kind=sp_kind), intent(in) :: c
aee8aaaf1   wdaniau   Added a function ...
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
    character(len=80) :: umfpack_return_code
    select case(c)
      case(0)
        umfpack_return_code="UMFPACK_OK"
      case(1)
        umfpack_return_code="UMFPACK_WARNING_singular_matrix"
      case(2)
        umfpack_return_code="UMFPACK_WARNING_determinant_underflow"
      case(3)
        umfpack_return_code="UMFPACK_WARNING_determinant_overflow"
      case(-1)
        umfpack_return_code="UMFPACK_ERROR_out_of_memory"
      case(-3)
        umfpack_return_code="UMFPACK_ERROR_invalid_Numeric_object"
      case(-4)
        umfpack_return_code="UMFPACK_ERROR_invalid_Symbolic_object"
      case(-5)
        umfpack_return_code="UMFPACK_ERROR_argument_missing"
      case(-6)
        umfpack_return_code="UMFPACK_ERROR_n_nonpositive"
      case(-8)
        umfpack_return_code="UMFPACK_ERROR_invalid_matrix"
      case(-11)
        umfpack_return_code="UMFPACK_ERROR_different_pattern"
      case(-13)
        umfpack_return_code="UMFPACK_ERROR_invalid_system"
      case(-15)
        umfpack_return_code="UMFPACK_ERROR_invalid_permutation"
      case(-911)
        umfpack_return_code="UMFPACK_ERROR_internal_error"
      case(-17)
        umfpack_return_code="UMFPACK_ERROR_file_IO"
      case default
        umfpack_return_code="Unknown return code"
    end select
  end function
14c923613   wdaniau   Return explicit u...
53

b93026039   daniau   git-svn-id: https...
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
  !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
  !
  ! SPARSE RESOLUTION
  !
  !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
  ! 
  ! Sparse resolution is done by interfaçing Tim Davi's UMFPACK
  ! http://www.cise.ufl.edu/research/sparse/SuiteSparse/
  ! Used packages from SuiteSparse : AMD,UMFPACK,UFconfig
  !
  ! Solve Ax=B using UMFPACK
  !
  ! Where A is a sparse matrix given in its triplet form 
  ! T -> non zero elements
  ! Ti,Tj -> row and column index (1-based) of the given elt
  ! n : rank of matrix A
  ! nz : number of non zero elts
  !
  ! fvn_*_sparse_solve
f6bacaf83   cwaterkeyn   ChW 11/09: ANSI c...
73
74
  ! * = zl : double complex + integer(kind=dp_kind)
  ! * = zi : double complex + integer(kind=sp_kind)
b93026039   daniau   git-svn-id: https...
75
  !
2b83390c6   wdaniau   1) added fvn_spar...
76
  subroutine fvn_zl_sparse_solve(n,nz,T,Ti,Tj,B,x,status,det)
b93026039   daniau   git-svn-id: https...
77
  implicit none
f6bacaf83   cwaterkeyn   ChW 11/09: ANSI c...
78
79
80
81
82
83
  integer(kind=dp_kind), intent(in) :: n,nz
  complex(kind=dp_kind),dimension(nz),intent(in) :: T
  integer(kind=dp_kind),dimension(nz),intent(in)  :: Ti,Tj
  complex(kind=dp_kind),dimension(n),intent(in) :: B
  complex(kind=dp_kind),dimension(n),intent(out) :: x
  integer(kind=dp_kind), intent(out) :: status
c64a05f8a   wdaniau   Sparse Determinan...
84
  real(kind=dp_kind), dimension(3), optional, intent(out) :: det
f6bacaf83   cwaterkeyn   ChW 11/09: ANSI c...
85
86
87
88
89
90
91
92
93
94
  
  integer(kind=dp_kind),dimension(:),allocatable :: wTi,wTj
  real(kind=dp_kind),dimension(:),allocatable :: Tx,Tz
  real(kind=dp_kind),dimension(:),allocatable :: Ax,Az
  integer(kind=dp_kind),dimension(:),allocatable :: Ap,Ai
  integer(kind=dp_kind) :: symbolic,numeric
  real(kind=dp_kind),dimension(:),allocatable :: xx,xz,bx,bz
  real(kind=dp_kind),dimension(90) :: info
  real(kind=dp_kind),dimension(20) :: control
  integer(kind=dp_kind) :: sys
b93026039   daniau   git-svn-id: https...
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
  
  
  status=0
  
  ! we use a working copy of Ti and Tj to perform 1-based to 0-based translation
  ! Tx and Tz are the real and imaginary parts of T
  allocate(wTi(nz),wTj(nz))
  allocate(Tx(nz),Tz(nz))
  Tx=dble(T)
  Tz=aimag(T)
  wTi=Ti-1
  wTj=Tj-1
  allocate(Ax(nz),Az(nz))
  allocate(Ap(n+1),Ai(nz))
  
  ! perform the triplet to compressed column form -> Ap,Ai,Ax,Az
  call umfpack_zl_triplet_to_col(n,n,nz,wTi,wTj,Tx,Tz,Ap,Ai,Ax,Az,status)
  ! if status is not zero a problem has occured
  if (status /= 0) then
7dc628d15   wdaniau   Added error outpu...
114
      write(*,*) "Problem during umfpack_zl_triplet_to_col : ",trim(umfpack_return_code(int(status,kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
115
116
117
118
119
120
121
122
123
  endif
  
  ! Define defaults control values
  call umfpack_zl_defaults(control)
  
  ! Symbolic analysis
  call umfpack_zl_symbolic(n,n,Ap,Ai,Ax,Az,symbolic, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
124
      write(*,*) "Problem during symbolic analysis : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
125
126
127
128
129
130
131
      status=info(1)
  endif
  
  ! Numerical factorization
  call umfpack_zl_numeric (Ap, Ai, Ax, Az, symbolic, numeric, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
132
      write(*,*) "Problem during numerical factorization : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
133
134
135
136
137
      status=info(1)
  endif
  
  ! free the C symbolic pointer
  call umfpack_zl_free_symbolic (symbolic)
2b83390c6   wdaniau   1) added fvn_spar...
138
139
  ! if parameter det is present, the determinant of the matrix is calculated
  if (present(det) ) then
c64a05f8a   wdaniau   Sparse Determinan...
140
    call umfpack_zl_get_determinant(det(1),det(2),det(3),numeric,info,status)  
2b83390c6   wdaniau   1) added fvn_spar...
141
142
    ! info(1) should be zero
    if (info(1) /= 0) then
ba393a269   wdaniau   1) updated docume...
143
        if ( (info(1) < 1) .or. (info(1) >3) ) then ! not a warning
7dc628d15   wdaniau   Added error outpu...
144
            write(*,*) "Problem during sparse determinant : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
ba393a269   wdaniau   1) updated docume...
145
        endif
2b83390c6   wdaniau   1) added fvn_spar...
146
147
148
        status=info(1)
    endif
  endif
b93026039   daniau   git-svn-id: https...
149
150
151
152
153
154
155
156
157
158
  allocate(bx(n),bz(n),xx(n),xz(n))
  bx=dble(B)
  bz=aimag(B)
  sys=0
  ! sys may be used to define type of solving -> see umfpack.h
  
  ! Solving
  call umfpack_zl_solve (sys, Ap, Ai, Ax,Az, xx,xz, bx,bz, numeric, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
159
      write(*,*) "Problem during solving : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
160
161
162
163
164
165
      status=info(1)
  endif
  
  
  ! free the C numeric pointer
  call umfpack_zl_free_numeric (numeric)
f6bacaf83   cwaterkeyn   ChW 11/09: ANSI c...
166
  x=cmplx(xx,xz,dp_kind)
b93026039   daniau   git-svn-id: https...
167
168
169
170
171
172
  
  deallocate(bx,bz,xx,xz)
  deallocate(Ax,Az)
  deallocate(Tx,Tz)
  deallocate(wTi,wTj)
  end subroutine
2b83390c6   wdaniau   1) added fvn_spar...
173
  subroutine fvn_zi_sparse_solve(n,nz,T,Ti,Tj,B,x,status,det)
b93026039   daniau   git-svn-id: https...
174
  implicit none
f6bacaf83   cwaterkeyn   ChW 11/09: ANSI c...
175
176
177
178
179
180
  integer(kind=sp_kind), intent(in) :: n,nz
  complex(kind=dp_kind),dimension(nz),intent(in) :: T
  integer(kind=sp_kind),dimension(nz),intent(in)  :: Ti,Tj
  complex(kind=dp_kind),dimension(n),intent(in) :: B
  complex(kind=dp_kind),dimension(n),intent(out) :: x
  integer(kind=sp_kind), intent(out) :: status
c64a05f8a   wdaniau   Sparse Determinan...
181
  real(kind=dp_kind), dimension(3), optional, intent(out) :: det
f6bacaf83   cwaterkeyn   ChW 11/09: ANSI c...
182
183
184
185
186
187
188
  
  integer(kind=sp_kind),dimension(:),allocatable :: wTi,wTj
  real(kind=dp_kind),dimension(:),allocatable :: Tx,Tz
  real(kind=dp_kind),dimension(:),allocatable :: Ax,Az
  integer(kind=sp_kind),dimension(:),allocatable :: Ap,Ai
  !integer(kind=dp_kind) :: symbolic,numeric
  integer(kind=sp_kind),dimension(2) :: symbolic,numeric
b93026039   daniau   git-svn-id: https...
189
  ! As symbolic and numeric are used to store a C pointer, it is necessary to
f6bacaf83   cwaterkeyn   ChW 11/09: ANSI c...
190
191
192
193
194
195
  ! still use an integer(kind=dp_kind) for 64bits machines
  ! An other possibility : integer(kind=sp_kind),dimension(2) :: symbolic,numeric
  real(kind=dp_kind),dimension(:),allocatable :: xx,xz,bx,bz
  real(kind=dp_kind),dimension(90) :: info
  real(kind=dp_kind),dimension(20) :: control
  integer(kind=sp_kind) :: sys
b93026039   daniau   git-svn-id: https...
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
  
  status=0
  ! we use a working copy of Ti and Tj to perform 1-based to 0-based translation
  ! Tx and Tz are the real and imaginary parts of T
  allocate(wTi(nz),wTj(nz))
  allocate(Tx(nz),Tz(nz))
  Tx=dble(T)
  Tz=aimag(T)
  wTi=Ti-1
  wTj=Tj-1
  allocate(Ax(nz),Az(nz))
  allocate(Ap(n+1),Ai(nz))
  
  ! perform the triplet to compressed column form -> Ap,Ai,Ax,Az
  call umfpack_zi_triplet_to_col(n,n,nz,wTi,wTj,Tx,Tz,Ap,Ai,Ax,Az,status)
  ! if status is not zero a problem has occured
  if (status /= 0) then
7dc628d15   wdaniau   Added error outpu...
213
      write(*,*) "Problem during umfpack_zl_triplet_to_col : ",trim(umfpack_return_code(status))
b93026039   daniau   git-svn-id: https...
214
215
216
217
218
219
220
221
222
  endif
  
  ! Define defaults control values
  call umfpack_zi_defaults(control)
  
  ! Symbolic analysis
  call umfpack_zi_symbolic(n,n,Ap,Ai,Ax,Az,symbolic, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
223
      write(*,*) "Problem during symbolic analysis : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
224
225
226
227
228
229
230
      status=info(1)
  endif
  
  ! Numerical factorization
  call umfpack_zi_numeric (Ap, Ai, Ax, Az, symbolic, numeric, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
231
      write(*,*) "Problem during numerical factorization : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
232
233
234
235
236
      status=info(1)
  endif
  
  ! free the C symbolic pointer
  call umfpack_zi_free_symbolic (symbolic)
2b83390c6   wdaniau   1) added fvn_spar...
237
238
  ! if parameter det is present, the determinant of the matrix is calculated
  if (present(det) ) then
c64a05f8a   wdaniau   Sparse Determinan...
239
    call umfpack_zi_get_determinant(det(1),det(2),det(3),numeric,info,status)
2b83390c6   wdaniau   1) added fvn_spar...
240
241
    ! info(1) should be zero
    if (info(1) /= 0) then
ba393a269   wdaniau   1) updated docume...
242
        if ( (info(1) < 1) .or. (info(1) >3) ) then ! not a warning
7dc628d15   wdaniau   Added error outpu...
243
            write(*,*) "Problem during sparse determinant : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
ba393a269   wdaniau   1) updated docume...
244
        endif
2b83390c6   wdaniau   1) added fvn_spar...
245
246
247
        status=info(1)
    endif
  endif
b93026039   daniau   git-svn-id: https...
248
249
250
251
252
253
254
255
256
257
  allocate(bx(n),bz(n),xx(n),xz(n))
  bx=dble(B)
  bz=aimag(B)
  sys=0
  ! sys may be used to define type of solving -> see umfpack.h
  
  ! Solving
  call umfpack_zi_solve (sys, Ap, Ai, Ax,Az, xx,xz, bx,bz, numeric, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
258
      write(*,*) "Problem during solving : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
259
260
261
262
263
      status=info(1)
  endif
  
  ! free the C numeric pointer
  call umfpack_zi_free_numeric (numeric)
f6bacaf83   cwaterkeyn   ChW 11/09: ANSI c...
264
  x=cmplx(xx,xz,dp_kind)
b93026039   daniau   git-svn-id: https...
265
266
267
268
269
270
  
  deallocate(bx,bz,xx,xz)
  deallocate(Ax,Az)
  deallocate(Tx,Tz)
  deallocate(wTi,wTj)
  end subroutine
2b83390c6   wdaniau   1) added fvn_spar...
271
  subroutine fvn_dl_sparse_solve(n,nz,T,Ti,Tj,B,x,status,det)
b93026039   daniau   git-svn-id: https...
272
  implicit none
f6bacaf83   cwaterkeyn   ChW 11/09: ANSI c...
273
274
275
276
277
278
  integer(kind=dp_kind), intent(in) :: n,nz
  real(kind=dp_kind),dimension(nz),intent(in) :: T
  integer(kind=dp_kind),dimension(nz),intent(in)  :: Ti,Tj
  real(kind=dp_kind),dimension(n),intent(in) :: B
  real(kind=dp_kind),dimension(n),intent(out) :: x
  integer(kind=dp_kind), intent(out) :: status
c64a05f8a   wdaniau   Sparse Determinan...
279
  real(kind=dp_kind), dimension(2), optional, intent(out) :: det
f6bacaf83   cwaterkeyn   ChW 11/09: ANSI c...
280
281
282
283
284
285
286
287
288
  
  integer(kind=dp_kind),dimension(:),allocatable :: wTi,wTj
  real(kind=dp_kind),dimension(:),allocatable :: A
  integer(kind=dp_kind),dimension(:),allocatable :: Ap,Ai
  !integer(kind=dp_kind) :: symbolic,numeric
  integer(kind=dp_kind) :: symbolic,numeric
  real(kind=dp_kind),dimension(90) :: info
  real(kind=dp_kind),dimension(20) :: control
  integer(kind=dp_kind) :: sys
b93026039   daniau   git-svn-id: https...
289
290
291
292
293
294
295
296
297
298
299
300
301
  
  status=0
  ! we use a working copy of Ti and Tj to perform 1-based to 0-based translation
  allocate(wTi(nz),wTj(nz))
  wTi=Ti-1
  wTj=Tj-1
  allocate(A(nz))
  allocate(Ap(n+1),Ai(nz))
  
  ! perform the triplet to compressed column form -> Ap,Ai,Ax,Az
  call umfpack_dl_triplet_to_col(n,n,nz,wTi,wTj,T,Ap,Ai,A,status)
  ! if status is not zero a problem has occured
  if (status /= 0) then
7dc628d15   wdaniau   Added error outpu...
302
      write(*,*) "Problem during umfpack_dl_triplet_to_col : ",trim(umfpack_return_code(int(status,kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
303
304
305
306
307
308
309
310
311
  endif
  
  ! Define defaults control values
  call umfpack_dl_defaults(control)
  
  ! Symbolic analysis
  call umfpack_dl_symbolic(n,n,Ap,Ai,A,symbolic, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
312
      write(*,*) "Problem during symbolic analysis : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
313
314
315
316
317
318
319
      status=info(1)
  endif
  
  ! Numerical factorization
  call umfpack_dl_numeric (Ap, Ai, A, symbolic, numeric, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
320
      write(*,*) "Problem during numerical factorization : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
321
322
323
324
325
      status=info(1)
  endif
  
  ! free the C symbolic pointer
  call umfpack_dl_free_symbolic (symbolic)
2b83390c6   wdaniau   1) added fvn_spar...
326
327
  ! if parameter det is present, the determinant of the matrix is calculated
  if (present(det) ) then
c64a05f8a   wdaniau   Sparse Determinan...
328
    call umfpack_dl_get_determinant(det(1),det(2),numeric,info,status)
2b83390c6   wdaniau   1) added fvn_spar...
329
330
    ! info(1) should be zero
    if (info(1) /= 0) then
ba393a269   wdaniau   1) updated docume...
331
        if ( (info(1) < 1) .or. (info(1) >3) ) then ! not a warning
7dc628d15   wdaniau   Added error outpu...
332
            write(*,*) "Problem during sparse determinant : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
ba393a269   wdaniau   1) updated docume...
333
        endif
2b83390c6   wdaniau   1) added fvn_spar...
334
335
336
        status=info(1)
    endif
  endif
b93026039   daniau   git-svn-id: https...
337
338
339
340
341
342
343
  sys=0
  ! sys may be used to define type of solving -> see umfpack.h
  
  ! Solving
  call umfpack_dl_solve (sys, Ap, Ai, A, x, B, numeric, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
344
      write(*,*) "Problem during solving : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
345
346
347
348
349
350
351
352
353
      status=info(1)
  endif
  
  ! free the C numeric pointer
  call umfpack_dl_free_numeric (numeric)
  
  deallocate(A)
  deallocate(wTi,wTj)
  end subroutine
2b83390c6   wdaniau   1) added fvn_spar...
354
  subroutine fvn_di_sparse_solve(n,nz,T,Ti,Tj,B,x,status,det)
b93026039   daniau   git-svn-id: https...
355
  implicit none
f6bacaf83   cwaterkeyn   ChW 11/09: ANSI c...
356
357
358
359
360
361
  integer(kind=sp_kind), intent(in) :: n,nz
  real(kind=dp_kind),dimension(nz),intent(in) :: T
  integer(kind=sp_kind),dimension(nz),intent(in)  :: Ti,Tj
  real(kind=dp_kind),dimension(n),intent(in) :: B
  real(kind=dp_kind),dimension(n),intent(out) :: x
  integer(kind=sp_kind), intent(out) :: status
c64a05f8a   wdaniau   Sparse Determinan...
362
  real(kind=dp_kind), dimension(2), optional, intent(out) :: det
f6bacaf83   cwaterkeyn   ChW 11/09: ANSI c...
363
364
365
366
367
368
  
  integer(kind=sp_kind),dimension(:),allocatable :: wTi,wTj
  real(kind=dp_kind),dimension(:),allocatable :: A
  integer(kind=sp_kind),dimension(:),allocatable :: Ap,Ai
  !integer(kind=dp_kind) :: symbolic,numeric
  integer(kind=sp_kind),dimension(2) :: symbolic,numeric
b93026039   daniau   git-svn-id: https...
369
  ! As symbolic and numeric are used to store a C pointer, it is necessary to
f6bacaf83   cwaterkeyn   ChW 11/09: ANSI c...
370
371
372
373
374
  ! still use an integer(kind=dp_kind) for 64bits machines
  ! An other possibility : integer(kind=sp_kind),dimension(2) :: symbolic,numeric
  real(kind=dp_kind),dimension(90) :: info
  real(kind=dp_kind),dimension(20) :: control
  integer(kind=sp_kind) :: sys
b93026039   daniau   git-svn-id: https...
375
376
377
378
379
380
381
382
383
384
385
386
387
  
  status=0
  ! we use a working copy of Ti and Tj to perform 1-based to 0-based translation
  allocate(wTi(nz),wTj(nz))
  wTi=Ti-1
  wTj=Tj-1
  allocate(A(nz))
  allocate(Ap(n+1),Ai(nz))
  
  ! perform the triplet to compressed column form -> Ap,Ai,Ax,Az
  call umfpack_di_triplet_to_col(n,n,nz,wTi,wTj,T,Ap,Ai,A,status)
  ! if status is not zero a problem has occured
  if (status /= 0) then
7dc628d15   wdaniau   Added error outpu...
388
      write(*,*) "Problem during umfpack_di_triplet_to_col : ",trim(umfpack_return_code(status))
b93026039   daniau   git-svn-id: https...
389
390
391
392
393
394
395
396
397
  endif
  
  ! Define defaults control values
  call umfpack_di_defaults(control)
  
  ! Symbolic analysis
  call umfpack_di_symbolic(n,n,Ap,Ai,A,symbolic, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
398
      write(*,*) "Problem during symbolic analysis : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
399
400
401
402
403
404
405
      status=info(1)
  endif
  
  ! Numerical factorization
  call umfpack_di_numeric (Ap, Ai, A, symbolic, numeric, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
406
      write(*,*) "Problem during numerical factorization : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
407
408
409
410
411
      status=info(1)
  endif
  
  ! free the C symbolic pointer
  call umfpack_di_free_symbolic (symbolic)
2b83390c6   wdaniau   1) added fvn_spar...
412
413
  ! if parameter det is present, the determinant of the matrix is calculated
  if (present(det) ) then
c64a05f8a   wdaniau   Sparse Determinan...
414
    call umfpack_di_get_determinant(det(1),det(2),numeric,info,status)  
2b83390c6   wdaniau   1) added fvn_spar...
415
416
    ! info(1) should be zero
    if (info(1) /= 0) then
ba393a269   wdaniau   1) updated docume...
417
        if ( (info(1) < 1) .or. (info(1) >3) ) then ! not a warning
7dc628d15   wdaniau   Added error outpu...
418
            write(*,*) "Problem during sparse determinant : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
ba393a269   wdaniau   1) updated docume...
419
        endif
2b83390c6   wdaniau   1) added fvn_spar...
420
421
422
        status=info(1)
    endif
  endif
b93026039   daniau   git-svn-id: https...
423
424
  sys=0
  ! sys may be used to define type of solving -> see umfpack.h
b93026039   daniau   git-svn-id: https...
425
426
427
428
  ! Solving
  call umfpack_di_solve (sys, Ap, Ai, A, x, B, numeric, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
429
      write(*,*) "Problem during solving : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
b93026039   daniau   git-svn-id: https...
430
431
432
433
434
435
436
437
438
      status=info(1)
  endif
  
  ! free the C numeric pointer
  call umfpack_di_free_numeric (numeric)
  
  deallocate(A)
  deallocate(wTi,wTj)
  end subroutine
2b83390c6   wdaniau   1) added fvn_spar...
439
440
441
442
443
444
445
446
447
448
  !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
  !
  ! SPARSE DETERMINANT
  !
  !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
  subroutine fvn_zl_sparse_det(n,nz,T,Ti,Tj,det,status)
  implicit none
  integer(kind=dp_kind), intent(in) :: n,nz
  complex(kind=dp_kind),dimension(nz),intent(in) :: T
  integer(kind=dp_kind),dimension(nz),intent(in)  :: Ti,Tj
c64a05f8a   wdaniau   Sparse Determinan...
449
  real(kind=dp_kind), dimension(3), intent(out) :: det
2b83390c6   wdaniau   1) added fvn_spar...
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
  integer(kind=dp_kind), intent(out) :: status
  
  integer(kind=dp_kind),dimension(:),allocatable :: wTi,wTj
  real(kind=dp_kind),dimension(:),allocatable :: Tx,Tz
  real(kind=dp_kind),dimension(:),allocatable :: Ax,Az
  integer(kind=dp_kind),dimension(:),allocatable :: Ap,Ai
  integer(kind=dp_kind) :: symbolic,numeric
  real(kind=dp_kind),dimension(90) :: info
  real(kind=dp_kind),dimension(20) :: control
  real(kind=dp_kind) :: Mx,Mz,Ex
  
  
  status=0
  
  ! we use a working copy of Ti and Tj to perform 1-based to 0-based translation
  ! Tx and Tz are the real and imaginary parts of T
  allocate(wTi(nz),wTj(nz))
  allocate(Tx(nz),Tz(nz))
  Tx=dble(T)
  Tz=aimag(T)
  wTi=Ti-1
  wTj=Tj-1
  allocate(Ax(nz),Az(nz))
  allocate(Ap(n+1),Ai(nz))
  
  ! perform the triplet to compressed column form -> Ap,Ai,Ax,Az
  call umfpack_zl_triplet_to_col(n,n,nz,wTi,wTj,Tx,Tz,Ap,Ai,Ax,Az,status)
  ! if status is not zero a problem has occured
  if (status /= 0) then
7dc628d15   wdaniau   Added error outpu...
479
      write(*,*) "Problem during umfpack_zl_triplet_to_col ",trim(umfpack_return_code(int(status,kind=sp_kind)))
2b83390c6   wdaniau   1) added fvn_spar...
480
481
482
483
484
485
486
487
488
  endif
  
  ! Define defaults control values
  call umfpack_zl_defaults(control)
  
  ! Symbolic analysis
  call umfpack_zl_symbolic(n,n,Ap,Ai,Ax,Az,symbolic, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
489
      write(*,*) "Problem during symbolic analysis : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
2b83390c6   wdaniau   1) added fvn_spar...
490
491
492
493
494
495
496
      status=info(1)
  endif
  
  ! Numerical factorization
  call umfpack_zl_numeric (Ap, Ai, Ax, Az, symbolic, numeric, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
497
      write(*,*) "Problem during numerical factorization : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
2b83390c6   wdaniau   1) added fvn_spar...
498
499
500
501
502
      status=info(1)
  endif
  
  ! free the C symbolic pointer
  call umfpack_zl_free_symbolic (symbolic)
c64a05f8a   wdaniau   Sparse Determinan...
503
  call umfpack_zl_get_determinant(det(1),det(2),det(3),numeric,info,status)
2b83390c6   wdaniau   1) added fvn_spar...
504
505
  ! info(1) should be zero
  if (info(1) /= 0) then
ba393a269   wdaniau   1) updated docume...
506
        if ( (info(1) < 1) .or. (info(1) >3) ) then ! not a warning
7dc628d15   wdaniau   Added error outpu...
507
            write(*,*) "Problem during sparse determinant : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
ba393a269   wdaniau   1) updated docume...
508
        endif
2b83390c6   wdaniau   1) added fvn_spar...
509
510
      status=info(1)
  endif
2b83390c6   wdaniau   1) added fvn_spar...
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
  
  ! free the C numeric pointer
  call umfpack_zl_free_numeric (numeric)
  
  deallocate(Ax,Az)
  deallocate(Tx,Tz)
  deallocate(wTi,wTj)
  end subroutine
  
  
  subroutine fvn_zi_sparse_det(n,nz,T,Ti,Tj,det,status)
  implicit none
  integer(kind=sp_kind), intent(in) :: n,nz
  complex(kind=dp_kind),dimension(nz),intent(in) :: T
  integer(kind=sp_kind),dimension(nz),intent(in)  :: Ti,Tj
  integer(kind=sp_kind), intent(out) :: status
c64a05f8a   wdaniau   Sparse Determinan...
527
  real(kind=dp_kind), dimension(3), intent(out) :: det
2b83390c6   wdaniau   1) added fvn_spar...
528
529
530
531
532
533
534
535
536
537
538
539
  
  integer(kind=sp_kind),dimension(:),allocatable :: wTi,wTj
  real(kind=dp_kind),dimension(:),allocatable :: Tx,Tz
  real(kind=dp_kind),dimension(:),allocatable :: Ax,Az
  integer(kind=sp_kind),dimension(:),allocatable :: Ap,Ai
  !integer(kind=dp_kind) :: symbolic,numeric
  integer(kind=sp_kind),dimension(2) :: symbolic,numeric
  ! As symbolic and numeric are used to store a C pointer, it is necessary to
  ! still use an integer(kind=dp_kind) for 64bits machines
  ! An other possibility : integer(kind=sp_kind),dimension(2) :: symbolic,numeric
  real(kind=dp_kind),dimension(90) :: info
  real(kind=dp_kind),dimension(20) :: control
2b83390c6   wdaniau   1) added fvn_spar...
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
  
  status=0
  ! we use a working copy of Ti and Tj to perform 1-based to 0-based translation
  ! Tx and Tz are the real and imaginary parts of T
  allocate(wTi(nz),wTj(nz))
  allocate(Tx(nz),Tz(nz))
  Tx=dble(T)
  Tz=aimag(T)
  wTi=Ti-1
  wTj=Tj-1
  allocate(Ax(nz),Az(nz))
  allocate(Ap(n+1),Ai(nz))
  
  ! perform the triplet to compressed column form -> Ap,Ai,Ax,Az
  call umfpack_zi_triplet_to_col(n,n,nz,wTi,wTj,Tx,Tz,Ap,Ai,Ax,Az,status)
  ! if status is not zero a problem has occured
  if (status /= 0) then
7dc628d15   wdaniau   Added error outpu...
557
      write(*,*) "Problem during umfpack_zl_triplet_to_col : ",trim(umfpack_return_code(status))
2b83390c6   wdaniau   1) added fvn_spar...
558
559
560
561
562
563
564
565
566
  endif
  
  ! Define defaults control values
  call umfpack_zi_defaults(control)
  
  ! Symbolic analysis
  call umfpack_zi_symbolic(n,n,Ap,Ai,Ax,Az,symbolic, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
567
      write(*,*) "Problem during symbolic analysis : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
2b83390c6   wdaniau   1) added fvn_spar...
568
569
570
571
572
573
574
      status=info(1)
  endif
  
  ! Numerical factorization
  call umfpack_zi_numeric (Ap, Ai, Ax, Az, symbolic, numeric, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
575
      write(*,*) "Problem during numerical factorization : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
2b83390c6   wdaniau   1) added fvn_spar...
576
577
578
579
580
      status=info(1)
  endif
  
  ! free the C symbolic pointer
  call umfpack_zi_free_symbolic (symbolic)
c64a05f8a   wdaniau   Sparse Determinan...
581
  call umfpack_zi_get_determinant(det(1),det(2),det(3),numeric,info,status)
2b83390c6   wdaniau   1) added fvn_spar...
582
583
  ! info(1) should be zero
  if (info(1) /= 0) then
ba393a269   wdaniau   1) updated docume...
584
        if ( (info(1) < 1) .or. (info(1) >3) ) then ! not a warning
7dc628d15   wdaniau   Added error outpu...
585
            write(*,*) "Problem during sparse determinant : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
ba393a269   wdaniau   1) updated docume...
586
        endif
2b83390c6   wdaniau   1) added fvn_spar...
587
588
      status=info(1)
  endif
2b83390c6   wdaniau   1) added fvn_spar...
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
  
  ! free the C numeric pointer
  call umfpack_zi_free_numeric (numeric)
  
  deallocate(Ax,Az)
  deallocate(Tx,Tz)
  deallocate(wTi,wTj)
  end subroutine
  
  subroutine fvn_dl_sparse_det(n,nz,T,Ti,Tj,det,status)
  implicit none
  integer(kind=dp_kind), intent(in) :: n,nz
  real(kind=dp_kind),dimension(nz),intent(in) :: T
  integer(kind=dp_kind),dimension(nz),intent(in)  :: Ti,Tj
  integer(kind=dp_kind), intent(out) :: status
c64a05f8a   wdaniau   Sparse Determinan...
604
  real(kind=dp_kind), dimension(2), intent(out) :: det
2b83390c6   wdaniau   1) added fvn_spar...
605
606
607
608
609
610
611
612
  
  integer(kind=dp_kind),dimension(:),allocatable :: wTi,wTj
  real(kind=dp_kind),dimension(:),allocatable :: A
  integer(kind=dp_kind),dimension(:),allocatable :: Ap,Ai
  !integer(kind=dp_kind) :: symbolic,numeric
  integer(kind=dp_kind) :: symbolic,numeric
  real(kind=dp_kind),dimension(90) :: info
  real(kind=dp_kind),dimension(20) :: control
2b83390c6   wdaniau   1) added fvn_spar...
613
614
615
616
617
618
619
620
621
622
623
624
625
  
  status=0
  ! we use a working copy of Ti and Tj to perform 1-based to 0-based translation
  allocate(wTi(nz),wTj(nz))
  wTi=Ti-1
  wTj=Tj-1
  allocate(A(nz))
  allocate(Ap(n+1),Ai(nz))
  
  ! perform the triplet to compressed column form -> Ap,Ai,Ax,Az
  call umfpack_dl_triplet_to_col(n,n,nz,wTi,wTj,T,Ap,Ai,A,status)
  ! if status is not zero a problem has occured
  if (status /= 0) then
7dc628d15   wdaniau   Added error outpu...
626
      write(*,*) "Problem during umfpack_dl_triplet_to_col : ",trim(umfpack_return_code(int(status,kind=sp_kind)))
2b83390c6   wdaniau   1) added fvn_spar...
627
628
629
630
631
632
633
634
635
  endif
  
  ! Define defaults control values
  call umfpack_dl_defaults(control)
  
  ! Symbolic analysis
  call umfpack_dl_symbolic(n,n,Ap,Ai,A,symbolic, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
636
      write(*,*) "Problem during symbolic analysis : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
2b83390c6   wdaniau   1) added fvn_spar...
637
638
639
640
641
642
643
      status=info(1)
  endif
  
  ! Numerical factorization
  call umfpack_dl_numeric (Ap, Ai, A, symbolic, numeric, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
644
      write(*,*) "Problem during numerical factorization : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
2b83390c6   wdaniau   1) added fvn_spar...
645
646
647
648
649
      status=info(1)
  endif
  
  ! free the C symbolic pointer
  call umfpack_dl_free_symbolic (symbolic)
c64a05f8a   wdaniau   Sparse Determinan...
650
  call umfpack_dl_get_determinant(det(1),det(2),numeric,info,status)
2b83390c6   wdaniau   1) added fvn_spar...
651
652
  ! info(1) should be zero
  if (info(1) /= 0) then
ba393a269   wdaniau   1) updated docume...
653
        if ( (info(1) < 1) .or. (info(1) >3) ) then ! not a warning
7dc628d15   wdaniau   Added error outpu...
654
            write(*,*) "Problem during sparse determinant : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
ba393a269   wdaniau   1) updated docume...
655
        endif
2b83390c6   wdaniau   1) added fvn_spar...
656
657
      status=info(1)
  endif
2b83390c6   wdaniau   1) added fvn_spar...
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
  
  ! free the C numeric pointer
  call umfpack_dl_free_numeric (numeric)
  
  deallocate(A)
  deallocate(wTi,wTj)
  end subroutine
  
  
  subroutine fvn_di_sparse_det(n,nz,T,Ti,Tj,det,status)
  implicit none
  integer(kind=sp_kind), intent(in) :: n,nz
  real(kind=dp_kind),dimension(nz),intent(in) :: T
  integer(kind=sp_kind),dimension(nz),intent(in)  :: Ti,Tj
  integer(kind=sp_kind), intent(out) :: status
c64a05f8a   wdaniau   Sparse Determinan...
673
  real(kind=dp_kind), dimension(2), intent(out) :: det
2b83390c6   wdaniau   1) added fvn_spar...
674
675
676
677
678
679
680
681
682
683
684
685
  
  integer(kind=sp_kind),dimension(:),allocatable :: wTi,wTj
  real(kind=dp_kind),dimension(:),allocatable :: A
  integer(kind=sp_kind),dimension(:),allocatable :: Ap,Ai
  !integer(kind=dp_kind) :: symbolic,numeric
  integer(kind=sp_kind),dimension(2) :: symbolic,numeric
  ! As symbolic and numeric are used to store a C pointer, it is necessary to
  ! still use an integer(kind=dp_kind) for 64bits machines
  ! An other possibility : integer(kind=sp_kind),dimension(2) :: symbolic,numeric
  real(kind=dp_kind),dimension(90) :: info
  real(kind=dp_kind),dimension(20) :: control
  integer(kind=sp_kind) :: sys
2b83390c6   wdaniau   1) added fvn_spar...
686
687
688
689
690
691
692
693
694
695
696
697
698
699
  
  
  status=0
  ! we use a working copy of Ti and Tj to perform 1-based to 0-based translation
  allocate(wTi(nz),wTj(nz))
  wTi=Ti-1
  wTj=Tj-1
  allocate(A(nz))
  allocate(Ap(n+1),Ai(nz))
  
  ! perform the triplet to compressed column form -> Ap,Ai,Ax,Az
  call umfpack_di_triplet_to_col(n,n,nz,wTi,wTj,T,Ap,Ai,A,status)
  ! if status is not zero a problem has occured
  if (status /= 0) then
7dc628d15   wdaniau   Added error outpu...
700
      write(*,*) "Problem during umfpack_di_triplet_to_col : ",trim(umfpack_return_code(status))
2b83390c6   wdaniau   1) added fvn_spar...
701
702
703
704
705
706
707
708
709
  endif
  
  ! Define defaults control values
  call umfpack_di_defaults(control)
  
  ! Symbolic analysis
  call umfpack_di_symbolic(n,n,Ap,Ai,A,symbolic, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
710
      write(*,*) "Problem during symbolic analysis : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
2b83390c6   wdaniau   1) added fvn_spar...
711
712
713
714
715
716
717
      status=info(1)
  endif
  
  ! Numerical factorization
  call umfpack_di_numeric (Ap, Ai, A, symbolic, numeric, control, info)
  ! info(1) should be zero
  if (info(1) /= 0) then
7dc628d15   wdaniau   Added error outpu...
718
      write(*,*) "Problem during numerical factorization : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
2b83390c6   wdaniau   1) added fvn_spar...
719
720
721
722
723
724
      status=info(1)
  endif
  
  
  ! free the C symbolic pointer
  call umfpack_di_free_symbolic (symbolic)
c64a05f8a   wdaniau   Sparse Determinan...
725
  call umfpack_di_get_determinant(det(1),det(2),numeric,info,status)
2b83390c6   wdaniau   1) added fvn_spar...
726
727
  ! info(1) should be zero
  if (info(1) /= 0) then
ba393a269   wdaniau   1) updated docume...
728
        if ( (info(1) < 1) .or. (info(1) >3) ) then ! not a warning
7dc628d15   wdaniau   Added error outpu...
729
            write(*,*) "Problem during sparse determinant : ",trim(umfpack_return_code(int(info(1),kind=sp_kind)))
ba393a269   wdaniau   1) updated docume...
730
        endif
2b83390c6   wdaniau   1) added fvn_spar...
731
732
733
734
735
736
737
738
739
      status=info(1)
  endif
  
  ! free the C numeric pointer
  call umfpack_di_free_numeric (numeric)
  
  deallocate(A)
  deallocate(wTi,wTj)
  end subroutine
b93026039   daniau   git-svn-id: https...
740
  end module fvn_sparse