shampoo_example.py

from __future__ import (absolute_import, division, print_function,
                        unicode_literals)

from shampoo.reconstruction import Hologram
import matplotlib.pyplot as plt
hologram_path = 'data/Hologram.tif'
propagation_distance = 0.0369  # m

# Full res: 2 s, Rebin 2x2: 0.5 s, 4x4: 0.125

binning = 1
if binning == 1:
    n = 1024
    interval = 4
    import numpy as np
    h = Hologram(hologram_path, background_interval=interval)
    waveform = h.reconstruct(propagation_distance,
                             plot_aberration_correction=False)
elif binning == 2:
    import numpy as np
    n = 512
    interval = 2
    h = Hologram(hologram_path, rebin_factor=2, background_interval=interval)
    waveform = h.reconstruct(propagation_distance,
                             plot_aberration_correction=False)

fig, ax = waveform.plot(all=True)
plt.show()