Changeset 1924 for branches/polybatch/python

Timestamp:

09/14/10 12:17:16 (14 years ago)

Author:

Malte Marquarding

Message:

Ticket #206: use STFitEntry as return objetc instead of pointer wrnagling.

File:

• branches/polybatch/python/scantable.py (modified) (9 diffs)

branches/polybatch/python/scantable.py

@@ -1158 +1158 @@
         return msk
 
-    def get_masklist(self, mask=None, row=0):
+    def get_masklist(self, mask=None, row=0, silent=False):
         """\
         Compute and return a list of mask windows, [min, max].
@@ -1192 +1192 @@
         if not i:
             msg += "\nThis mask is only valid for IF=%d" % (self.getif(i))
-        asaplog.push(msg)
+        if not silent:
+            asaplog.push(msg)
         masklist=[]
         ist, ien = None, None
@@ -1934 +1935 @@
                 workscan._setspectrum(f.fitter.getresidual(), r)
                 self.blpars.append(f.get_parameters())
-                self.masklists.append(workscan.get_masklist(f.mask, row=r))
+                self.masklists.append(workscan.get_masklist(f.mask, row=r, silent=True))
                 self.actualmask.append(f.mask)
 
@@ -1949 +1950 @@
             raise RuntimeError(msg)
 
-
-    def poly_baseline(self, mask=None, order=0, plot=False, batch=False, insitu=None, rows=None):
+    @asaplog_post_dec
+    def poly_baseline(self, mask=None, order=0, plot=False, batch=False,
+                      insitu=None, rows=None):
         """\
         Return a scan which has been baselined (all rows) by a polynomial.
@@ -1973 +1975 @@
         """
         if insitu is None: insitu = rcParams["insitu"]
+        varlist = vars()
         if insitu:
             workscan = self
@@ -1978 +1981 @@
             workscan = self.copy()
 
-        varlist = vars()
         nchan = workscan.nchan()
         
@@ -1991 +1993 @@
             
             if len(rows) > 0:
-                self.blpars = []
-                self.masklists = []
-                self.actualmask = []
+                workscan.blpars = []
+                workscan.masklists = []
+                workscan.actualmask = []
 
             if batch:
-                for r in rows:
-                    workscan._poly_baseline_batch(mask, order, r)
+                workscan._poly_baseline_batch(mask, order)
             elif plot:
                 f = fitter()
@@ -2016 +2017 @@
                         continue
                     workscan._setspectrum(f.fitter.getresidual(), r)
-                    self.blpars.append(f.get_parameters())
-                    self.masklists.append(workscan.get_masklist(f.mask, row=r))
-                    self.actualmask.append(f.mask)
+                    workscan.blpars.append(f.get_parameters())
+                    workscan.masklists.append(workscan.get_masklist(f.mask, row=r))
+                    workscan.actualmask.append(f.mask)
                     
                 f._p.unmap()
                 f._p = None
             else:
-                import array
                 for r in rows:
-                    pars = array.array("f", [0.0 for i in range(order+1)])
-                    pars_adr = pars.buffer_info()[0]
-                    pars_len = pars.buffer_info()[1]
+                    fitparams = workscan._poly_baseline(mask, order, r)
+                    params = fitparams.getparameters()
+                    fmtd = ", ".join(["p%d = %3.6f" % (i, v) for i, v in enumerate(params)])
+                    errors = fitparams.geterrors()
+                    fmask = mask_and(mask, workscan._getmask(r))
+
+                    workscan.blpars.append({"params":params,
+                                            "fixed": fitparams.getfixedparameters(),
+                                            "formatted":fmtd, "errors":errors})
+                    workscan.masklists.append(workscan.get_masklist(fmask, r, silent=True))
+                    workscan.actualmask.append(fmask)
                     
-                    errs = array.array("f", [0.0 for i in range(order+1)])
-                    errs_adr = errs.buffer_info()[0]
-                    errs_len = errs.buffer_info()[1]
-                    
-                    fmsk = array.array("i", [1 for i in range(nchan)])
-                    fmsk_adr = fmsk.buffer_info()[0]
-                    fmsk_len = fmsk.buffer_info()[1]
-                    
-                    workscan._poly_baseline(mask, order, r, pars_adr, pars_len, errs_adr, errs_len, fmsk_adr, fmsk_len)
-                    
-                    params = pars.tolist()
-                    fmtd = ""
-                    for i in xrange(len(params)): fmtd += "  p%d= %3.6f," % (i, params[i])
-                    fmtd = fmtd[:-1]  # remove trailing ","
-                    errors = errs.tolist()
-                    fmask = fmsk.tolist()
-                    for i in xrange(len(fmask)): fmask[i] = (fmask[i] > 0)    # transform (1/0) -> (True/False)
-                    
-                    self.blpars.append({"params":params, "fixed":[], "formatted":fmtd, "errors":errors})
-                    self.masklists.append(workscan.get_masklist(fmask, r))
-                    self.actualmask.append(fmask)
-                    
-                    asaplog.push(str(fmtd))
+                    asaplog.push(fmtd)
             
             workscan._add_history("poly_baseline", varlist)
@@ -2202 +2188 @@
 
             # Show mask list
-            masklist=workscan.get_masklist(f.mask, row=r)
+            masklist=workscan.get_masklist(f.mask, row=r, silent=True)
             msg = "mask range: "+str(masklist)
             asaplog.push(msg, False)