[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")); |
---|
| 72 | table_.addColumn(ArrayColumnDesc<Bool>("PARMASKS")); |
---|
| 73 | table_.addColumn(ArrayColumnDesc<String>("FRAMEINFO")); |
---|
[468] | 74 | |
---|
[960] | 75 | // new cached columns |
---|
| 76 | funcCol_.attach(table_,"FUNCTIONS"); |
---|
| 77 | compCol_.attach(table_,"COMPONENTS"); |
---|
| 78 | parCol_.attach(table_,"PARAMETERS"); |
---|
| 79 | maskCol_.attach(table_,"PARMASKS"); |
---|
| 80 | frameCol_.attach(table_,"FRAMEINFO"); |
---|
[468] | 81 | } |
---|
| 82 | |
---|
[960] | 83 | uInt STFit::addEntry( const STFitEntry& fit, Int id ) |
---|
[468] | 84 | { |
---|
[960] | 85 | uInt rno = table_.nrow(); |
---|
[972] | 86 | uInt resultid = 0; |
---|
[960] | 87 | bool foundentry = false; |
---|
[972] | 88 | // replace |
---|
[960] | 89 | if ( id > -1 ) { |
---|
| 90 | Table t = table_(table_.col("ID") == id ); |
---|
| 91 | if (t.nrow() > 0) { |
---|
[1000] | 92 | rno = t.rowNumbers(table_)[0]; |
---|
[960] | 93 | resultid = id; |
---|
| 94 | foundentry = true; |
---|
| 95 | } |
---|
| 96 | } |
---|
[972] | 97 | // doesn't exist |
---|
[960] | 98 | if ( rno > 0 && !foundentry ) { |
---|
| 99 | idCol_.get(rno-1, resultid); |
---|
| 100 | resultid++; |
---|
| 101 | } |
---|
[972] | 102 | // add new row if new id |
---|
| 103 | if ( !foundentry ) table_.addRow(); |
---|
[960] | 104 | funcCol_.put(rno, mathutil::toVectorString(fit.getFunctions())); |
---|
| 105 | compCol_.put(rno, Vector<Int>(fit.getComponents())); |
---|
| 106 | parCol_.put(rno, Vector<Double>(fit.getParameters())); |
---|
| 107 | maskCol_.put(rno, Vector<Bool>(fit.getParmasks())); |
---|
| 108 | frameCol_.put(rno, mathutil::toVectorString(fit.getFrameinfo())); |
---|
| 109 | idCol_.put(rno, resultid); |
---|
| 110 | return resultid; |
---|
[468] | 111 | } |
---|
| 112 | |
---|
[972] | 113 | void STFit::getEntry( STFitEntry& fit, uInt id ) const |
---|
[468] | 114 | { |
---|
[960] | 115 | Table t = table_(table_.col("ID") == Int(id) ); |
---|
| 116 | if (t.nrow() == 0 ) { |
---|
| 117 | throw(AipsError("STFit::getEntry - id out of range")); |
---|
| 118 | } |
---|
| 119 | ROTableRow row(t); |
---|
| 120 | // get first row - there should only be one matching id |
---|
| 121 | const TableRecord& rec = row.get(0); |
---|
| 122 | std::vector<std::string> outstr; |
---|
| 123 | Vector<String> vec; |
---|
| 124 | rec.get("FUNCTIONS", vec); |
---|
| 125 | fit.setFunctions(mathutil::tovectorstring(vec)); |
---|
| 126 | Vector<Int> ivec; |
---|
| 127 | std::vector<int> istl; |
---|
| 128 | rec.get("COMPONENTS", ivec); |
---|
| 129 | ivec.tovector(istl); |
---|
| 130 | fit.setComponents(istl); |
---|
| 131 | Vector<Double> dvec; |
---|
| 132 | std::vector<double> dstl; |
---|
| 133 | rec.get("PARAMETERS", dvec); |
---|
| 134 | dvec.tovector(dstl); |
---|
| 135 | fit.setParameters(dstl); |
---|
| 136 | Vector<Bool> bvec; |
---|
| 137 | std::vector<bool> bstl; |
---|
| 138 | rec.get("PARMASKS", bvec); |
---|
| 139 | bvec.tovector(bstl); |
---|
| 140 | fit.setParmasks(bstl); |
---|
| 141 | vec.resize(); |
---|
| 142 | rec.get("FRAMEINFO", vec); |
---|
| 143 | fit.setFrameinfo(mathutil::tovectorstring(vec)); |
---|
[468] | 144 | } |
---|
| 145 | |
---|
[960] | 146 | } //namespace |
---|