Added obj_flag check for ngmix catalogs.

sweverett · Apr 6, 2018 · d774700 · d774700
1 parent 4a87afc
commit d774700
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/balrog/ngmix_catalog.py b/balrog/ngmix_catalog.py
@@ -226,6 +226,7 @@ def getFlags(self):
 
         # General flags
         self.flags = self.catalog['flags']
+        self.obj_flags = self.catalog['obj_flags']
 
         # ngmix catalog-specific flags
         self.ngmix_flags = self.catalog[self.col_prefix+'_flags']
@@ -251,6 +252,7 @@ def makeMask(self):
 
         # For now, remove objects with any flags present
         mask[self.flags != 0] = False
+        mask[self.obj_flags !=0] = False
         mask[self.ngmix_flags !=0] = False
         # Extra flags for 'mof' catalogs
         if self.cat_type == 'mof':