[bug] further fix of multi-source replay, #215

fangq · Mar 13, 2024 · e8cd4cd · e8cd4cd
1 parent abdee14
commit e8cd4cd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/mcx_core.cu b/src/mcx_core.cu
@@ -1153,6 +1153,10 @@ __device__ inline int launchnewphoton(MCXpos* p, MCXdir* v, Stokes* s, MCXtime*
         }
     }
 
+    if (gcfg->issaveseed) {
+        copystate(t, photonseed);
+    }
+
     if (gcfg->extrasrclen && gcfg->srcid != 1) {
         if (gcfg->srcid > 1) {
             launchsrc = (MCXSrc*)(gproperty + gcfg->maxmedia + 1 + gcfg->detnum + ((gcfg->srcid - 2) * 4));
@@ -1177,10 +1181,6 @@ __device__ inline int launchnewphoton(MCXpos* p, MCXdir* v, Stokes* s, MCXtime*
         *idx1d = *((uint*)&launchsrc->param2.z);      /**< pre-computed 1D index of the photon at launch for pencil/isotropic beams */
         *mediaid = *((uint*)&launchsrc->param2.w);    /**< pre-computed media index of the photon at launch for pencil/isotropic beams */
 
-        if (gcfg->issaveseed) {
-            copystate(t, photonseed);
-        }
-
         *rv = float3(launchsrc->pos.x, launchsrc->pos.y, launchsrc->pos.z); //< reuse as the origin of the src, needed for focusable sources
 
         if (issvmc) {