source: branches/asap-3.x/src/STFit.cpp@ 2676

Last change on this file since 2676 was 1000, checked in by mar637, 19 years ago

fixed bug due to incorrect use of Table::rowNumbers (RTFM!).
added flagging

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 3.7 KB
Line 
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
28using namespace casa;
29
30namespace asap {
31
32const casa::String STFit::name_ = "FIT";
33
34STFit::STFit(const Scantable& parent) :
35 STSubTable( parent, name_ )
36{
37 setup();
38}
39
40STFit& 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
53asap::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
62STFit::~STFit()
63{
64}
65
66void 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
83uInt 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
113void 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
Note: See TracBrowser for help on using the repository browser.