fftfreq.py
56e1e9fb1c3cci-fix-static-checks

fftfreq.py
View Options

			from __future__ import division

			__author__ = "Lucas Frérot"

			import numpy as np
			from numpy.fft import fftfreq


			def frequencies1D(frequencies):
			n = len(frequencies)
			frequencies[:] = n * fftfreq(n)


			def assign2DFrequencies(qx, qy, frequencies):
			qx, qy = np.meshgrid(qx, qy, indexing='ij')

			frequencies[:, :, 0] = qx
			frequencies[:, :, 1] = qy


			def frequencies2D(frequencies):
			n = frequencies.shape
			qx = fftfreq(n[0]) * n[0]
			qy = fftfreq(n[1]) * n[1]
			assign2DFrequencies(qx, qy, frequencies)


			def hfrequencies2D(frequencies):
			n = frequencies.shape
			qx = fftfreq(n[0]) * n[0]
			qy = np.arange(n[1])
			assign2DFrequencies(qx, qy, frequencies)