[468] | 1 | |
---|
[960] | 2 | // |
---|
| 3 | // C++ Implementation: STFit |
---|
| 4 | // |
---|
| 5 | // Description: |
---|
| 6 | // |
---|
| 7 | // |
---|
| 8 | // Author: Malte Marquarding <asap@atnf.csiro.au>, (C) 2006 |
---|
| 9 | // |
---|
| 10 | // Copyright: See COPYING file that comes with this distribution |
---|
| 11 | // |
---|
| 12 | // |
---|
| 13 | #include <casa/Exceptions/Error.h> |
---|
| 14 | #include <tables/Tables/TableDesc.h> |
---|
| 15 | #include <tables/Tables/SetupNewTab.h> |
---|
| 16 | #include <tables/Tables/ScaColDesc.h> |
---|
| 17 | #include <tables/Tables/ArrColDesc.h> |
---|
| 18 | #include <tables/Tables/TableRecord.h> |
---|
| 19 | #include <tables/Tables/TableParse.h> |
---|
| 20 | #include <tables/Tables/TableRow.h> |
---|
| 21 | #include <casa/Containers/RecordField.h> |
---|
| 22 | |
---|
[468] | 23 | #include "MathUtils.h" |
---|
[960] | 24 | #include "STFitEntry.h" |
---|
| 25 | #include "STFit.h" |
---|
[468] | 26 | |
---|
[960] | 27 | |
---|
[468] | 28 | using namespace casa; |
---|
| 29 | |
---|
[960] | 30 | namespace asap { |
---|
[468] | 31 | |
---|
[960] | 32 | const casa::String STFit::name_ = "FIT"; |
---|
[468] | 33 | |
---|
[960] | 34 | STFit::STFit(const Scantable& parent) : |
---|
| 35 | STSubTable( parent, name_ ) |
---|
[468] | 36 | { |
---|
[960] | 37 | setup(); |
---|
[468] | 38 | } |
---|
| 39 | |
---|
[960] | 40 | STFit& asap::STFit::operator =( const STFit & other ) |
---|
[468] | 41 | { |
---|
[960] | 42 | if ( this != &other ) { |
---|
| 43 | static_cast<STSubTable&>(*this) = other; |
---|
| 44 | funcCol_.attach(table_,"FUNCTIONS"); |
---|
| 45 | compCol_.attach(table_,"COMPONENTS"); |
---|
| 46 | parCol_.attach(table_,"PARAMETERS"); |
---|
| 47 | maskCol_.attach(table_,"PARMASKS"); |
---|
| 48 | frameCol_.attach(table_,"FRAMEINFO"); |
---|
| 49 | } |
---|
| 50 | return *this; |
---|
[468] | 51 | } |
---|
| 52 | |
---|
[960] | 53 | asap::STFit::STFit( casa::Table tab ) : STSubTable(tab, name_) |
---|
[468] | 54 | { |
---|
[960] | 55 | funcCol_.attach(table_,"FUNCTIONS"); |
---|
| 56 | compCol_.attach(table_,"COMPONENTS"); |
---|
| 57 | parCol_.attach(table_,"PARAMETERS"); |
---|
| 58 | maskCol_.attach(table_,"PARMASKS"); |
---|
| 59 | frameCol_.attach(table_,"FRAMEINFO"); |
---|
[468] | 60 | } |
---|
| 61 | |
---|
[960] | 62 | STFit::~STFit() |
---|
[468] | 63 | { |
---|
| 64 | } |
---|
| 65 | |
---|
[960] | 66 | void asap::STFit::setup( ) |
---|
[468] | 67 | { |
---|
[960] | 68 | // add to base class table |
---|
| 69 | table_.addColumn(ArrayColumnDesc<String>("FUNCTIONS")); |
---|
| 70 | table_.addColumn(ArrayColumnDesc<Int>("COMPONENTS")); |
---|
| 71 | table_.addColumn(ArrayColumnDesc<Double>("PARAMETERS")); |
---|
[1932] | 72 | // table_.addColumn(ArrayColumnDesc<Double>("ERRORS")); |
---|
[960] | 73 | table_.addColumn(ArrayColumnDesc<Bool>("PARMASKS")); |
---|
| 74 | table_.addColumn(ArrayColumnDesc<String>("FRAMEINFO")); |
---|
[468] | 75 | |
---|
[960] | 76 | // new cached columns |
---|
| 77 | funcCol_.attach(table_,"FUNCTIONS"); |
---|
| 78 | compCol_.attach(table_,"COMPONENTS"); |
---|
| 79 | parCol_.attach(table_,"PARAMETERS"); |
---|
[1932] | 80 | // errCol_.attach(table_,"ERRORS"); |
---|
[960] | 81 | maskCol_.attach(table_,"PARMASKS"); |
---|
| 82 | frameCol_.attach(table_,"FRAMEINFO"); |
---|
[468] | 83 | } |
---|
| 84 | |
---|
[960] | 85 | uInt STFit::addEntry( const STFitEntry& fit, Int id ) |
---|
[468] | 86 | { |
---|
[960] | 87 | uInt rno = table_.nrow(); |
---|
[972] | 88 | uInt resultid = 0; |
---|
[960] | 89 | bool foundentry = false; |
---|
[972] | 90 | // replace |
---|
[960] | 91 | if ( id > -1 ) { |
---|
[2243] | 92 | Table t = table_(table_.col("ID") == id, 1 ); |
---|
[960] | 93 | if (t.nrow() > 0) { |
---|
[1000] | 94 | rno = t.rowNumbers(table_)[0]; |
---|
[960] | 95 | resultid = id; |
---|
| 96 | foundentry = true; |
---|
| 97 | } |
---|
| 98 | } |
---|
[972] | 99 | // doesn't exist |
---|
[960] | 100 | if ( rno > 0 && !foundentry ) { |
---|
| 101 | idCol_.get(rno-1, resultid); |
---|
| 102 | resultid++; |
---|
| 103 | } |
---|
[972] | 104 | // add new row if new id |
---|
| 105 | if ( !foundentry ) table_.addRow(); |
---|
[1932] | 106 | |
---|
[960] | 107 | funcCol_.put(rno, mathutil::toVectorString(fit.getFunctions())); |
---|
| 108 | compCol_.put(rno, Vector<Int>(fit.getComponents())); |
---|
[1932] | 109 | const std::vector<float>& pvec = fit.getParameters(); |
---|
| 110 | Vector<Double> dvec(pvec.size()); |
---|
| 111 | for (size_t i=0; i < dvec.nelements(); ++i) { |
---|
| 112 | dvec[i] = Double(pvec[i]); |
---|
| 113 | } |
---|
| 114 | parCol_.put(rno, dvec); |
---|
| 115 | /* |
---|
| 116 | const std::vector<double>& evec = fit.getErrors(); |
---|
| 117 | for (size_t i=0; i < dvec.nelements(); ++i) { |
---|
| 118 | dvec[i] = Double(evec[i]); |
---|
| 119 | } |
---|
| 120 | errCol_.put(rno, dvec); |
---|
| 121 | */ |
---|
[960] | 122 | maskCol_.put(rno, Vector<Bool>(fit.getParmasks())); |
---|
| 123 | frameCol_.put(rno, mathutil::toVectorString(fit.getFrameinfo())); |
---|
| 124 | idCol_.put(rno, resultid); |
---|
[1932] | 125 | |
---|
[960] | 126 | return resultid; |
---|
[468] | 127 | } |
---|
| 128 | |
---|
[972] | 129 | void STFit::getEntry( STFitEntry& fit, uInt id ) const |
---|
[468] | 130 | { |
---|
[2243] | 131 | Table t = table_(table_.col("ID") == Int(id), 1 ); |
---|
[960] | 132 | if (t.nrow() == 0 ) { |
---|
| 133 | throw(AipsError("STFit::getEntry - id out of range")); |
---|
| 134 | } |
---|
| 135 | ROTableRow row(t); |
---|
| 136 | // get first row - there should only be one matching id |
---|
| 137 | const TableRecord& rec = row.get(0); |
---|
| 138 | std::vector<std::string> outstr; |
---|
| 139 | Vector<String> vec; |
---|
| 140 | rec.get("FUNCTIONS", vec); |
---|
| 141 | fit.setFunctions(mathutil::tovectorstring(vec)); |
---|
| 142 | Vector<Int> ivec; |
---|
| 143 | std::vector<int> istl; |
---|
| 144 | rec.get("COMPONENTS", ivec); |
---|
| 145 | ivec.tovector(istl); |
---|
| 146 | fit.setComponents(istl); |
---|
| 147 | Vector<Double> dvec; |
---|
| 148 | rec.get("PARAMETERS", dvec); |
---|
[1932] | 149 | std::vector<float> dstl(dvec.begin(), dvec.end()); |
---|
| 150 | fit.setParameters(dstl); |
---|
| 151 | /* |
---|
[960] | 152 | dvec.tovector(dstl); |
---|
| 153 | fit.setParameters(dstl); |
---|
[1932] | 154 | dvec.resize(); |
---|
| 155 | rec.get("ERRORS", dvec); |
---|
| 156 | dvec.tovector(dstl); |
---|
| 157 | fit.setErrors(dstl); |
---|
| 158 | */ |
---|
[960] | 159 | Vector<Bool> bvec; |
---|
| 160 | std::vector<bool> bstl; |
---|
| 161 | rec.get("PARMASKS", bvec); |
---|
| 162 | bvec.tovector(bstl); |
---|
| 163 | fit.setParmasks(bstl); |
---|
| 164 | vec.resize(); |
---|
| 165 | rec.get("FRAMEINFO", vec); |
---|
| 166 | fit.setFrameinfo(mathutil::tovectorstring(vec)); |
---|
[468] | 167 | } |
---|
| 168 | |
---|
[960] | 169 | } //namespace |
---|