[bug] store reflection position in trajectory

fangq · Jun 22, 2024 · f959c71 · f959c71
1 parent 4ff5b60
commit f959c71
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/mcx_core.cu b/src/mcx_core.cu
@@ -2368,6 +2368,14 @@ __global__ void mcx_main_loop(uint media[], OutputType field[], float genergy[],
                         GPUDEBUG(("Rtotal=%f\n", Rtotal));
                     } //< else, total internal reflection
 
+                    if (gcfg->debuglevel & (MCX_DEBUG_MOVE | MCX_DEBUG_MOVE_ONLY)) {
+                        if (ispolarized && gcfg->istrajstokes) {
+                            savedebugstokes(&p, &s, (uint)f.ndone + idx * gcfg->threadphoton + umin(idx, gcfg->oddphotons), gdebugdata, (int)ppath[gcfg->w0offset - 1]);
+                        } else {
+                            savedebugdata(&p, (uint)f.ndone + idx * gcfg->threadphoton + umin(idx, gcfg->oddphotons), gdebugdata, (int)ppath[gcfg->w0offset - 1]);
+                        }
+                    }
+
                     if (Rtotal < 1.f // if total internal reflection does not happen
                             && (!(mediaid == 0 && ((isdet & 0xF) == bcMirror))) // if out of bbx and cfg.bc is not 'm'
                             && rand_next_reflect(t) > Rtotal) { // and if photon chooses the transmission path, then do transmission