Merge pull request #75 from SWIFTSIM/centrals-mask

Added mask for centrals and satellites to the catalogue.

Author: JBorrow

velociraptor/catalogue/catalogue.py

@@ -412,3 +412,17 @@ def register_derived_quantities(self, registration_file_path: str) -> None:
         self.derived_quantities = DerivedQuantities(registration_file_path, self)
 
         return
+
+    @property
+    def centrals(self):
+        if hasattr(self.structure_type, "structuretype"):
+            return self.structure_type.structuretype == 10
+        else:
+            return np.s_[:]
+
+    @property
+    def satellites(self):
+        if hasattr(self.structure_type, "structuretype"):
+            return self.structure_type.structuretype != 10
+        else:
+            return np.s_[:]