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

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

merged Takeshi's fix to asapgrid and STGrid class form trunk
File size: 7.0 KB
RevLine 
[2356]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>
[2360]18#include <vector>
[2356]19
20#include <casa/BasicSL/String.h>
[2381]21#include <casa/Arrays/Array.h>
[2356]22#include <casa/Arrays/Vector.h>
[2381]23#include <casa/Containers/RecordField.h>
[2356]24
25#include <tables/Tables/Table.h>
[2379]26#include <tables/Tables/ScalarColumn.h>
27#include <tables/Tables/ArrayColumn.h>
[2393]28
29#include "concurrent.h"
[2356]30
31using namespace std ;
32using namespace casa ;
33
34namespace asap {
35class STGrid
36{
37public:
38  STGrid() ;
39  STGrid( const string infile ) ;
[2390]40  STGrid( const vector<string> infile ) ;
[2356]41  virtual ~STGrid() {} ;
42
43  void setFileIn( const string infile ) ;
[2390]44  void setFileList( const vector<string> infile ) ;
[2356]45
[2362]46  void setIF( unsigned int ifno ) { ifno_ = ifno ; } ;
47
[2360]48  void setPolList( vector<unsigned int> pols ) ;
49
[2364]50  void setScanList( vector<unsigned int> scans ) ;
51
[2356]52  void defineImage( int nx=-1,
53                    int ny=-1,
54                    string scellx="",
55                    string scelly="",
56                    string scenter="" ) ;
[2364]57  void setFunc( string convtype="box",
58                int convsupport=-1 ) ;
[2361]59
60  void setWeight( const string wType="uniform" ) ;
61
[2396]62  void enableClip() { doclip_ = True ; } ;
63  void disableClip() { doclip_ = False ; } ;
64
[2356]65  void grid() ;
66 
67  string saveData( string outfile="" ) ;
68
69private:
70  void init() ;
71
[2382]72  // actual gridding
73  void gridPerRow() ;
[2396]74  void gridPerRowWithClipping() ;
[2382]75
[2396]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
[2388]88  void setupGrid() ;
[2356]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 ) ;
[2388]98  void mapExtent( Double &xmin, Double &xmax,
99                  Double &ymin, Double &ymax ) ;
[2368]100
[2378]101  void setData( Array<Complex> &gdata,
[2368]102                Array<Float> &gwgt ) ;
[2356]103 
[2393]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 ) ;
[2379]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 ) ;
[2356]122
[2375]123  void getWeight( Array<Float> &w,
124                  Array<Float> &tsys,
125                  Array<Double> &tint ) ;
[2382]126 
[2375]127  void toInt( Array<uChar> &u, Array<Int> &v ) ;
128  void toInt( Array<uInt> &u, Array<Int> &v ) ;
[2356]129
[2375]130  void toPixel( Array<Double> &world, Array<Double> &pixel ) ;
[2356]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
[2371]138  void prepareTable( Table &tab, String &name ) ;
139
[2379]140  void selectData() ;
141  void setupArray() ;
[2378]142
[2398]143  void updateChunkShape() ;
[2382]144  void attach( Table &tab ) ;
[2379]145
[2384]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,
[2381]164                     Int *chanMap,
165                     Int *polMap ) ;
[2396]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 ) ;
[2379]193
[2382]194  void initPol( Int ipol ) ;
[2390]195  void initTable( uInt idx ) ;
[2386]196  Bool isMultiIF( Table &tab ) ;
[2459]197  void fillMainColumns( Table &tab ) ;
[2393]198  static bool produceChunk(void *ctx) throw(concurrent::PCException);
199  static void consumeChunk(void *ctx) throw(concurrent::PCException);
[2396]200  static void consumeChunkWithClipping(void *ctx) throw(concurrent::PCException);
[2381]201
[2382]202
[2388]203  // user input
204  Int nxUI_ ;
205  Int nyUI_ ;
206  String cellxUI_ ;
207  String cellyUI_ ;
208  String centerUI_ ;
209
[2389]210  Block<String> infileList_ ;
[2390]211  uInt nfile_ ;
[2362]212  Int ifno_ ;
[2396]213  Bool doclip_ ;
[2393]214
[2356]215  Int nx_ ;
216  Int ny_ ;
[2361]217  Int npol_ ;
[2386]218  Int npolOrg_ ;
[2361]219  Int nchan_ ;
[2356]220  Double cellx_ ;
221  Double celly_ ;
222  Vector<Double> center_ ;
223  String convType_ ;
224  Int convSupport_ ;
225  Int userSupport_ ;
226  Int convSampling_ ;
[2360]227  Vector<uInt> pollist_ ;
[2364]228  Vector<uInt> scanlist_ ;
[2361]229  String wtype_ ;
[2393]230  Block<Table> tableList_ ;
231  Vector<uInt> rows_ ;
232  Int nchunk_ ;
[2356]233
[2393]234  /////////////// gridPerRow variable
235  IPosition vshape_;
236  IPosition wshape_;
237  IPosition dshape_;
238  // loop variable
239  Int nrow_ ;
240  Array<Float> data_ ;
241
[2356]242  Table tab_ ;
[2393]243  // per pol
[2382]244  Table ptab_ ;
[2379]245  ROArrayColumn<Float> spectraCol_ ;
246  ROArrayColumn<uChar> flagtraCol_ ;
247  ROArrayColumn<Double> directionCol_ ;
248  ROScalarColumn<uInt> flagRowCol_ ;
249  ROArrayColumn<Float> tsysCol_ ;
250  ROScalarColumn<Double> intervalCol_ ;
[2393]251
[2379]252  Int nprocessed_ ;
[2381]253
254
[2393]255  double eGetData_;
256  double eToPixel_;
257  double eGGridSD_;
[2356]258};
259}
260#endif
Note: See TracBrowser for help on using the repository browser.