Coverage for kwave/kWaveSimulation_helper/retract_transducer_grid_size.py: 18%

1import numpy as np 

2 

3from kwave.ktransducer import NotATransducer 

4 

5 

6def retract_transducer_grid_size(source, sensor, retract_size, pml_inside: bool): 

7 # resize the transducer object if the grid has been expanded 

8 is_source_kwave_transducer = isinstance(source, NotATransducer) 

9 is_sensor_kwave_transducer = isinstance(sensor, NotATransducer) 

10 retract_size = np.array(retract_size) 

11 

12 if not pml_inside and (is_source_kwave_transducer or is_sensor_kwave_transducer): 

13 # check if the sensor is a transducer 

14 if is_sensor_kwave_transducer: 

15 # retract the transducer mask 

16 sensor.retract_grid(retract_size) 

17 

18 # check if the source is a transducer, and if so, and different 

19 # transducer to the sensor 

20 if is_source_kwave_transducer and not (is_sensor_kwave_transducer and sensor == source): 

21 # retract the transducer mask 

22 source.retract_grid(retract_size)