Blame view

fvn_test/test_sparse_zi.f90 1.64 KB
2b83390c6   wdaniau   1) added fvn_spar...
1
2
3
4
5
6
7
8
9
10
  program test_sparse
  use fvn
  implicit none
  integer(kind=sp_kind), parameter :: nz=12
  integer(kind=sp_kind), parameter :: n=5
  complex(kind=dp_kind),dimension(nz) :: A
  complex(kind=dp_kind),dimension(n,n) :: As
  integer(kind=sp_kind),dimension(nz) :: Ti,Tj
  complex(kind=dp_kind),dimension(n) :: B,x
  integer(kind=sp_kind) :: status,i
c64a05f8a   wdaniau   Sparse Determinan...
11
  real(kind=dp_kind),dimension(3) :: det
2b83390c6   wdaniau   1) added fvn_spar...
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
  character(len=80) :: fmcmplx
  
  fmcmplx='(5("(",f8.5,",",f8.5,")  "))'
  
  ! Description of the matrix in triplet form
  A = (/ (2.,-1.),(3.,2.),(3.,1.),(-1.,5.),(4.,-7.),(4.,0.),(-3.,-4.),(1.,3.),(2.,0.),(2.,-2.),(6.,4.),(1.,0.) /)
  B = (/ (8.,3.), (45.,1.), (-3.,-2.), (3.,0.), (19.,2.) /)
  Ti = (/ 1,2,1,3,5,2,3,4,5,3,2,5 /)
  Tj = (/ 1,1,2,2,2,3,3,3,3,4,5,5 /)
  
  ! Reconstruction of the matrix in standard form
  As=0.
  do i=1,nz
      As(Ti(i),Tj(i))=A(i)
  end do
  
  write(*,*) "Matrix in standard representation :"
  do i=1,5
      write(*,fmcmplx) As(i,:)
  end do
  write(*,*)
  write(*,*) "Standard determinant : ",fvn_det(5,As)
  write(*,*)
  write(*,*) "Right hand side :"
  write(*,fmcmplx) B
  
  ! can use either specific interface, fvn_zi_sparse_det
  ! either generic one fvn_sparse_det
  call fvn_zi_sparse_det(n,nz,A,Ti,Tj,det,status)
  write(*,*)
c64a05f8a   wdaniau   Sparse Determinan...
42
  write(*,*) "Sparse Det = ",cmplx(det(1),det(2),kind=dp_kind)*10**det(3)
2b83390c6   wdaniau   1) added fvn_spar...
43
44
45
46
47
  ! can use either specific interface fvn_zi_sparse_solve
  ! either generic one fvn_sparse_solve
  ! parameter det is optional
  call fvn_zi_sparse_solve(n,nz,A,Ti,Tj,B,x,status,det)
  write(*,*)
c64a05f8a   wdaniau   Sparse Determinan...
48
  write(*,*) "Sparse Det as solve option= ",cmplx(det(1),det(2),kind=dp_kind)*10**det(3)
2b83390c6   wdaniau   1) added fvn_spar...
49
50
51
52
53
54
55
  write(*,*)
  write(*,*) "Solution :"
  write(*,fmcmplx) x
  write(*,*)
  write(*,*) "Product matrix Solution :"
  write(*,fmcmplx) matmul(As,x)
  end program