Changeset 175

Timestamp:

01/06/05 16:40:42 (20 years ago)

Author:

kil064

Message:

Add cursor selection to function 'hanning' (arg. 'all')

Location:

trunk/src

Files:

• SDMath.cc (modified) (3 diffs)
• SDMath.h (modified) (2 diffs)
• SDMathWrapper.cc (modified) (1 diff)
• SDMathWrapper.h (modified) (1 diff)

trunk/src/SDMath.cc

@@ -391 +391 @@
 
 
-SDMemTable* SDMath::hanning(const SDMemTable& in) 
+SDMemTable* SDMath::hanning(const SDMemTable& in, Bool doAll)
 //
 // Hanning smooth each row
@@ -397 +397 @@
 //
 {
+
+// New Table
+
   SDMemTable* pTabOut = new SDMemTable(in,True);
 
+// Get cursor location
+         
+  IPosition start, end;
+  getCursorLocation (start, end, in);
+//
+  const uInt chanAxis = 3;                    // Spectrum axis
+  IPosition shapeOut(4,1);
+
 // Loop over rows in Table
 
   for (uInt ri=0; ri < in.nRow(); ++ri) {
 
-// Get data
+// Get copy of data
     
-    const MaskedArray<Float>& marr(in.rowAsMaskedArray(ri));
-    Array<Float> arr = marr.getArray();
-    Array<Bool> barr = marr.getMask();
+    const MaskedArray<Float>& dataIn(in.rowAsMaskedArray(ri));
+    Array<Float> valuesIn = dataIn.getArray();
+    Array<Bool> maskIn = dataIn.getMask();
 
 // Smooth along the channels axis
 
-    uInt axis = 3;
-    VectorIterator<Float> itData(arr, axis);
-    VectorIterator<Bool> itMask(barr, axis);
-    Vector<Float> outv;
-    Vector<Bool> outm;
-    while (!itData.pastEnd()) {
-       mathutil::hanning(outv, outm, itData.vector(), itMask.vector());
-       itData.vector() = outv;                                                 
-       itMask.vector() = outm;
-//
-       itData.next();
-       itMask.next();
+    Vector<Float> outValues;
+    Vector<Bool> outMask;
+    if (doAll) {
+       uInt axis = 3;
+       VectorIterator<Float> itValues(valuesIn, axis);
+       VectorIterator<Bool> itMask(maskIn, axis);
+       while (!itValues.pastEnd()) {
+          mathutil::hanning(outValues, outMask, itValues.vector(), itMask.vector());
+          itValues.vector() = outValues; 
+          itMask.vector() = outMask;
+//
+          itValues.next();
+          itMask.next();
+       }
+    } else {
+
+// Set multi-dim Vector shape
+
+       shapeOut(chanAxis) = valuesIn.shape()(chanAxis);
+
+// Stuff about with shapes so that we don't have conformance run-time errors
+
+       Vector<Float> valuesIn2 = valuesIn(start,end).nonDegenerate();
+       Vector<Bool> maskIn2 = maskIn(start,end).nonDegenerate();
+       mathutil::hanning(outValues, outMask, valuesIn2, maskIn2);
+
+// Write back
+
+       valuesIn(start,end) = outValues.reform(shapeOut);
+       maskIn(start,end) = outMask.reform(shapeOut);
     }
 
@@ -428 +457 @@
 
     SDContainer sc = in.getSDContainer(ri);
-    putDataInSDC (sc, arr, barr);
+    putDataInSDC (sc, valuesIn, maskIn);
 //
     pTabOut->putSDContainer(sc);

trunk/src/SDMath.h

@@ -59 +59 @@
 
 // Average in time
-   casa::CountedPtr<SDMemTable>  average (const casa::Block<casa::CountedPtr<SDMemTable> >& in,
-                                          const casa::Vector<casa::Bool>& mask,
-                                          bool scanAverage, const std::string& weightStr);
+   casa::CountedPtr<SDMemTable>  average(const casa::Block<casa::CountedPtr<SDMemTable> >& in,
+                                         const casa::Vector<casa::Bool>& mask,
+                                         bool scanAverage, const std::string& weightStr);
 
 // Statistics
@@ -67 +67 @@
                                 const std::vector<bool>& mask, const std::string& which);
 
-//  Hanning
-   SDMemTable* hanning(const SDMemTable& in);
-
+// Hanning
+   SDMemTable* hanning(const SDMemTable& in, casa::Bool doAll);
 
 // Bin up spectra
-   SDMemTable* bin (const SDMemTable& in, casa::Int width);
+   SDMemTable* bin(const SDMemTable& in, casa::Int width);
 
 // Simple mathematical operations.  what=0 (mul) or 1 (add)
-   SDMemTable* simpleOperate (const SDMemTable& in, casa::Float offset, 
-                              casa::Bool doAll, casa::uInt what);
+   SDMemTable* simpleOperate(const SDMemTable& in, casa::Float offset, 
+                             casa::Bool doAll, casa::uInt what);
 
 // Average polarizations
-   SDMemTable* averagePol (const SDMemTable& in, const casa::Vector<casa::Bool>& mask);
+   SDMemTable* averagePol(const SDMemTable& in, const casa::Vector<casa::Bool>& mask);
 
  private:

trunk/src/SDMathWrapper.cc

@@ -88 +88 @@
 }
 
-void SDMathWrapper::hanningInSitu(SDMemTableWrapper& in)
+void SDMathWrapper::hanningInSitu(SDMemTableWrapper& in, bool doAll)
 {
   SDMemTable* pIn = in.getPtr();
   SDMath sdm;
-  SDMemTable* pOut = sdm.hanning (*pIn);
+  SDMemTable* pOut = sdm.hanning (*pIn, Bool(doAll));
   *pIn = *pOut;
    delete pOut;
 }
 
-SDMemTableWrapper SDMathWrapper::hanning (const SDMemTableWrapper& in)
+SDMemTableWrapper SDMathWrapper::hanning (const SDMemTableWrapper& in, bool doAll)
 {
   const CountedPtr<SDMemTable>& pIn = in.getCP();
   SDMath sdm;
-  return CountedPtr<SDMemTable>(sdm.hanning(*pIn));
+  return CountedPtr<SDMemTable>(sdm.hanning(*pIn, Bool(doAll)));
 }
 

trunk/src/SDMathWrapper.h

@@ -53 +53 @@
 
 // Add
-  void addInSitu(SDMemTableWrapper& in, float offset, bool all);
+  void addInSitu(SDMemTableWrapper& in, float offset, bool doAall);
   SDMemTableWrapper add(const SDMemTableWrapper& in, float offset, bool all);
 
 // Hanning
-  void hanningInSitu (SDMemTableWrapper& in);
-  SDMemTableWrapper hanning(const SDMemTableWrapper& in);
+  void hanningInSitu (SDMemTableWrapper& in, bool doAll);
+  SDMemTableWrapper hanning(const SDMemTableWrapper& in, bool doAll);
 
 // Bin up