introduce in_place

Author: Yoshinori Sueno

sotodlib/preprocess/processes.py

@@ -1273,7 +1273,7 @@ def save(self, proc_aman, source_aman):
         if self.save_cfgs:
             proc_aman.wrap(self.source_flags_name, source_aman)
 
-    def select(self, meta, proc_aman=None):
+    def select(self, meta, proc_aman=None, in_place=True):
         if self.select_cfgs is None:
             return meta
         if proc_aman is None:
@@ -1300,13 +1300,18 @@ def select(self, meta, proc_aman=None):
         elif len(kinds) != len(select_list):
             raise ValueError("Length of kinds must match length of select_source, or just str")
 
-        keep = np.zeros(meta.dets.count)
+        keep_all = np.ones(meta.dets.count, dtype=bool)
+
         for source, kind, cut in zip(select_list, kinds, cuts):
             if source in source_flags._fields:
-                keep += flag_cut_select(source_flags[source], cut, kind)
-        meta.restrict("dets", meta.dets.vals[keep.astype('bool')])
-        source_flags.restrict("dets", source_flags.dets.vals[keep.astype('bool')])
-        return meta
+                keep_all &= flag_cut_select(source_flags[source], cut, kind)
+
+        if in_place:
+            meta.restrict("dets", meta.dets.vals[keep_all])
+            source_flags.restrict("dets", source_flags.dets.vals[keep_all])
+            return meta
+        else:
+            return keep_all
 
 class HWPAngleModel(_Preprocess):
     """Apply hwp angle model to the TOD.