Merge pull request PyLops#546 from mrava87/patch-contdeb

fix: change BlendingContinuous to work with pylops solvers
mrava87 · Nov 15, 2023 · 99f5aa8 · 99f5aa8
2 parents e69a172 + 1c7c9e0
commit 99f5aa8
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/pylops/waveeqprocessing/blending.py b/pylops/waveeqprocessing/blending.py
@@ -160,7 +160,11 @@ def _matvec_largerecs(self, x: NDArray) -> NDArray:
             if self.ShiftOps[i] is None:
                 blended_data[:, shift_int : shift_int + self.nt] += x[i, :, :]
             else:
-                shifted_data = self.ShiftOps[i] * self.PadOp * x[i, :, :]
+                shifted_data = (
+                    self.ShiftOps[i]
+                    .matvec(self.PadOp.matvec(x[i, :, :].ravel()))
+                    .reshape(self.ShiftOps[i].dimsd)
+                )
                 blended_data[:, shift_int : shift_int + self.nt + 1] += shifted_data
         return blended_data
 
@@ -172,11 +176,11 @@ def _rmatvec_largerecs(self, x: NDArray) -> NDArray:
             if self.ShiftOps[i] is None:
                 deblended_data[i, :, :] = x[:, shift_int : shift_int + self.nt]
             else:
-                shifted_data = (
-                    self.PadOp.H
-                    * self.ShiftOps[i].H
-                    * x[:, shift_int : shift_int + self.nt + 1]
-                )
+                shifted_data = self.PadOp.rmatvec(
+                    self.ShiftOps[i].rmatvec(
+                        x[:, shift_int : shift_int + self.nt + 1].ravel()
+                    )
+                ).reshape(self.PadOp.dims)
                 deblended_data[i, :, :] = shifted_data
         return deblended_data