fix q padding with non zero t0

slimgroup · Mar 19, 2024 · 4f8b15e · 4f8b15e
1 parent 000d669
commit 4f8b15e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/TimeModeling/Modeling/python_interface.jl b/src/TimeModeling/Modeling/python_interface.jl
@@ -56,6 +56,7 @@ function devito_interface(modelPy::PyObject, srcGeometry::Geometry, srcData::Arr
     # Interpolate input data to computational grid
     dtComp = convert(Float32, modelPy."critical_dt")
     qIn = time_resample(srcData, srcGeometry, dtComp)
+    qIn = _maybe_pad_t0(qIn, srcGeometry, recGeometry)
 
     # Set up coordinates with devito dimensions
     src_coords = setup_grid(srcGeometry, modelPy.shape)
@@ -109,6 +110,7 @@ function devito_interface(modelPy::PyObject, srcGeometry::Geometry, srcData::Arr
     # Interpolate input data to computational grid
     dtComp = convert(Float32, modelPy."critical_dt")
     qIn = time_resample(srcData, srcGeometry, dtComp)
+    qIn = _maybe_pad_t0(qIn, srcGeometry, recGeometry)
 
     # Set up coordinates with devito dimensions
     #origin = get_origin(modelPy)
@@ -167,6 +169,7 @@ function devito_interface(modelPy::PyObject, recGeometry::Geometry, recData::Arr
     dtComp = convert(Float32, modelPy."critical_dt")
     dIn = time_resample(recData, recGeometry, dtComp)
     qIn = time_resample(srcData, recGeometry, dtComp)
+    qIn, dIn = _maybe_pad_t0(qIn, recGeometry, dIn, recGeometry)
 
     # Set up coordinates with devito dimensions
     rec_coords = setup_grid(recGeometry, modelPy.shape)

diff --git a/src/TimeModeling/Utils/time_utilities.jl b/src/TimeModeling/Utils/time_utilities.jl
@@ -139,3 +139,6 @@ end
 
 _maybe_pad_t0(qIn::judiVector{T, AT}, dObserved::judiVector{T, AT}) where{T<:Number, AT} =
     _maybe_pad_t0(get_data(qIn), get_data(dObserved))
+
+_maybe_pad_t0(qIn::Matrix{T}, qGeom::Geometry, dataGeom::Geometry) where T<:Number =
+    _maybe_pad_t0(qIn, qGeom, zeros(T, dataGeom.nt[1], 1), dataGeom)[1]