external/fftpack/fftpack5/mradb3.F

   1 subroutine mradb3 ( m, ido, l1, cc, im1, in1, ch, im2, in2, wa1, wa2 )
   2
   3 !*****************************************************************************80
   4 !
   5 !! MRADB3 is an FFTPACK5 auxiliary routine.
   6 !
   7 !
   8 !    Copyright (C) 1995-2004, Scientific Computing Division,
   9 !    University Corporation for Atmospheric Research
  10 !
  11 !  Modified:
  12 !
  13 !    27 March 2009
  14 !
  15 !  Author:
  16 !
  17 !    Paul Swarztrauber
  18 !    Richard Valent
  19 !
  20 !  Reference:
  21 !
  22 !    Paul Swarztrauber,
  23 !    Vectorizing the Fast Fourier Transforms,
  24 !    in Parallel Computations,
  25 !    edited by G. Rodrigue,
  26 !    Academic Press, 1982.
  27 !
  28 !    Paul Swarztrauber,
  29 !    Fast Fourier Transform Algorithms for Vector Computers,
  30 !    Parallel Computing, pages 45-63, 1984.
  31 !
  32 !  Parameters:
  33 !
  34   implicit none
  35
  36   integer ( kind = 4 ) ido
  37   integer ( kind = 4 ) in1
  38   integer ( kind = 4 ) in2
  39   integer ( kind = 4 ) l1
  40
  41   real ( kind = 4 ) arg
  42   real ( kind = 4 ) cc(in1,ido,3,l1)
  43   real ( kind = 4 ) ch(in2,ido,l1,3)
  44   integer ( kind = 4 ) i
  45   integer ( kind = 4 ) ic
  46   integer ( kind = 4 ) idp2
  47   integer ( kind = 4 ) im1
  48   integer ( kind = 4 ) im2
  49   integer ( kind = 4 ) k
  50   integer ( kind = 4 ) m
  51   integer ( kind = 4 ) m1
  52   integer ( kind = 4 ) m1d
  53   integer ( kind = 4 ) m2
  54   integer ( kind = 4 ) m2s
  55   real ( kind = 4 ) taui
  56   real ( kind = 4 ) taur
  57   real ( kind = 4 ) wa1(ido)
  58   real ( kind = 4 ) wa2(ido)
  59
  60   m1d = ( m - 1 ) * im1 + 1
  61   m2s = 1 - im2
  62   arg = 2.0E+00 * 4.0E+00 * atan ( 1.0E+00 ) / 3.0E+00
  63   taur = cos ( arg )
  64   taui = sin ( arg )
  65
  66   do k = 1, l1
  67     m2 = m2s
  68     do m1 = 1, m1d, im1
  69       m2 = m2 + im2
  70       ch(m2,1,k,1) = cc(m1,1,1,k) + 2.0E+00 * cc(m1,ido,2,k)
  71       ch(m2,1,k,2) = cc(m1,1,1,k) + ( 2.0E+00 * taur ) * cc(m1,ido,2,k) &
  72         - ( 2.0E+00 * taui ) * cc(m1,1,3,k)
  73       ch(m2,1,k,3) = cc(m1,1,1,k) + ( 2.0E+00 * taur ) * cc(m1,ido,2,k) &
  74         + 2.0E+00 * taui * cc(m1,1,3,k)
  75     end do
  76   end do
  77
  78   if ( ido == 1 ) then
  79     return
  80   end if
  81
  82   idp2 = ido + 2
  83
  84   do k = 1, l1
  85     do i = 3, ido, 2
  86       ic = idp2 - i
  87       m2 = m2s
  88       do m1 = 1, m1d, im1
  89
  90         m2 = m2 + im2
  91
  92         ch(m2,i-1,k,1) = cc(m1,i-1,1,k)+(cc(m1,i-1,3,k)+cc(m1,ic-1,2,k))
  93
  94         ch(m2,i,k,1) = cc(m1,i,1,k)+(cc(m1,i,3,k)-cc(m1,ic,2,k))
  95
  96         ch(m2,i-1,k,2) = wa1(i-2)* &
  97           ((cc(m1,i-1,1,k)+taur*(cc(m1,i-1,3,k)+cc(m1,ic-1,2,k)))- &
  98           (taui*(cc(m1,i,3,k)+cc(m1,ic,2,k)))) - wa1(i-1)* &
  99           ((cc(m1,i,1,k)+taur*(cc(m1,i,3,k)-cc(m1,ic,2,k)))+ &
 100           (taui*(cc(m1,i-1,3,k)-cc(m1,ic-1,2,k))))
 101
 102         ch(m2,i,k,2) = wa1(i-2)* &
 103           ((cc(m1,i,1,k)+taur*(cc(m1,i,3,k)-cc(m1,ic,2,k)))+ &
 104           (taui*(cc(m1,i-1,3,k)-cc(m1,ic-1,2,k)))) + wa1(i-1)* &
 105           ((cc(m1,i-1,1,k)+taur*(cc(m1,i-1,3,k)+cc(m1,ic-1,2,k)))- &
 106           (taui*(cc(m1,i,3,k)+cc(m1,ic,2,k))))
 107
 108         ch(m2,i-1,k,3) = wa2(i-2)* &
 109           ((cc(m1,i-1,1,k)+taur*(cc(m1,i-1,3,k)+cc(m1,ic-1,2,k)))+ &
 110           (taui*(cc(m1,i,3,k)+cc(m1,ic,2,k)))) - wa2(i-1)* &
 111           ((cc(m1,i,1,k)+taur*(cc(m1,i,3,k)-cc(m1,ic,2,k)))- &
 112           (taui*(cc(m1,i-1,3,k)-cc(m1,ic-1,2,k))))
 113
 114         ch(m2,i,k,3) = wa2(i-2)* &
 115           ((cc(m1,i,1,k)+taur*(cc(m1,i,3,k)-cc(m1,ic,2,k)))- &
 116           (taui*(cc(m1,i-1,3,k)-cc(m1,ic-1,2,k)))) + wa2(i-1)* &
 117           ((cc(m1,i-1,1,k)+taur*(cc(m1,i-1,3,k)+cc(m1,ic-1,2,k)))+ &
 118           (taui*(cc(m1,i,3,k)+cc(m1,ic,2,k))))
 119
 120       end do
 121     end do
 122   end do
 123
 124   return
 125 end