| 38 |
|
!! University of Notre Dame has been advised of the possibility of |
| 39 |
|
!! such damages. |
| 40 |
|
!! |
| 41 |
< |
|
| 41 |
> |
!!$ |
| 42 |
|
module switcheroo |
| 43 |
|
|
| 44 |
|
use definitions |
| 45 |
|
use interpolation |
| 46 |
+ |
use status |
| 47 |
|
|
| 48 |
|
implicit none |
| 49 |
|
PRIVATE |
| 50 |
|
|
| 51 |
|
#define __FORTRAN90 |
| 51 |
– |
#include "UseTheForce/fSwitchingFunction.h" |
| 52 |
|
#include "UseTheForce/DarkSide/fSwitchingFunctionType.h" |
| 53 |
+ |
|
| 54 |
+ |
!! number of points for the spline approximations |
| 55 |
+ |
INTEGER, PARAMETER :: np = 150 |
| 56 |
|
|
| 57 |
< |
real ( kind = dp ), dimension(NSWITCHTYPES) :: rin |
| 58 |
< |
real ( kind = dp ), dimension(NSWITCHTYPES) :: rout |
| 59 |
< |
real ( kind = dp ), dimension(NSWITCHTYPES) :: rin2 |
| 60 |
< |
real ( kind = dp ), dimension(NSWITCHTYPES) :: rout2 |
| 58 |
< |
real ( kind = dp ), save :: c0, c1, c2, c3, c4, c5 |
| 57 |
> |
real ( kind = dp ), save :: rin |
| 58 |
> |
real ( kind = dp ), save :: rout |
| 59 |
> |
real ( kind = dp ), save :: rin2 |
| 60 |
> |
real ( kind = dp ), save :: rout2 |
| 61 |
|
|
| 62 |
< |
logical, dimension(NSWITCHTYPES) :: isOK |
| 63 |
< |
logical, save :: haveFunctionType = .false. |
| 62 |
< |
logical, save :: haveSqrtSpline = .false. |
| 63 |
< |
logical, save :: useSpline = .true. |
| 62 |
> |
logical, save :: haveSplines = .false. |
| 63 |
> |
logical, save :: switchIsCubic = .true. |
| 64 |
|
integer, save :: functionType = CUBIC |
| 65 |
|
|
| 66 |
< |
! spline structure |
| 67 |
< |
type(cubicSpline), save :: splineSqrt |
| 66 |
> |
! spline structures |
| 67 |
> |
type(cubicSpline), save :: r2Spline |
| 68 |
> |
type(cubicSpline), save :: switchSpline |
| 69 |
|
|
| 70 |
+ |
public::set_switch_type |
| 71 |
|
public::set_switch |
| 70 |
– |
public::set_function_type |
| 72 |
|
public::get_switch |
| 73 |
+ |
public::delete_switch |
| 74 |
|
|
| 75 |
|
contains |
| 76 |
|
|
| 77 |
< |
subroutine setupSqrtSpline(rmin, rmax, np) |
| 76 |
< |
real( kind = dp ), intent(in) :: rmin, rmax |
| 77 |
< |
integer, intent(in) :: np |
| 78 |
< |
real( kind = dp ) :: rvals(np), yvals(np) |
| 79 |
< |
real( kind = dp ) :: dr, r |
| 80 |
< |
real( kind = dp ) :: yprime1, yprimen |
| 81 |
< |
integer :: i |
| 77 |
> |
subroutine set_switch(rinner, router) |
| 78 |
|
|
| 83 |
– |
dr = (rmax-rmin) / float(np-1) |
| 84 |
– |
|
| 85 |
– |
do i = 1, np |
| 86 |
– |
r = rmin + float(i-1)*dr |
| 87 |
– |
rvals(i) = r |
| 88 |
– |
yvals(i) = dsqrt(r) |
| 89 |
– |
enddo |
| 90 |
– |
|
| 91 |
– |
yprime1 = 1.0d0 / ( 2.0d0 * dsqrt( rmin ) ) |
| 92 |
– |
yprimeN = 1.0d0 / ( 2.0d0 * dsqrt( rmax ) ) |
| 93 |
– |
|
| 94 |
– |
call newSpline(splineSqrt, rvals, yvals, yprime1, yprimen, .true.) |
| 95 |
– |
|
| 96 |
– |
return |
| 97 |
– |
end subroutine setupSqrtSpline |
| 98 |
– |
|
| 99 |
– |
subroutine set_switch(SwitchType, rinner, router) |
| 100 |
– |
|
| 79 |
|
real ( kind = dp ), intent(in):: rinner, router |
| 80 |
< |
integer, intent(in) :: SwitchType |
| 80 |
> |
real ( kind = dp ), dimension(np) :: xvals, yvals |
| 81 |
> |
real ( kind = dp ), dimension(2) :: rCubVals, sCubVals |
| 82 |
> |
real ( kind = dp ) :: rval, rval2, rval3, rval4, rval5 |
| 83 |
> |
real ( kind = dp ) :: rvaldi, rvaldi2, rvaldi3, rvaldi4, rvaldi5 |
| 84 |
> |
real ( kind = dp ) :: c0, c3, c4, c5, dx, r, r2 |
| 85 |
|
integer :: i |
| 86 |
|
|
| 87 |
< |
if (SwitchType .gt. NSWITCHTYPES) then |
| 88 |
< |
write(default_error, *) & |
| 107 |
< |
'set_switch: not that many switch types! ' |
| 87 |
> |
if (router .lt. rinner) then |
| 88 |
> |
call handleError("set_switch", "router is less than rinner") |
| 89 |
|
return |
| 90 |
|
endif |
| 91 |
|
|
| 92 |
< |
isOK(SwitchType) = .false. |
| 93 |
< |
|
| 113 |
< |
if (router .lt. rinner) then |
| 114 |
< |
write(default_error, *) & |
| 115 |
< |
'set_switch: router is less than rinner ' |
| 92 |
> |
if ((router .lt. 0.0_dp) .or. (rinner .lt. 0.0_dp)) then |
| 93 |
> |
call handleError("set_switch", "one of the switches is negative!") |
| 94 |
|
return |
| 95 |
|
endif |
| 96 |
|
|
| 97 |
< |
if ((router .lt. 0.0d0) .or. (rinner .lt. 0.0d0)) then |
| 98 |
< |
write(default_error, *) & |
| 99 |
< |
'set_switch: one of the switches is negative!' |
| 97 |
> |
rin = rinner |
| 98 |
> |
rout = router |
| 99 |
> |
rin2 = rinner * rinner |
| 100 |
> |
rout2 = router * router |
| 101 |
> |
|
| 102 |
> |
if ((router-rinner) .lt. 1e-8) then |
| 103 |
> |
! no reason to set up splines if the switching region is tiny |
| 104 |
|
return |
| 105 |
|
endif |
| 106 |
|
|
| 107 |
< |
rin(SwitchType) = rinner |
| 108 |
< |
rout(SwitchType) = router |
| 109 |
< |
rin2(SwitchType) = rinner * rinner |
| 110 |
< |
rout2(SwitchType) = router * router |
| 111 |
< |
isOK(SwitchType) = .true. |
| 107 |
> |
dx = (rout2-rin2) / dble(np-1) |
| 108 |
> |
|
| 109 |
> |
do i = 1, np |
| 110 |
> |
r2 = rin2 + dble(i-1)*dx |
| 111 |
> |
xvals(i) = r2 |
| 112 |
> |
yvals(i) = sqrt(r2) |
| 113 |
> |
enddo |
| 114 |
|
|
| 115 |
< |
if (.not.haveSqrtSpline) then |
| 116 |
< |
! fill arrays for building the spline |
| 117 |
< |
call setupSqrtSpline(1.0d0, rout2(switchType), SPLINE_SEGMENTS) |
| 118 |
< |
haveSqrtSpline = .true. |
| 115 |
> |
call newSpline(r2spline, xvals, yvals, .true.) |
| 116 |
> |
|
| 117 |
> |
if (functionType .eq. FIFTH_ORDER_POLY) then |
| 118 |
> |
c0 = 1.0_dp |
| 119 |
> |
c3 = -10.0_dp |
| 120 |
> |
c4 = 15.0_dp |
| 121 |
> |
c5 = -6.0_dp |
| 122 |
> |
|
| 123 |
> |
dx = (rout-rin) / dble(np-1) |
| 124 |
> |
|
| 125 |
> |
do i = 1, np |
| 126 |
> |
r = rin + dble(i-1)*dx |
| 127 |
> |
xvals(i) = r |
| 128 |
> |
|
| 129 |
> |
rval = ( r - rin ) |
| 130 |
> |
rval2 = rval*rval |
| 131 |
> |
rval3 = rval2*rval |
| 132 |
> |
rval4 = rval2*rval2 |
| 133 |
> |
rval5 = rval3*rval2 |
| 134 |
> |
rvaldi = 1.0_dp/( rout - rin ) |
| 135 |
> |
rvaldi2 = rvaldi*rvaldi |
| 136 |
> |
rvaldi3 = rvaldi2*rvaldi |
| 137 |
> |
rvaldi4 = rvaldi2*rvaldi2 |
| 138 |
> |
rvaldi5 = rvaldi3*rvaldi2 |
| 139 |
> |
yvals(i)= c0 + c3*rval3*rvaldi3 + c4*rval4*rvaldi4 + c5*rval5*rvaldi5 |
| 140 |
> |
enddo |
| 141 |
> |
|
| 142 |
> |
call newSpline(switchSpline, xvals, yvals, .true.) |
| 143 |
> |
|
| 144 |
> |
switchIsCubic = .false. |
| 145 |
> |
else |
| 146 |
> |
rCubVals(1) = rin |
| 147 |
> |
rCubVals(2) = rout |
| 148 |
> |
sCubVals(1) = 1.0_dp |
| 149 |
> |
sCubVals(2) = 0.0_dp |
| 150 |
> |
call newSpline(switchSpline, rCubVals, sCubVals, .true.) |
| 151 |
|
endif |
| 152 |
|
|
| 153 |
+ |
haveSplines = .true. |
| 154 |
+ |
return |
| 155 |
|
end subroutine set_switch |
| 156 |
|
|
| 157 |
< |
subroutine set_function_type(functionForm) |
| 157 |
> |
subroutine set_switch_type(functionForm) |
| 158 |
|
integer, intent(in) :: functionForm |
| 159 |
|
functionType = functionForm |
| 160 |
|
|
| 161 |
< |
if (functionType .eq. FIFTH_ORDER_POLY) then |
| 162 |
< |
c0 = 1.0d0 |
| 163 |
< |
c1 = 0.0d0 |
| 164 |
< |
c2 = 0.0d0 |
| 165 |
< |
c3 = -10.0d0 |
| 166 |
< |
c4 = 15.0d0 |
| 167 |
< |
c5 = -6.0d0 |
| 161 |
> |
if ((functionType.eq.FIFTH_ORDER_POLY).or.(functionType.eq.CUBIC)) then |
| 162 |
> |
if (haveSplines) then |
| 163 |
> |
call delete_switch() |
| 164 |
> |
call set_switch(rin, rout) |
| 165 |
> |
endif |
| 166 |
> |
else |
| 167 |
> |
call handleError("set_switch_type", & |
| 168 |
> |
"Unknown type of switching function!") |
| 169 |
> |
return |
| 170 |
|
endif |
| 171 |
< |
end subroutine set_function_type |
| 171 |
> |
end subroutine set_switch_type |
| 172 |
> |
|
| 173 |
> |
subroutine delete_switch() |
| 174 |
> |
call deleteSpline(switchSpline) |
| 175 |
> |
call deleteSpline(r2spline) |
| 176 |
> |
return |
| 177 |
> |
end subroutine delete_switch |
| 178 |
> |
|
| 179 |
> |
subroutine get_switch(r2, sw, dswdr, r, in_switching_region) |
| 180 |
|
|
| 153 |
– |
subroutine get_switch(r2, sw, dswdr, r, SwitchType, in_switching_region) |
| 154 |
– |
|
| 181 |
|
real( kind = dp ), intent(in) :: r2 |
| 182 |
|
real( kind = dp ), intent(inout) :: sw, dswdr, r |
| 157 |
– |
real( kind = dp ) :: ron, roff, a, b, c, d, dx |
| 158 |
– |
real( kind = dp ) :: rval, rval2, rval3, rval4, rval5 |
| 159 |
– |
real( kind = dp ) :: rvaldi, rvaldi2, rvaldi3, rvaldi4, rvaldi5 |
| 160 |
– |
integer, intent(in) :: SwitchType |
| 183 |
|
logical, intent(inout) :: in_switching_region |
| 184 |
|
integer :: j |
| 185 |
+ |
real ( kind = dp ) :: a, b, c, d, dx |
| 186 |
|
|
| 187 |
< |
sw = 0.0d0 |
| 188 |
< |
dswdr = 0.0d0 |
| 187 |
> |
sw = 1.0_dp |
| 188 |
> |
dswdr = 0.0_dp |
| 189 |
|
in_switching_region = .false. |
| 190 |
|
|
| 191 |
< |
if (.not.isOK(SwitchType)) then |
| 192 |
< |
write(default_error, *) & |
| 170 |
< |
'get_switch: this switching function has not been set up!' |
| 171 |
< |
return |
| 172 |
< |
endif |
| 191 |
> |
if (r2.gt.rin2) then |
| 192 |
> |
if (r2.gt.rout2) then |
| 193 |
|
|
| 194 |
< |
if (r2.lt.rout2(SwitchType)) then |
| 195 |
< |
if (r2.lt.rin2(SwitchType)) then |
| 176 |
< |
|
| 177 |
< |
sw = 1.0d0 |
| 178 |
< |
dswdr = 0.0d0 |
| 194 |
> |
sw = 0.0_dp |
| 195 |
> |
dswdr = 0.0_dp |
| 196 |
|
return |
| 197 |
< |
|
| 198 |
< |
else |
| 199 |
< |
if (useSpline) then |
| 200 |
< |
j = MAX(1, MIN(splineSqrt%np, idint((r2-splineSqrt%x(1)) * splineSqrt%dx_i) + 1)) |
| 201 |
< |
|
| 202 |
< |
dx = r2 - splineSqrt%x(j) |
| 203 |
< |
|
| 204 |
< |
a = splineSqrt%c(1,j) |
| 205 |
< |
b = splineSqrt%c(2,j) |
| 206 |
< |
c = splineSqrt%c(3,j) |
| 207 |
< |
d = splineSqrt%c(4,j) |
| 191 |
< |
|
| 192 |
< |
r = c + dx * d |
| 193 |
< |
r = b + dx * r |
| 194 |
< |
r = a + dx * r |
| 197 |
> |
|
| 198 |
> |
else |
| 199 |
> |
|
| 200 |
> |
call lookupUniformSpline(r2Spline, r2, r) |
| 201 |
> |
if (switchIsCubic) then |
| 202 |
> |
! super zippy automated use of precomputed spline coefficients |
| 203 |
> |
dx = r - rin |
| 204 |
> |
sw = switchSpline%y(1) + dx*(dx*(switchSpline%c(1) + & |
| 205 |
> |
dx*switchSpline%d(1))) |
| 206 |
> |
dswdr = dx*(2.0_dp * switchSpline%c(1) + & |
| 207 |
> |
3.0_dp * dx * switchSpline%d(1)) |
| 208 |
|
else |
| 209 |
< |
r = dsqrt(r2) |
| 209 |
> |
call lookupUniformSpline1d(switchSpline, r, sw, dswdr) |
| 210 |
|
endif |
| 211 |
< |
|
| 199 |
< |
ron = rin(SwitchType) |
| 200 |
< |
roff = rout(SwitchType) |
| 201 |
< |
|
| 202 |
< |
if (functionType .eq. FIFTH_ORDER_POLY) then |
| 203 |
< |
rval = ( r - ron ) |
| 204 |
< |
rval2 = rval*rval |
| 205 |
< |
rval3 = rval2*rval |
| 206 |
< |
rval4 = rval2*rval2 |
| 207 |
< |
rval5 = rval3*rval2 |
| 208 |
< |
rvaldi = 1.0d0/( roff - ron ) |
| 209 |
< |
rvaldi2 = rvaldi*rvaldi |
| 210 |
< |
rvaldi3 = rvaldi2*rvaldi |
| 211 |
< |
rvaldi4 = rvaldi2*rvaldi2 |
| 212 |
< |
rvaldi5 = rvaldi3*rvaldi2 |
| 213 |
< |
sw = c0 + c1*rval*rvaldi + c2*rval2*rvaldi2 + c3*rval3*rvaldi3 & |
| 214 |
< |
+ c4*rval4*rvaldi4 + c5*rval5*rvaldi5 |
| 215 |
< |
dswdr = c1*rvaldi + 2.0d0*c2*rval*rvaldi2 & |
| 216 |
< |
+ 3.0d0*c3*rval2*rvaldi3 + 4.0d0*c4*rval3*rvaldi4 & |
| 217 |
< |
+ 5.0d0*c5*rval4*rvaldi5 |
| 218 |
< |
|
| 219 |
< |
else |
| 220 |
< |
sw = (roff + 2.0d0*r - 3.0d0*ron)*(roff-r)**2/ ((roff-ron)**3) |
| 221 |
< |
dswdr = 6.0d0*(r*r - r*ron - r*roff +roff*ron)/((roff-ron)**3) |
| 222 |
< |
|
| 223 |
< |
endif |
| 211 |
> |
|
| 212 |
|
in_switching_region = .true. |
| 213 |
+ |
|
| 214 |
|
return |
| 215 |
|
endif |
| 216 |
|
else |
| 217 |
|
return |
| 218 |
|
endif |
| 219 |
< |
|
| 219 |
> |
|
| 220 |
|
end subroutine get_switch |
| 221 |
|
|
| 222 |
|
end module switcheroo |
