[2696] | 1 | // |
---|
| 2 | // C++ Implementation: STCalTsys |
---|
| 3 | // |
---|
| 4 | // Description: |
---|
| 5 | // |
---|
| 6 | // |
---|
[2703] | 7 | // Author: Takeshi Nakazato <takeshi.nakazato@nao.ac.jp> (C) 2012 |
---|
[2696] | 8 | // |
---|
| 9 | // Copyright: See COPYING file that comes with this distribution |
---|
| 10 | // |
---|
| 11 | // |
---|
| 12 | |
---|
[2703] | 13 | #include <vector> |
---|
[2786] | 14 | |
---|
| 15 | #include <casa/Arrays/ArrayMath.h> |
---|
| 16 | #include <casa/Logging/LogIO.h> |
---|
| 17 | |
---|
[2703] | 18 | #include "STSelector.h" |
---|
[2696] | 19 | #include "STCalTsys.h" |
---|
[2703] | 20 | #include "STDefs.h" |
---|
| 21 | #include <atnf/PKSIO/SrcType.h> |
---|
[2696] | 22 | |
---|
[2703] | 23 | using namespace std; |
---|
[2696] | 24 | using namespace casa; |
---|
| 25 | |
---|
| 26 | namespace asap { |
---|
[2923] | 27 | STCalTsys::STCalTsys(CountedPtr<Scantable> &s, vector<int> &iflist) |
---|
| 28 | : STCalibration(s, "TSYS"), |
---|
| 29 | iflist_(iflist), |
---|
| 30 | tsysspw_(), |
---|
| 31 | do_average_(false) |
---|
[2696] | 32 | { |
---|
[2703] | 33 | applytable_ = new STCalTsysTable(*s); |
---|
[2696] | 34 | } |
---|
| 35 | |
---|
[2923] | 36 | STCalTsys::STCalTsys(CountedPtr<Scantable> &s, Record &iflist, bool average) |
---|
| 37 | : STCalibration(s, "TSYS"), |
---|
| 38 | iflist_(), |
---|
| 39 | tsysspw_(iflist), |
---|
| 40 | do_average_(average) |
---|
| 41 | { |
---|
| 42 | iflist_.resize(tsysspw_.nfields()); |
---|
| 43 | for (uInt i = 0; i < tsysspw_.nfields(); ++i) { |
---|
| 44 | iflist_[i] = std::atoi(tsysspw_.name(i).c_str()); |
---|
| 45 | } |
---|
| 46 | applytable_ = new STCalTsysTable(*s); |
---|
| 47 | } |
---|
| 48 | |
---|
[2786] | 49 | void STCalTsys::setupSelector(const STSelector &sel) |
---|
[2696] | 50 | { |
---|
[2786] | 51 | sel_ = sel; |
---|
| 52 | vector<int> ifnos = sel_.getIFs(); |
---|
| 53 | if (ifnos.size() > 0) { |
---|
| 54 | int nif = 0; |
---|
| 55 | int nifOrg = iflist_.size(); |
---|
| 56 | vector<int> iflistNew(iflist_); |
---|
| 57 | for (int i = 0; i < nifOrg; i++) { |
---|
| 58 | if (find(ifnos.begin(), ifnos.end(), iflist_[i]) != ifnos.end()) { |
---|
| 59 | iflistNew[nif] = iflist_[i]; |
---|
| 60 | ++nif; |
---|
| 61 | } |
---|
| 62 | } |
---|
| 63 | if (nif == 0) { |
---|
| 64 | LogIO os(LogOrigin("STCalTsys", "setupSelector", WHERE)); |
---|
| 65 | os << LogIO::SEVERE << "Selection contains no data." << LogIO::EXCEPTION; |
---|
| 66 | } |
---|
| 67 | sel_.setIFs(iflistNew); |
---|
| 68 | } |
---|
| 69 | else { |
---|
| 70 | sel_.setIFs(iflist_); |
---|
| 71 | } |
---|
[2696] | 72 | } |
---|
| 73 | |
---|
[2915] | 74 | void STCalTsys::appenddata(uInt scanno, uInt cycleno, |
---|
| 75 | uInt beamno, uInt ifno, uInt polno, |
---|
| 76 | uInt freqid, Double time, Float elevation, |
---|
[2955] | 77 | const Vector<Float> &any_data, |
---|
| 78 | const Vector<uChar> &channel_flag) |
---|
[2696] | 79 | { |
---|
[2915] | 80 | STCalTsysTable *p = dynamic_cast<STCalTsysTable *>(&(*applytable_)); |
---|
[2923] | 81 | if (do_average_ && tsysspw_.isDefined(String::toString(ifno))) { |
---|
| 82 | LogIO os(LogOrigin("STCalTsys", "appenddata", WHERE)); |
---|
| 83 | Vector<Float> averaged_data(any_data.size()); |
---|
[2955] | 84 | Vector<uChar> averaged_flag(any_data.size(), 0); |
---|
[2923] | 85 | Float averaged_value = 0.0; |
---|
[2924] | 86 | size_t num_value = 0; |
---|
[2923] | 87 | Vector<Double> channel_range = tsysspw_.asArrayDouble(String::toString(ifno)); |
---|
| 88 | os << LogIO::DEBUGGING << "do averaging: channel range for IFNO " << ifno << " is " << channel_range << LogIO::POST; |
---|
| 89 | for (uInt i = 1; i < channel_range.size(); i += 2) { |
---|
[2924] | 90 | size_t start = (size_t)channel_range[i-1]; |
---|
| 91 | size_t end = std::min((size_t)channel_range[i] + 1, averaged_data.size()); |
---|
[2923] | 92 | os << LogIO::DEBUGGING << "start=" << start << ", end=" << end << LogIO::POST; |
---|
[2924] | 93 | float sum_per_segment = 0.0; |
---|
[2958] | 94 | size_t count = 0; |
---|
[2924] | 95 | for (size_t j = start; j < end; ++j) { |
---|
[2958] | 96 | if (channel_flag[j] == 0) { |
---|
| 97 | sum_per_segment += any_data[j]; |
---|
| 98 | count += 1; |
---|
| 99 | } |
---|
[2923] | 100 | } |
---|
[2924] | 101 | averaged_value += sum_per_segment; |
---|
[2958] | 102 | num_value += count; |
---|
[2923] | 103 | } |
---|
| 104 | averaged_value /= (Float)num_value; |
---|
| 105 | averaged_data = averaged_value; |
---|
| 106 | os << LogIO::DEBUGGING << "averaged_data = " << averaged_data << LogIO::POST; |
---|
| 107 | os << LogIO::DEBUGGING << "any_data = " << any_data << LogIO::POST; |
---|
| 108 | p->appenddata(scanno, cycleno, beamno, ifno, polno, |
---|
[2955] | 109 | freqid, time, elevation, averaged_data, |
---|
| 110 | averaged_flag); |
---|
[2923] | 111 | } |
---|
| 112 | else { |
---|
| 113 | p->appenddata(scanno, cycleno, beamno, ifno, polno, |
---|
[2955] | 114 | freqid, time, elevation, any_data, |
---|
| 115 | channel_flag); |
---|
[2923] | 116 | } |
---|
[2696] | 117 | } |
---|
| 118 | |
---|
| 119 | } |
---|