--- trunk/dyn3d/fxhyp_loop_ik.f 2015/01/28 16:10:02 121 +++ trunk/Sources/dyn3d/invert_zoom_x.f 2015/08/24 16:30:33 167 @@ -1,175 +1,98 @@ -module fxhyp_loop_ik_m +module invert_zoom_x_m implicit none INTEGER, PARAMETER:: nmax = 30000 + DOUBLE PRECISION abs_y + + private abs_y, funcd contains - subroutine fxhyp_loop_ik(ik, decalx, xf, xtild, Xprimt, xzoom, xlon, & - xprimm, xuv) + subroutine invert_zoom_x(beta, xf, xtild, G, xlon, xprim, xuv) - use coefpoly_m, only: coefpoly + use coefpoly_m, only: coefpoly, a1, a2, a3 USE dimens_m, ONLY: iim - use nr_util, only: pi_d, twopi_d, twopi - use serre, only: grossismx + use dynetat0_m, only: clon, grossismx + use nr_util, only: pi_d, twopi_d + use numer_rec_95, only: hunt, rtsafe + + DOUBLE PRECISION, intent(in):: beta, Xf(0:), xtild(0:), G(0:) ! (0:nmax) - INTEGER, intent(in):: ik - DOUBLE PRECISION, intent(in):: decalx - DOUBLE PRECISION, intent(in):: Xf(0:), xtild(0:), Xprimt(0:) ! (0:2 * nmax) - DOUBLE PRECISION, intent(in):: xzoom - real, intent(out):: xlon(:), xprimm(:) ! (iim + 1) + real, intent(out):: xlon(:), xprim(:) ! (iim) DOUBLE PRECISION, intent(in):: xuv + ! between - 0.25 and 0.5 ! 0. si calcul aux points scalaires - ! 0.5 si calcul aux points U + ! 0.5 si calcul aux points U ! Local: + DOUBLE PRECISION Y + DOUBLE PRECISION h ! step of the uniform grid + integer i, it - DOUBLE PRECISION xo1, Xfi, xi, a0, a1, a2, a3, Xf1, Xprimin, xi2 - integer ii1, ii2, i, it, iter, idif, ii - DOUBLE PRECISION, parameter:: my_eps = 1e-3 - DOUBLE PRECISION xxprim(iim + 1), xvrai(iim + 1) - INTEGER:: is2 = 0 + DOUBLE PRECISION xvrai(iim), Gvrai(iim) + ! intermediary variables because xlon and xprim are simple precision !------------------------------------------------------------------ - xo1 = 0. - - IF (ik == 1 .and. grossismx == 1.) THEN - ii1 = 2 - ii2 = iim + 1 - else - ii1=1 - ii2=iim - END IF - - DO i = ii1, ii2 - Xfi = - pi_d + (i + xuv - decalx) * twopi_d / iim - - it = 2 * nmax - do while (xfi < xf(it) .and. it >= 1) - it = it - 1 - end do - - ! Calcul de Xf(xi) - - xi = xtild(it) + print *, "Call sequence information: invert_zoom_x" + it = 0 ! initial guess + h = twopi_d / iim - IF (it == 2 * nmax) THEN - it = 2 * nmax -1 - END IF - - ! Appel de la routine qui calcule les coefficients a0, a1, - ! a2, a3 d'un polynome de degre 3 qui passe par les points - ! (Xf(it), xtild(it)) et (Xf(it + 1), xtild(it + 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 + 2. * a2 * xi + 3. * a3 * xi**2 - - iter = 1 - - do - xi = xi - (Xf1 - Xfi) / Xprimin - IF (ABS(xi - xo1) <= my_eps .or. iter == 300) exit - xo1 = xi - xi2 = xi * xi - Xf1 = a0 + a1 * xi + a2 * xi2 + a3 * xi2 * xi - Xprimin = a1 + 2. * a2 * xi + 3. * a3 * xi2 - end DO - - if (ABS(xi - xo1) > my_eps) then - ! iter == 300 - print *, 'Pas de solution.' - print *, i, xfi - STOP 1 + DO i = 1, iim + Y = - pi_d + (i + xuv - 0.75d0) * h + ! - pi <= y < pi + abs_y = abs(y) + + ! Distinguish boundaries in order to avoid roundoff error. + ! funcd should be exactly equal to 0 at xtild(it) or xtild(it + + ! 1) and could be very small with the wrong sign so rtsafe + ! would fail. + if (abs_y == 0d0) then + xvrai(i) = 0d0 + gvrai(i) = grossismx + else if (abs_y == pi_d) then + xvrai(i) = pi_d + gvrai(i) = 2d0 * beta - grossismx + else + call hunt(xf, abs_y, it, my_lbound = 0) + ! {0 <= it <= nmax - 1} + + ! Calcul de xvrai(i) et Gvrai(i) + CALL coefpoly(Xf(it), Xf(it + 1), G(it), G(it + 1), xtild(it), & + xtild(it + 1)) + xvrai(i) = rtsafe(funcd, xtild(it), xtild(it + 1), xacc = 1d-6) + Gvrai(i) = a1 + xvrai(i) * (2d0 * a2 + xvrai(i) * 3d0 * a3) end if - xxprim(i) = twopi_d / (REAL(iim) * Xprimin) - xvrai(i) = xi + xzoom + if (y < 0d0) xvrai(i) = - xvrai(i) end DO - IF (ik == 1 .and. grossismx == 1.) THEN - xvrai(1) = xvrai(iim + 1)-twopi_d - xxprim(1) = xxprim(iim + 1) - END IF - - DO i = 1, iim - xlon(i) = xvrai(i) - xprimm(i) = xxprim(i) - END DO - DO i = 1, iim -1 IF (xvrai(i + 1) < xvrai(i)) THEN - print *, 'rlonu(', i + 1, ') < rlonu(', i, ')' + print *, 'xvrai(', i + 1, ') < xvrai(', i, ')' STOP 1 END IF END DO - IF (.not. (MINval(xvrai(:iim)) >= - pi_d - 1d-5 & - .and. MAXval(xvrai(:iim)) <= pi_d + 1d-5)) THEN - IF (xzoom <= 0.) THEN - IF (ik == 1) THEN - i = 1 - - do while (xvrai(i) < - pi_d .and. i < iim) - i = i + 1 - end do - - if (xvrai(i) < - pi_d) then - print *, 'Xvrai plus petit que - pi !' - STOP 1 - end if - - is2 = i - END IF - - IF (is2 /= 1) THEN - DO ii = is2, iim - xlon(ii-is2 + 1) = xvrai(ii) - xprimm(ii-is2 + 1) = xxprim(ii) - END DO - DO ii = 1, is2 -1 - xlon(ii + iim-is2 + 1) = xvrai(ii) + twopi_d - xprimm(ii + iim-is2 + 1) = xxprim(ii) - END DO - END IF - ELSE - IF (ik == 1) THEN - i = iim - - do while (xvrai(i) > pi_d .and. i > 1) - i = i - 1 - end do - - if (xvrai(i) > pi_d) then - print *, 'Xvrai plus grand que pi !' - STOP 1 - end if - - is2 = i - END IF - - idif = iim -is2 - - DO ii = 1, is2 - xlon(ii + idif) = xvrai(ii) - xprimm(ii + idif) = xxprim(ii) - END DO - - DO ii = 1, idif - xlon(ii) = xvrai(ii + is2) - twopi_d - xprimm(ii) = xxprim(ii + is2) - END DO - END IF - END IF + xlon = xvrai + clon + xprim = h / Gvrai + + end subroutine invert_zoom_x + + !********************************************************************** + + SUBROUTINE funcd(x, fval, fderiv) + + use coefpoly_m, only: a0, a1, a2, a3 + + DOUBLE PRECISION, INTENT(IN):: x + DOUBLE PRECISION, INTENT(OUT):: fval, fderiv - xlon(iim + 1) = xlon(1) + twopi - xprimm(iim + 1) = xprimm(1) + fval = a0 + x * (a1 + x * (a2 + x * a3)) - abs_y + fderiv = a1 + x * (2d0 * a2 + x * 3d0 * a3) - end subroutine fxhyp_loop_ik + END SUBROUTINE funcd -end module fxhyp_loop_ik_m +end module invert_zoom_x_m