source: trunk/src/STFit.cpp@ 1355

Last change on this file since 1355 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
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"));
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]83uInt 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]113void 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
Note: See TracBrowser for help on using the repository browser.