From ccb246bb15b3732e2cb55f3484346a1d8fd732bb Mon Sep 17 00:00:00 2001
From: olender <olender@usp.br>
Date: Sun, 25 Aug 2024 11:44:35 -0300
Subject: [PATCH] creating real outside of fwi

---
 test_small_marmousi.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/test_small_marmousi.py b/test_small_marmousi.py
index 1721c37c..1d636a11 100644
--- a/test_small_marmousi.py
+++ b/test_small_marmousi.py
@@ -1,4 +1,5 @@
 import spyro
+from copy import deepcopy
 import sys
 
 
@@ -25,7 +26,6 @@
     "Lz": 3.5,  # depth in km - always positive   # Como ver isso sem ler a malha?
     "Lx": 17.0,  # width in km - always positive
     "Ly": 0.0,  # thickness in km - always positive
-    "mesh_file": "real5hz.msh",
 }
 dictionary["acquisition"] = {
     "source_type": "ricker",
@@ -64,11 +64,13 @@
 
 
 def test_real_shot_record_generation_parallel():
-    dictionary["mesh"]["mesh_file"] = "meshes/real5hz.msh"
+    real_dictionary = deepcopy(dictionary)
+    real_dictionary["mesh"]["mesh_file"] = "meshes/real5hz.msh"
     
-    fwi = spyro.FullWaveformInversion(dictionary=dictionary)
-    fwi.set_real_velocity_model(new_file="velocity_models/vp_marmousi-ii.hdf5")
-    fwi.generate_real_shot_record(plot_model=True, save_shot_record=True)
+    real_wave = spyro.AcousticWave(dictionary=real_dictionary)
+    real_wave.set_initial_velocity_model(new_file="velocity_models/vp_marmousi-ii.hdf5")
+    real_wave.forward_solve()
+    spyro.io.save_shots(real_wave)
 
 
 def test_realistic_fwi():