Merge pull request #82 from slimgroup/vtifix

Fix VTI FD

Project.toml

@@ -1,7 +1,7 @@
 name = "JUDI"
 uuid = "f3b833dc-6b2e-5b9c-b940-873ed6319979"
 authors = ["Philipp Witte, Mathias Louboutin"]
-version = "2.6.2"
+version = "2.6.3"
 
 [deps]
 DSP = "717857b8-e6f2-59f4-9121-6e50c889abd2"

src/pysource/FD_utils.py

@@ -21,9 +21,13 @@ def divs(func, so_fact=1, side=-1):
     GrDivergenceadient shifted by half a grid point, only to be used in combination
     with grads.
     """
-    so = func.space_order // so_fact
-    return sum([getattr(func[i], 'd%s' % d.name)(x0=d + side * d.spacing/2, fd_order=so)
-                for i, d in enumerate(func.space_dimensions)])
+    res = 0
+    zfd = lambda *ar, **kw: 0
+    for i, d in enumerate(func.space_dimensions):
+        so = getattr(func[i], 'space_order', 0) // so_fact
+        res += getattr(func[i], 'd%s' % d.name, zfd)(x0=d+side*d.spacing/2, fd_order=so)
+
+    return res
 
 
 def laplacian(v, irho):

test/utils.jl

@@ -43,7 +43,7 @@ function setup_model(tti=false, nlayer=2; n=(301, 151), d=(10., 10.))
         println("TTI Model")
         epsilon = smooth((v0[:, :] .- 1.5f0)/12f0, 3)
         delta =  smooth((v0[:, :] .- 1.5f0)/14f0, 3)
-        theta =  smooth((v0[:, :] .- 1.5f0)/4, 3)
+        theta =  smooth((v0[:, :] .- 1.5f0)/4, 3) .* rand([0, 1]) # makes sure VTI is tested
         model0 = Model(n, d, o, m0; epsilon=epsilon, delta=delta, theta=theta)
         model = Model(n, d, o, m; epsilon=epsilon, delta=delta, theta=theta)
     else