Changeset 3010 for trunk/src

Timestamp:

10/31/14 21:26:51 (10 years ago)

Author:

WataruKawasaki

Message:

New Development: No

JIRA Issue: Yes CAS-6599

Ready for Test: Yes

Interface Changes: No

What Interface Changed:

Test Programs:

Put in Release Notes: No

Module(s): sd

Description: modified so that sd.scantable.stats() returns None for flagged rows and rows with all channels flagged.

Location:

trunk/src

Files:

• STMath.cpp (modified) (4 diffs)
• Scantable.cpp (modified) (1 diff)

trunk/src/STMath.cpp

@@ -11 +11 @@
 //
 
+//#include <math.h>
+//#include <bits/nan.h>
 #include <sstream>
 #include <iostream>
@@ -1993 +1995 @@
                                         const std::string& which )
 {
-
   Vector<Bool> m(mask);
   const Table& tab = in->table();
@@ -1999 +2000 @@
   ROArrayColumn<uChar> flagCol(tab, "FLAGTRA");
   std::vector<float> out;
-  for (uInt i=0; i < tab.nrow(); ++i ) {
-    Vector<Float> spec; specCol.get(i, spec);
-    Vector<uChar> flag; flagCol.get(i, flag);
-    MaskedArray<Float> ma  = maskedArray(spec, flag);
-    float outstat = 0.0;
-    if ( spec.nelements() == m.nelements() ) {
-      outstat = mathutil::statistics(which, ma(m));
+  for (uInt i = 0; i < tab.nrow(); ++i) {
+    if (in->isAllChannelsFlagged(i)) {
+      out.push_back(NAN);
     } else {
-      outstat = mathutil::statistics(which, ma);
-    }
-    out.push_back(outstat);
+      float outstat = 0.0;
+      Vector<Float> spec; specCol.get(i, spec);
+      Vector<uChar> flag; flagCol.get(i, flag);
+      MaskedArray<Float> ma  = maskedArray(spec, flag);
+      if (spec.nelements() == m.nelements()) {
+        outstat = mathutil::statistics(which, ma(m));
+      } else {
+        outstat = mathutil::statistics(which, ma);
+      }
+      out.push_back(outstat);
+    }
   }
   return out;
@@ -2051 +2056 @@
   std::vector<int> out;
   for (uInt i=0; i < tab.nrow(); ++i ) {
-    Vector<Float> spec; specCol.get(i, spec);
-    Vector<uChar> flag; flagCol.get(i, flag);
-    MaskedArray<Float> ma  = maskedArray(spec, flag);
-    if (ma.ndim() != 1) {
-      throw (ArrayError(
-          "std::vector<int> STMath::minMaxChan("
-          "ContedPtr<Scantable> &in, std::vector<bool> &mask, "
-          " std::string &which)"
-          " - MaskedArray is not 1D"));
-    }
-    IPosition outpos(1,0);
-    if ( spec.nelements() == m.nelements() ) {
-      outpos = mathutil::minMaxPos(which, ma(m));
+    if (in->isAllChannelsFlagged(i)) {
+      out.push_back(NAN);
     } else {
-      outpos = mathutil::minMaxPos(which, ma);
-    }
-    out.push_back(outpos[0]);
+      Vector<Float> spec; specCol.get(i, spec);
+      Vector<uChar> flag; flagCol.get(i, flag);
+      MaskedArray<Float> ma  = maskedArray(spec, flag);
+      if (ma.ndim() != 1) {
+        throw (ArrayError(
+                          "std::vector<int> STMath::minMaxChan("
+                          "ContedPtr<Scantable> &in, std::vector<bool> &mask, "
+                          " std::string &which)"
+                          " - MaskedArray is not 1D"));
+      }
+      IPosition outpos(1,0);
+      if ( spec.nelements() == m.nelements() ) {
+        outpos = mathutil::minMaxPos(which, ma(m));
+      } else {
+        outpos = mathutil::minMaxPos(which, ma);
+      }
+      out.push_back(outpos[0]);
+    }
   }
   return out;

trunk/src/Scantable.cpp

@@ -5004 +5004 @@
   lineFinder.setData(getSpectrum(whichrow));
   lineFinder.findLines(getCompositeChanMask(whichrow, inMask), currEdge, whichrow);
-
   return lineFinder.getMask();
 }