Change to get source position directly from solver instead of self.geometry

Author: Gustavo Coelho

pylops/waveeqprocessing/twoway.py

@@ -250,7 +250,7 @@ def _srcillumination_oneshot(self, solver: AcousticWaveSolverType, isrc: int) ->
 
         # assign source location to source object with custom wavelet
         if hasattr(self, "wav"):
-            self.wav.coordinates.data[0, :] = self.geometry.src_positions[isrc, :]
+            self.wav.coordinates.data[0, :] = solver.geometry.src_positions[:]
 
         # source wavefield
         u0 = solver.forward(
@@ -388,7 +388,7 @@ def _bornadj_oneshot(self, solver: AcousticWaveSolverType, isrc, dobs):
 
         # assign source location to source object with custom wavelet
         if hasattr(self, "wav"):
-            self.wav.coordinates.data[0, :] = self.geometry.src_positions[isrc, :]
+            self.wav.coordinates.data[0, :] = solver.geometry.src_positions[:]
 
         # source wavefield
         if hasattr(self, "src_wavefield"):