From 9cfaf9c79324682ae55b783c49702da4edd89fc9 Mon Sep 17 00:00:00 2001
From: olender <olender@usp.br>
Date: Thu, 17 Oct 2024 13:08:15 -0300
Subject: [PATCH] debugging

---
 initial_guess_cpw.py | 97 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 initial_guess_cpw.py

diff --git a/initial_guess_cpw.py b/initial_guess_cpw.py
new file mode 100644
index 00000000..10c553c6
--- /dev/null
+++ b/initial_guess_cpw.py
@@ -0,0 +1,97 @@
+import firedrake as fire
+import numpy as np
+
+import spyro
+
+L = 500  # [m]
+
+rho = 7850          # [kg/m3]
+lambda_in = 6.86e9  # [Pa]
+lambda_out = 9.88e9  # [Pa]
+mu_in = 3.86e9      # [Pa]
+mu_out = 5.86e9     # [Pa]
+
+smag = 1e6
+freq = 2  # Central frequency of Ricker wavelet [Hz]
+hf = 90  # [m]
+hs = 100  # [m]
+source_locations = spyro.create_transect((-hf, 0.2*L), (-hf, 0.8*L), 3)
+receiver_locations = spyro.create_transect((-hs, 0), (-hs, L), 40)
+source_locations = [[-hf, 0.5*L]]
+
+time_step = 2e-4  # [s]
+final_time = 1.5  # [s]
+out_freq = int(0.01/time_step)
+
+n = 20
+mesh = fire.RectangleMesh(n, n, 0, L, originX=-L, diagonal='crossed')
+z, x = fire.SpatialCoordinate(mesh)
+
+zc = 250  # [m]
+xc = 250  # [m]
+ri = 50  # [m]
+camembert = lambda v_inside, v_outside: fire.conditional(
+    (z - zc) ** 2 + (x - xc) ** 2 < ri**2, v_inside, v_outside)
+
+d = {}
+
+d["options"] = {
+    "cell_type": "T",
+    "variant": "lumped",
+    "degree": 4,
+    "dimension": 2,
+}
+
+d["parallelism"] = {
+    "type": "automatic",
+}
+
+d["mesh"] = {
+    "user_mesh": mesh,
+}
+
+d["acquisition"] = {
+    "source_type": "ricker",
+    "source_locations": source_locations,
+    "frequency": freq,
+    "delay": 0,
+    "delay_type": "time",
+    "amplitude": np.array([0, smag]),
+    #"amplitude": smag * np.eye(2),
+    #"amplitude": smag * np.array([[0, 1], [-1, 0]]),
+    "receiver_locations": receiver_locations,
+}
+
+d["synthetic_data"] = {
+    "type": "object",
+    "density": fire.Constant(rho),
+    "lambda": camembert(lambda_in, lambda_out),
+    "mu": camembert(mu_in, mu_out),
+    "real_velocity_file": None,
+}
+
+d["time_axis"] = {
+    "initial_time": 0,
+    "final_time": final_time,
+    "dt": time_step,
+    "output_frequency": out_freq,
+    "gradient_sampling_frequency": 1,
+}
+
+d["visualization"] = {
+    "forward_output": True,
+    "forward_output_filename": "results/forward_output.pvd",
+    "fwi_velocity_model_output": False,
+    "gradient_output": False,
+    "adjoint_output": False,
+    "debug_output": False,
+}
+
+d["absorving_boundary_conditions"] = {
+    "status": True,
+    "damping_type": "local",
+}
+
+wave = spyro.IsotropicWave(d)
+wave.set_mesh(user_mesh=mesh, mesh_parameters={})
+wave.forward_solve()