From 12daf7588aa55078fcd67315ccf29ffa0851dd1d Mon Sep 17 00:00:00 2001 From: Noe Brucy Date: Fri, 23 Jul 2021 00:09:58 +0200 Subject: [PATCH] [plotter] fix mask bug in particles + new default for tight_layout --- params.yml | 3 ++- plotter.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/params.yml b/params.yml index 14d4a69..58efde0 100644 --- a/params.yml +++ b/params.yml @@ -7,7 +7,8 @@ plot : # Plot parameters # Overlays vel_red : 40 # Take point each vel_red for velocities - time_fmt : "time = {:.3g} {}" # Time format string, 1st field is time and 2nd is unit + tight_layout : False # Wheter to apply tight_layout + time_fmt : "time = {:.3g} {}" # Time format string, 1st field is time and 2nd is unit disk: # Disk speficic parameters enable : False # Enable specific disk analysis diff --git a/plotter.py b/plotter.py index 48858fe..d640a0d 100644 --- a/plotter.py +++ b/plotter.py @@ -386,7 +386,7 @@ class Plotter(Aggregator, BaseProcessor): if self._needs_computation(overwrite, plot_filename): plot_info = rule.process(arg, **kwargs) - if not self.params.out.interactive and close: + if self.params.plot.tight_layout and close: plt.tight_layout(pad=1) if self.params.out.save: @@ -838,7 +838,7 @@ class Plotter(Aggregator, BaseProcessor): part_v = part_v[mask] # Scatter plot - plt.scatter(part_h, part_v, s=mass / 5e3, **kwargs) + plt.scatter(part_h, part_v, s=mass[mask] / 5e3, **kwargs) def _overlay_speed( self, ax_los, im_extent, unit=U.km_s, unit_coeff=1.0, key_v=None, **kwargs