external/fftpack/fftpack5/r1f3kb.F

   1 subroutine r1f3kb ( ido, l1, cc, in1, ch, in2, wa1, wa2 )
   2
   3 !*****************************************************************************80
   4 !
   5 !! R1F3KB 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 ) k
  48   real ( kind = 4 ) taui
  49   real ( kind = 4 ) taur
  50   real ( kind = 4 ) wa1(ido)
  51   real ( kind = 4 ) wa2(ido)
  52
  53   arg = 2.0E+00 * 4.0E+00 * atan ( 1.0E+00 ) / 3.0E+00
  54   taur = cos ( arg )
  55   taui = sin ( arg )
  56
  57   do k = 1, l1
  58     ch(1,1,k,1) = cc(1,1,1,k) + 2.0E+00 * cc(1,ido,2,k)
  59     ch(1,1,k,2) = cc(1,1,1,k) + 2.0E+00 * taur * cc(1,ido,2,k) &
  60                               - 2.0E+00 * taui * cc(1,1,3,k)
  61     ch(1,1,k,3) = cc(1,1,1,k) + 2.0E+00 * taur * cc(1,ido,2,k) &
  62                               + 2.0E+00 * taui * cc(1,1,3,k)
  63   end do
  64
  65   if ( ido == 1 ) then
  66     return
  67   end if
  68
  69   idp2 = ido + 2
  70
  71   do k = 1, l1
  72     do i = 3, ido, 2
  73       ic = idp2 - i
  74       ch(1,i-1,k,1) = cc(1,i-1,1,k)+(cc(1,i-1,3,k)+cc(1,ic-1,2,k))
  75       ch(1,i,k,1) = cc(1,i,1,k)+(cc(1,i,3,k)-cc(1,ic,2,k))
  76       ch(1,i-1,k,2) = wa1(i-2)* &
  77         ((cc(1,i-1,1,k)+taur*(cc(1,i-1,3,k)+cc(1,ic-1,2,k)))- &
  78         (taui*(cc(1,i,3,k)+cc(1,ic,2,k)))) -wa1(i-1)* &
  79         ((cc(1,i,1,k)+taur*(cc(1,i,3,k)-cc(1,ic,2,k)))+ &
  80         (taui*(cc(1,i-1,3,k)-cc(1,ic-1,2,k))))
  81       ch(1,i,k,2) = wa1(i-2)* &
  82         ((cc(1,i,1,k)+taur*(cc(1,i,3,k)-cc(1,ic,2,k)))+ &
  83         (taui*(cc(1,i-1,3,k)-cc(1,ic-1,2,k)))) +wa1(i-1)* &
  84         ((cc(1,i-1,1,k)+taur*(cc(1,i-1,3,k)+cc(1,ic-1,2,k)))- &
  85         (taui*(cc(1,i,3,k)+cc(1,ic,2,k))))
  86       ch(1,i-1,k,3) = wa2(i-2)* &
  87         ((cc(1,i-1,1,k)+taur*(cc(1,i-1,3,k)+cc(1,ic-1,2,k)))+ &
  88         (taui*(cc(1,i,3,k)+cc(1,ic,2,k)))) -wa2(i-1)* &
  89         ((cc(1,i,1,k)+taur*(cc(1,i,3,k)-cc(1,ic,2,k)))- &
  90         (taui*(cc(1,i-1,3,k)-cc(1,ic-1,2,k))))
  91       ch(1,i,k,3) = wa2(i-2)* &
  92         ((cc(1,i,1,k)+taur*(cc(1,i,3,k)-cc(1,ic,2,k)))- &
  93         (taui*(cc(1,i-1,3,k)-cc(1,ic-1,2,k)))) +wa2(i-1)* &
  94         ((cc(1,i-1,1,k)+taur*(cc(1,i-1,3,k)+cc(1,ic-1,2,k)))+ &
  95         (taui*(cc(1,i,3,k)+cc(1,ic,2,k))))
  96     end do
  97   end do
  98
  99   return
 100 end