external/fftpack/fftpack5/zmf2kf.F

   1 subroutine zmf2kf ( lot, ido, l1, na, cc, im1, in1, ch, im2, in2, wa )
   2
   3 !*****************************************************************************80
   4 !
   5 !! ZMF2KF is an FFTPACK5 auxiliary routine.
   6 !
   7 !
   8 !
   9 !  Modified:
  10 !
  11 !    26 Ausust 2009
  12 !
  13 !  Author:
  14 !
  15 !    Original complex single precision by Paul Swarztrauber, Richard Valent.
  16 !    Complex double precision version by John Burkardt.
  17 !
  18 !  Reference:
  19 !
  20 !    Paul Swarztrauber,
  21 !    Vectorizing the Fast Fourier Transforms,
  22 !    in Parallel Computations,
  23 !    edited by G. Rodrigue,
  24 !    Academic Press, 1982.
  25 !
  26 !    Paul Swarztrauber,
  27 !    Fast Fourier Transform Algorithms for Vector Computers,
  28 !    Parallel Computing, pages 45-63, 1984.
  29 !
  30 !  Parameters:
  31 !
  32   implicit none
  33
  34   integer ( kind = 4 ) ido
  35   integer ( kind = 4 ) in1
  36   integer ( kind = 4 ) in2
  37   integer ( kind = 4 ) l1
  38
  39   real ( kind = 8 ) cc(2,in1,l1,ido,2)
  40   real ( kind = 8 ) ch(2,in2,l1,2,ido)
  41   real ( kind = 8 ) chold1
  42   real ( kind = 8 ) chold2
  43   integer ( kind = 4 ) i
  44   integer ( kind = 4 ) im1
  45   integer ( kind = 4 ) im2
  46   integer ( kind = 4 ) k
  47   integer ( kind = 4 ) lid
  48   integer ( kind = 4 ) lot
  49   integer ( kind = 4 ) m1
  50   integer ( kind = 4 ) m1d
  51   integer ( kind = 4 ) m2
  52   integer ( kind = 4 ) m2s
  53   integer ( kind = 4 ) n
  54   integer ( kind = 4 ) na
  55   real ( kind = 8 ) sn
  56   real ( kind = 8 ) ti2
  57   real ( kind = 8 ) tr2
  58   real ( kind = 8 ) wa(ido,1,2)
  59
  60   m1d = ( lot - 1 ) * im1 + 1
  61   m2s = 1 - im2
  62
  63   if ( 1 < ido ) then
  64
  65     do k = 1, l1
  66       m2 = m2s
  67       do m1 = 1, m1d, im1
  68         m2 = m2 + im2
  69         ch(1,m2,k,1,1) = cc(1,m1,k,1,1)+cc(1,m1,k,1,2)
  70         ch(1,m2,k,2,1) = cc(1,m1,k,1,1)-cc(1,m1,k,1,2)
  71         ch(2,m2,k,1,1) = cc(2,m1,k,1,1)+cc(2,m1,k,1,2)
  72         ch(2,m2,k,2,1) = cc(2,m1,k,1,1)-cc(2,m1,k,1,2)
  73       end do
  74     end do
  75
  76     do i = 2, ido
  77       do k = 1, l1
  78         m2 = m2s
  79         do m1 = 1, m1d, im1
  80           m2 = m2 + im2
  81           ch(1,m2,k,1,i) = cc(1,m1,k,i,1)+cc(1,m1,k,i,2)
  82           tr2 = cc(1,m1,k,i,1)-cc(1,m1,k,i,2)
  83           ch(2,m2,k,1,i) = cc(2,m1,k,i,1)+cc(2,m1,k,i,2)
  84           ti2 = cc(2,m1,k,i,1)-cc(2,m1,k,i,2)
  85           ch(2,m2,k,2,i) = wa(i,1,1)*ti2-wa(i,1,2)*tr2
  86           ch(1,m2,k,2,i) = wa(i,1,1)*tr2+wa(i,1,2)*ti2
  87         end do
  88       end do
  89     end do
  90
  91   else if ( na == 1 ) then
  92
  93     sn = 1.0D+00 / real ( 2 * l1, kind = 8 )
  94
  95     do k = 1, l1
  96       m2 = m2s
  97       do m1 = 1, m1d, im1
  98         m2 = m2 + im2
  99         ch(1,m2,k,1,1) = sn * ( cc(1,m1,k,1,1) + cc(1,m1,k,1,2) )
 100         ch(1,m2,k,2,1) = sn * ( cc(1,m1,k,1,1) - cc(1,m1,k,1,2) )
 101         ch(2,m2,k,1,1) = sn * ( cc(2,m1,k,1,1) + cc(2,m1,k,1,2) )
 102         ch(2,m2,k,2,1) = sn * ( cc(2,m1,k,1,1) - cc(2,m1,k,1,2) )
 103       end do
 104     end do
 105
 106   else
 107
 108     sn = 1.0D+00 / real ( 2 * l1, kind = 8 )
 109
 110     do k = 1, l1
 111       do m1 = 1, m1d, im1
 112
 113         chold1         = sn * ( cc(1,m1,k,1,1) + cc(1,m1,k,1,2) )
 114         cc(1,m1,k,1,2) = sn * ( cc(1,m1,k,1,1) - cc(1,m1,k,1,2) )
 115         cc(1,m1,k,1,1) = chold1
 116
 117         chold2         = sn * ( cc(2,m1,k,1,1) + cc(2,m1,k,1,2) )
 118         cc(2,m1,k,1,2) = sn * ( cc(2,m1,k,1,1) - cc(2,m1,k,1,2) )
 119         cc(2,m1,k,1,1) = chold2
 120
 121       end do
 122     end do
 123
 124   end if
 125
 126   return
 127 end