Changeset 896 for trunk/src/Scantable.cpp

Timestamp:

03/10/06 12:09:29 (18 years ago)

Author:

mar637

Message:

enable polarimetry in asap2

File:

• trunk/src/Scantable.cpp (modified) (9 diffs)

trunk/src/Scantable.cpp

@@ -54 +54 @@
 
 #include "Scantable.h"
+#include "STPolLinear.h"
 #include "STAttr.h"
 
@@ -59 +60 @@
 
 namespace asap {
+
+std::map<std::string, STPol::STPolFactory *> Scantable::factories_;
+
+void Scantable::initFactories() {
+  if ( factories_.empty() ) {
+    Scantable::factories_["linear"] = &STPolLinear::myFactory;
+  }
+}
 
 Scantable::Scantable(Table::TableType ttype) :
   type_(ttype)
 {
+  initFactories();
   setupMainTable();
   freqTable_ = STFrequencies(*this);
@@ -85 +95 @@
   type_(ttype)
 {
+  initFactories();
   Table tab(name, Table::Update);
   Int version;
@@ -566 +577 @@
   if ( selector_.empty() )
     throw(AipsError("Trying to flag whole scantable. Aborted."));
+  TableVector<uChar> tvec(table_, "FLAGTRA");
+  uChar userflag = 1 << 7;
+  tvec = userflag;
 }
 
@@ -580 +594 @@
 }
 
-std::vector<float> Scantable::getSpectrum(int whichrow) const
-{
+std::vector<float> Scantable::getSpectrum( int whichrow,
+                                           const std::string& poltype) const
+{
+  std::vector<float> out;
   Vector<Float> arr;
-  specCol_.get(whichrow, arr);
-  std::vector<float> out;
+  uInt requestedpol = polCol_(whichrow);
+  String basetype = getPolType();
+  if ( String(poltype) == basetype) {
+    specCol_.get(whichrow, arr);
+  } else {
+    STPol* stpol = 0;
+    stpol =STPol::getPolClass(Scantable::factories_, basetype);
+    try {
+      uInt row = uInt(whichrow);
+      stpol->setSpectra(getPolMatrix(row));
+      Float frot,fang,ftan;
+      focusTable_.getEntry(frot, fang, ftan, row);
+      stpol->setPhaseCorrections(frot, fang, ftan);
+      arr = stpol->getSpectrum(requestedpol, poltype);
+      delete stpol;
+    } catch (AipsError& e) {
+      delete stpol;
+      throw(e);
+    }
+  }
   arr.tovector(out);
   return out;
 }
-
 
 void asap::Scantable::setSpectrum( const std::vector<float>& spec,
@@ -597 +630 @@
   specCol_.get(whichrow, arr);
   if ( spectrum.nelements() != arr.nelements() )
-    throw AipsError("The specturm has incorrect number of channels.");
+    throw AipsError("The spectrum has incorrect number of channels.");
   specCol_.put(whichrow, spectrum);
 }
@@ -616 +649 @@
   return table_;
 }
+
+std::string Scantable::getPolType() const
+{
+  return table_.keywordSet().asString("POLTYPE");
+}
+
 
 std::string Scantable::getPolarizationLabel(bool linear, bool stokes,
@@ -655 +694 @@
   oss << setw(15) << "Beams:" << setw(4) << nbeam() << endl
       << setw(15) << "IFs:" << setw(4) << nif() << endl
-      << setw(15) << "Polarisations:" << setw(4) << npol() << endl
+      << setw(15) << "Polarisations:" << setw(4) << npol()
+      << "(" << getPolType() << ")" << endl
       << setw(15) << "Channels:"  << setw(4) << nchan() << endl;
   oss << endl;
@@ -839 +879 @@
 
 
+Matrix<Float> asap::Scantable::getPolMatrix( uInt whichrow ) const
+{
+  ROTableRow row(table_);
+  const TableRecord& rec = row.get(whichrow);
+  Table t =
+    originalTable_( originalTable_.col("SCANNO") == Int(rec.asuInt("SCANNO"))
+                    && originalTable_.col("BEAMNO") == Int(rec.asuInt("BEAMNO"))
+                    && originalTable_.col("IFNO") == Int(rec.asuInt("IFNO"))
+                    && originalTable_.col("CYCLENO") == Int(rec.asuInt("CYCLENO")) );
+  ROArrayColumn<Float> speccol(t, "SPECTRA");
+  return speccol.getColumn();
+}
+
 
 }//namespace asap