Changeset 514 for trunk/python

Timestamp:

02/28/05 15:29:06 (20 years ago)

Author:

mar637

Message:

moved auto_poly... to scantable

File:

• trunk/python/asaplinefind.py (modified) (1 diff)

trunk/python/asaplinefind.py

@@ -122 +122 @@
             return self.finder.getlinerangesinchannels()
 
-def auto_poly_baseline(scan, mask=None, edge=(0,0), order=0,
-    threshold=3,insitu=None):
-    """
-    Return a scan which has been baselined (all rows) by a polynomial.
-    Spectral lines are detected first using linefinder and masked out
-    to avoid them affecting the baseline solution.
-
-    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
-                    specified, the same number will be dropped from
-                    both sides of the spectrum. Default is to keep
-                    all channels
-        order:      the order of the polynomial (default is 0)
-        threshold:  the threshold used by line finder. It is better to
-                    keep it large as only strong lines affect the
-                    baseline solution.
-        insitu:     if False a new scantable is returned.
-                    Otherwise, the scaling is done in-situ
-                    The default is taken from .asaprc (False)
-
-    Example:
-        sc2=auto_poly_baseline(sc,order=7)
-    """
-    from asap.asapfitter import fitter
-    from asap import scantable
-
-    # setup fitter
-
-    f = fitter()
-    f._verbose(True)
-    f.set_function(poly=order)
-
-    # setup line finder
-
-    fl=linefinder()
-    fl.set_options(threshold=threshold)
-
-    if not insitu:
-        workscan=scan.copy()
-    else:
-        workscan=scan
-
-    vb=workscan._vb
-    # remember the verbose parameter and selection
-    workscan._vb=False
-    sel=workscan.get_cursor()
-    rows=range(workscan.nrow()) 
-    for i in range(workscan.nbeam()):
-        workscan.setbeam(i)
-        for j in range(workscan.nif()):
-            workscan.setif(j)
-            for k in range(workscan.npol()):
-                workscan.setpol(k)
-                if f._vb:
-                   print "Processing:"
-                   print 'Beam[%d], IF[%d], Pol[%d]' % (i,j,k)
-                for iRow in rows:
-                   fl.set_scan(workscan,mask,edge)
-                   fl.find_lines(iRow)
-                   f.set_scan(workscan, fl.get_mask())
-                   f.x=workscan._getabcissa(iRow)
-                   f.y=workscan._getspectrum(iRow)
-                   f.data=None
-                   f.fit()
-                   x=f.get_parameters()
-                   workscan._setspectrum(f.fitter.getresidual(),iRow)
-    workscan.set_cursor(sel[0],sel[1],sel[2])
-    workscan._vb = vb
-    if not insitu:
-       return workscan