edited kick to account for no massive bodies

MintonGroup · Dec 12, 2023 · 9ba8f74 · 9ba8f74
1 parent dea961a
commit 9ba8f74
Showing 1 changed file with 17 additions and 15 deletions.
diff --git a/src/whm/whm_kick.f90 b/src/whm/whm_kick.f90
@@ -92,29 +92,31 @@ module subroutine whm_kick_getacch_tp(self, nbody_system, param, t, lbeg)
       associate(tp => self, pl => nbody_system%pl, cb => nbody_system%cb)
          npl = nbody_system%pl%nbody
          ntp = self%nbody
-         if (ntp == 0 .and. npl == 0) return
+         if (ntp == 0) return
          nbody_system%lbeg = lbeg
 
-         if (lbeg) then
-            ah0(:) = whm_kick_getacch_ah0(pl%Gmass(1:npl), pl%rbeg(:, 1:npl), npl)
+         if(npl > 0) then
+            if (lbeg) then
+               ah0(:) = whm_kick_getacch_ah0(pl%Gmass(1:npl), pl%rbeg(:, 1:npl), npl)
 #ifdef DOCONLOC
-            do concurrent(i = 1:ntp, tp%lmask(i)) shared(tp,ah0)
+               do concurrent(i = 1:ntp, tp%lmask(i)) shared(tp,ah0)
 #else
-            do concurrent(i = 1:ntp, tp%lmask(i))
+               do concurrent(i = 1:ntp, tp%lmask(i))
 #endif
-               tp%ah(:, i) = tp%ah(:, i) + ah0(:)
-            end do
-            call tp%accel_int(param, pl%Gmass(1:npl), pl%rbeg(:, 1:npl), npl)
-         else
-            ah0(:) = whm_kick_getacch_ah0(pl%Gmass(1:npl), pl%rend(:, 1:npl), npl)
+                  tp%ah(:, i) = tp%ah(:, i) + ah0(:)
+               end do
+               call tp%accel_int(param, pl%Gmass(1:npl), pl%rbeg(:, 1:npl), npl)
+            else
+               ah0(:) = whm_kick_getacch_ah0(pl%Gmass(1:npl), pl%rend(:, 1:npl), npl)
 #ifdef DOCONLOC
-            do concurrent(i = 1:ntp, tp%lmask(i)) shared(tp,ah0)
+               do concurrent(i = 1:ntp, tp%lmask(i)) shared(tp,ah0)
 #else
-            do concurrent(i = 1:ntp, tp%lmask(i))
+               do concurrent(i = 1:ntp, tp%lmask(i))
 #endif
-               tp%ah(:, i) = tp%ah(:, i) + ah0(:)
-            end do
-            call tp%accel_int(param, pl%Gmass(1:npl), pl%rend(:, 1:npl), npl)
+                  tp%ah(:, i) = tp%ah(:, i) + ah0(:)
+               end do
+               call tp%accel_int(param, pl%Gmass(1:npl), pl%rend(:, 1:npl), npl)
+            end if
          end if
 
          if (param%loblatecb) call tp%accel_obl(nbody_system)