source: branches/hpc33/src/STGrid.h@ 2461

Last change on this file since 2461 was 2459, checked in by Kana Sugimoto, 13 years ago

merged Takeshi's fix to asapgrid and STGrid class form trunk
File size: 7.0 KB
Line 
1//
2// C++ Interface: STGrid
3//
4// Description:
5//
6//
7// Author: Takeshi Nakazato <takeshi.nakazato@nao.ac.jp>, (C) 2011
8//
9// Copyright: See COPYING file that comes with this distribution
10//
11//
12#ifndef ASAPSTGRID_H
13#define ASAPSTGRID_H
14
15#include <iostream>
16#include <fstream>
17#include <string>
18#include <vector>
19
20#include <casa/BasicSL/String.h>
21#include <casa/Arrays/Array.h>
22#include <casa/Arrays/Vector.h>
23#include <casa/Containers/RecordField.h>
24
25#include <tables/Tables/Table.h>
26#include <tables/Tables/ScalarColumn.h>
27#include <tables/Tables/ArrayColumn.h>
28
29#include "concurrent.h"
30
31using namespace std ;
32using namespace casa ;
33
34namespace asap {
35class STGrid
36{
37public:
38 STGrid() ;
39 STGrid( const string infile ) ;
40 STGrid( const vector<string> infile ) ;
41 virtual ~STGrid() {} ;
42
43 void setFileIn( const string infile ) ;
44 void setFileList( const vector<string> infile ) ;
45
46 void setIF( unsigned int ifno ) { ifno_ = ifno ; } ;
47
48 void setPolList( vector<unsigned int> pols ) ;
49
50 void setScanList( vector<unsigned int> scans ) ;
51
52 void defineImage( int nx=-1,
53 int ny=-1,
54 string scellx="",
55 string scelly="",
56 string scenter="" ) ;
57 void setFunc( string convtype="box",
58 int convsupport=-1 ) ;
59
60 void setWeight( const string wType="uniform" ) ;
61
62 void enableClip() { doclip_ = True ; } ;
63 void disableClip() { doclip_ = False ; } ;
64
65 void grid() ;
66
67 string saveData( string outfile="" ) ;
68
69private:
70 void init() ;
71
72 // actual gridding
73 void gridPerRow() ;
74 void gridPerRowWithClipping() ;
75
76 // clipping
77 void clipMinMax( Array<Complex> &data,
78 Array<Float> &weight,
79 Array<Int> &npoints,
80 Array<Complex> &clipmin,
81 Array<Float> &clipwmin,
82 Array<Float> &clipcmin,
83 Array<Complex> &clipmax,
84 Array<Float> &clipwmax,
85 Array<Float> &clipcmax ) ;
86
87
88 void setupGrid() ;
89 void setupGrid( Int &nx,
90 Int &ny,
91 String &cellx,
92 String &celly,
93 Double &xmin,
94 Double &xmax,
95 Double &ymin,
96 Double &ymax,
97 String &center ) ;
98 void mapExtent( Double &xmin, Double &xmax,
99 Double &ymin, Double &ymax ) ;
100
101 void setData( Array<Complex> &gdata,
102 Array<Float> &gwgt ) ;
103
104 Int getDataChunk( IPosition const &wshape,
105 IPosition const &vshape,
106 IPosition const &dshape,
107 Array<Complex> &spectra,
108 Array<Double> &direction,
109 Array<Int> &flagtra,
110 Array<Int> &rflag,
111 Array<Float> &weight ) ;
112 Int getDataChunk( Array<Complex> &spectra,
113 Array<Double> &direction,
114 Array<Int> &flagtra,
115 Array<Int> &rflag,
116 Array<Float> &weight ) ;
117 Int getDataChunk( Array<Float> &spectra,
118 Array<Double> &direction,
119 Array<uChar> &flagtra,
120 Array<uInt> &rflag,
121 Array<Float> &weight ) ;
122
123 void getWeight( Array<Float> &w,
124 Array<Float> &tsys,
125 Array<Double> &tint ) ;
126
127 void toInt( Array<uChar> &u, Array<Int> &v ) ;
128 void toInt( Array<uInt> &u, Array<Int> &v ) ;
129
130 void toPixel( Array<Double> &world, Array<Double> &pixel ) ;
131
132 void boxFunc( Vector<Float> &convFunc, Int &convSize ) ;
133 void spheroidalFunc( Vector<Float> &convFunc ) ;
134 void gaussFunc( Vector<Float> &convFunc ) ;
135 void pbFunc( Vector<Float> &convFunc ) ;
136 void setConvFunc( Vector<Float> &convFunc ) ;
137
138 void prepareTable( Table &tab, String &name ) ;
139
140 void selectData() ;
141 void setupArray() ;
142
143 void updateChunkShape() ;
144 void attach( Table &tab ) ;
145
146 void call_ggridsd( Array<Double> &xy,
147 Array<Complex> &values,
148 Int &nvispol,
149 Int &nvischan,
150 Array<Int> &flag,
151 Array<Int> &rflag,
152 Array<Float> &weight,
153 Int &nrow,
154 Int &irow,
155 Array<Complex> &grid,
156 Array<Float> &wgrid,
157 Int &nx,
158 Int &ny,
159 Int &npol,
160 Int &nchan,
161 Int &support,
162 Int &sampling,
163 Vector<Float> &convFunc,
164 Int *chanMap,
165 Int *polMap ) ;
166 void call_ggridsd2( Array<Double> &xy,
167 Array<Complex> &values,
168 Int &nvispol,
169 Int &nvischan,
170 Array<Int> &flag,
171 Array<Int> &rflag,
172 Array<Float> &weight,
173 Int &nrow,
174 Int &irow,
175 Array<Complex> &grid,
176 Array<Float> &wgrid,
177 Array<Int> &npoints,
178 Array<Complex> &clipmin,
179 Array<Float> &clipwmin,
180 Array<Float> &clipcmin,
181 Array<Complex> &clipmax,
182 Array<Float> &clipwmax,
183 Array<Float> &clipcmax,
184 Int &nx,
185 Int &ny,
186 Int &npol,
187 Int &nchan,
188 Int &support,
189 Int &sampling,
190 Vector<Float> &convFunc,
191 Int *chanMap,
192 Int *polMap ) ;
193
194 void initPol( Int ipol ) ;
195 void initTable( uInt idx ) ;
196 Bool isMultiIF( Table &tab ) ;
197 void fillMainColumns( Table &tab ) ;
198 static bool produceChunk(void *ctx) throw(concurrent::PCException);
199 static void consumeChunk(void *ctx) throw(concurrent::PCException);
200 static void consumeChunkWithClipping(void *ctx) throw(concurrent::PCException);
201
202
203 // user input
204 Int nxUI_ ;
205 Int nyUI_ ;
206 String cellxUI_ ;
207 String cellyUI_ ;
208 String centerUI_ ;
209
210 Block<String> infileList_ ;
211 uInt nfile_ ;
212 Int ifno_ ;
213 Bool doclip_ ;
214
215 Int nx_ ;
216 Int ny_ ;
217 Int npol_ ;
218 Int npolOrg_ ;
219 Int nchan_ ;
220 Double cellx_ ;
221 Double celly_ ;
222 Vector<Double> center_ ;
223 String convType_ ;
224 Int convSupport_ ;
225 Int userSupport_ ;
226 Int convSampling_ ;
227 Vector<uInt> pollist_ ;
228 Vector<uInt> scanlist_ ;
229 String wtype_ ;
230 Block<Table> tableList_ ;
231 Vector<uInt> rows_ ;
232 Int nchunk_ ;
233
234 /////////////// gridPerRow variable
235 IPosition vshape_;
236 IPosition wshape_;
237 IPosition dshape_;
238 // loop variable
239 Int nrow_ ;
240 Array<Float> data_ ;
241
242 Table tab_ ;
243 // per pol
244 Table ptab_ ;
245 ROArrayColumn<Float> spectraCol_ ;
246 ROArrayColumn<uChar> flagtraCol_ ;
247 ROArrayColumn<Double> directionCol_ ;
248 ROScalarColumn<uInt> flagRowCol_ ;
249 ROArrayColumn<Float> tsysCol_ ;
250 ROScalarColumn<Double> intervalCol_ ;
251
252 Int nprocessed_ ;
253
254
255 double eGetData_;
256 double eToPixel_;
257 double eGGridSD_;
258};
259}
260#endif
Note: See TracBrowser for help on using the repository browser.