def test_default_window_2D():
x = np.linspace(0, 1, 200).reshape(-1, 1)
heights = np.cos(2 * np.pi * x * 8.3) * np.ones((1, 200))
topography = Topography(heights, physical_sizes=(1, 1), periodic=True)
if True:
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.plot(*topography.power_spectrum_1D(), label="periodic=True")
ax.plot(*topography.power_spectrum_2D(nbins=20),
label="2D, periodic=True")
ax.set_xscale("log")
ax.set_yscale("log")
topography = Topography(heights, physical_sizes=(1, 1), periodic=False)
if True:
import matplotlib.pyplot as plt
ax.plot(*topography.power_spectrum_1D(), label="periodic=False")
ax.plot(*topography.power_spectrum_2D(nbins=20),
label="2D, periodic=False")
ax.set_xscale("log")
ax.set_yscale("log")
# ax.set_ylim(bottom=1e-6)
ax.legend()
plt.show(block=True)
def test_power_spectrum_1D():
X = np.arange(3).reshape(1, 3)
Y = np.arange(4).reshape(4, 1)
h = X + Y
t = Topography(h, (8, 6))
q1, C1 = t.power_spectrum_1D(window='hann')