source: trunk/src/STGrid.h @ 2398

Last change on this file since 2398 was 2398, checked in by Takeshi Nakazato, 12 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...

Reorganization of procedure:

  • Dropped STGrid::gridPerPol
  • Broker is created per input table
  • STGrid::examine is redefined as STGrid::updateChunkShape
File size: 6.9 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  static bool produceChunk(void *ctx) throw(concurrent::PCException);
198  static void consumeChunk(void *ctx) throw(concurrent::PCException);
199  static void consumeChunkWithClipping(void *ctx) throw(concurrent::PCException);
200
201
202  // user input
203  Int nxUI_ ;
204  Int nyUI_ ;
205  String cellxUI_ ;
206  String cellyUI_ ;
207  String centerUI_ ;
208
209  Block<String> infileList_ ;
210  uInt nfile_ ;
211  Int ifno_ ;
212  Bool doclip_ ;
213
214  Int nx_ ;
215  Int ny_ ;
216  Int npol_ ;
217  Int npolOrg_ ;
218  Int nchan_ ;
219  Double cellx_ ;
220  Double celly_ ;
221  Vector<Double> center_ ;
222  String convType_ ;
223  Int convSupport_ ;
224  Int userSupport_ ;
225  Int convSampling_ ;
226  Vector<uInt> pollist_ ;
227  Vector<uInt> scanlist_ ;
228  String wtype_ ;
229  Block<Table> tableList_ ;
230  Vector<uInt> rows_ ;
231  Int nchunk_ ;
232
233  /////////////// gridPerRow variable
234  IPosition vshape_;
235  IPosition wshape_;
236  IPosition dshape_;
237  // loop variable
238  Int nrow_ ;
239  Array<Float> data_ ;
240
241  Table tab_ ;
242  // per pol
243  Table ptab_ ;
244  ROArrayColumn<Float> spectraCol_ ;
245  ROArrayColumn<uChar> flagtraCol_ ;
246  ROArrayColumn<Double> directionCol_ ;
247  ROScalarColumn<uInt> flagRowCol_ ;
248  ROArrayColumn<Float> tsysCol_ ;
249  ROScalarColumn<Double> intervalCol_ ;
250
251  Int nprocessed_ ;
252
253
254  double eGetData_;
255  double eToPixel_;
256  double eGGridSD_;
257};
258}
259#endif
Note: See TracBrowser for help on using the repository browser.