acoular 24.3__py3-none-any.whl → 24.7__py3-none-any.whl

acoular/demo/acoular_demo.py

@@ -1,80 +1,98 @@
-#! /usr/bin/env python
-# -*- coding: utf-8 -*-
-#pylint: disable-msg=E0611, E1101, C0103, R0901, R0902, R0903, R0904, W0232
-#------------------------------------------------------------------------------
+# ------------------------------------------------------------------------------
 # Copyright (c) Acoular Development Team.
-#------------------------------------------------------------------------------
-"""Demo for Acoular
+# ------------------------------------------------------------------------------
+"""Demo for Acoular.
+
+To run the demo, execute the following commands:
+
+.. code-block:: python
+
+    import acoular
+
+    acoular.demo.acoular_demo.run()
+
 
 Generates a test data set for three sources, analyzes them and generates a
 map of the three sources.
- 
+
 The simulation generates the sound pressure at 64 microphones that are
 arrangend in the 'array64' geometry, which is part of the package. The sound
 pressure signals are sampled at 51200 Hz for a duration of 1 second.
 
-Source location (relative to array center) and levels:
+Source location (relative to array center) and RMS in 1 m distance:
 
 ====== =============== ======
-Source Location        Level 
+Source Location        RMS
 ====== =============== ======
 1      (-0.1,-0.1,0.3) 1.0 Pa
-2      (0.15,0,0.3)    0.7 Pa 
+2      (0.15,0,0.3)    0.7 Pa
 3      (0,0.1,0.3)     0.5 Pa
 ====== =============== ======
+
 """
 
+
 def run():
+    """Run the Acoular demo."""
+    from pathlib import Path
+
+    import acoular as ac
+
+    ac.config.global_caching = 'none'
 
-    from os import path
-    from acoular import __file__ as bpath, MicGeom, WNoiseGenerator, PointSource,\
-     Mixer, WriteH5, TimeSamples, PowerSpectra, RectGrid, SteeringVector,\
-     BeamformerBase, L_p
-    from pylab import figure, plot, axis, imshow, colorbar, show
-    
     # set up the parameters
-    sfreq = 51200 
+    sfreq = 51200
     duration = 1
-    nsamples = duration*sfreq
-    micgeofile = path.join(path.split(bpath)[0],'xml','array_64.xml')
+    nsamples = duration * sfreq
+    micgeofile = Path(ac.__file__).parent / 'xml' / 'array_64.xml'
     h5savefile = 'three_sources.h5'
-    
+
     # generate test data, in real life this would come from an array measurement
-    mg = MicGeom( from_file=micgeofile )
-    n1 = WNoiseGenerator( sample_freq=sfreq, numsamples=nsamples, seed=1 )
-    n2 = WNoiseGenerator( sample_freq=sfreq, numsamples=nsamples, seed=2, rms=0.7 )
-    n3 = WNoiseGenerator( sample_freq=sfreq, numsamples=nsamples, seed=3, rms=0.5 )
-    p1 = PointSource( signal=n1, mics=mg,  loc=(-0.1,-0.1,0.3) )
-    p2 = PointSource( signal=n2, mics=mg,  loc=(0.15,0,0.3) )
-    p3 = PointSource( signal=n3, mics=mg,  loc=(0,0.1,0.3) )
-    pa = Mixer( source=p1, sources=[p2,p3] )
-    wh5 = WriteH5( source=pa, name=h5savefile )
+    mg = ac.MicGeom(from_file=micgeofile)
+    n1 = ac.WNoiseGenerator(sample_freq=sfreq, numsamples=nsamples, seed=1)
+    n2 = ac.WNoiseGenerator(sample_freq=sfreq, numsamples=nsamples, seed=2, rms=0.7)
+    n3 = ac.WNoiseGenerator(sample_freq=sfreq, numsamples=nsamples, seed=3, rms=0.5)
+    p1 = ac.PointSource(signal=n1, mics=mg, loc=(-0.1, -0.1, 0.3))
+    p2 = ac.PointSource(signal=n2, mics=mg, loc=(0.15, 0, 0.3))
+    p3 = ac.PointSource(signal=n3, mics=mg, loc=(0, 0.1, 0.3))
+    pa = ac.Mixer(source=p1, sources=[p2, p3])
+    wh5 = ac.WriteH5(source=pa, name=h5savefile)
     wh5.save()
-    
+
     # analyze the data and generate map
-    
-    ts = TimeSamples( name=h5savefile )
-    ps = PowerSpectra( time_data=ts, block_size=128, window='Hanning' )
-    
-    rg = RectGrid( x_min=-0.2, x_max=0.2, y_min=-0.2, y_max=0.2, z=0.3, \
-    increment=0.01 )
-    st = SteeringVector(grid=rg, mics=mg)
-    
-    bb = BeamformerBase( freq_data=ps, steer=st )
-    pm = bb.synthetic( 8000, 3 )
-    Lm = L_p( pm )
-    
-    # show map
-    imshow( Lm.T, origin='lower', vmin=Lm.max()-10, extent=rg.extend(), \
-    interpolation='bicubic')
-    colorbar()
-    
-    # plot microphone geometry
-    figure(2)
-    plot(mg.mpos[0],mg.mpos[1],'o')
-    axis('equal')
-    
-    show()
-
-if __name__ == "__main__":
+
+    ps = ac.PowerSpectra(time_data=pa, block_size=128, window='Hanning')
+
+    rg = ac.RectGrid(x_min=-0.2, x_max=0.2, y_min=-0.2, y_max=0.2, z=0.3, increment=0.01)
+    st = ac.SteeringVector(grid=rg, mics=mg)
+
+    bb = ac.BeamformerBase(freq_data=ps, steer=st)
+    pm = bb.synthetic(8000, 3)
+    spl = ac.L_p(pm)
+
+    if ac.config.have_matplotlib:
+        from pylab import axis, colorbar, figure, imshow, plot, show
+
+        # show map
+        imshow(spl.T, origin='lower', vmin=spl.max() - 10, extent=rg.extend(), interpolation='bicubic')
+        colorbar()
+
+        # plot microphone geometry
+        figure(2)
+        plot(mg.mpos[0], mg.mpos[1], 'o')
+        axis('equal')
+
+        show()
+
+    else:
+        print('Matplotlib not found! Please install matplotlib if you want to plot the results.')
+        print('For consolation we do an ASCII map plot of the results here.')
+        grayscale = '@%#*+=-:. '[::-1]
+        ind = ((spl.T - spl.max() + 9).clip(0, 9)).astype(int)[::-1]
+        print(78 * '-')
+        print('|\n'.join([' '.join(['|'] + [grayscale[i] for i in row[2:-1]]) for row in ind]) + '|')
+        print(7 * '-', ''.join([f'{grayscale[i]}={int(spl.max())-9+i}dB ' for i in range(1, 10)]), 6 * '-')
+
+
+if __name__ == '__main__':
     run()