source: trunk/src/ScantableWrapper.h@ 1451

Last change on this file since 1451 was 1430, checked in by Malte Marquarding, 16 years ago

merged in unflag option from alma branch

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 6.6 KB
Line 
1//
2// C++ Interface: Scantable
3//
4// Description:
5//
6//
7// Author: Malte Marquarding <asap@atnf.csiro.au>, (C) 2006
8//
9// Copyright: See COPYING file that comes with this distribution
10//
11//
12#ifndef ASAPSCANTABLEWRAPPER_H
13#define ASAPSCANTABLEWRAPPER_H
14
15#include <vector>
16#include <string>
17#include <casa/Arrays/Vector.h>
18
19#include "MathUtils.h"
20#include "STFit.h"
21#include "Scantable.h"
22
23namespace asap {
24/**
25 * This class contains and wraps a CountedPtr<Scantable>, as the CountedPtr
26 * class does not provide the dor operator which is need for references
27 * in boost::python
28 * see Scantable for interfce description
29 *
30 * @brief The main ASAP data container wrapper
31 * @author Malte Marquarding
32 * @date 2006-02-23
33 * @version 2.0a
34*/
35class ScantableWrapper {
36
37public:
38 explicit ScantableWrapper( const std::string& name,
39 int type=0)
40 {
41 casa::Table::TableType tp = casa::Table::Memory;
42 if ( type == 1 ) tp = casa::Table::Plain;
43 table_ = new Scantable(name, tp);
44 }
45
46 explicit ScantableWrapper(int type=0)
47 {
48 casa::Table::TableType tp = casa::Table::Memory;
49 if ( type == 1) tp = casa::Table::Plain;
50 table_= new Scantable(tp);
51 }
52
53 explicit ScantableWrapper(casa::CountedPtr<Scantable> cp) : table_(cp) {;}
54
55 ScantableWrapper(const ScantableWrapper& mt) :
56 table_(mt.getCP()) {;}
57
58 void assign(const ScantableWrapper& mt)
59 { table_= mt.getCP(); }
60
61 ScantableWrapper copy() {
62 return ScantableWrapper(new Scantable(*(this->getCP()), false));
63 }
64
65 std::vector<float> getSpectrum( int whichrow=0,
66 const std::string& poltype="" ) const {
67 return table_->getSpectrum(whichrow, poltype);
68 }
69 // std::string getPolarizationLabel(bool linear, bool stokes, bool linPol, int polIdx) const {
70 // Boost fails with 4 arguments.
71 std::string getPolarizationLabel(int index, const std::string& ptype) const {
72 return table_->getPolarizationLabel(index, ptype);
73 }
74
75 std::vector<double> getAbcissa(int whichrow=0) const
76 { return table_->getAbcissa(whichrow); }
77
78 std::string getAbcissaLabel(int whichrow=0) const
79 { return table_->getAbcissaLabel(whichrow); }
80
81 float getTsys(int whichrow=0) const
82 { return table_->getTsys(whichrow); }
83
84 std::string getTime(int whichrow=0) const
85 { return table_->getTime(whichrow); }
86
87 double getIntTime(int whichrow=0) const
88 { return table_->getIntTime(whichrow); }
89
90 std::string getDirectionString(int whichrow=0) const
91 { return table_->getDirectionString(whichrow); }
92
93 std::string getFluxUnit() const { return table_->getFluxUnit(); }
94
95 void setFluxUnit(const std::string& unit) { table_->setFluxUnit(unit); }
96
97 void setInstrument(const std::string& name) {table_->setInstrument(name);}
98 void setFeedType(const std::string& ftype) {table_->setFeedType(ftype);}
99
100 std::vector<bool> getMask(int whichrow=0) const
101 { return table_->getMask(whichrow); }
102
103 /**
104 void flag(const std::vector<bool>& msk=std::vector<bool>())
105 { table_->flag(msk); }
106 **/
107 void flag(const std::vector<bool>& msk=std::vector<bool>(), bool unflag=false)
108 { table_->flag(msk, unflag); }
109
110 std::string getSourceName(int whichrow=0) const
111 { return table_->getSourceName(whichrow); }
112
113 float getElevation(int whichrow=0) const
114 { return table_->getElevation(whichrow); }
115
116 float getAzimuth(int whichrow=0) const
117 { return table_->getAzimuth(whichrow); }
118
119 float getParAngle(int whichrow=0) const
120 { return table_->getParAngle(whichrow); }
121
122
123 void setSpectrum(std::vector<float> spectrum, int whichrow=0)
124 { table_->setSpectrum(spectrum, whichrow); }
125
126 std::vector<uint> getIFNos() { return table_->getIFNos(); }
127 int getIF(int whichrow) const {return table_->getIF(whichrow);}
128 std::vector<uint> getBeamNos() { return table_->getBeamNos(); }
129 int getBeam(int whichrow) const {return table_->getBeam(whichrow);}
130 std::vector<uint> getPolNos() { return table_->getPolNos(); }
131 int getPol(int whichrow) const {return table_->getPol(whichrow);}
132 int getCycle(int whichrow) const {return table_->getCycle(whichrow);}
133 std::vector<uint> getScanNos() { return table_->getScanNos(); }
134 int getScan(int whichrow) const {return table_->getScan(whichrow);}
135
136 STSelector getSelection() const { return table_->getSelection(); }
137 void setSelection(const STSelector& sts)
138 { return table_->setSelection(sts);}
139
140 std::string getPolType() const { return table_->getPolType(); }
141
142 int nif(int scanno=-1) const {return table_->nif(scanno);}
143 int nbeam(int scanno=-1) const {return table_->nbeam(scanno);}
144 int npol(int scanno=-1) const {return table_->npol(scanno);}
145 int nchan(int ifno=-1) const {return table_->nchan(ifno);}
146 int nscan() const {return table_->nscan();}
147 int nrow() const {return table_->nrow();}
148 int ncycle(int scanno) const {return table_->ncycle(scanno);}
149
150 void makePersistent(const std::string& fname)
151 { table_->makePersistent(fname); }
152
153 void setSourceType(int stype)
154 { table_->setSourceType(stype); }
155
156 void shift(int npix)
157 { table_->shift(npix); }
158
159 void setRestFrequencies(double rf, const std::string& name,
160 const std::string& unit)
161 { table_->setRestFrequencies(rf, name, unit); }
162/*
163 void setRestFrequencies(const std::string& name) {
164 table_->setRestFrequencies(name);
165 }
166*/
167
168 std::vector<double> getRestFrequencies() const
169 { return table_->getRestFrequencies(); }
170
171 void setCoordInfo(std::vector<string> theinfo) {
172 table_->setCoordInfo(theinfo);
173 }
174 std::vector<string> getCoordInfo() const {
175 return table_->getCoordInfo();
176 }
177
178 void setDirection(const std::string& refstr="")
179 { table_->setDirectionRefString(refstr); }
180
181 casa::CountedPtr<Scantable> getCP() const {return table_;}
182 Scantable* getPtr() {return &(*table_);}
183
184 std::string summary(bool verbose=false) const {
185 return table_->summary(verbose);
186 }
187
188 std::vector<std::string> getHistory()const
189 { return table_->getHistory(); }
190
191 void addHistory(const std::string& hist)
192 { table_->addHistory(hist); }
193
194 void addFit(const STFitEntry& fit, int row)
195 { table_->addFit(fit, row); }
196
197 STFitEntry getFit(int whichrow) const
198 { return table_->getFit(whichrow); }
199
200 void calculateAZEL() { table_->calculateAZEL(); };
201
202 std::vector<std::string> columnNames() const
203 { return table_->columnNames(); }
204
205 std::string getAntennaName() const
206 { return table_->getAntennaName(); }
207
208 int checkScanInfo(const vector<int>& scanlist) const
209 { return table_->checkScanInfo(scanlist); }
210
211 std::vector<double> getDirectionVector(int whichrow) const
212 { return table_->getDirectionVector(whichrow); }
213
214private:
215 casa::CountedPtr<Scantable> table_;
216};
217
218} // namespace
219#endif
220
Note: See TracBrowser for help on using the repository browser.