Changeset 992 for trunk/python

Timestamp:

04/06/06 10:12:03 (18 years ago)

Author:

mar637

Message:

added function to convert between polarisation types

Location:

trunk/python

Files:

• __init__.py (modified) (3 diffs)
• scantable.py (modified) (1 diff)

trunk/python/__init__.py

@@ -232 +232 @@
 
 def unique(x):
+    """
+    Return the unique values in a list
+    Parameters:
+        x:      the list to reduce
+    Examples:
+        x = [1,2,3,3,4]
+        print unique(x)
+        [1,2,3,4]
+    """
     return dict([ (val, 1) for val in x]).keys()
+
+def list_files(path=".",suffix="rpf"):
+    """
+    Return a list files readable by asap, such as rpf, sdfits, mbf, asap
+    Parameters:
+        path:     The directory to list (default '.')
+        suffix:   The file extension (default rpf)
+    Example:
+        files = list_files("data/","sdfits")
+        print files
+        ['data/2001-09-01_0332_P363.sdfits',
+        'data/2003-04-04_131152_t0002.sdfits',
+        'data/Sgr_86p262_best_SPC.sdfits']
+    """
+    import os
+    if not os.path.isdir(path):
+        return None
+    valid = "rpf sdf sdfits mbf asap".split()
+    if not suffix in valid:
+        return None
+    files = [os.path.expanduser(os.path.expandvars(path+"/"+f)) for f in os.listdir(path)]
+    return filter(lambda x: x.endswith(suffix),files)
 
 # workaround for ipython, which redirects this if banner=0 in ipythonrc
@@ -335 +366 @@
                               all polarisations will contain the
                               averaged spectrum.
+            convert_pol     - convert to a different polarisation type
             auto_quotient   - return the on/off quotient with
                               automatic detection of the on/off scans
@@ -417 +449 @@
         print               - print details about a variable
         list_scans          - list all scantables created bt the user
+        list_files          - list all files readable by asap (default rpf)
         del                 - delete the given variable from memory
         range               - create a list of values, e.g.

trunk/python/scantable.py

@@ -974 +974 @@
         if mask is None:
             mask = ()
-        s = self._math._averagepol(self, mask, weight)
+        s = scantable(self._math._averagepol(self, mask, weight))
         s._add_history("average_pol",varlist)
         print_log()
-        return scantable(s)
+        return s
+
+    def convert_pol(self, poltype=None):
+        """
+        Convert the data to a different polarisation type.
+        Parameters:
+            poltype:    The new polarisation type. Valid types are:
+                        "linear", "stokes" and "circular"
+        """
+        varlist = vars()
+        try:
+            s = scantable(self._math._convertpol(self, poltype))
+        except RuntimeError,msg:
+            if rcParams['verbose']:
+              print msg
+              return
+            else:
+                raise
+        s._add_history("convert_pol",varlist)
+        print_log()
+        return s
 
     def smooth(self, kernel="hanning", width=5.0, insitu=None):