Changeset 907 for trunk/python/asaplinefind.py

Timestamp:

03/20/06 15:17:06 (18 years ago)

Author:

vor010

Message:

LineFinder? & auto_poly_baseline: a support of
new scantable format has been added. Now findLines accept a mask and edge
parameters, which can therefore be different for different rows. Constructor
need now just a scan table. Boost types removed from STLineFinder. auto_poly_baseline can accept a nested tuple of edges, which would be interpreted as
different edge parameters for different IFs

File:

• trunk/python/asaplinefind.py (modified) (3 diffs)

trunk/python/asaplinefind.py

@@ -7 +7 @@
     Example:
        fl=linefinder()
-       fl.set_scan(sc,edge=(50,))
+       fl.set_scan(sc)
        fl.set_options(threshold=3)
-       nlines=fl.find_lines()
+       nlines=fl.find_lines(edge=(50,0))
        if nlines!=0:
           print "Found ",nlines," spectral lines"
@@ -65 +65 @@
         return
 
-    def set_scan(self, scan, mask=None, edge=(0,0)):
+    def set_scan(self, scan):
         """
         Set the 'data' (scantable) to work with.
         Parameters:
              scan:    a scantable
-             mask:       an optional mask retreived from scantable
-             edge:       an optional number of channel to drop at
-                         the edge of spectrum. If only one value is
+        """
+        if not scan:
+           raise RuntimeError, 'Please give a correct scan'
+        return
+    def find_lines(self,nRow=0,mask=None,edge=(0,0)):
+        """
+        Search for spectral lines in the scan assigned in set_scan.
+        Parameters:
+             nRow:       a row in the scantable to work with
+             mask:       an optional mask (e.g. retreived from scantable)
+             edge:       an optional number of channels to drop at
+                         the edge of the spectrum. If only one value is
                          specified, the same number will be dropped from
                          both sides of the spectrum. Default is to keep
                          all channels
+        A number of lines found will be returned
         """
-        if not scan:
-           raise RuntimeError, 'Please give a correct scan'
-        if not scan._check_ifs():
-           raise RuntimeError, 'IFs with different numbers of channels are not yet supported'
-
         if isinstance(edge,int):
            edge=(edge,)
@@ -96 +101 @@
         if mask is None:
             from numarray import ones
-            self.finder.setscan(scan,ones(scan.nchan(-1)),tuple(edge))
+            return self.finder.findlines(ones(scan.nchan(nRow)),list(edge),nRow)
         else:
-            self.finder.setscan(scan,mask,tuple(edge))
-        return
-    def find_lines(self,nRow=0):
-        """
-        Search for spectral lines in the scan assigned in set_scan.
-        Current Beam/IF/Pol is used, Row is specified by parameter
-        A number of lines found will be returned
-        """
-        return self.finder.findlines(nRow)
+            return self.finder.setscan(mask,list(edge),nRow)
     def get_mask(self,invert=False):
         """