diff --git a/tools/plot-tools/abacus_plot/dipole.py b/tools/plot-tools/abacus_plot/dipole.py index f3f7fc88a76..236e11e792a 100644 --- a/tools/plot-tools/abacus_plot/dipole.py +++ b/tools/plot-tools/abacus_plot/dipole.py @@ -113,11 +113,7 @@ def alpha(self,dirc: int): efield=self.padding(self.efield_data[dirc]) dipole_fft = np.fft.fft(dipole) efield_fft = np.fft.fft(efield) - numerator=dipole_fft*efield_fft.conjugate() - denominator=efield_fft*efield_fft.conjugate() - #added to avoid zero division - denominator=0.01*max(denominator)+denominator - alpha = np.abs((numerator/denominator).imag) + alpha = np.abs((dipole_fft/efield_fft).imag) return alpha def plot_abs(self, fig: Figure, ax: axes.Axes, directions: list = [0, 1, 2], x_range: list = [], unit: str = 'eV'):