Changeset 1576 for trunk/python/scantable.py

Timestamp:

06/29/09 13:40:55 (15 years ago)

Author:

Malte Marquarding

Message:

Ticket #169: allow direct settings of selections

File:

• trunk/python/scantable.py (modified) (2 diffs)

trunk/python/scantable.py

@@ -312 +312 @@
         return selector(self._getselection())
 
-    def set_selection(self, selection=selector()):
+    def set_selection(self, selection=None, **kw):
         """
         Select a subset of the data. All following operations on this scantable
@@ -325 +325 @@
             scan.set_selection()     # unset the selection
         """
+        if selection is None:
+            # reset
+            if len(kw) == 0:
+                selection = selector()
+            else:
+                # try keywords
+                for k in kw:
+                    if k not in selector.fields:
+                        raise KeyError("Invalid selection key '%s', valid keys are %s" % (k, selector.fields))
+                selection = selector(**kw)
         self._setselection(selection)