Plot improvements

shimmingtoolbox/shim/sequencer.py

@@ -602,21 +602,23 @@ def plot_full_mask(self, unshimmed, shimmed_masked, mask):
         metric_shimmed_std = calculate_metric_within_mask(shimmed_masked, mask, metric='std')
         metric_unshimmed_mean = calculate_metric_within_mask(unshimmed, mask, metric='mean')
         metric_shimmed_mean = calculate_metric_within_mask(shimmed_masked, mask, metric='mean')
-        metric_unshimmed_absmean = calculate_metric_within_mask(np.abs(unshimmed), mask, metric='mean')
-        metric_shimmed_absmean = calculate_metric_within_mask(np.abs(shimmed_masked), mask, metric='mean')
+        metric_unshimmed_mae = calculate_metric_within_mask(unshimmed, mask, metric='mae')
+        metric_shimmed_mae = calculate_metric_within_mask(shimmed_masked, mask, metric='mae')
+        metric_unshimmed_rmse = calculate_metric_within_mask(unshimmed, mask, metric='rmse')
+        metric_shimmed_rmse = calculate_metric_within_mask(shimmed_masked, mask, metric='rmse')
 
         min_value = min(mt_unshimmed_masked.min(), mt_shimmed_masked.min())
         max_value = max(mt_unshimmed_masked.max(), mt_shimmed_masked.max())
 
-        fig = Figure(figsize=(9, 6))
+        fig = Figure(figsize=(8, 5))
         fig.suptitle(f"Fieldmaps\nFieldmap Coordinate System")
 
         ax = fig.add_subplot(1, 2, 1)
         ax.imshow(mt_unshimmed, cmap='gray')
         mt_unshimmed_masked[mt_unshimmed_masked == 0] = np.nan
         im = ax.imshow(mt_unshimmed_masked, vmin=min_value, vmax=max_value, cmap='viridis')
-        ax.set_title(f"Before shimming\nSTD: {metric_unshimmed_std:.3}, mean: {metric_unshimmed_mean:.3}, "
-                     f"abs mean: {metric_unshimmed_absmean:.3}")
+        ax.set_title(f"Before shimming\nstd: {metric_unshimmed_std:.3}, mean: {metric_unshimmed_mean:.3}\n"
+                     f"mae: {metric_unshimmed_mae:.3}, rmse: {metric_unshimmed_rmse:.3}")
         ax.get_xaxis().set_visible(False)
         ax.get_yaxis().set_visible(False)
         divider = make_axes_locatable(ax)
@@ -627,17 +629,14 @@ def plot_full_mask(self, unshimmed, shimmed_masked, mask):
         ax.imshow(mt_unshimmed, cmap='gray')
         mt_shimmed_masked[mt_shimmed_masked == 0] = np.nan
         im = ax.imshow(mt_shimmed_masked, vmin=min_value, vmax=max_value, cmap='viridis')
-        ax.set_title(f"After shimming\nSTD: {metric_shimmed_std:.3}, mean: {metric_shimmed_mean:.3}, "
-                     f"abs mean: {metric_shimmed_absmean:.3}")
+        ax.set_title(f"After shimming\nstd: {metric_shimmed_std:.3}, mean: {metric_shimmed_mean:.3}\n"
+                     f"mae: {metric_shimmed_mae:.3}, rmse: {metric_shimmed_rmse:.3}")
         ax.get_xaxis().set_visible(False)
         ax.get_yaxis().set_visible(False)
         divider = make_axes_locatable(ax)
         cax = divider.append_axes('right', size='5%', pad=0.05)
         fig.colorbar(im, cax=cax)
 
-        # Lower suptitle
-        fig.subplots_adjust(top=0.85)
-
         # Save
         fname_figure = os.path.join(self.path_output, 'fig_shimmed_vs_unshimmed.png')
         fig.savefig(fname_figure, bbox_inches='tight')