Changeset 2320 for trunk/python/asapmath.py

Timestamp:

10/04/11 13:40:03 (13 years ago)

Author:

Malte Marquarding

Message:

Ticket #251: fixed copy of input scantable which breaks scantable selection

File:

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

trunk/python/asapmath.py

@@ -1012 +1012 @@
 
 @asaplog_post_dec
-def _array2dOp( scan, value, mode="ADD", tsys=False ):
+def _array2dOp( scan, value, mode="ADD", tsys=False, insitu=None):
     """
     This function is workaround on the basic operation of scantable
@@ -1024 +1024 @@
     nrow = scan.nrow()
     s = None
+    from asap._asap import stmath
+    stm = stmath()
+    stm._setinsitu(insitu)
     if len( value ) == 1:
-        from asap._asap import stmath
-        stm = stmath()
-        s = scantable( stm._arrayop( scan.copy(), value[0], mode, tsys ) )
-        del stm
+        s = scantable( stm._arrayop( scan, value[0], mode, tsys ) )
     elif len( value ) != nrow:
         raise ValueError( 'len(value) must be 1 or conform to scan.nrow()' )
     else:
         from asap._asap import stmath
-        stm = stmath()
-        # insitu must be True
+        if not insitu:
+            s = scan.copy()
+        else:
+            s = scan
+        # insitu must be True as we go row by row on the same data
         stm._setinsitu( True )
-        s = scan.copy()
-        sel = selector()
+        basesel = s.get_selection()
+        sel = selector()+basesel
         for irow in range( nrow ):
             sel.set_rows( irow )
@@ -1046 +1049 @@
                 #stm._arrayop( s, value[irow], mode, tsys, 'channel' )
                 stm._arrayop( s, value[irow], mode, tsys )
-            s.set_selection()
-            sel.reset()
-        del sel
-        del stm
+        s.set_selection(basesel)
     return s