Sources/dyn3d/invert_zoom_x.f

module invert_zoom_x_m

  implicit none

  INTEGER, PARAMETER:: nmax = 30000

contains

  subroutine invert_zoom_x(xf, xtild, Xprimt, xlon, xprimm, xuv)

    use coefpoly_m, only: coefpoly
    USE dimens_m, ONLY: iim
    use dynetat0_m, only: clon
    use nr_util, only: pi_d, twopi_d

    DOUBLE PRECISION, intent(in):: Xf(0:), xtild(0:), Xprimt(0:) ! (0:2 * nmax)
    real, intent(out):: xlon(:), xprimm(:) ! (iim)

    DOUBLE PRECISION, intent(in):: xuv
    ! 0. si calcul aux points scalaires 
    ! 0.5 si calcul aux points U 

    ! Local:
    DOUBLE PRECISION xo1, Xfi, a0, a1, a2, a3, Xf1, Xprimin
    integer i, it, iter
    DOUBLE PRECISION, parameter:: my_eps = 1d-6

    DOUBLE PRECISION xxprim(iim), xvrai(iim) 
    ! intermediary variables because xlon and xprimm are simple precision

    !------------------------------------------------------------------

    DO i = 1, iim
       Xfi = - pi_d + (i + xuv - 0.75d0) * twopi_d / iim

       it = 2 * nmax
       do while (xfi < xf(it) .and. it >= 1)
          it = it - 1
       end do

       ! Calcul de Xf(xvrai(i)) 

       xvrai(i) = xtild(it)
       IF (it == 2 * nmax) it = 2 * nmax -1
       CALL coefpoly(Xf(it), Xf(it + 1), Xprimt(it), Xprimt(it + 1), &
            xtild(it), xtild(it + 1), a0, a1, a2, a3)
       Xf1 = Xf(it)
       Xprimin = a1 + xvrai(i) * (2d0 * a2 + xvrai(i) * 3d0 * a3)
       xo1 = xvrai(i)
       iter = 1

       do
          xvrai(i) = xvrai(i) - (Xf1 - Xfi) / Xprimin
          IF (ABS(xvrai(i) - xo1) <= my_eps .or. iter == 300) exit
          xo1 = xvrai(i)
          Xf1 = a0 + xvrai(i) * (a1 + xvrai(i) * (a2 + xvrai(i) * a3))
          Xprimin = a1 + xvrai(i) * (2d0 * a2 + xvrai(i) * 3d0 * a3)
       end DO

       if (ABS(xvrai(i) - xo1) > my_eps) then
          ! iter == 300
          print *, 'Pas de solution.'
          print *, i, xfi
          STOP 1
       end if

       xxprim(i) = twopi_d / (iim * Xprimin)
    end DO

    DO i = 1, iim -1
       IF (xvrai(i + 1) < xvrai(i)) THEN
          print *, 'xvrai(', i + 1, ') < xvrai(', i, ')'
          STOP 1
       END IF
    END DO

    xlon = xvrai + clon
    xprimm = xxprim

  end subroutine invert_zoom_x

end module invert_zoom_x_m
1	module invert_zoom_x_m
2
3	implicit none
4
5	INTEGER, PARAMETER:: nmax = 30000
6
7	contains
8
9	subroutine invert_zoom_x(xf, xtild, Xprimt, xlon, xprimm, xuv)
10
11	use coefpoly_m, only: coefpoly
12	USE dimens_m, ONLY: iim
13	use dynetat0_m, only: clon
14	use nr_util, only: pi_d, twopi_d
15
16	DOUBLE PRECISION, intent(in):: Xf(0:), xtild(0:), Xprimt(0:) ! (0:2 * nmax)
17	real, intent(out):: xlon(:), xprimm(:) ! (iim)
18
19	DOUBLE PRECISION, intent(in):: xuv
20	! 0. si calcul aux points scalaires
21	! 0.5 si calcul aux points U
22
23	! Local:
24	DOUBLE PRECISION xo1, Xfi, a0, a1, a2, a3, Xf1, Xprimin
25	integer i, it, iter
26	DOUBLE PRECISION, parameter:: my_eps = 1d-6
27
28	DOUBLE PRECISION xxprim(iim), xvrai(iim)
29	! intermediary variables because xlon and xprimm are simple precision
30
31	!------------------------------------------------------------------
32
33	DO i = 1, iim
34	Xfi = - pi_d + (i + xuv - 0.75d0) * twopi_d / iim
35
36	it = 2 * nmax
37	do while (xfi < xf(it) .and. it >= 1)
38	it = it - 1
39	end do
40
41	! Calcul de Xf(xvrai(i))
42
43	xvrai(i) = xtild(it)
44	IF (it == 2 * nmax) it = 2 * nmax -1
45	CALL coefpoly(Xf(it), Xf(it + 1), Xprimt(it), Xprimt(it + 1), &
46	xtild(it), xtild(it + 1), a0, a1, a2, a3)
47	Xf1 = Xf(it)
48	Xprimin = a1 + xvrai(i) * (2d0 * a2 + xvrai(i) * 3d0 * a3)
49	xo1 = xvrai(i)
50	iter = 1
51
52	do
53	xvrai(i) = xvrai(i) - (Xf1 - Xfi) / Xprimin
54	IF (ABS(xvrai(i) - xo1) <= my_eps .or. iter == 300) exit
55	xo1 = xvrai(i)
56	Xf1 = a0 + xvrai(i) * (a1 + xvrai(i) * (a2 + xvrai(i) * a3))
57	Xprimin = a1 + xvrai(i) * (2d0 * a2 + xvrai(i) * 3d0 * a3)
58	end DO
59
60	if (ABS(xvrai(i) - xo1) > my_eps) then
61	! iter == 300
62	print *, 'Pas de solution.'
63	print *, i, xfi
64	STOP 1
65	end if
66
67	xxprim(i) = twopi_d / (iim * Xprimin)
68	end DO
69
70	DO i = 1, iim -1
71	IF (xvrai(i + 1) < xvrai(i)) THEN
72	print *, 'xvrai(', i + 1, ') < xvrai(', i, ')'
73	STOP 1
74	END IF
75	END DO
76
77	xlon = xvrai + clon
78	xprimm = xxprim
79
80	end subroutine invert_zoom_x
81
82	end module invert_zoom_x_m