From 77d16d0861723e6065c2c57f72fcb7e30207b77e Mon Sep 17 00:00:00 2001 From: Mike Hadmack Date: Fri, 17 Nov 2023 11:23:57 -1000 Subject: [PATCH] Update plot_intensity.py Stopped slice plotting from computing `I**2` so that the `square_root` option is respected. Updated slice plot label to match the 2D plot. --- diffractsim/visualization/plot_intensity.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/diffractsim/visualization/plot_intensity.py b/diffractsim/visualization/plot_intensity.py index 60b4c59..92dc4ae 100644 --- a/diffractsim/visualization/plot_intensity.py +++ b/diffractsim/visualization/plot_intensity.py @@ -116,8 +116,11 @@ def plot_intensity(self, I, square_root = False, figsize=(7, 6), x = self.x y = self.y - ax_slice.plot(x/units, I[np.argmin(abs(y-slice_y_pos)),:]**2) - ax_slice.set_ylabel(r'Intensity $\left[W / m^2 \right]$') + ax_slice.plot(x/units, I[np.argmin(abs(y-slice_y_pos)),:]) + if square_root == False: + ax_slice.set_ylabel(r'Intensity $\left[W / m^2 \right]$') + else: + ax_slice.set_ylabel(r'Square Root Intensity $\left[ \sqrt{W / m^2 } \right]$') if grid == True: ax_slice.grid(alpha =0.2) @@ -149,8 +152,11 @@ def plot_intensity(self, I, square_root = False, figsize=(7, 6), x = self.x y = self.y - ax_slice.plot(y/units, I[:, np.argmin(abs(x-slice_x_pos))]**2) - ax_slice.set_ylabel(r'Intensity $\left[W / m^2 \right]$') + ax_slice.plot(y/units, I[:, np.argmin(abs(x-slice_x_pos))]) + if square_root == False: + ax_slice.set_ylabel(r'Intensity $\left[W / m^2 \right]$') + else: + ax_slice.set_ylabel(r'Square Root Intensity $\left[ \sqrt{W / m^2 } \right]$') if grid == True: ax_slice.grid(alpha =0.2)