1 | |
---|
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 | |
---|
23 | #include "MathUtils.h" |
---|
24 | #include "STFitEntry.h" |
---|
25 | #include "STFit.h" |
---|
26 | |
---|
27 | |
---|
28 | using namespace casa; |
---|
29 | |
---|
30 | namespace asap { |
---|
31 | |
---|
32 | const casa::String STFit::name_ = "FIT"; |
---|
33 | |
---|
34 | STFit::STFit(const Scantable& parent) : |
---|
35 | STSubTable( parent, name_ ) |
---|
36 | { |
---|
37 | setup(); |
---|
38 | } |
---|
39 | |
---|
40 | STFit& asap::STFit::operator =( const STFit & other ) |
---|
41 | { |
---|
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; |
---|
51 | } |
---|
52 | |
---|
53 | asap::STFit::STFit( casa::Table tab ) : STSubTable(tab, name_) |
---|
54 | { |
---|
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"); |
---|
60 | } |
---|
61 | |
---|
62 | STFit::~STFit() |
---|
63 | { |
---|
64 | } |
---|
65 | |
---|
66 | void asap::STFit::setup( ) |
---|
67 | { |
---|
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")); |
---|
74 | |
---|
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"); |
---|
81 | } |
---|
82 | |
---|
83 | uInt STFit::addEntry( const STFitEntry& fit, Int id ) |
---|
84 | { |
---|
85 | uInt rno = table_.nrow(); |
---|
86 | uInt resultid = 0; |
---|
87 | bool foundentry = false; |
---|
88 | // replace |
---|
89 | if ( id > -1 ) { |
---|
90 | Table t = table_(table_.col("ID") == id ); |
---|
91 | if (t.nrow() > 0) { |
---|
92 | rno = t.rowNumbers(table_)[0]; |
---|
93 | resultid = id; |
---|
94 | foundentry = true; |
---|
95 | } |
---|
96 | } |
---|
97 | // doesn't exist |
---|
98 | if ( rno > 0 && !foundentry ) { |
---|
99 | idCol_.get(rno-1, resultid); |
---|
100 | resultid++; |
---|
101 | } |
---|
102 | // add new row if new id |
---|
103 | if ( !foundentry ) table_.addRow(); |
---|
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; |
---|
111 | } |
---|
112 | |
---|
113 | void STFit::getEntry( STFitEntry& fit, uInt id ) const |
---|
114 | { |
---|
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)); |
---|
144 | } |
---|
145 | |
---|
146 | } //namespace |
---|