source: trunk/src/STGrid.h@ 2396

Last change on this file since 2396 was 2396, checked in by Takeshi Nakazato, 13 years ago

New Development: No

JIRA Issue: Yes CAS-2816

Ready for Test: Yes

Interface Changes: No

What Interface Changed: Please list interface changes

Test Programs: List test programs

Put in Release Notes: Yes/No

Module(s): Module Names change impacts.

Description: Describe your changes here...

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