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
|
subroutine wmsum(flag,ar,ai,na,m,n,vr,vi,nv)
c!purpose
c computes the sum of the entries of a complexmatrix according to flag
c!calling sequence
c subroutine wmsum(flag,ar,ai,na,m,n,vr,vi,nv)
c double precision ar(na,n),ai(na,n),vr(*),vi(*)
c integer na,n,m,nv
c integer flag
c!parameters
c flag : indicates operation to perform
c 0 : returns in v(1) the sum of all entries of a
c 1 : returns in v(j) the sum of jth column of a
c 2 : returns in v(i) the sum of ith row of a
c a : array containing the a matrix
c na : a matrix leading dimension
c m : a matrix row dimension
c n : a matrix column dimension
c v : array containing the result, may be confused with a row or
c a column of the a matrix
c if flag==0 size(v)>=1
c if flag==1 size(v)>=n*nv
c if flag==1 size(v)>=m*nv
c nv : increment between to consecutive entries ov v
c!
c Copyright INRIA
double precision ar(na,n),ai(na,n),vr(*),vi(*)
integer na,n,m,nv
integer flag
c
double precision tr,ti,dsum
integer iv
c
iv=1
if(flag.eq.0) then
c sum of all the entries
tr=0.0d0
ti=0.0d0
do 10 j=1,n
tr=tr+dsum(m,ar(1,j),1)
ti=ti+dsum(m,ai(1,j),1)
10 continue
vr(1)=tr
vi(1)=ti
elseif(flag.eq.1) then
do 20 j=1,n
tr=dsum(m,ar(1,j),1)
ti=dsum(m,ai(1,j),1)
vr(iv)=tr
vi(iv)=ti
iv=iv+nv
20 continue
elseif(flag.eq.2) then
do 30 i=1,m
tr=dsum(n,ar(i,1),m)
ti=dsum(n,ai(i,1),m)
vr(iv)=tr
vi(iv)=ti
iv=iv+nv
30 continue
endif
return
end
|
