source: trunk/src/SDContainer.h@ 459

Last change on this file since 459 was 453, checked in by mar637, 20 years ago

Added SDFitTable

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 9.7 KB
Line 
1//#---------------------------------------------------------------------------
2//# SDContainer.h: A container class for single dish integrations
3//#---------------------------------------------------------------------------
4//# Copyright (C) 2004
5//# ATNF
6//#
7//# This program is free software; you can redistribute it and/or modify it
8//# under the terms of the GNU General Public License as published by the Free
9//# Software Foundation; either version 2 of the License, or (at your option)
10//# any later version.
11//#
12//# This program is distributed in the hope that it will be useful, but
13//# WITHOUT ANY WARRANTY; without even the implied warranty of
14//# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
15//# Public License for more details.
16//#
17//# You should have received a copy of the GNU General Public License along
18//# with this program; if not, write to the Free Software Foundation, Inc.,
19//# 675 Massachusetts Ave, Cambridge, MA 02139, USA.
20//#
21//# Correspondence concerning this software should be addressed as follows:
22//# Internet email: Malte.Marquarding@csiro.au
23//# Postal address: Malte Marquarding,
24//# Australia Telescope National Facility,
25//# P.O. Box 76,
26//# Epping, NSW, 2121,
27//# AUSTRALIA
28//#
29//# $Id:
30//#---------------------------------------------------------------------------
31#ifndef SDCONTAINER_H
32#define SDCONTAINER_H
33
34#include <vector>
35
36#include <casa/aips.h>
37#include <casa/BasicSL/String.h>
38#include <casa/Arrays/Array.h>
39#include <casa/Arrays/Vector.h>
40#include <casa/Containers/Block.h>
41#include <measures/Measures/MDirection.h>
42
43template<class T> class casa::Matrix;
44
45namespace asap {
46
47
48struct SDHeader {
49 casa::Int nchan;
50 casa::Int npol;
51 casa::Int nif;
52 casa::Int nbeam;
53 casa::String observer;
54 casa::String project;
55 casa::String obstype;
56 casa::String antennaname;
57 casa::Vector<casa::Double> antennaposition;
58 casa::Float equinox;
59 casa::String freqref;
60 casa::Double reffreq;
61 casa::Double bandwidth;
62 casa::Double utc;
63 casa::String fluxunit;
64 casa::String epoch;
65 void print() const ;
66};
67
68class SDFrequencyTable {
69
70public:
71
72 SDFrequencyTable() : nFreq_(0) {;}
73 virtual ~SDFrequencyTable() {;}
74
75 // Add a new entry or match an existing one. Returns the index into
76 // the table
77 casa::uInt addFrequency(casa::Double refPix, casa::Double refVal,
78 casa::Double inc);
79
80 casa::Int length() const { return nFreq_;} // # of stored Frequencies
81 void setLength(casa::uInt length) {nFreq_ = length;}
82
83 // Get attributes
84 casa::Double referencePixel(casa::uInt which) const { return refPix_[which];}
85 casa::Double referenceValue(casa::uInt which) const { return refVal_[which];}
86 casa::Double increment(casa::uInt which) const { return increment_[which];}
87 casa::Float equinox() const { return equinox_; }
88 casa::String refFrame() const { return refFrame_; }
89
90 void restFrequencies(casa::Vector<casa::Double>& rfs,
91 casa::String& rfunit ) const ;
92
93 // Set attributes
94 void setEquinox(casa::Float eq) { equinox_ = eq; }
95 void setRefFrame(const casa::String& reff) { refFrame_ = reff; }
96
97 void deleteRestFrequencies() {restFreqs_.resize(0);}
98 casa::uInt addRestFrequency(casa::Double);
99 void setRestFrequencyUnit(const casa::String& theunit)
100 { restFreqUnit_ = theunit;}
101
102private:
103 casa::uInt nFreq_;
104 casa::Vector<casa::Double> refPix_;
105 casa::Vector<casa::Double> refVal_; // Hz
106 casa::Vector<casa::Double> increment_; // Hz
107 casa::Float equinox_;
108 casa::String refFrame_;
109 casa::Vector<casa::Double> restFreqs_; // Hz
110 casa::String restFreqUnit_;
111};
112
113
114class SDContainer {
115
116public:
117 SDContainer(casa::uInt nBeam, casa::uInt nIF, casa::uInt nPol,
118 casa::uInt nChan);
119 SDContainer(casa::IPosition shp);
120
121 virtual ~SDContainer();
122
123 casa::Bool resize(casa::IPosition shp);
124
125 casa::Bool setSpectrum(const casa::Matrix<casa::Float>& spec,
126 casa::uInt whichBeam, casa::uInt whichIF);
127 casa::Bool setSpectrum(const casa::Matrix<casa::Float>& spec,
128 const casa::Vector<casa::Complex>& cSpec,
129 casa::uInt whichBeam, casa::uInt whichIF);
130 casa::Bool putSpectrum(const casa::Array<casa::Float>& spec);
131
132 casa::Bool setFlags(const casa::Matrix<casa::uChar>& flgs,
133 casa::uInt whichBeam, casa::uInt whichIF,
134 casa::Bool hasXPol=casa::False);
135 casa::Bool putFlags(const casa::Array<casa::uChar>& spec);
136
137 casa::Bool setTsys(const casa::Vector<casa::Float>& ts,
138 casa::uInt whichBeam, casa::uInt whichIF,
139 casa::Bool hasXpol);
140 casa::Bool putTsys(const casa::Array<casa::Float>& spec);
141
142 casa::Bool setDirection(const casa::Vector<casa::Double>& point,
143 casa::uInt whichBeam);
144 casa::Bool putDirection(const casa::Array<casa::Double>& dir);
145
146 casa::Bool setFrequencyMap(casa::uInt freqslot, casa::uInt whichIF);
147 casa::Bool putFreqMap(const casa::Vector<casa::uInt>& freqs);
148
149 casa::Bool setRestFrequencyMap(casa::uInt freqslot, casa::uInt whichIF);
150 casa::Bool putRestFreqMap(const casa::Vector<casa::uInt>& freqs);
151
152 casa::Array<casa::Float> getSpectrum(casa::uInt whichBeam,
153 casa::uInt whichIF);
154 casa::Array<casa::uChar> getFlags(casa::uInt whichBeam,
155 casa::uInt whichIF);
156 casa::Array<casa::Float> getTsys(casa::uInt whichBeam,
157 casa::uInt whichIF);
158 casa::Array<casa::Double> getDirection(casa::uInt whichBeam) const;
159
160 const casa::Array<casa::Float>& getSpectrum() const { return spectrum_; }
161 const casa::Array<casa::uChar>& getFlags() const { return flags_; }
162 const casa::Array<casa::Float>& getTsys() const { return tsys_; }
163 const casa::Array<casa::Double>& getDirection() const { return direction_; }
164
165 const casa::Vector<casa::uInt>& getFreqMap() const { return freqidx_; }
166 const casa::Vector<casa::uInt>& getRestFreqMap() const
167 { return restfreqidx_; }
168
169 const casa::Vector<casa::String>& getHistory() const { return history_; }
170 casa::Bool putHistory(const casa::Vector<casa::String>& hist);
171 casa::Bool appendHistory(const casa::String& hist);
172
173 casa::Bool putFitMap(const casa::Array<casa::Int>& arr);
174
175 const casa::Array<casa::Int>& getFitMap() const { return fitIDMap_; }
176
177
178 casa::Double timestamp;
179 //Double bandwidth;
180 casa::String sourcename;
181 casa::String fieldname;
182 casa::Double interval;
183 casa::Int scanid;
184 casa::Vector<casa::Float> tcal;
185 casa::String tcaltime;
186 casa::Float azimuth;
187 casa::Float elevation;
188 casa::Float parangle;
189 casa::Int refbeam;
190
191private:
192 casa::uInt nBeam_,nIF_,nPol_,nChan_;
193
194 // (nBeam,nIF,nPol,nChannel)
195 casa::Array<casa::Float> spectrum_;
196 casa::Array<casa::uChar> flags_;
197
198 // (nBeam,nIF,nPol,[nChannel]) Tsys is not really a function of
199 // channel, but this makes it easier to work with at the expense of
200 // a little memory
201 casa::Array<casa::Float> tsys_;
202 casa::Array<casa::Float> tcal_;
203
204 //(nIF) indx into "global" frequency table
205 casa::Vector<casa::uInt> freqidx_;
206
207 // (nIF) indx into "global" rest frequency table
208 casa::Vector<casa::uInt> restfreqidx_;
209
210 //(nBeam,2) maybe use Measures here...
211 casa::Array<casa::Double> direction_;
212 casa::Vector<casa::String> history_;
213 casa::Array<casa::Int> fitIDMap_;
214
215 void setSlice(casa::IPosition& start, casa::IPosition& end,
216 const casa::IPosition& shpIn, const casa::IPosition& shpOut,
217 casa::uInt whichBeam, casa::uInt whichIF, casa::Bool checkPol,
218 casa::Bool xPol) const;
219 void setSlice(casa::IPosition& start, casa::IPosition& end,
220 const casa::IPosition& shape,
221 casa::uInt whichBeam, casa::uInt whichIF) const;
222};
223
224
225
226class SDDataDesc {
227
228public:
229
230 // Constructor
231 SDDataDesc() : n_(0) {;}
232 ~SDDataDesc() {;}
233
234 // Add an entry if source name and Integer ID (can be anything you
235 // like, such as FreqID) are unique. You can add secondary entries
236 // direction and another integer index which are just stored along
237 // with the the primary entries
238 casa::uInt addEntry(const casa::String& source, casa::uInt ID,
239 const casa::MDirection& secDir, casa::uInt secID);
240
241 // Number of entries
242 casa::Int length() const { return n_;}
243
244 // Get attributes
245 casa::String source(casa::uInt which) const {return source_[which];}
246 casa::uInt ID(casa::uInt which) const {return ID_[which];}
247 casa::uInt secID(casa::uInt which) const {return secID_[which];}
248 casa::MDirection secDir(casa::uInt which) const {return secDir_[which];}
249
250 // Summary
251 void summary() const;
252
253private:
254 casa::uInt n_;
255 casa::Vector<casa::String> source_;
256 casa::Vector<casa::uInt> ID_, secID_;
257 casa::Block<casa::MDirection> secDir_;
258
259 SDDataDesc(const SDDataDesc& other);
260
261};
262
263class SDFitTable {
264public:
265 // Create a FitTable with "n" rows
266 SDFitTable(casa::uInt n=0) : n_(n) {;};
267
268 casa::uInt length() const { return n_; };
269 const casa::Vector<casa::Double>& getFitParameters(casa::uInt which) const;
270 const casa::Vector<casa::Bool>& getFitParameterMask(casa::uInt which) const;
271 const casa::Vector<casa::String>& getFitFunctions(casa::uInt which) const;
272 const casa::Vector<casa::Int>& getFitComponents(casa::uInt which) const;
273
274 void putFitParameters(casa::uInt whichRow,
275 const casa::Vector<casa::Double>& arr);
276 void putFitParameterMask(casa::uInt whichRow,
277 const casa::Vector<casa::Bool>& arr);
278 void putFitFunctions(casa::uInt whichRow,
279 const casa::Vector<casa::String>& arr);
280 void putFitComponents(casa::uInt whichRow,
281 const casa::Vector<casa::Int>& arr);
282
283private:
284
285 casa::uInt n_;
286 casa::Vector<casa::Vector<casa::Double> > fitParms_;
287 // (npars,nrows)
288 casa::Vector<casa::Vector<casa::Bool> > parMask_;
289 // the fit function names (nnames,nrows)
290 casa::Vector<casa::Vector<casa::String> > fitFuncs_;
291 // the number of components of the function (ncomps, nrows)
292 casa::Vector<casa::Vector<casa::Int> > fitComps_;
293
294};
295
296} // namespace
297#endif
Note: See TracBrowser for help on using the repository browser.