Changeset 167 for trunk/src/SDMath.cc

Timestamp:

12/27/04 22:14:55 (19 years ago)

Author:

kil064

Message:

Reimplement 'bin' with insitu version as well

File:

• trunk/src/SDMath.cc (modified) (3 diffs)

trunk/src/SDMath.cc

@@ -462 +462 @@
 }
 
-CountedPtr<SDMemTable> SDMath::bin(const CountedPtr<SDMemTable>& in,
-                                   Int width) 
-{
-  SDHeader sh = in->getSDHeader();
-  SDMemTable* sdmt = new SDMemTable(*in,True);
-
-// Bin up SpectralCoordinates
-
-  IPosition factors(1);
-  factors(0) = width;
-  for (uInt j=0; j<in->nCoordinates(); ++j) {
-    CoordinateSystem cSys;
-    cSys.addCoordinate(in->getCoordinate(j));
-    CoordinateSystem cSysBin = 
-      CoordinateUtil::makeBinnedCoordinateSystem (factors, cSys, False);
-//
-    SpectralCoordinate sCBin = cSysBin.spectralCoordinate(0);
-    sdmt->setCoordinate(sCBin, j);
-  }
-
-// Use RebinLattice to find shape
-
-  IPosition shapeIn(1,sh.nchan);
-  IPosition shapeOut = RebinLattice<Float>::rebinShape (shapeIn, factors);
-  sh.nchan = shapeOut(0);
-  sdmt->putSDHeader(sh);
-
-
-// Loop over rows and bin along channel axis
-  
-  const uInt axis = 3;
-  for (uInt i=0; i < in->nRow(); ++i) {
-    SDContainer sc = in->getSDContainer(i);
-//
-    Array<Float> tSys(sc.getTsys());                           // Get it out before sc changes shape
-
-// Bin up spectrum
-
-    MaskedArray<Float> marr(in->rowAsMaskedArray(i));
-    MaskedArray<Float> marrout;
-    LatticeUtilities::bin(marrout, marr, axis, width);
-
-// Put back the binned data and flags
-
-    IPosition ip2 = marrout.shape();
-    sc.resize(ip2);
-//
-    putDataInSDC (sc, marrout.getArray(), marrout.getMask());
-
-// Bin up Tsys.  
-
-    Array<Bool> allGood(tSys.shape(),True);
-    MaskedArray<Float> tSysIn(tSys, allGood, True);
-//
-    MaskedArray<Float> tSysOut;    
-    LatticeUtilities::bin(tSysOut, tSysIn, axis, width);
-    sc.putTsys(tSysOut.getArray());
-//
-    sdmt->putSDContainer(sc);
-  }
-  return CountedPtr<SDMemTable>(sdmt);
-}
+
+void SDMath::binInSitu (SDMemTable* pIn, int width)
+{
+  const uInt what = 1;
+  SDMemTable* pOut = localBin (*pIn, Int(width));
+  *pIn = *pOut;
+   delete pOut;
+}  
+
+
+CountedPtr<SDMemTable> SDMath::bin (const CountedPtr<SDMemTable>& in, int width)
+{
+  const uInt what = 1;
+  return CountedPtr<SDMemTable>(localBin(*in, Int(width)));
+}
+
 
 
@@ -866 +820 @@
 // 
   const Bool useMask = (mask.nelements() == shapeIn(chanAxis));
-cerr << "nEl=" << mask.nelements() << endl;
-cerr << "useMask=" << useMask << endl;
 
 // Loop over rows
@@ -962 +914 @@
   return pTabOut;
 }
+
+SDMemTable* SDMath::localBin (const SDMemTable& in, Int width)
+{
+  SDHeader sh = in.getSDHeader();
+  SDMemTable* pTabOut = new SDMemTable(in, True);
+
+// Bin up SpectralCoordinates
+
+  IPosition factors(1);
+  factors(0) = width;
+  for (uInt j=0; j<in.nCoordinates(); ++j) {
+    CoordinateSystem cSys;
+    cSys.addCoordinate(in.getCoordinate(j));
+    CoordinateSystem cSysBin = 
+      CoordinateUtil::makeBinnedCoordinateSystem (factors, cSys, False);
+//
+    SpectralCoordinate sCBin = cSysBin.spectralCoordinate(0);
+    pTabOut->setCoordinate(sCBin, j);
+  }
+
+// Use RebinLattice to find shape
+
+  IPosition shapeIn(1,sh.nchan);
+  IPosition shapeOut = RebinLattice<Float>::rebinShape (shapeIn, factors);
+  sh.nchan = shapeOut(0);
+  pTabOut->putSDHeader(sh);
+
+
+// Loop over rows and bin along channel axis
+  
+  const uInt axis = 3;
+  for (uInt i=0; i < in.nRow(); ++i) {
+    SDContainer sc = in.getSDContainer(i);
+//
+    Array<Float> tSys(sc.getTsys());                           // Get it out before sc changes shape
+
+// Bin up spectrum
+
+    MaskedArray<Float> marr(in.rowAsMaskedArray(i));
+    MaskedArray<Float> marrout;
+    LatticeUtilities::bin(marrout, marr, axis, width);
+
+// Put back the binned data and flags
+
+    IPosition ip2 = marrout.shape();
+    sc.resize(ip2);
+//
+    putDataInSDC (sc, marrout.getArray(), marrout.getMask());
+
+// Bin up Tsys.  
+
+    Array<Bool> allGood(tSys.shape(),True);
+    MaskedArray<Float> tSysIn(tSys, allGood, True);
+//
+    MaskedArray<Float> tSysOut;    
+    LatticeUtilities::bin(tSysOut, tSysIn, axis, width);
+    sc.putTsys(tSysOut.getArray());
+//
+    pTabOut->putSDContainer(sc);
+  }
+  return pTabOut;
+}
+
+