source: branches/polybatch/src/STFit.cpp@ 2161

Last change on this file since 2161 was 1924, checked in by Malte Marquarding, 14 years ago

Ticket #206: use STFitEntry as return objetc instead of pointer wrnagling.

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 4.3 KB
RevLine 
[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]28using namespace casa;
29
[960]30namespace asap {
[468]31
[960]32const casa::String STFit::name_ = "FIT";
[468]33
[960]34STFit::STFit(const Scantable& parent) :
35 STSubTable( parent, name_ )
[468]36{
[960]37 setup();
[468]38}
39
[960]40STFit& 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]53asap::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]62STFit::~STFit()
[468]63{
64}
65
[960]66void 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"));
[1924]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");
[1924]80 // errCol_.attach(table_,"ERRORS");
[960]81 maskCol_.attach(table_,"PARMASKS");
82 frameCol_.attach(table_,"FRAMEINFO");
[468]83}
84
[960]85uInt 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 ) {
92 Table t = table_(table_.col("ID") == id );
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();
[1924]106
[960]107 funcCol_.put(rno, mathutil::toVectorString(fit.getFunctions()));
108 compCol_.put(rno, Vector<Int>(fit.getComponents()));
[1924]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);
[1924]125
[960]126 return resultid;
[468]127}
128
[972]129void STFit::getEntry( STFitEntry& fit, uInt id ) const
[468]130{
[960]131 Table t = table_(table_.col("ID") == Int(id) );
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);
[1924]149 std::vector<float> dstl(dvec.begin(), dvec.end());
150 fit.setParameters(dstl);
151 /*
[960]152 dvec.tovector(dstl);
153 fit.setParameters(dstl);
[1924]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
Note: See TracBrowser for help on using the repository browser.