Changeset 443 for trunk/src/SDFITSImageWriter.cc

Timestamp:

02/15/05 11:59:01 (19 years ago)

Author:

kil064

Message:

add stokes conversion

File:

• trunk/src/SDFITSImageWriter.cc (modified) (8 diffs)

trunk/src/SDFITSImageWriter.cc

@@ -52 +52 @@
 
 #include <measures/Measures/MEpoch.h>
+#include <measures/Measures/Stokes.h>
 
 #include <tables/Tables/Table.h>
@@ -60 +61 @@
 #include "SDContainer.h"
 #include "SDMemTable.h"
+#include "SDPol.h"
 
 using namespace casa;
@@ -73 +75 @@
 
 Bool SDFITSImageWriter::write(const SDMemTable& sdTable, 
-                              const String& dirName, Bool verbose)
+                              const String& dirName, Bool toStokes,
+                              Bool verbose)
 {
 
@@ -131 +134 @@
    Vector<Double> refPixLonLat(2,0.0);
 
+// Do we have linear or circular ?  No way to know yet...
+
+   Bool linear = True;
+
 // Loop over rows
 
@@ -147 +154 @@
    for (uInt iRow=0; iRow<nRows; iRow++) {
 
-// Get data
-
-      const MaskedArray<Float>& dataIn(sdTable.rowAsMaskedArray(iRow));
+// Get data converted to Stokes
+
+      const MaskedArray<Float>& dataIn(sdTable.rowAsMaskedArray(iRow,False,toStokes));
       const Array<Float>& values = dataIn.getArray();
       const Array<Bool>& mask = dataIn.getMask();
@@ -196 +203 @@
                                refPixLonLat[0], refPixLonLat[1]);
 
-// Form Stokes Coordinate (no true Stokes info yet);
-
-         whichStokes(0) = convertStokes(pos(stokesAxis));
+// Form Stokes Coordinate (Stokes info still sketchy);
+
+         Stokes::StokesTypes stokes = SDPolUtil::convertStokes(pos(stokesAxis), toStokes, linear);
+         String stokesName = Stokes::name(stokes);
+         whichStokes(0) = Int(stokes);
          StokesCoordinate stC(whichStokes);
 
@@ -226 +235 @@
          ostringstream oss;
          oss << "row" << iRow << "_beam" << pos(0) << "_if" 
-             << pos(1) << "_pol" << pos(2) << "_" << srcCol(iRow) << ".fits";
+             << pos(1) << "_" << stokesName  << "_" << srcCol(iRow) << ".fits";
          String tS(oss);
          String fileName = dirName2 + String("/") + tS;
@@ -256 +265 @@
 
 
-Int SDFITSImageWriter::convertStokes(Int val)
-{
-   Stokes::StokesTypes stokes = Stokes::RR;
-   if (val==0) {
-      stokes = Stokes::XX;
-   } else if (val==1) {
-      stokes = Stokes::YY;
-   } else if (val==2) {
-      stokes = Stokes::XY;
-   } else if (val==3) {
-      stokes = Stokes::YX;
-   } else {
-      stokes = Stokes::Undefined;
-   }
-//
-   return Int(stokes);
-}