Changeset 1310 for branches/Release2.1.2/src/STMath.cpp

Timestamp:

12/14/06 11:21:48 (17 years ago)

Author:

mar637

Message:

merge from trunk, to get binary operator changes

File:

• branches/Release2.1.2/src/STMath.cpp (modified) (4 diffs)

branches/Release2.1.2/src/STMath.cpp

@@ -281 +281 @@
                                               bool tsys )
 {
-  // modes are "ADD" and "MUL"
   CountedPtr< Scantable > out = getScantable(in, false);
   Table& tab = out->table();
@@ -291 +290 @@
     specCol.get(i, spec);
     tsysCol.get(i, ts);
-    if (mode == "MUL") {
+    if (mode == "MUL" || mode == "DIV") {
+      if (mode == "DIV") val = 1.0/val;
       spec *= val;
       specCol.put(i, spec);
@@ -298 +298 @@
         tsysCol.put(i, ts);
       }
-    } else if ( mode == "ADD" ) {
+    } else if ( mode == "ADD"  || mode == "SUB") {
+      if (mode == "SUB") val *= -1.0;
       spec += val;
       specCol.put(i, spec);
@@ -309 +310 @@
   return out;
 }
+
+CountedPtr<Scantable> STMath::binaryOperate(const CountedPtr<Scantable>& left, 
+                                            const CountedPtr<Scantable>& right, 
+                                            const std::string& mode)
+{
+  bool insitu = insitu_;
+  if ( ! left->conformant(*right) ) {
+    throw(AipsError("'left' and 'right' scantables are not conformant."));
+  }
+  setInsitu(false);
+  CountedPtr< Scantable > out = getScantable(left, false);
+  setInsitu(insitu);
+  Table& tout = out->table();
+  Block<String> coln(5);
+  coln[0] = "SCANNO";  coln[1] = "CYCLENO";  coln[2] = "BEAMNO";
+  coln[3] = "IFNO";  coln[4] = "POLNO";
+  Table tmpl = tout.sort(coln);
+  Table tmpr = right->table().sort(coln);
+  ArrayColumn<Float> lspecCol(tmpl,"SPECTRA");
+  ROArrayColumn<Float> rspecCol(tmpr,"SPECTRA");
+  ArrayColumn<uChar> lflagCol(tmpl,"FLAGTRA");
+  ROArrayColumn<uChar> rflagCol(tmpr,"FLAGTRA");
+
+  for (uInt i=0; i<tout.nrow(); ++i) {
+    Vector<Float> lspecvec, rspecvec;
+    Vector<uChar> lflagvec, rflagvec;
+    lspecvec = lspecCol(i);    rspecvec = rspecCol(i);
+    lflagvec = lflagCol(i);    rflagvec = rflagCol(i);
+    MaskedArray<Float> mleft = maskedArray(lspecvec, lflagvec);
+    MaskedArray<Float> mright = maskedArray(rspecvec, rflagvec);
+    if (mode == "ADD") {
+      mleft += mright;
+    } else if ( mode == "SUB") {
+      mleft -= mright;
+    } else if ( mode == "MUL") {
+      mleft *= mright;
+    } else if ( mode == "DIV") {
+      mleft /= mright;
+    } else {
+      throw(AipsError("Illegal binary operator"));
+    }
+    lspecCol.put(i, mleft.getArray());
+  }
+  return out;
+}
+
+
 
 MaskedArray<Float> STMath::maskedArray( const Vector<Float>& s,