From 101c9d8b6b8953fa223d665c025e4ac8e89f7485 Mon Sep 17 00:00:00 2001 From: lyb9812 <88070104+lyb9812@users.noreply.github.com> Date: Wed, 1 Nov 2023 17:12:29 +0800 Subject: [PATCH 1/2] Update dipole.py --- tools/plot-tools/abacus_plot/dipole.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tools/plot-tools/abacus_plot/dipole.py b/tools/plot-tools/abacus_plot/dipole.py index f3f7fc88a76..5c610cab680 100644 --- a/tools/plot-tools/abacus_plot/dipole.py +++ b/tools/plot-tools/abacus_plot/dipole.py @@ -113,10 +113,8 @@ 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 + numerator=dipole_fft + denominator=efield_fft alpha = np.abs((numerator/denominator).imag) return alpha From 3c1d27d066b80a6f4d964a87196a9990f5ca5d3c Mon Sep 17 00:00:00 2001 From: lyb9812 <88070104+lyb9812@users.noreply.github.com> Date: Wed, 1 Nov 2023 17:16:41 +0800 Subject: [PATCH 2/2] Update dipole.py --- tools/plot-tools/abacus_plot/dipole.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tools/plot-tools/abacus_plot/dipole.py b/tools/plot-tools/abacus_plot/dipole.py index 5c610cab680..236e11e792a 100644 --- a/tools/plot-tools/abacus_plot/dipole.py +++ b/tools/plot-tools/abacus_plot/dipole.py @@ -113,9 +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 - denominator=efield_fft - 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'):