AOT-biomaps 2.9.377__tar.gz → 2.9.378__tar.gz

{aot_biomaps-2.9.377 → aot_biomaps-2.9.378}/AOT_biomaps/AOT_Acoustic/_mainAcoustic.py

@@ -513,7 +513,7 @@ class AcousticField(ABC):
             dx = self.params['dx']
             air_margin = 20 
             if dx >= self.params['element_width']:
-                dx = self.params['element_width'] / 2
+                dx = self.params['element_width']/2
                 if self.params['width_phantom'] is not None:
                     pva_nx = int(np.round((self.params['width_phantom'])/dx))
                     Nx = pva_nx + 2 * air_margin
@@ -530,29 +530,46 @@ class AcousticField(ABC):
                 else:
                     Nx = int(round((self.params['Xrange'][1] - self.params['Xrange'][0]) / self.params['dx']))
                 if self.params['height_phantom'] is not None:
-                    Nz = int(np.round((self.params['height_phantom'])/self.params['dz']))
+                    Nz = int(np.round((self.params['height_phantom'])/self.params['dx']))
                 else:
-                    Nz = int(round((self.params['Zrange'][1] - self.params['Zrange'][0]) / self.params['dz']))
-           
+                    Nz = int(round((self.params['Zrange'][1] - self.params['Zrange'][0]) / self.params['dx']))
+            
+            Nx_final = int(round((self.params['Xrange'][1] - self.params['Xrange'][0]) / self.params['dx']))
+            Nz_final = int(round((self.params['Zrange'][1] - self.params['Zrange'][0]) / self.params['dx']))
             # --- 2. Time and space factors ---
             factorT = int(np.ceil(self.params['f_AQ'] / self.params['f_saving']))
-            factorX = int(np.ceil(Nx / self.params['Nx']))
-            factorZ = int(np.ceil(Nz / self.params['Nz']))
+            factorX = int(np.ceil(self.params['dx'] / dx))
+            factorZ = factorX
 
             # --- 3. Grid and source initialization ---
             kgrid = kWaveGrid([Nx, Nz], [dx, dx])
-            kgrid.setTime(self.kgrid.Nt, 1 / self.params['f_AQ'])
-            
-            medium = kWaveMedium()
-            medium.sound_speed = np.full((Nx, Nz), 343.0)
-            medium.density = np.full((Nx, Nz), 1.2)
-            x_start, x_end = air_margin, air_margin + pva_nx
-            medium.sound_speed[x_start:x_end, :] = self.params['c0'] # 1540 m/s
-            medium.density[x_start:x_end, :] = self.params.get('density', 1000.0)
+
+            # --- CORRECTIF : Définition du Medium avec hétérogénéité ---
+            # Initialisation en float32 (essentiel pour GPU)
+            c_map = np.full((Nx, Nz), 343.0, dtype=np.float32)  # Air par défaut
+            rho_map = np.full((Nx, Nz), 1.2, dtype=np.float32)
+
+            # Insertion du bloc de PVA
+            x_start = air_margin
+            x_end = air_margin + pva_nx
+            c_map[x_start:x_end, :] = self.params['c0']
+            rho_map[x_start:x_end, :] = self.params.get('density', 1000.0)
+
+            medium = kWaveMedium(sound_speed=c_map, density=rho_map)
+
+            # --- CORRECTIF : Recalcul du pas de temps (CFL) ---
+            # k-Wave a besoin d'un dt basé sur la vitesse MAXIMALE (1540 m/s)
+            c_mean = np.mean(c_map[:, 0]) # mean speed at z=0
+            c_max = np.max(c_map)
+            cfl = 0.3 # Valeur de sécurité
+            dt = cfl * dx / c_max
+            kgrid.setTime(self.kgrid.Nt, dt)
 
 
             source = kSource()
             source.p_mask = np.zeros((Nx, Nz))
+            # Appel à la méthode spécialisée
+            self._SetUpSource(self,source, Nx, dt, dx, c_mean,factorT)  # factorT=1 pour simplifier
 
             # --- 4. Sensor setup ---
             sensor = kSensor()
@@ -564,11 +581,14 @@ class AcousticField(ABC):
             # --- 7. Simulation options ---
             simulation_options = SimulationOptions(
             pml_inside=False, # PML ajoutée autour de la grille Air+PVA
-            pml_size=[pml_size, pml_size],
+            pml_size=[1, pml_size],
             use_sg=False,
             save_to_disk=True,
             input_filename=os.path.join(gettempdir(), "KwaveIN.h5"),
-            output_filename=os.path.join(gettempdir(), "KwaveOUT.h5")
+            output_filename=os.path.join(gettempdir(), "KwaveOUT.h5"),
+            smooth_c0 = True,
+            smooth_rho0 = True,
+            smooth_p0 = True
             )
 
             execution_options = SimulationExecutionOptions(
@@ -577,9 +597,6 @@ class AcousticField(ABC):
                 show_sim_log=show_log
             )
 
-            # --- 7. Call specialized function to set up source.p_mask and source.p ---
-            self._SetUpSource(source, Nx, dx, factorT)
-
             # --- 8. Run simulation ---
             sensor_data = kspaceFirstOrder2D(
                 kgrid=kgrid,
@@ -592,16 +609,18 @@ class AcousticField(ABC):
 
             # --- 9. Post-process results ---
             data = sensor_data['p'].reshape(kgrid.Nt, Nz, Nx)
-
             if factorT != 1 or factorX != 1 or factorZ != 1:
-                return reshape_field(data, [factorT, factorX, factorZ])
+                data = reshape_field(data, [factorT, factorX, factorZ])
+                xStart = (Nx//2)//factorX - (Nx_final//2)
+                return data[:, :Nz_final, xStart:xStart+Nx_final]
             else:
-                return data
+                return data[:, :Nz_final, xStart:xStart+Nx_final]
             
         except Exception as e:
             print(f"Error generating 2D acoustic field: {e}")
             return None
     
+    
     def _generate_acoustic_field_KWAVE_3D(self, isGPU=True, show_log=True):
         """
         Generate a 3D acoustic field using k-Wave.

{aot_biomaps-2.9.377 → aot_biomaps-2.9.378}/AOT_biomaps/__init__.py

@@ -85,7 +85,7 @@ from .AOT_Recon.AOT_PotentialFunctions.RelativeDifferences import *
 from .Config import config
 from .Settings import *
 
-__version__ = '2.9.377'
+__version__ = '2.9.378'
 __process__ = config.get_process()
 
 def initialize(process=None):
@@ -239,5 +239,6 @@ def initialize(process=None):
 
 
 
+
 
 

{aot_biomaps-2.9.377 → aot_biomaps-2.9.378}/AOT_biomaps.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: AOT_biomaps
-Version: 2.9.377
+Version: 2.9.378
 Summary: Acousto-Optic Tomography
 Home-page: https://github.com/LucasDuclos/AcoustoOpticTomography
 Author: Lucas Duclos

{aot_biomaps-2.9.377 → aot_biomaps-2.9.378}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: AOT_biomaps
-Version: 2.9.377
+Version: 2.9.378
 Summary: Acousto-Optic Tomography
 Home-page: https://github.com/LucasDuclos/AcoustoOpticTomography
 Author: Lucas Duclos

{aot_biomaps-2.9.377 → aot_biomaps-2.9.378}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='AOT_biomaps',
-    version='2.9.377',
+    version='2.9.378',
     packages=find_packages(),
     include_package_data=True,
     
@@ -397,5 +397,6 @@ setup(
 
 
 
+