external/fftpack/fftpack5/zmf2kb.F

   1 subroutine zmf2kb ( lot, ido, l1, na, cc, im1, in1, ch, im2, in2, wa )
   2
   3 !*****************************************************************************80
   4 !
   5 !! ZMF2KB 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 ) lot
  48   integer ( kind = 4 ) m1
  49   integer ( kind = 4 ) m1d
  50   integer ( kind = 4 ) m2
  51   integer ( kind = 4 ) m2s
  52   integer ( kind = 4 ) na
  53   real ( kind = 8 ) ti2
  54   real ( kind = 8 ) tr2
  55   real ( kind = 8 ) wa(ido,1,2)
  56
  57   m1d = ( lot - 1 ) * im1 + 1
  58   m2s = 1 - im2
  59
  60   if ( 1 < ido .or. na == 1 ) then
  61
  62     do k = 1, l1
  63       m2 = m2s
  64       do m1 = 1, m1d, im1
  65         m2 = m2 + im2
  66         ch(1,m2,k,1,1) = cc(1,m1,k,1,1) + cc(1,m1,k,1,2)
  67         ch(1,m2,k,2,1) = cc(1,m1,k,1,1) - cc(1,m1,k,1,2)
  68         ch(2,m2,k,1,1) = cc(2,m1,k,1,1) + cc(2,m1,k,1,2)
  69         ch(2,m2,k,2,1) = cc(2,m1,k,1,1) - cc(2,m1,k,1,2)
  70       end do
  71     end do
  72
  73     do i = 2, ido
  74       do k = 1, l1
  75         m2 = m2s
  76         do m1 = 1, m1d, im1
  77           m2 = m2 + im2
  78           ch(1,m2,k,1,i) = cc(1,m1,k,i,1)+cc(1,m1,k,i,2)
  79           tr2 = cc(1,m1,k,i,1)-cc(1,m1,k,i,2)
  80           ch(2,m2,k,1,i) = cc(2,m1,k,i,1)+cc(2,m1,k,i,2)
  81           ti2 = cc(2,m1,k,i,1)-cc(2,m1,k,i,2)
  82
  83           ch(2,m2,k,2,i) = wa(i,1,1) * ti2 + wa(i,1,2) * tr2
  84           ch(1,m2,k,2,i) = wa(i,1,1) * tr2 - wa(i,1,2) * ti2
  85
  86         end do
  87       end do
  88     end do
  89
  90   else
  91
  92     do k = 1, l1
  93       do m1 = 1, m1d, im1
  94
  95         chold1         = cc(1,m1,k,1,1) + cc(1,m1,k,1,2)
  96         cc(1,m1,k,1,2) = cc(1,m1,k,1,1) - cc(1,m1,k,1,2)
  97         cc(1,m1,k,1,1) = chold1
  98
  99         chold2         = cc(2,m1,k,1,1) + cc(2,m1,k,1,2)
 100         cc(2,m1,k,1,2) = cc(2,m1,k,1,1) - cc(2,m1,k,1,2)
 101         cc(2,m1,k,1,1) = chold2
 102
 103       end do
 104     end do
 105
 106   end if
 107
 108   return
 109 end