where OmegaXY=0.0, mask as outside disk [set nan]

1 files changed, 2 insertions, 1 deletions

diff --git a/DiskXYZ.py b/DiskXYZ.py
index e743287..605f32f 100644
--- a/DiskXYZ.py
+++ b/DiskXYZ.py
@@ -373,6 +373,7 @@ class DiskReduce(DiskReduceBase):
     def integrate_to_rho_xy(self):
         """Z-Integrate Volume Density. Gives Surface Density."""
         self.rho_xy = self.integrate(self.dset["rho"])
+        self.rho_xy[np.isnan(self.Omega_xy)] = np.nan
         self.rho_xy_max = np.nanmax(self.rho_xy)
         self.rho_xy_min = np.nanmin(self.rho_xy)
 
@@ -528,9 +529,9 @@ class DiskReduce(DiskReduceBase):
     def reduce_all(self):
         self.average_velocities_to_xy()
         self.average_velocities_to_r()
+        self.compute_Omega_xy()
         self.integrate_to_P_xy()
         self.integrate_to_rho_xy()
-        self.compute_Omega_xy()
         self.compute_cs2_xy()
         self.compute_T_mid_xy()
         self.compute_h_xy()