source: trunk/src/MSFiller.cpp@ 2237

Last change on this file since 2237 was 2237, checked in by Takeshi Nakazato, 14 years ago

New Development: No

JIRA Issue: No

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...

Refactoring MSFiller::fill() to detect bottle neck of the process.
There are no changes effectively.
File size: 66.0 KB
Line 
1//
2// C++ Interface: MSFiller
3//
4// Description:
5//
6// This class is specific filler for MS format
7//
8// Takeshi Nakazato <takeshi.nakazato@nao.ac.jp>, (C) 2010
9//
10// Copyright: See COPYING file that comes with this distribution
11//
12//
13
14#include <iostream>
15#include <map>
16
17#include <tables/Tables/ExprNode.h>
18#include <tables/Tables/TableIter.h>
19#include <tables/Tables/TableColumn.h>
20#include <tables/Tables/ScalarColumn.h>
21#include <tables/Tables/ArrayColumn.h>
22#include <tables/Tables/TableParse.h>
23#include <tables/Tables/TableRow.h>
24
25#include <casa/Containers/RecordField.h>
26#include <casa/Logging/LogIO.h>
27#include <casa/Arrays/Slicer.h>
28#include <casa/Quanta/MVTime.h>
29#include <casa/OS/Path.h>
30
31#include <measures/Measures/Stokes.h>
32#include <measures/Measures/MEpoch.h>
33#include <measures/Measures/MCEpoch.h>
34#include <measures/Measures/MFrequency.h>
35#include <measures/Measures/MCFrequency.h>
36#include <measures/Measures/MPosition.h>
37#include <measures/Measures/MCPosition.h>
38#include <measures/Measures/MDirection.h>
39#include <measures/Measures/MCDirection.h>
40#include <measures/Measures/MeasConvert.h>
41#include <measures/TableMeasures/ScalarMeasColumn.h>
42#include <measures/TableMeasures/ArrayMeasColumn.h>
43#include <measures/TableMeasures/ScalarQuantColumn.h>
44#include <measures/TableMeasures/ArrayQuantColumn.h>
45
46#include <ms/MeasurementSets/MSAntennaIndex.h>
47
48#include <atnf/PKSIO/SrcType.h>
49
50#include "MSFiller.h"
51#include "STHeader.h"
52
53#include <ctime>
54#include <sys/time.h>
55
56using namespace casa ;
57using namespace std ;
58
59namespace asap {
60double MSFiller::gettimeofday_sec()
61{
62 struct timeval tv ;
63 gettimeofday( &tv, NULL ) ;
64 return tv.tv_sec + (double)tv.tv_usec*1.0e-6 ;
65}
66
67MSFiller::MSFiller( casa::CountedPtr<Scantable> stable )
68 : table_( stable ),
69 tablename_( "" ),
70 antenna_( -1 ),
71 antennaStr_(""),
72 getPt_( True ),
73 isFloatData_( False ),
74 isData_( False ),
75 isDoppler_( False ),
76 isFlagCmd_( False ),
77 isFreqOffset_( False ),
78 isHistory_( False ),
79 isProcessor_( False ),
80 isSysCal_( False ),
81 isWeather_( False ),
82 colTsys_( "TSYS_SPECTRUM" ),
83 colTcal_( "TCAL_SPECTRUM" )
84{
85 os_ = LogIO() ;
86 os_.origin( LogOrigin( "MSFiller", "MSFiller()", WHERE ) ) ;
87}
88
89MSFiller::~MSFiller()
90{
91 os_.origin( LogOrigin( "MSFiller", "~MSFiller()", WHERE ) ) ;
92}
93
94bool MSFiller::open( const std::string &filename, const casa::Record &rec )
95{
96 os_.origin( LogOrigin( "MSFiller", "open()", WHERE ) ) ;
97 //double startSec = gettimeofday_sec() ;
98 //os_ << "start MSFiller::open() startsec=" << startSec << LogIO::POST ;
99 //os_ << " filename = " << filename << endl ;
100
101 // parsing MS options
102 if ( rec.isDefined( "ms" ) ) {
103 Record msrec = rec.asRecord( "ms" ) ;
104 if ( msrec.isDefined( "getpt" ) ) {
105 getPt_ = msrec.asBool( "getpt" ) ;
106 }
107 if ( msrec.isDefined( "antenna" ) ) {
108 if ( msrec.type( msrec.fieldNumber( "antenna" ) ) == TpInt ) {
109 antenna_ = msrec.asInt( "antenna" ) ;
110 }
111 else {
112 //antenna_ = atoi( msrec.asString( "antenna" ).c_str() ) ;
113 antennaStr_ = msrec.asString( "antenna" ) ;
114 }
115 }
116 else {
117 antenna_ = 0 ;
118 }
119 }
120
121 MeasurementSet *tmpMS = new MeasurementSet( filename, Table::Old ) ;
122 //mstable_ = (*tmpMS)( tmpMS->col("ANTENNA1") == antenna_
123 // && tmpMS->col("ANTENNA1") == tmpMS->col("ANTENNA2") ) ;
124 tablename_ = tmpMS->tableName() ;
125 if ( antenna_ == -1 && antennaStr_.size() > 0 ) {
126 MSAntennaIndex msAntIdx( tmpMS->antenna() ) ;
127 Vector<Int> id = msAntIdx.matchAntennaName( antennaStr_ ) ;
128 if ( id.size() > 0 )
129 antenna_ = id[0] ;
130 }
131
132 os_ << "Parsing MS options" << endl ;
133 os_ << " getPt = " << getPt_ << endl ;
134 os_ << " antenna = " << antenna_ << endl ;
135 os_ << " antennaStr = " << antennaStr_ << LogIO::POST ;
136
137 mstable_ = MeasurementSet( (*tmpMS)( tmpMS->col("ANTENNA1") == antenna_
138 && tmpMS->col("ANTENNA1") == tmpMS->col("ANTENNA2") ) ) ;
139// stringstream ss ;
140// ss << "SELECT FROM $1 WHERE ANTENNA1 == ANTENNA2 && ANTENNA1 == " << antenna_ ;
141// String taql( ss.str() ) ;
142// mstable_ = MeasurementSet( tableCommand( taql, *tmpMS ) ) ;
143 delete tmpMS ;
144
145 // check which data column exists
146 isFloatData_ = mstable_.tableDesc().isColumn( "FLOAT_DATA" ) ;
147 isData_ = mstable_.tableDesc().isColumn( "DATA" ) ;
148
149 //double endSec = gettimeofday_sec() ;
150 //os_ << "end MSFiller::open() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
151 return true ;
152}
153
154void MSFiller::fill()
155{
156 os_.origin( LogOrigin( "MSFiller", "fill()", WHERE ) ) ;
157 //double startSec = gettimeofday_sec() ;
158 //os_ << "start MSFiller::fill() startSec=" << startSec << LogIO::POST ;
159
160 //double time0 = gettimeofday_sec() ;
161 //os_ << "start init fill: " << time0 << LogIO::POST ;
162
163 // Initialize header
164 STHeader sdh ;
165 initHeader( sdh ) ;
166
167 // check if optional table exists
168 //const TableRecord msrec = tablesel_.keywordSet() ;
169 const TableRecord msrec = mstable_.keywordSet() ;
170 isDoppler_ = msrec.isDefined( "DOPPLER" ) ;
171 if ( isDoppler_ )
172 if ( mstable_.doppler().nrow() == 0 )
173 isDoppler_ = False ;
174 isFlagCmd_ = msrec.isDefined( "FLAG_CMD" ) ;
175 if ( isFlagCmd_ )
176 if ( mstable_.flagCmd().nrow() == 0 )
177 isFlagCmd_ = False ;
178 isFreqOffset_ = msrec.isDefined( "FREQ_OFFSET" ) ;
179 if ( isFreqOffset_ )
180 if ( mstable_.freqOffset().nrow() == 0 )
181 isFreqOffset_ = False ;
182 isHistory_ = msrec.isDefined( "HISTORY" ) ;
183 if ( isHistory_ )
184 if ( mstable_.history().nrow() == 0 )
185 isHistory_ = False ;
186 isProcessor_ = msrec.isDefined( "PROCESSOR" ) ;
187 if ( isProcessor_ )
188 if ( mstable_.processor().nrow() == 0 )
189 isProcessor_ = False ;
190 isSysCal_ = msrec.isDefined( "SYSCAL" ) ;
191 if ( isSysCal_ )
192 if ( mstable_.sysCal().nrow() == 0 )
193 isSysCal_ = False ;
194 isWeather_ = msrec.isDefined( "WEATHER" ) ;
195 if ( isWeather_ )
196 if ( mstable_.weather().nrow() == 0 )
197 isWeather_ = False ;
198
199 // Access to MS subtables
200 MSField fieldtab = mstable_.field() ;
201 MSPolarization poltab = mstable_.polarization() ;
202 MSDataDescription ddtab = mstable_.dataDescription() ;
203 MSObservation obstab = mstable_.observation() ;
204 MSSource srctab = mstable_.source() ;
205 MSSpectralWindow spwtab = mstable_.spectralWindow() ;
206 MSSysCal caltab = mstable_.sysCal() ;
207 if ( caltab.nrow() == 0 )
208 isSysCal_ = False ;
209 else {
210 if ( !caltab.tableDesc().isColumn( colTcal_ ) ) {
211 colTcal_ = "TCAL" ;
212 if ( !caltab.tableDesc().isColumn( colTcal_ ) )
213 colTcal_ = "NONE" ;
214 }
215 if ( !caltab.tableDesc().isColumn( colTsys_ ) ) {
216 colTsys_ = "TSYS" ;
217 if ( !caltab.tableDesc().isColumn( colTcal_ ) )
218 colTsys_ = "NONE" ;
219 }
220 }
221// colTcal_ = "TCAL" ;
222// colTsys_ = "TSYS" ;
223 MSPointing pointtab = mstable_.pointing() ;
224 if ( mstable_.weather().nrow() == 0 )
225 isWeather_ = False ;
226 MSState stattab = mstable_.state() ;
227 MSAntenna anttab = mstable_.antenna() ;
228
229 // TEST
230 // memory allocation by boost::object_pool
231 boost::object_pool<ROTableColumn> *tpoolr = new boost::object_pool<ROTableColumn> ;
232 //
233
234 // SUBTABLES: FREQUENCIES
235 //string freqFrame = getFrame() ;
236 string freqFrame = "LSRK" ;
237 table_->frequencies().setFrame( freqFrame ) ;
238 table_->frequencies().setFrame( freqFrame, True ) ;
239
240 // SUBTABLES: WEATHER
241 fillWeather() ;
242
243 // SUBTABLES: FOCUS
244 fillFocus() ;
245
246 // SUBTABLES: TCAL
247 fillTcal( tpoolr ) ;
248
249 // SUBTABLES: FIT
250 //fillFit() ;
251
252 // SUBTABLES: HISTORY
253 //fillHistory() ;
254
255 // MAIN
256 // Iterate over several ids
257 map<Int, uInt> ifmap ; // (IFNO, FREQ_ID) pair
258 ROArrayQuantColumn<Double> *sharedQDArrCol = new ROArrayQuantColumn<Double>( anttab, "POSITION" ) ;
259 Vector< Quantum<Double> > antpos = (*sharedQDArrCol)( antenna_ ) ;
260 delete sharedQDArrCol ;
261 MPosition mp( MVPosition( antpos ), MPosition::ITRF ) ;
262 if ( getPt_ ) {
263 pointtab = MSPointing( pointtab( pointtab.col("ANTENNA_ID")==antenna_ ).sort("TIME") ) ;
264 }
265 String stationName = asString( "STATION", antenna_, anttab, tpoolr ) ;
266 String antennaName = asString( "NAME", antenna_, anttab, tpoolr ) ;
267 sdh.antennaposition.resize( 3 ) ;
268 for ( int i = 0 ; i < 3 ; i++ )
269 sdh.antennaposition[i] = antpos[i].getValue( "m" ) ;
270 String telescopeName = "" ;
271
272 //double time1 = gettimeofday_sec() ;
273 //os_ << "end fill init: " << time1 << " (" << time1-time0 << "sec)" << LogIO::POST ;
274
275 // row based
276 Table &stab = table_->table() ;
277 TableRow row( stab ) ;
278 TableRecord &trec = row.record() ;
279 RecordFieldPtr< Array<Float> > spRF( trec, "SPECTRA" ) ;
280 RecordFieldPtr< Array<uChar> > ucarrRF( trec, "FLAGTRA" ) ;
281 RecordFieldPtr<Double> timeRF( trec, "TIME" ) ;
282 RecordFieldPtr< Array<Float> > tsysRF( trec, "TSYS" ) ;
283 RecordFieldPtr<Double> intervalRF( trec, "INTERVAL" ) ;
284 RecordFieldPtr< Array<Double> > dirRF( trec, "DIRECTION" ) ;
285 RecordFieldPtr<Float> azRF( trec, "AZIMUTH" ) ;
286 RecordFieldPtr<Float> elRF( trec, "ELEVATION" ) ;
287 RecordFieldPtr< Array<Double> > scrRF( trec, "SCANRATE" ) ;
288 RecordFieldPtr<uInt> cycleRF( trec, "CYCLENO" ) ;
289 RecordFieldPtr<uInt> flrRF( trec, "FLAGROW" ) ;
290 RecordFieldPtr<uInt> tcalidRF( trec, "TCAL_ID" ) ;
291 RecordFieldPtr<uInt> widRF( trec, "WEATHER_ID" ) ;
292 RecordFieldPtr<uInt> polnoRF( trec, "POLNO" ) ;
293
294
295 // REFBEAMNO
296 RecordFieldPtr<Int> intRF( trec, "REFBEAMNO" ) ;
297 *intRF = -1 ;
298
299 // FIT_ID
300 intRF.attachToRecord( trec, "FIT_ID" ) ;
301 *intRF = -1 ;
302
303 // OPACITY
304 RecordFieldPtr<Float> floatRF( trec, "OPACITY" ) ;
305 *floatRF = 0.0 ;
306
307 //
308 // ITERATION: OBSERVATION_ID
309 //
310 TableIterator iter0( mstable_, "OBSERVATION_ID" ) ;
311 while( !iter0.pastEnd() ) {
312 //time0 = gettimeofday_sec() ;
313 //os_ << "start 0th iteration: " << time0 << LogIO::POST ;
314 Table t0 = iter0.table() ;
315 Int obsId = asInt( "OBSERVATION_ID", 0, t0, tpoolr ) ;
316 if ( sdh.observer == "" ) {
317 sdh.observer = asString( "OBSERVER", obsId, obstab, tpoolr ) ;
318 }
319 if ( sdh.project == "" ) {
320 sdh.project = asString( "PROJECT", obsId, obstab, tpoolr ) ;
321 }
322 ROArrayMeasColumn<MEpoch> *tmpMeasCol = new ROArrayMeasColumn<MEpoch>( obstab, "TIME_RANGE" ) ;
323 MEpoch me = (*tmpMeasCol)( obsId )( IPosition(1,0) ) ;
324 delete tmpMeasCol ;
325 if ( sdh.utc == 0.0 ) {
326 sdh.utc = me.get( "d" ).getValue() ;
327 }
328 if ( telescopeName == "" ) {
329 telescopeName = asString( "TELESCOPE_NAME", obsId, obstab, tpoolr ) ;
330 }
331 Int nbeam = 0 ;
332 //time1 = gettimeofday_sec() ;
333 //os_ << "end 0th iteration init: " << time1 << " (" << time1-time0 << "sec)" << LogIO::POST ;
334 //
335 // ITERATION: FEED1
336 //
337 TableIterator iter1( t0, "FEED1" ) ;
338 while( !iter1.pastEnd() ) {
339 //time0 = gettimeofday_sec() ;
340 //os_ << "start 1st iteration: " << time0 << LogIO::POST ;
341 Table t1 = iter1.table() ;
342 // assume FEED1 == FEED2
343 Int feedId = asInt( "FEED1", 0, t1, tpoolr ) ;
344 nbeam++ ;
345
346 // BEAMNO
347 RecordFieldPtr<uInt> uintRF( trec, "BEAMNO" ) ;
348 *uintRF = feedId ;
349
350 // FOCUS_ID
351 uintRF.attachToRecord( trec, "FOCUS_ID" ) ;
352 *uintRF = 0 ;
353
354 //time1 = gettimeofday_sec() ;
355 //os_ << "end 1st iteration init: " << time1 << " (" << time1-time0 << "sec)" << LogIO::POST ;
356 //
357 // ITERATION: FIELD_ID
358 //
359 TableIterator iter2( t1, "FIELD_ID" ) ;
360 while( !iter2.pastEnd() ) {
361 //time0 = gettimeofday_sec() ;
362 //os_ << "start 2nd iteration: " << time0 << LogIO::POST ;
363 Table t2 = iter2.table() ;
364 Int fieldId = asInt( "FIELD_ID", 0, t2, tpoolr ) ;
365 Int srcId = asInt( "SOURCE_ID", fieldId, fieldtab, tpoolr ) ;
366 String fieldName = asString( "NAME", fieldId, fieldtab, tpoolr ) ;
367 fieldName += "__" + String::toString(fieldId) ;
368 ROArrayMeasColumn<MDirection> *delayDirCol = new ROArrayMeasColumn<MDirection>( fieldtab, "DELAY_DIR" ) ;
369 Vector<MDirection> delayDir = (*delayDirCol)( fieldId ) ;
370 delete delayDirCol ;
371
372 // FIELDNAME
373 RecordFieldPtr<String> strRF( trec, "FIELDNAME" ) ;
374 *strRF = fieldName ;
375
376
377 //time1 = gettimeofday_sec() ;
378 //os_ << "end 2nd iteration init: " << time1 << " (" << time1-time0 << "sec)" << LogIO::POST ;
379 //
380 // ITERATION: DATA_DESC_ID
381 //
382 TableIterator iter3( t2, "DATA_DESC_ID" ) ;
383 while( !iter3.pastEnd() ) {
384 //time0 = gettimeofday_sec() ;
385 //os_ << "start 3rd iteration: " << time0 << LogIO::POST ;
386 Table t3 = iter3.table() ;
387 Int ddId = asInt( "DATA_DESC_ID", 0, t3, tpoolr ) ;
388 Int polId = asInt( "POLARIZATION_ID", ddId, ddtab, tpoolr ) ;
389 Int spwId = asInt( "SPECTRAL_WINDOW_ID", ddId, ddtab, tpoolr ) ;
390
391 // IFNO
392 uintRF.attachToRecord( trec, "IFNO" ) ;
393 *uintRF = (uInt)spwId ;
394
395 // polarization information
396 Int npol = asInt( "NUM_CORR", polId, poltab, tpoolr ) ;
397 ROArrayColumn<Int> *roArrICol = new ROArrayColumn<Int>( poltab, "CORR_TYPE" ) ;
398 Vector<Int> corrtype = (*roArrICol)( polId ) ;
399 delete roArrICol ;
400 String srcName( "" ) ;
401 Vector<Double> srcPM( 2, 0.0 ) ;
402 Vector<Double> srcDir( 2, 0.0 ) ;
403 MDirection md ;
404 Vector<Double> restFreqs ;
405 Vector<String> transitionName ;
406 Vector<Double> sysVels ;
407// os_ << "npol = " << npol << LogIO::POST ;
408// os_ << "corrtype = " << corrtype << LogIO::POST ;
409
410 // source and molecular transition
411 sourceInfo( srcId, spwId, srcName, md, srcPM, restFreqs, transitionName, sysVels, tpoolr ) ;
412// os_ << "srcId = " << srcId << ", spwId = " << spwId << LogIO::POST ;
413
414 // SRCNAME
415 strRF.attachToRecord( trec, "SRCNAME" ) ;
416 *strRF = srcName ;
417
418// os_ << "srcName = " << srcName << LogIO::POST ;
419
420 // SRCPROPERMOTION
421 RecordFieldPtr< Array<Double> > darrRF( trec, "SRCPROPERMOTION" ) ;
422 *darrRF = srcPM ;
423
424 //os_ << "srcPM = " << srcPM << LogIO::POST ;
425
426 // SRCDIRECTION
427 darrRF.attachToRecord( trec, "SRCDIRECTION" ) ;
428// *darrRF = srcDir ;
429 *darrRF = md.getAngle().getValue( "rad" ) ;
430
431 //os_ << "srcDir = " << srcDir << LogIO::POST ;
432
433 // SRCVELOCITY
434 Double sysVel = 0.0 ;
435 if ( !sysVels.empty() )
436 sysVel = sysVels[0] ;
437 RecordFieldPtr<Double> doubleRF( trec, "SRCVELOCITY" ) ;
438 *doubleRF = sysVel ;
439
440// os_ << "sysVel = " << sysVel << LogIO::POST ;
441
442 uInt molId = table_->molecules().addEntry( restFreqs, transitionName, transitionName ) ;
443
444 // MOLECULE_ID
445 uintRF.attachToRecord( trec, "MOLECULE_ID" ) ;
446 *uintRF = molId ;
447
448 // spectral setup
449 uInt freqId ;
450 Int nchan = asInt( "NUM_CHAN", spwId, spwtab, tpoolr ) ;
451 Bool iswvr = False ;
452 if ( nchan == 4 ) iswvr = True ;
453 sdh.nchan = max( sdh.nchan, nchan ) ;
454 map<Int,uInt>::iterator iter = ifmap.find( spwId ) ;
455 if ( iter == ifmap.end() ) {
456 ROScalarMeasColumn<MEpoch> *tmpMeasCol = new ROScalarMeasColumn<MEpoch>( t3, "TIME" ) ;
457 me = (*tmpMeasCol)( 0 ) ;
458 delete tmpMeasCol ;
459 Double refpix ;
460 Double refval ;
461 Double increment ;
462 spectralSetup( spwId,
463 me,
464 mp,
465 md,
466 refpix,
467 refval,
468 increment,
469 nchan,
470 sdh.freqref,
471 sdh.reffreq,
472 sdh.bandwidth,
473 tpoolr ) ;
474 freqId = table_->frequencies().addEntry( refpix, refval, increment ) ;
475 ifmap.insert( pair<Int, uInt>(spwId,freqId) ) ;
476 //os_ << "added to ifmap: (" << spwId << "," << freqId << ")" << LogIO::POST ;
477 }
478 else {
479 freqId = iter->second ;
480 }
481
482 // FREQ_ID
483 uintRF.attachToRecord( trec, "FREQ_ID" ) ;
484 *uintRF = freqId ;
485
486 // for TSYS and TCAL
487 Vector<MEpoch> scTime ;
488 Vector<Double> scInterval ;
489 ROArrayColumn<Float> scTsysCol ;
490 MSSysCal caltabsel ;
491 if ( isSysCal_ ) {
492 caltabsel = caltab( caltab.col("ANTENNA_ID") == antenna_ && caltab.col("FEED_ID") == feedId && caltab.col("SPECTRAL_WINDOW_ID") == spwId ).sort("TIME") ;
493 ROScalarMeasColumn<MEpoch> scTimeCol( caltabsel, "TIME" ) ;
494 scTime.resize( caltabsel.nrow() ) ;
495 for ( uInt irow = 0 ; irow < caltabsel.nrow() ; irow++ )
496 scTime[irow] = scTimeCol( irow ) ;
497 ROScalarColumn<Double> *scIntervalCol = new ROScalarColumn<Double>( caltabsel, "INTERVAL" ) ;
498 scInterval = scIntervalCol->getColumn() ;
499 delete scIntervalCol ;
500 if ( colTsys_ != "NONE" )
501 scTsysCol.attach( caltabsel, colTsys_ ) ;
502 }
503
504 sdh.npol = max( sdh.npol, npol ) ;
505 if ( !iswvr && sdh.poltype == "" ) sdh.poltype = getPolType( corrtype[0] ) ;
506
507 //time1 = gettimeofday_sec() ;
508 //os_ << "end 3rd iteration init: " << time1 << " (" << time1-time0 << "sec)" << LogIO::POST ;
509 //
510 // ITERATION: SCAN_NUMBER
511 //
512 TableIterator iter4( t3, "SCAN_NUMBER" ) ;
513 while( !iter4.pastEnd() ) {
514 //time0 = gettimeofday_sec() ;
515 //os_ << "start 4th iteration: " << time0 << LogIO::POST ;
516 Table t4 = iter4.table() ;
517 Int scanNum = asInt( "SCAN_NUMBER", 0, t4, tpoolr ) ;
518
519 // SCANNO
520 uintRF.attachToRecord( trec, "SCANNO" ) ;
521 *uintRF = scanNum - 1 ;
522
523 //time1 = gettimeofday_sec() ;
524 //os_ << "end 4th iteration init: " << time1 << " (" << time1-time0 << "sec)" << LogIO::POST ;
525 //
526 // ITERATION: STATE_ID
527 //
528 TableIterator iter5( t4, "STATE_ID" ) ;
529 while( !iter5.pastEnd() ) {
530 //time0 = gettimeofday_sec() ;
531 //os_ << "start 5th iteration: " << time0 << LogIO::POST ;
532 Table t5 = iter5.table() ;
533 Int stateId = asInt( "STATE_ID", 0, t5, tpoolr ) ;
534 String obstype = asString( "OBS_MODE", 0, stattab, tpoolr ) ;
535 if ( sdh.obstype == "" ) sdh.obstype = obstype ;
536
537 Int nrow = t5.nrow() ;
538 //time1 = gettimeofday_sec() ;
539 //os_ << "end 5th iteration init: " << time1 << " (" << time1-time0 << "sec)" << LogIO::POST ;
540
541 uInt cycle = 0 ;
542
543 Cube<Float> spArr ;
544 Cube<Bool> flArr ;
545 reshapeSpectraAndFlagtra( spArr,
546 flArr,
547 t5,
548 npol,
549 nchan,
550 nrow,
551 corrtype ) ;
552 if ( sdh.fluxunit == "" ) {
553 String colName = "FLOAT_DATA" ;
554 if ( isData_ ) colName = "DATA" ;
555 ROTableColumn dataCol( t5, colName ) ;
556 const TableRecord &dataColKeys = dataCol.keywordSet() ;
557 if ( dataColKeys.isDefined( "UNIT" ) )
558 sdh.fluxunit = dataColKeys.asString( "UNIT" ) ;
559 else if ( dataColKeys.isDefined( "QuantumUnits" ) )
560 sdh.fluxunit = dataColKeys.asString( "QuantumUnits" ) ;
561 }
562
563 ROScalarMeasColumn<MEpoch> *mTimeCol = new ROScalarMeasColumn<MEpoch>( t5, "TIME" ) ;
564 Block<MEpoch> mTimeB( nrow ) ;
565 for ( Int irow = 0 ; irow < nrow ; irow++ )
566 mTimeB[irow] = (*mTimeCol)( irow ) ;
567// ROTableColumn *mIntervalCol = tpoolr->construct( t5, "INTERVAL" ) ;
568// ROTableColumn *mFlagRowCol = tpoolr->construct( t5, "FLAG_ROW" ) ;
569 Block<Int> sysCalIdx( nrow, -1 ) ;
570 if ( isSysCal_ ) {
571 getSysCalTime( scTime, scInterval, mTimeB, sysCalIdx ) ;
572 }
573 delete mTimeCol ;
574 Matrix<Float> defaulttsys( npol, 1, 1.0 ) ;
575 Int srcType = getSrcType( stateId, tpoolr ) ;
576 uInt diridx = 0 ;
577 uInt wid = 0 ;
578 Int pidx = 0 ;
579 Bool crossOK = False ;
580 Block<uInt> polnos( npol, 99 ) ;
581 for ( Int ipol = 0 ; ipol < npol ; ipol++ ) {
582 Block<uInt> p = getPolNo( corrtype[ipol] ) ;
583 if ( p.size() > 1 ) {
584 if ( crossOK ) continue ;
585 else {
586 polnos[pidx] = p[0] ;
587 pidx++ ;
588 polnos[pidx] = p[1] ;
589 pidx++ ;
590 crossOK = True ;
591 }
592 }
593 else {
594 polnos[pidx] = p[0] ;
595 pidx++ ;
596 }
597 }
598
599 // SRCTYPE
600 intRF.attachToRecord( trec, "SRCTYPE" ) ;
601 *intRF = srcType ;
602
603 for ( Int irow = 0 ; irow < nrow ; irow++ ) {
604 // CYCLENO
605 *cycleRF = cycle ;
606
607 // FLAGROW
608// *flrRF = (uInt)mFlagRowCol->asBool( irow ) ;
609 *flrRF = (uInt)asBool( "FLAG_ROW", irow, t5, tpoolr ) ;
610
611 // SPECTRA, FLAG
612 Matrix<Float> sp = spArr.xyPlane( irow ) ;
613 Matrix<Bool> flb = flArr.xyPlane( irow ) ;
614 Matrix<uChar> fl( flb.shape() ) ;
615 convertArray( fl, flb ) ;
616
617 // TIME
618 *timeRF = mTimeB[irow].get("d").getValue() ;
619
620 // INTERVAL
621// *intervalRF = (Double)(mIntervalCol->asdouble( irow )) ;
622 *intervalRF = asDouble( "INTERVAL", irow, t5, tpoolr ) ;
623
624 // TSYS
625 Matrix<Float> tsys ;
626 if ( sysCalIdx[irow] != -1 && colTsys_ != "NONE" )
627 tsys = scTsysCol( sysCalIdx[irow] ) ;
628 else
629 tsys = defaulttsys ;
630
631 // TCAL_ID
632 Block<uInt> tcalids( npol, 0 ) ;
633 if ( sysCalIdx[irow] != -1 && colTcal_ != "NONE" ) {
634 tcalids = getTcalId( feedId, spwId, scTime[sysCalIdx[irow]] ) ;
635 }
636
637 // WEATHER_ID
638 if ( isWeather_ ) {
639 wid = getWeatherId( wid, mTimeB[irow].get("s").getValue() ) ;
640 *widRF = mwIndex_[wid] ;
641 }
642 else {
643 *widRF = wid ;
644 }
645
646
647 // DIRECTION, AZEL, SCANRATE
648 Vector<Double> dir ;
649 Vector<Double> azel ;
650 Vector<Double> scanrate ;
651 String refString ;
652 if ( getPt_ )
653 diridx = getDirection( diridx, dir, azel, scanrate, pointtab, mTimeB[irow], mp ) ;
654 else
655 getSourceDirection( dir, azel, scanrate, mTimeB[irow], mp, delayDir ) ;
656 *dirRF = dir ;
657 *azRF = azel[0] ;
658 *elRF = azel[1] ;
659 *scrRF = scanrate ;
660
661 // Polarization dependent things
662 for ( Int ipol = 0 ; ipol < npol ; ipol++ ) {
663 // POLNO
664 *polnoRF = polnos[ipol] ;
665
666 spRF.define( sp.row( ipol ) ) ;
667 ucarrRF.define( fl.row( ipol ) ) ;
668 tsysRF.define( tsys.row( ipol ) ) ;
669 *tcalidRF = tcalids[ipol] ;
670
671 // Commit row
672 stab.addRow() ;
673 row.put( stab.nrow()-1 ) ;
674 }
675
676 cycle++ ;
677 }
678
679 //time1 = gettimeofday_sec() ;
680 //os_ << "end 5th iteration: " << time1 << " (" << time1-time0 << "sec)" << LogIO::POST ;
681
682 iter5.next() ;
683 }
684 iter4.next() ;
685 }
686 iter3.next() ;
687 }
688 iter2.next() ;
689 }
690 iter1.next() ;
691 }
692 if ( sdh.nbeam < nbeam ) sdh.nbeam = nbeam ;
693
694 iter0.next() ;
695 }
696
697
698 delete tpoolr ;
699
700
701 // Table Keywords
702 sdh.nif = ifmap.size() ;
703 if ( ( telescopeName == "" ) || ( antennaName == telescopeName ) ) {
704 sdh.antennaname = antennaName ;
705 }
706 else {
707 sdh.antennaname = telescopeName + "//" + antennaName ;
708 }
709 if ( stationName != "" && stationName != antennaName ) {
710 sdh.antennaname += "@" + stationName ;
711 }
712 ROArrayColumn<Double> pdirCol( pointtab, "DIRECTION" ) ;
713 String dirref = pdirCol.keywordSet().asRecord("MEASINFO").asString("Ref") ;
714 if ( dirref == "AZELGEO" || dirref == "AZEL" ) {
715 dirref = "J2000" ;
716 }
717 sscanf( dirref.chars()+1, "%f", &sdh.equinox ) ;
718 sdh.epoch = "UTC" ;
719 if (sdh.freqref == "TOPO") {
720 sdh.freqref = "TOPOCENT";
721 } else if (sdh.freqref == "GEO") {
722 sdh.freqref = "GEOCENTR";
723 } else if (sdh.freqref == "BARY") {
724 sdh.freqref = "BARYCENT";
725 } else if (sdh.freqref == "GALACTO") {
726 sdh.freqref = "GALACTOC";
727 } else if (sdh.freqref == "LGROUP") {
728 sdh.freqref = "LOCALGRP";
729 } else if (sdh.freqref == "CMB") {
730 sdh.freqref = "CMBDIPOL";
731 } else if (sdh.freqref == "REST") {
732 sdh.freqref = "SOURCE";
733 }
734
735 if ( sdh.fluxunit == "" || sdh.fluxunit == "CNTS" )
736 sdh.fluxunit = "K" ;
737 table_->setHeader( sdh ) ;
738
739 // save path to POINTING table
740 // 2011/07/06 TN
741 // Path to POINTING table in original MS will not be written
742 // if getPt_ is True
743 Path datapath( tablename_ ) ;
744 if ( !getPt_ ) {
745 String pTabName = datapath.absoluteName() + "/POINTING" ;
746 stab.rwKeywordSet().define( "POINTING", pTabName ) ;
747 }
748
749 // for GBT
750 if ( antennaName.contains( "GBT" ) ) {
751 String goTabName = datapath.absoluteName() + "/GBT_GO" ;
752 stab.rwKeywordSet().define( "GBT_GO", goTabName ) ;
753 }
754
755 // for MS created from ASDM
756 //mstable_.keywordSet().print(cout) ;
757 const TableRecord &msKeys = mstable_.keywordSet() ;
758 uInt nfields = msKeys.nfields() ;
759 for ( uInt ifield = 0 ; ifield < nfields ; ifield++ ) {
760 String name = msKeys.name( ifield ) ;
761 //os_ << "name = " << name << LogIO::POST ;
762 if ( name.find( "ASDM" ) != String::npos ) {
763 String asdmpath = msKeys.asTable( ifield ).tableName() ;
764 os_ << "ASDM table: " << asdmpath << LogIO::POST ;
765 stab.rwKeywordSet().define( name, asdmpath ) ;
766 }
767 }
768
769 //double endSec = gettimeofday_sec() ;
770 //os_ << "end MSFiller::fill() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
771}
772
773void MSFiller::close()
774{
775 //tablesel_.closeSubTables() ;
776 mstable_.closeSubTables() ;
777 //tablesel_.unlock() ;
778 mstable_.unlock() ;
779}
780
781Int MSFiller::getSrcType( Int stateId, boost::object_pool<ROTableColumn> *tpool )
782{
783 //double startSec = gettimeofday_sec() ;
784 //os_ << "start MSFiller::getSrcType() startSec=" << startSec << LogIO::POST ;
785
786 MSState statetab = mstable_.state() ;
787 String obsMode = asString( "OBS_MODE", stateId, statetab, tpool ) ;
788 Bool sig = asBool( "SIG", stateId, statetab, tpool ) ;
789 Bool ref = asBool( "REF", stateId, statetab, tpool ) ;
790 Double cal = asDouble( "CAL", stateId, statetab, tpool ) ;
791 //os_ << "OBS_MODE = " << obsMode << LogIO::POST ;
792
793 // determine separator
794 String sep = "" ;
795 String tmpStr = obsMode.substr( 0, obsMode.find_first_of( "," ) ) ;
796 //os_ << "tmpStr = " << tmpStr << LogIO::POST ;
797 //if ( obsMode.find( ":" ) != String::npos ) {
798 if ( tmpStr.find( ":" ) != String::npos ) {
799 sep = ":" ;
800 }
801 //else if ( obsMode.find( "." ) != String::npos ) {
802 else if ( tmpStr.find( "." ) != String::npos ) {
803 sep = "." ;
804 }
805 else if ( tmpStr.find( "#" ) != String::npos ) {
806 sep = "#" ;
807 }
808 //else if ( obsMode.find( "_" ) != String::npos ) {
809 else if ( tmpStr.find( "_" ) != String::npos ) {
810 sep = "_" ;
811 }
812 //os_ << "separator = " << sep << LogIO::POST ;
813
814 // determine SRCTYPE
815 Int srcType = SrcType::NOTYPE ;
816 if ( sep == ":" ) {
817 // sep == ":"
818 //
819 // GBT case
820 //
821 // obsMode1=Nod
822 // NOD
823 // obsMode1=OffOn
824 // obsMode2=PSWITCHON: PSON
825 // obsMode2=PSWITCHOFF: PSOFF
826 // obsMode1=??
827 // obsMode2=FSWITCH:
828 // SIG=1: FSON
829 // REF=1: FSOFF
830 // Calibration scan if CAL != 0
831 Int epos = obsMode.find_first_of( sep ) ;
832 Int nextpos = obsMode.find_first_of( sep, epos+1 ) ;
833 String obsMode1 = obsMode.substr( 0, epos ) ;
834 String obsMode2 = obsMode.substr( epos+1, nextpos-epos-1 ) ;
835 if ( obsMode1 == "Nod" ) {
836 srcType = SrcType::NOD ;
837 }
838 else if ( obsMode1 == "OffOn" ) {
839 if ( obsMode2 == "PSWITCHON" ) srcType = SrcType::PSON ;
840 if ( obsMode2 == "PSWITCHOFF" ) srcType = SrcType::PSOFF ;
841 }
842 else {
843 if ( obsMode2 == "FSWITCH" ) {
844 if ( sig ) srcType = SrcType::FSON ;
845 if ( ref ) srcType = SrcType::FSOFF ;
846 }
847 }
848 if ( cal > 0.0 ) {
849 if ( srcType == SrcType::NOD )
850 srcType = SrcType::NODCAL ;
851 else if ( srcType == SrcType::PSON )
852 srcType = SrcType::PONCAL ;
853 else if ( srcType == SrcType::PSOFF )
854 srcType = SrcType::POFFCAL ;
855 else if ( srcType == SrcType::FSON )
856 srcType = SrcType::FONCAL ;
857 else if ( srcType == SrcType::FSOFF )
858 srcType = SrcType::FOFFCAL ;
859 else
860 srcType = SrcType::CAL ;
861 }
862 }
863 else if ( sep == "." || sep == "#" ) {
864 // sep == "." or "#"
865 //
866 // ALMA & EVLA case (MS via ASDM) before3.1
867 //
868 // obsMode1=CALIBRATE_*
869 // obsMode2=ON_SOURCE: PONCAL
870 // obsMode2=OFF_SOURCE: POFFCAL
871 // obsMode1=OBSERVE_TARGET
872 // obsMode2=ON_SOURCE: PON
873 // obsMode2=OFF_SOURCE: POFF
874 string substr[2] ;
875 int numSubstr = split( obsMode, substr, 2, "," ) ;
876 //os_ << "numSubstr = " << numSubstr << LogIO::POST ;
877 //for ( int i = 0 ; i < numSubstr ; i++ )
878 //os_ << "substr[" << i << "] = " << substr[i] << LogIO::POST ;
879 String obsType( substr[0] ) ;
880 //os_ << "obsType = " << obsType << LogIO::POST ;
881 Int epos = obsType.find_first_of( sep ) ;
882 Int nextpos = obsType.find_first_of( sep, epos+1 ) ;
883 String obsMode1 = obsType.substr( 0, epos ) ;
884 String obsMode2 = obsType.substr( epos+1, nextpos-epos-1 ) ;
885 //os_ << "obsMode1 = " << obsMode1 << LogIO::POST ;
886 //os_ << "obsMode2 = " << obsMode2 << LogIO::POST ;
887 if ( obsMode1.find( "CALIBRATE_" ) == 0 ) {
888 if ( obsMode2 == "ON_SOURCE" ) srcType = SrcType::PONCAL ;
889 if ( obsMode2 == "OFF_SOURCE" ) srcType = SrcType::POFFCAL ;
890 }
891 else if ( obsMode1 == "OBSERVE_TARGET" ) {
892 if ( obsMode2 == "ON_SOURCE" ) srcType = SrcType::PSON ;
893 if ( obsMode2 == "OFF_SOURCE" ) srcType = SrcType::PSOFF ;
894 }
895 }
896 else if ( sep == "_" ) {
897 // sep == "_"
898 //
899 // ALMA & EVLA case (MS via ASDM) after 3.2
900 //
901 // obsMode1=CALIBRATE_*
902 // obsMode2=ON_SOURCE: PONCAL
903 // obsMode2=OFF_SOURCE: POFFCAL
904 // obsMode1=OBSERVE_TARGET
905 // obsMode2=ON_SOURCE: PON
906 // obsMode2=OFF_SOURCE: POFF
907 string substr[2] ;
908 int numSubstr = split( obsMode, substr, 2, "," ) ;
909 //os_ << "numSubstr = " << numSubstr << LogIO::POST ;
910 //for ( int i = 0 ; i < numSubstr ; i++ )
911 //os_ << "substr[" << i << "] = " << substr[i] << LogIO::POST ;
912 String obsType( substr[0] ) ;
913 //os_ << "obsType = " << obsType << LogIO::POST ;
914 string substr2[4] ;
915 int numSubstr2 = split( obsType, substr2, 4, sep ) ;
916 //Int epos = obsType.find_first_of( sep ) ;
917 //Int nextpos = obsType.find_first_of( sep, epos+1 ) ;
918 //String obsMode1 = obsType.substr( 0, epos ) ;
919 //String obsMode2 = obsType.substr( epos+1, nextpos-epos-1 ) ;
920 String obsMode1( substr2[0] ) ;
921 String obsMode2( substr2[2] ) ;
922 //os_ << "obsMode1 = " << obsMode1 << LogIO::POST ;
923 //os_ << "obsMode2 = " << obsMode2 << LogIO::POST ;
924 if ( obsMode1.find( "CALIBRATE" ) == 0 ) {
925 if ( obsMode2 == "ON" ) srcType = SrcType::PONCAL ;
926 if ( obsMode2 == "OFF" ) srcType = SrcType::POFFCAL ;
927 }
928 else if ( obsMode1 == "OBSERVE" ) {
929 if ( obsMode2 == "ON" ) srcType = SrcType::PSON ;
930 if ( obsMode2 == "OFF" ) srcType = SrcType::PSOFF ;
931 }
932 }
933 else {
934 if ( sig ) srcType = SrcType::SIG ;
935 if ( ref ) srcType = SrcType::REF ;
936 }
937
938 //os_ << "srcType = " << srcType << LogIO::POST ;
939 //double endSec = gettimeofday_sec() ;
940 //os_ << "end MSFiller::getSrcType() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
941 return srcType ;
942}
943
944//Vector<uInt> MSFiller::getPolNo( Int corrType )
945Block<uInt> MSFiller::getPolNo( Int corrType )
946{
947 //double startSec = gettimeofday_sec() ;
948 //os_ << "start MSFiller::getPolNo() startSec=" << startSec << LogIO::POST ;
949 Block<uInt> polno( 1 ) ;
950
951 if ( corrType == Stokes::I || corrType == Stokes::RR || corrType == Stokes::XX ) {
952 polno = 0 ;
953 }
954 else if ( corrType == Stokes::Q || corrType == Stokes::LL || corrType == Stokes::YY ) {
955 polno = 1 ;
956 }
957 else if ( corrType == Stokes::U ) {
958 polno = 2 ;
959 }
960 else if ( corrType == Stokes::V ) {
961 polno = 3 ;
962 }
963 else if ( corrType == Stokes::RL || corrType == Stokes::XY || corrType == Stokes::LR || corrType == Stokes::RL ) {
964 polno.resize( 2 ) ;
965 polno[0] = 2 ;
966 polno[1] = 3 ;
967 }
968 else if ( corrType == Stokes::Plinear ) {
969 polno[0] = 1 ;
970 }
971 else if ( corrType == Stokes::Pangle ) {
972 polno[0] = 2 ;
973 }
974 else {
975 polno = 99 ;
976 }
977 //os_ << "polno = " << polno << LogIO::POST ;
978 //double endSec = gettimeofday_sec() ;
979 //os_ << "end MSFiller::getPolNo() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
980
981 return polno ;
982}
983
984String MSFiller::getPolType( Int corrType )
985{
986 //double startSec = gettimeofday_sec() ;
987 //os_ << "start MSFiller::getPolType() startSec=" << startSec << LogIO::POST ;
988 String poltype = "" ;
989
990 if ( corrType == Stokes::I || corrType == Stokes::Q || corrType == Stokes::U || corrType == Stokes::V )
991 poltype = "stokes" ;
992 else if ( corrType == Stokes::XX || corrType == Stokes::YY || corrType == Stokes::XY || corrType == Stokes::YX )
993 poltype = "linear" ;
994 else if ( corrType == Stokes::RR || corrType == Stokes::LL || corrType == Stokes::RL || corrType == Stokes::LR )
995 poltype = "circular" ;
996 else if ( corrType == Stokes::Plinear || corrType == Stokes::Pangle )
997 poltype = "linpol" ;
998
999 //double endSec = gettimeofday_sec() ;
1000 //os_ << "end MSFiller::getPolType() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
1001 return poltype ;
1002}
1003
1004void MSFiller::fillWeather()
1005{
1006 //double startSec = gettimeofday_sec() ;
1007 //os_ << "start MSFiller::fillWeather() startSec=" << startSec << LogIO::POST ;
1008
1009 if ( !isWeather_ ) {
1010 // add dummy row
1011 table_->weather().table().addRow(1,True) ;
1012 return ;
1013 }
1014
1015 Table mWeather = mstable_.weather() ;
1016 //Table mWeatherSel = mWeather( mWeather.col("ANTENNA_ID") == antenna_ ).sort("TIME") ;
1017 Table mWeatherSel( mWeather( mWeather.col("ANTENNA_ID") == antenna_ ).sort("TIME") ) ;
1018 //os_ << "mWeatherSel.nrow() = " << mWeatherSel.nrow() << LogIO::POST ;
1019 if ( mWeatherSel.nrow() == 0 ) {
1020 os_ << "No rows with ANTENNA_ID = " << antenna_ << " in WEATHER table, Try -1..." << LogIO::POST ;
1021 mWeatherSel = Table( MSWeather( mWeather( mWeather.col("ANTENNA_ID") == -1 ) ) ) ;
1022 if ( mWeatherSel.nrow() == 0 ) {
1023 os_ << "No rows in WEATHER table" << LogIO::POST ;
1024 }
1025 }
1026 uInt wnrow = mWeatherSel.nrow() ;
1027 //os_ << "wnrow = " << wnrow << LogIO::POST ;
1028
1029 if ( wnrow == 0 )
1030 return ;
1031
1032 Table wtab = table_->weather().table() ;
1033 wtab.addRow( wnrow ) ;
1034
1035 Bool stationInfoExists = mWeatherSel.tableDesc().isColumn( "NS_WX_STATION_ID" ) ;
1036 Int stationId = -1 ;
1037 if ( stationInfoExists ) {
1038 // determine which station is closer
1039 ROScalarColumn<Int> stationCol( mWeatherSel, "NS_WX_STATION_ID" ) ;
1040 ROArrayColumn<Double> stationPosCol( mWeatherSel, "NS_WX_STATION_POSITION" ) ;
1041 Vector<Int> stationIds = stationCol.getColumn() ;
1042 Vector<Int> stationIdList( 0 ) ;
1043 Matrix<Double> stationPosList( 0, 3, 0.0 ) ;
1044 uInt numStation = 0 ;
1045 for ( uInt i = 0 ; i < stationIds.size() ; i++ ) {
1046 if ( !anyEQ( stationIdList, stationIds[i] ) ) {
1047 numStation++ ;
1048 stationIdList.resize( numStation, True ) ;
1049 stationIdList[numStation-1] = stationIds[i] ;
1050 stationPosList.resize( numStation, 3, True ) ;
1051 stationPosList.row( numStation-1 ) = stationPosCol( i ) ;
1052 }
1053 }
1054 //os_ << "staionIdList = " << stationIdList << endl ;
1055 Table mAntenna = mstable_.antenna() ;
1056 ROArrayColumn<Double> antposCol( mAntenna, "POSITION" ) ;
1057 Vector<Double> antpos = antposCol( antenna_ ) ;
1058 Double minDiff = -1.0 ;
1059 for ( uInt i = 0 ; i < stationIdList.size() ; i++ ) {
1060 Double diff = sum( square( antpos - stationPosList.row( i ) ) ) ;
1061 if ( minDiff < 0.0 || minDiff > diff ) {
1062 minDiff = diff ;
1063 stationId = stationIdList[i] ;
1064 }
1065 }
1066 }
1067 //os_ << "stationId = " << stationId << endl ;
1068
1069 ScalarColumn<Float> *fCol ;
1070 ROScalarColumn<Float> *sharedFloatCol ;
1071 if ( mWeatherSel.tableDesc().isColumn( "TEMPERATURE" ) ) {
1072 fCol = new ScalarColumn<Float>( wtab, "TEMPERATURE" ) ;
1073 sharedFloatCol = new ROScalarColumn<Float>( mWeatherSel, "TEMPERATURE" ) ;
1074 fCol->putColumn( *sharedFloatCol ) ;
1075 delete sharedFloatCol ;
1076 delete fCol ;
1077 }
1078 if ( mWeatherSel.tableDesc().isColumn( "PRESSURE" ) ) {
1079 fCol = new ScalarColumn<Float>( wtab, "PRESSURE" ) ;
1080 sharedFloatCol = new ROScalarColumn<Float>( mWeatherSel, "PRESSURE" ) ;
1081 fCol->putColumn( *sharedFloatCol ) ;
1082 delete sharedFloatCol ;
1083 delete fCol ;
1084 }
1085 if ( mWeatherSel.tableDesc().isColumn( "REL_HUMIDITY" ) ) {
1086 fCol = new ScalarColumn<Float>( wtab, "HUMIDITY" ) ;
1087 sharedFloatCol = new ROScalarColumn<Float>( mWeatherSel, "REL_HUMIDITY" ) ;
1088 fCol->putColumn( *sharedFloatCol ) ;
1089 delete sharedFloatCol ;
1090 delete fCol ;
1091 }
1092 if ( mWeatherSel.tableDesc().isColumn( "WIND_SPEED" ) ) {
1093 fCol = new ScalarColumn<Float>( wtab, "WINDSPEED" ) ;
1094 sharedFloatCol = new ROScalarColumn<Float>( mWeatherSel, "WIND_SPEED" ) ;
1095 fCol->putColumn( *sharedFloatCol ) ;
1096 delete sharedFloatCol ;
1097 delete fCol ;
1098 }
1099 if ( mWeatherSel.tableDesc().isColumn( "WIND_DIRECTION" ) ) {
1100 fCol = new ScalarColumn<Float>( wtab, "WINDAZ" ) ;
1101 sharedFloatCol = new ROScalarColumn<Float>( mWeatherSel, "WIND_DIRECTION" ) ;
1102 fCol->putColumn( *sharedFloatCol ) ;
1103 delete sharedFloatCol ;
1104 delete fCol ;
1105 }
1106 ScalarColumn<uInt> idCol( wtab, "ID" ) ;
1107 for ( uInt irow = 0 ; irow < wnrow ; irow++ )
1108 idCol.put( irow, irow ) ;
1109
1110 ROScalarQuantColumn<Double> tqCol( mWeatherSel, "TIME" ) ;
1111 ROScalarColumn<Double> tCol( mWeatherSel, "TIME" ) ;
1112 String tUnit = tqCol.getUnits() ;
1113 Vector<Double> mwTime = tCol.getColumn() ;
1114 if ( tUnit == "d" )
1115 mwTime *= 86400.0 ;
1116 tqCol.attach( mWeatherSel, "INTERVAL" ) ;
1117 tCol.attach( mWeatherSel, "INTERVAL" ) ;
1118 String iUnit = tqCol.getUnits() ;
1119 Vector<Double> mwInterval = tCol.getColumn() ;
1120 if ( iUnit == "d" )
1121 mwInterval *= 86400.0 ;
1122
1123 if ( stationId > 0 ) {
1124 ROScalarColumn<Int> stationCol( mWeatherSel, "NS_WX_STATION_ID" ) ;
1125 Vector<Int> stationVec = stationCol.getColumn() ;
1126 uInt wsnrow = ntrue( stationVec == stationId ) ;
1127 mwTime_.resize( wsnrow ) ;
1128 mwInterval_.resize( wsnrow ) ;
1129 mwIndex_.resize( wsnrow ) ;
1130 uInt wsidx = 0 ;
1131 for ( uInt irow = 0 ; irow < wnrow ; irow++ ) {
1132 if ( stationId == stationVec[irow] ) {
1133 mwTime_[wsidx] = mwTime[irow] ;
1134 mwInterval_[wsidx] = mwInterval[irow] ;
1135 mwIndex_[wsidx] = irow ;
1136 wsidx++ ;
1137 }
1138 }
1139 }
1140 else {
1141 mwTime_ = mwTime ;
1142 mwInterval_ = mwInterval ;
1143 mwIndex_.resize( mwTime_.size() ) ;
1144 indgen( mwIndex_ ) ;
1145 }
1146 //os_ << "mwTime[0] = " << mwTime_[0] << " mwInterval[0] = " << mwInterval_[0] << LogIO::POST ;
1147 //double endSec = gettimeofday_sec() ;
1148 //os_ << "end MSFiller::fillWeather() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
1149}
1150
1151void MSFiller::fillFocus()
1152{
1153 //double startSec = gettimeofday_sec() ;
1154 //os_ << "start MSFiller::fillFocus() startSec=" << startSec << LogIO::POST ;
1155 // tentative
1156 table_->focus().addEntry( 0.0, 0.0, 0.0, 0.0 ) ;
1157 //double endSec = gettimeofday_sec() ;
1158 //os_ << "end MSFiller::fillFocus() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
1159}
1160
1161void MSFiller::fillTcal( boost::object_pool<ROTableColumn> *tpoolr )
1162{
1163 //double startSec = gettimeofday_sec() ;
1164 //os_ << "start MSFiller::fillTcal() startSec=" << startSec << LogIO::POST ;
1165
1166 if ( !isSysCal_ ) {
1167 // add dummy row
1168 os_ << "No SYSCAL rows" << LogIO::POST ;
1169 table_->tcal().table().addRow(1,True) ;
1170 Vector<Float> defaultTcal( 1, 1.0 ) ;
1171 ArrayColumn<Float> tcalCol( table_->tcal().table(), "TCAL" ) ;
1172 tcalCol.put( 0, defaultTcal ) ;
1173 return ;
1174 }
1175
1176 if ( colTcal_ == "NONE" ) {
1177 // add dummy row
1178 os_ << "No TCAL column" << LogIO::POST ;
1179 table_->tcal().table().addRow(1,True) ;
1180 Vector<Float> defaultTcal( 1, 1.0 ) ;
1181 ArrayColumn<Float> tcalCol( table_->tcal().table(), "TCAL" ) ;
1182 tcalCol.put( 0, defaultTcal ) ;
1183 return ;
1184 }
1185
1186 Table sctab = mstable_.sysCal() ;
1187 if ( sctab.nrow() == 0 ) {
1188 os_ << "No SYSCAL rows" << LogIO::POST ;
1189 return ;
1190 }
1191 Table sctabsel( sctab( sctab.col("ANTENNA_ID") == antenna_ ) ) ;
1192 if ( sctabsel.nrow() == 0 ) {
1193 os_ << "No SYSCAL rows" << LogIO::POST ;
1194 return ;
1195 }
1196 ROArrayColumn<Float> *tmpTcalCol = new ROArrayColumn<Float>( sctabsel, colTcal_ ) ;
1197 // return if any rows without Tcal value exists
1198 Bool notDefined = False ;
1199 for ( uInt irow = 0 ; irow < sctabsel.nrow() ; irow++ ) {
1200 if ( !tmpTcalCol->isDefined( irow ) ) {
1201 notDefined = True ;
1202 break ;
1203 }
1204 }
1205 if ( notDefined ) {
1206 os_ << "No TCAL value" << LogIO::POST ;
1207 delete tmpTcalCol ;
1208 table_->tcal().table().addRow(1,True) ;
1209 Vector<Float> defaultTcal( 1, 1.0 ) ;
1210 ArrayColumn<Float> tcalCol( table_->tcal().table(), "TCAL" ) ;
1211 tcalCol.put( 0, defaultTcal ) ;
1212 return ;
1213 }
1214 uInt npol = tmpTcalCol->shape( 0 )(0) ;
1215 delete tmpTcalCol ;
1216 //os_ << "fillTcal(): npol = " << npol << LogIO::POST ;
1217 Table tab = table_->tcal().table() ;
1218 ArrayColumn<Float> tcalCol( tab, "TCAL" ) ;
1219 uInt oldnr = 0 ;
1220 uInt newnr = 0 ;
1221 TableRow row( tab ) ;
1222 TableRecord &trec = row.record() ;
1223 RecordFieldPtr<uInt> idRF( trec, "ID" ) ;
1224 RecordFieldPtr<String> timeRF( trec, "TIME" ) ;
1225 RecordFieldPtr< Array<Float> > tcalRF( trec, "TCAL" ) ;
1226 TableIterator iter0( sctabsel, "FEED_ID" ) ;
1227 while( !iter0.pastEnd() ) {
1228 Table t0 = iter0.table() ;
1229 Int feedId = asInt( "FEED_ID", 0, t0, tpoolr ) ;
1230 TableIterator iter1( t0, "SPECTRAL_WINDOW_ID" ) ;
1231 while( !iter1.pastEnd() ) {
1232 Table t1 = iter1.table() ;
1233 Int spwId = asInt( "SPECTRAL_WINDOW_ID", 0, t1, tpoolr ) ;
1234 tmpTcalCol = new ROArrayColumn<Float>( t1, colTcal_ ) ;
1235 ROScalarQuantColumn<Double> scTimeCol( t1, "TIME" ) ;
1236 Vector<uInt> idminmax( 2, oldnr ) ;
1237 for ( uInt irow = 0 ; irow < t1.nrow() ; irow++ ) {
1238 String sTime = MVTime( scTimeCol(irow) ).string( MVTime::YMD ) ;
1239 *timeRF = sTime ;
1240 uInt idx = oldnr ;
1241 Matrix<Float> subtcal = (*tmpTcalCol)( irow ) ;
1242 for ( uInt ipol = 0 ; ipol < npol ; ipol++ ) {
1243 *idRF = idx++ ;
1244 //*tcalRF = subtcal.row( ipol ) ;
1245 tcalRF.define( subtcal.row( ipol ) ) ;
1246
1247 // commit row
1248 tab.addRow() ;
1249 row.put( tab.nrow()-1 ) ;
1250
1251 newnr++ ;
1252 }
1253 idminmax[0] = oldnr ;
1254 idminmax[1] = newnr - 1 ;
1255 oldnr = newnr ;
1256
1257 String key = keyTcal( feedId, spwId, sTime ) ;
1258 tcalrec_.define( key, idminmax ) ;
1259 }
1260 delete tmpTcalCol ;
1261 iter1++ ;
1262 }
1263 iter0++ ;
1264 }
1265
1266 //tcalrec_.print( std::cout ) ;
1267 //double endSec = gettimeofday_sec() ;
1268 //os_ << "end MSFiller::fillTcal() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
1269}
1270
1271uInt MSFiller::getWeatherId( uInt idx, Double wtime )
1272{
1273 //double startSec = gettimeofday_sec() ;
1274 //os_ << "start MSFiller::getWeatherId() startSec=" << startSec << LogIO::POST ;
1275 uInt nrow = mwTime_.size() ;
1276 if ( nrow < 2 )
1277 return 0 ;
1278 uInt wid = nrow ;
1279 if ( idx == 0 ) {
1280 wid = 0 ;
1281 Double tStart = mwTime_[wid]-0.5*mwInterval_[wid] ;
1282 if ( wtime < tStart )
1283 return wid ;
1284 }
1285 for ( uInt i = idx ; i < nrow-1 ; i++ ) {
1286 Double tStart = mwTime_[i]-0.5*mwInterval_[i] ;
1287 // use of INTERVAL column is problematic
1288 // since there are "blank" time of weather monitoring
1289 //Double tEnd = tStart + mwInterval_[i] ;
1290 Double tEnd = mwTime_[i+1]-0.5*mwInterval_[i+1] ;
1291 //os_ << "tStart = " << tStart << " dtEnd = " << tEnd-tStart << " dwtime = " << wtime-tStart << LogIO::POST ;
1292 if ( wtime >= tStart && wtime <= tEnd ) {
1293 wid = i ;
1294 break ;
1295 }
1296 }
1297 if ( wid == nrow ) {
1298 uInt i = nrow - 1 ;
1299 Double tStart = mwTime_[i-1]+0.5*mwInterval_[i-1] ;
1300 Double tEnd = mwTime_[i]+0.5*mwInterval_[i] ;
1301 //os_ << "tStart = " << tStart << " dtEnd = " << tEnd-tStart << " dwtime = " << wtime-tStart << LogIO::POST ;
1302 if ( wtime >= tStart && wtime <= tEnd )
1303 wid = i-1 ;
1304 else
1305 wid = i ;
1306 }
1307
1308 //if ( wid == nrow )
1309 //os_ << LogIO::WARN << "Couldn't find correct WEATHER_ID for time " << wtime << LogIO::POST ;
1310
1311 //double endSec = gettimeofday_sec() ;
1312 //os_ << "end MSFiller::getWeatherId() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
1313 return wid ;
1314}
1315
1316void MSFiller::getSysCalTime( Vector<MEpoch> &scTime, Vector<Double> &scInterval, Block<MEpoch> &tcol, Block<Int> &tidx )
1317{
1318 //double startSec = gettimeofday_sec() ;
1319 //os_ << "start MSFiller::getSysCalTime() startSec=" << startSec << LogIO::POST ;
1320
1321 if ( !isSysCal_ )
1322 return ;
1323
1324 uInt nrow = tidx.nelements() ;
1325 if ( scTime.nelements() == 0 )
1326 return ;
1327 else if ( scTime.nelements() == 1 ) {
1328 tidx[0] = 0 ;
1329 return ;
1330 }
1331 uInt scnrow = scTime.nelements() ;
1332 uInt idx = 0 ;
1333 const Double half = 0.5e0 ;
1334 // execute binary search
1335 idx = binarySearch( scTime, tcol[0].get( "s" ).getValue() ) ;
1336 if ( idx != 0 )
1337 idx -= 1 ;
1338 for ( uInt i = 0 ; i < nrow ; i++ ) {
1339 Double t = tcol[i].get( "s" ).getValue() ;
1340 Double tsc = scTime[0].get( "s" ).getValue() ;
1341 if ( t < tsc ) {
1342 tidx[i] = 0 ;
1343 continue ;
1344 }
1345 for ( uInt j = idx ; j < scnrow-1 ; j++ ) {
1346 Double tsc1 = scTime[j].get( "s" ).getValue() ;
1347 Double dt1 = scInterval[j] ;
1348 Double tsc2 = scTime[j+1].get( "s" ).getValue() ;
1349 Double dt2 = scInterval[j+1] ;
1350 if ( t > tsc1-half*dt1 && t <= tsc2-half*dt2 ) {
1351 tidx[i] = j ;
1352 idx = j ;
1353 break ;
1354 }
1355 }
1356 if ( tidx[i] == -1 ) {
1357// Double tsc = scTime[scnrow-1].get( "s" ).getValue() ;
1358// Double dt = scInterval[scnrow-1] ;
1359// if ( t <= tsc+0.5*dt ) {
1360// tidx[i] = scnrow-1 ;
1361// }
1362 tidx[i] = scnrow-1 ;
1363 }
1364 }
1365 //double endSec = gettimeofday_sec() ;
1366 //os_ << "end MSFiller::getSysCalTime() endSec=" << endSec << " (" << endSec-startSec << "sec) scnrow = " << scnrow << " tcol.nelements = " << tcol.nelements() << LogIO::POST ;
1367 return ;
1368}
1369
1370Block<uInt> MSFiller::getTcalId( Int fid, Int spwid, MEpoch &t )
1371{
1372 //double startSec = gettimeofday_sec() ;
1373 //os_ << "start MSFiller::getTcalId() startSec=" << startSec << LogIO::POST ;
1374 //if ( table_->tcal().table().nrow() == 0 ) {
1375 if ( !isSysCal_ ) {
1376 os_ << "No TCAL rows" << LogIO::POST ;
1377 Block<uInt> tcalids( 4, 0 ) ;
1378 return tcalids ;
1379 }
1380 //String sctime = MVTime( Quantum<Double>(t,"s") ).string(MVTime::YMD) ;
1381 String sctime = MVTime( t.getValue() ).string(MVTime::YMD) ;
1382 String key = keyTcal( fid, spwid, sctime ) ;
1383 if ( !tcalrec_.isDefined( key ) ) {
1384 os_ << "No TCAL rows" << LogIO::POST ;
1385 Block<uInt> tcalids( 4, 0 ) ;
1386 return tcalids ;
1387 }
1388 Vector<uInt> ids = tcalrec_.asArrayuInt( key ) ;
1389 uInt npol = ids[1] - ids[0] + 1 ;
1390 Block<uInt> tcalids( npol ) ;
1391 tcalids[0] = ids[0] ;
1392 tcalids[1] = ids[1] ;
1393 for ( uInt ipol = 2 ; ipol < npol ; ipol++ )
1394 tcalids[ipol] = ids[0] + ipol - 1 ;
1395
1396 //double endSec = gettimeofday_sec() ;
1397 //os_ << "end MSFiller::getTcalId() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
1398 return tcalids ;
1399}
1400
1401uInt MSFiller::getDirection( uInt idx, Vector<Double> &dir, Vector<Double> &srate, String &ref, MSPointing &tab, Double t )
1402{
1403 //double startSec = gettimeofday_sec() ;
1404 //os_ << "start MSFiller::getDirection1() startSec=" << startSec << LogIO::POST ;
1405 // assume that cols is sorted by TIME
1406 Bool doInterp = False ;
1407 //uInt nrow = cols.nrow() ;
1408 uInt nrow = tab.nrow() ;
1409 if ( nrow == 0 )
1410 return 0 ;
1411 ROScalarMeasColumn<MEpoch> tcol( tab, "TIME" ) ;
1412 ROArrayMeasColumn<MDirection> dmcol( tab, "DIRECTION" ) ;
1413 ROArrayColumn<Double> dcol( tab, "DIRECTION" ) ;
1414 // binary search if idx == 0
1415 if ( idx == 0 ) {
1416 uInt nrowb = 75000 ;
1417 if ( nrow > nrowb ) {
1418 uInt nblock = nrow / nrowb + 1 ;
1419 for ( uInt iblock = 0 ; iblock < nblock ; iblock++ ) {
1420 uInt high = min( nrowb, nrow-iblock*nrowb ) ;
1421
1422 if ( tcol( high-1 ).get( "s" ).getValue() < t ) {
1423 idx = iblock * nrowb ;
1424 continue ;
1425 }
1426
1427 Vector<MEpoch> tarr( high ) ;
1428 for ( uInt irow = 0 ; irow < high ; irow++ ) {
1429 tarr[irow] = tcol( iblock*nrowb+irow ) ;
1430 }
1431
1432 uInt bidx = binarySearch( tarr, t ) ;
1433
1434 idx = iblock * nrowb + bidx ;
1435 break ;
1436 }
1437 }
1438 else {
1439 Vector<MEpoch> tarr( nrow ) ;
1440 for ( uInt irow = 0 ; irow < nrow ; irow++ ) {
1441 tarr[irow] = tcol( irow ) ;
1442 }
1443 idx = binarySearch( tarr, t ) ;
1444 }
1445 }
1446 // ensure that tcol(idx) < t
1447 //os_ << "tcol(idx) = " << tcol(idx).get("s").getValue() << " t = " << t << " diff = " << tcol(idx).get("s").getValue()-t << endl ;
1448 while ( tcol(idx).get("s").getValue() > t && idx > 0 )
1449 idx-- ;
1450 //os_ << "idx = " << idx << LogIO::POST ;
1451
1452 // index search
1453 for ( uInt i = idx ; i < nrow ; i++ ) {
1454 Double tref = tcol( i ).get( "s" ).getValue() ;
1455 if ( tref == t ) {
1456 idx = i ;
1457 break ;
1458 }
1459 else if ( tref > t ) {
1460 if ( i == 0 ) {
1461 idx = i ;
1462 }
1463 else {
1464 idx = i-1 ;
1465 doInterp = True ;
1466 }
1467 break ;
1468 }
1469 else {
1470 idx = nrow - 1 ;
1471 }
1472 }
1473 //os_ << "searched idx = " << idx << LogIO::POST ;
1474
1475 //os_ << "dmcol(idx).shape() = " << dmcol(idx).shape() << LogIO::POST ;
1476 IPosition ip( dmcol(idx).shape().nelements(), 0 ) ;
1477 //os_ << "ip = " << ip << LogIO::POST ;
1478 ref = dmcol(idx)(ip).getRefString() ;
1479 //os_ << "ref = " << ref << LogIO::POST ;
1480 if ( doInterp ) {
1481 //os_ << "do interpolation" << LogIO::POST ;
1482 //os_ << "dcol(idx).shape() = " << dcol(idx).shape() << LogIO::POST ;
1483 Double tref0 = tcol(idx).get("s").getValue() ;
1484 Double tref1 = tcol(idx+1).get("s").getValue() ;
1485 Matrix<Double> mdir0 = dcol( idx ) ;
1486 Matrix<Double> mdir1 = dcol( idx+1 ) ;
1487 Vector<Double> dir0 = mdir0.column( 0 ) ;
1488 //os_ << "dir0 = " << dir0 << LogIO::POST ;
1489 Vector<Double> dir1 = mdir1.column( 0 ) ;
1490 //os_ << "dir1 = " << dir1 << LogIO::POST ;
1491 Double dt0 = t - tref0 ;
1492 Double dt1 = tref1 - t ;
1493 dir.reference( (dt0*dir1+dt1*dir0)/(dt0+dt1) ) ;
1494 if ( mdir0.ncolumn() > 1 ) {
1495 if ( dt0 >= dt1 )
1496 srate.reference( mdir0.column( 1 ) ) ;
1497 else
1498 srate.reference( mdir1.column( 1 ) ) ;
1499 }
1500 //os_ << "dir = " << dir << LogIO::POST ;
1501 }
1502 else {
1503 //os_ << "no interpolation" << LogIO::POST ;
1504 Matrix<Double> mdir0 = dcol( idx ) ;
1505 dir.reference( mdir0.column( 0 ) ) ;
1506 if ( mdir0.ncolumn() > 1 )
1507 srate.reference( mdir0.column( 1 ) ) ;
1508 }
1509
1510 //double endSec = gettimeofday_sec() ;
1511 //os_ << "end MSFiller::getDirection1() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
1512 return idx ;
1513}
1514
1515String MSFiller::keyTcal( Int feedid, Int spwid, String stime )
1516{
1517 String sfeed = "FEED" + String::toString( feedid ) ;
1518 String sspw = "SPW" + String::toString( spwid ) ;
1519 return sfeed+":"+sspw+":"+stime ;
1520}
1521
1522uInt MSFiller::binarySearch( Vector<MEpoch> &timeList, Double target )
1523{
1524 Int low = 0 ;
1525 Int high = timeList.nelements() ;
1526 uInt idx = 0 ;
1527
1528 while ( low <= high ) {
1529 idx = (Int)( 0.5 * ( low + high ) ) ;
1530 Double t = timeList[idx].get( "s" ).getValue() ;
1531 if ( t < target )
1532 low = idx + 1 ;
1533 else if ( t > target )
1534 high = idx - 1 ;
1535 else
1536 return idx ;
1537 }
1538
1539 idx = max( 0, min( low, high ) ) ;
1540
1541 return idx ;
1542
1543}
1544
1545string MSFiller::getFrame()
1546{
1547 MFrequency::Types frame = MFrequency::DEFAULT ;
1548 ROTableColumn numChanCol( mstable_.spectralWindow(), "NUM_CHAN" ) ;
1549 ROTableColumn measFreqRefCol( mstable_.spectralWindow(), "MEAS_FREQ_REF" ) ;
1550 uInt nrow = numChanCol.nrow() ;
1551 Vector<Int> measFreqRef( nrow, MFrequency::DEFAULT ) ;
1552 uInt nref = 0 ;
1553 for ( uInt irow = 0 ; irow < nrow ; irow++ ) {
1554 if ( numChanCol.asInt( irow ) != 4 ) { // exclude WVR
1555 measFreqRef[nref] = measFreqRefCol.asInt( irow ) ;
1556 nref++ ;
1557 }
1558 }
1559 if ( nref > 0 )
1560 frame = (MFrequency::Types)measFreqRef[0] ;
1561
1562 return MFrequency::showType( frame ) ;
1563}
1564
1565void MSFiller::reshapeSpectraAndFlagtra( Cube<Float> &sp,
1566 Cube<Bool> &fl,
1567 Table &tab,
1568 Int &npol,
1569 Int &nchan,
1570 Int &nrow,
1571 Vector<Int> &corrtype )
1572{
1573 //double startSec = gettimeofday_sec() ;
1574 //os_ << "start MFiller::reshapeSpectraAndFlagtra() startSec=" << startSec << LogIO::POST ;
1575 if ( isFloatData_ ) {
1576 ROArrayColumn<Bool> mFlagCol( tab, "FLAG" ) ;
1577 ROArrayColumn<Float> mFloatDataCol( tab, "FLOAT_DATA" ) ;
1578 sp = mFloatDataCol.getColumn() ;
1579 fl = mFlagCol.getColumn() ;
1580 }
1581 else if ( isData_ ) {
1582 sp.resize( npol, nchan, nrow ) ;
1583 fl.resize( npol, nchan, nrow ) ;
1584 ROArrayColumn<Bool> mFlagCol( tab, "FLAG" ) ;
1585 ROArrayColumn<Complex> mDataCol( tab, "DATA" ) ;
1586 for ( Int irow = 0 ; irow < nrow ; irow++ ) {
1587 Bool crossOK = False ;
1588 Matrix<Complex> mSp = mDataCol( irow ) ;
1589 Matrix<Bool> mFl = mFlagCol( irow ) ;
1590 Matrix<Float> spxy = sp.xyPlane( irow ) ;
1591 Matrix<Bool> flxy = fl.xyPlane( irow ) ;
1592 for ( Int ipol = 0 ; ipol < npol ; ipol++ ) {
1593 if ( corrtype[ipol] == Stokes::XY || corrtype[ipol] == Stokes::YX
1594 || corrtype[ipol] == Stokes::RL || corrtype[ipol] == Stokes::LR ) {
1595 if ( !crossOK ) {
1596 spxy.row( ipol ) = real( mSp.row( ipol ) ) ;
1597 flxy.row( ipol ) = mFl.row( ipol ) ;
1598 if ( corrtype[ipol] == Stokes::XY || corrtype[ipol] == Stokes::RL ) {
1599 spxy.row( ipol+1 ) = imag( mSp.row( ipol ) ) ;
1600 flxy.row( ipol+1 ) = mFl.row( ipol ) ;
1601 }
1602 else {
1603 spxy.row( ipol+1 ) = imag( conj( mSp.row( ipol ) ) ) ;
1604 flxy.row( ipol+1 ) = mFl.row( ipol ) ;
1605 }
1606 crossOK = True ;
1607 }
1608 }
1609 else {
1610 spxy.row( ipol ) = real( mSp.row( ipol ) ) ;
1611 flxy.row( ipol ) = mFl.row( ipol ) ;
1612 }
1613 }
1614 }
1615 }
1616 //double endSec = gettimeofday_sec() ;
1617 //os_ << "end MSFiller::reshapeSpectraAndFlagtra() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
1618}
1619
1620uInt MSFiller::getDirection( uInt idx,
1621 Vector<Double> &dir,
1622 Vector<Double> &azel,
1623 Vector<Double> &srate,
1624 MSPointing &ptab,
1625 MEpoch &t,
1626 MPosition &antpos )
1627{
1628 //double startSec = gettimeofday_sec() ;
1629 //os_ << "start MFiller::getDirection2() startSec=" << startSec << LogIO::POST ;
1630 String refString ;
1631 MDirection::Types dirType ;
1632 uInt diridx = getDirection( idx, dir, srate, refString, ptab, t.get("s").getValue() ) ;
1633 MDirection::getType( dirType, refString ) ;
1634 MeasFrame mf( t, antpos ) ;
1635 if ( refString == "J2000" ) {
1636 MDirection::Convert toazel( dirType, MDirection::Ref( MDirection::AZEL, mf ) ) ;
1637 azel = toazel( dir ).getAngle("rad").getValue() ;
1638 }
1639 else if ( refString(0,4) == "AZEL" ) {
1640 azel = dir.copy() ;
1641 MDirection::Convert toj2000( dirType, MDirection::Ref( MDirection::J2000, mf ) ) ;
1642 dir = toj2000( dir ).getAngle("rad").getValue() ;
1643 }
1644 else {
1645 MDirection::Convert toazel( dirType, MDirection::Ref( MDirection::AZEL, mf ) ) ;
1646 azel = toazel( dir ).getAngle("rad").getValue() ;
1647 MDirection::Convert toj2000( dirType, MDirection::Ref( MDirection::J2000, mf ) ) ;
1648 dir = toj2000( dir ).getAngle("rad").getValue() ;
1649 }
1650 if ( srate.size() == 0 )
1651 srate = Vector<Double>( 2, 0.0 ) ;
1652 //double endSec = gettimeofday_sec() ;
1653 //os_ << "end MSFiller::getDirection2() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
1654 return diridx ;
1655}
1656
1657void MSFiller::getSourceDirection( Vector<Double> &dir,
1658 Vector<Double> &azel,
1659 Vector<Double> &srate,
1660 MEpoch &t,
1661 MPosition &antpos,
1662 Vector<MDirection> &srcdir )
1663{
1664 //double startSec = gettimeofday_sec() ;
1665 //os_ << "start MFiller::getSourceDirection() startSec=" << startSec << LogIO::POST ;
1666 Vector<Double> defaultScanrate( 2, 0.0 ) ;
1667 Vector<Double> defaultDir = srcdir[0].getAngle( "rad" ).getValue() ;
1668 if ( srcdir.nelements() > 1 )
1669 defaultScanrate = srcdir[1].getAngle( "rad" ).getValue() ;
1670 String ref = srcdir[0].getRefString() ;
1671 MDirection::Types dirType ;
1672 MDirection::getType( dirType, ref ) ;
1673 MeasFrame mf( t, antpos ) ;
1674 if ( ref != "J2000" ) {
1675 MDirection::Convert toj2000( dirType, MDirection::Ref( MDirection::J2000, mf ) ) ;
1676 dir = toj2000( defaultDir ).getAngle("rad").getValue() ;
1677 }
1678 else
1679 dir = defaultDir ;
1680 if ( ref != "AZELGEO" ) {
1681 MDirection::Convert toazel( dirType, MDirection::Ref( MDirection::AZELGEO, mf ) ) ;
1682 azel = toazel( defaultDir ).getAngle("rad").getValue() ;
1683 }
1684 else
1685 azel = defaultDir ;
1686 srate = defaultScanrate ;
1687 //double endSec = gettimeofday_sec() ;
1688 //os_ << "end MSFiller::getSourceDirection() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
1689}
1690
1691void MSFiller::initHeader( STHeader &header )
1692{
1693 header.nchan = 0 ;
1694 header.npol = 0 ;
1695 header.nif = 0 ;
1696 header.nbeam = 0 ;
1697 header.observer = "" ;
1698 header.project = "" ;
1699 header.obstype = "" ;
1700 header.antennaname = "" ;
1701 header.antennaposition.resize( 0 ) ;
1702 header.equinox = 0.0 ;
1703 header.freqref = "" ;
1704 header.reffreq = -1.0 ;
1705 header.bandwidth = 0.0 ;
1706 header.utc = 0.0 ;
1707 header.fluxunit = "" ;
1708 header.epoch = "" ;
1709 header.poltype = "" ;
1710}
1711
1712String MSFiller::asString( String name,
1713 uInt idx,
1714 Table tab,
1715 boost::object_pool<ROTableColumn> *pool )
1716{
1717 ROTableColumn *col = pool->construct( tab, name ) ;
1718 String v = col->asString( idx ) ;
1719 pool->destroy( col ) ;
1720 return v ;
1721}
1722
1723Bool MSFiller::asBool( String name,
1724 uInt idx,
1725 Table &tab,
1726 boost::object_pool<ROTableColumn> *pool )
1727{
1728 ROTableColumn *col = pool->construct( tab, name ) ;
1729 Bool v = col->asBool( idx ) ;
1730 pool->destroy( col ) ;
1731 return v ;
1732}
1733
1734uInt MSFiller::asuInt( String name,
1735 uInt idx,
1736 Table &tab,
1737 boost::object_pool<ROTableColumn> *pool )
1738{
1739 ROTableColumn *col = pool->construct( tab, name ) ;
1740 uInt v = col->asuInt( idx ) ;
1741 pool->destroy( col ) ;
1742 return v ;
1743}
1744
1745Int MSFiller::asInt( String name,
1746 uInt idx,
1747 Table &tab,
1748 boost::object_pool<ROTableColumn> *pool )
1749{
1750 ROTableColumn *col = pool->construct( tab, name ) ;
1751 Int v = col->asInt( idx ) ;
1752 pool->destroy( col ) ;
1753 return v ;
1754}
1755
1756Float MSFiller::asFloat( String name,
1757 uInt idx,
1758 Table &tab,
1759 boost::object_pool<ROTableColumn> *pool )
1760{
1761 ROTableColumn *col = pool->construct( tab, name ) ;
1762 Float v = col->asfloat( idx ) ;
1763 pool->destroy( col ) ;
1764 return v ;
1765}
1766
1767Double MSFiller::asDouble( String name,
1768 uInt idx,
1769 Table &tab,
1770 boost::object_pool<ROTableColumn> *pool )
1771{
1772 ROTableColumn *col = pool->construct( tab, name ) ;
1773 Double v = col->asdouble( idx ) ;
1774 pool->destroy( col ) ;
1775 return v ;
1776}
1777
1778void MSFiller::sourceInfo( Int sourceId,
1779 Int spwId,
1780 String &name,
1781 MDirection &direction,
1782 Vector<casa::Double> &properMotion,
1783 Vector<casa::Double> &restFreqs,
1784 Vector<casa::String> &transitions,
1785 Vector<casa::Double> &sysVels,
1786 boost::object_pool<ROTableColumn> *tpoolr )
1787{
1788 //double startSec = gettimeofday_sec() ;
1789 //os_ << "start MFiller::sourceInfo() startSec=" << startSec << LogIO::POST ;
1790
1791 MSSource srctab = mstable_.source() ;
1792 MSSource srctabSel = srctab( srctab.col("SOURCE_ID") == sourceId && srctab.col("SPECTRAL_WINDOW_ID") == spwId ) ;
1793 if ( srctabSel.nrow() == 0 ) {
1794 srctabSel = srctab( srctab.col("SOURCE_ID") == sourceId && srctab.col("SPECTRAL_WINDOW_ID") == -1 ) ;
1795 }
1796 Int numLines = 0 ;
1797 if ( srctabSel.nrow() > 0 ) {
1798 // source name
1799 name = asString( "NAME", 0, srctabSel, tpoolr ) ;
1800
1801 // source proper motion
1802 ROArrayColumn<Double> roArrDCol( srctabSel, "PROPER_MOTION" ) ;
1803 properMotion = roArrDCol( 0 ) ;
1804
1805 // source direction as MDirection object
1806 ROScalarMeasColumn<MDirection> tmpMeasCol( srctabSel, "DIRECTION" ) ;
1807 direction = tmpMeasCol( 0 ) ;
1808
1809 // number of lines
1810 numLines = asInt( "NUM_LINES", 0, srctabSel, tpoolr ) ;
1811 }
1812 else {
1813 name = "" ;
1814 properMotion = Vector<Double>( 2, 0.0 ) ;
1815 direction = MDirection( Quantum<Double>(0.0,Unit("rad")), Quantum<Double>(0.0,Unit("rad")) ) ;
1816 }
1817
1818 restFreqs.resize( numLines ) ;
1819 transitions.resize( numLines ) ;
1820 sysVels.resize( numLines ) ;
1821 if ( numLines > 0 ) {
1822 if ( srctabSel.tableDesc().isColumn( "REST_FREQUENCY" ) ) {
1823 ROArrayQuantColumn<Double> quantArrCol( srctabSel, "REST_FREQUENCY" ) ;
1824 Array< Quantum<Double> > qRestFreqs = quantArrCol( 0 ) ;
1825 for ( int i = 0 ; i < numLines ; i++ ) {
1826 restFreqs[i] = qRestFreqs( IPosition( 1, i ) ).getValue( "Hz" ) ;
1827 }
1828 }
1829 //os_ << "restFreqs = " << restFreqs << LogIO::POST ;
1830 if ( srctabSel.tableDesc().isColumn( "TRANSITION" ) ) {
1831 ROArrayColumn<String> transitionCol( srctabSel, "TRANSITION" ) ;
1832 if ( transitionCol.isDefined( 0 ) )
1833 transitions = transitionCol( 0 ) ;
1834 //os_ << "transitionNameCol.nrow() = " << transitionCol.nrow() << LogIO::POST ;
1835 }
1836 if ( srctabSel.tableDesc().isColumn( "SYSVEL" ) ) {
1837 ROArrayColumn<Double> roArrDCol( srctabSel, "SYSVEL" ) ;
1838 sysVels = roArrDCol( 0 ) ;
1839 }
1840 }
1841
1842 //double endSec = gettimeofday_sec() ;
1843 //os_ << "end MSFiller::sourceInfo() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
1844}
1845
1846void MSFiller::spectralSetup( Int spwId,
1847 MEpoch &me,
1848 MPosition &mp,
1849 MDirection &md,
1850 Double &refpix,
1851 Double &refval,
1852 Double &increment,
1853 Int &nchan,
1854 String &freqref,
1855 Double &reffreq,
1856 Double &bandwidth,
1857 boost::object_pool<ROTableColumn> *tpoolr )
1858{
1859 //double startSec = gettimeofday_sec() ;
1860 //os_ << "start MFiller::spectralSetup() startSec=" << startSec << LogIO::POST ;
1861
1862 MSSpectralWindow spwtab = mstable_.spectralWindow() ;
1863 MeasFrame mf( me, mp, md ) ;
1864 MFrequency::Types freqRef = MFrequency::castType( (uInt)asInt( "MEAS_FREQ_REF", spwId, spwtab, tpoolr ) ) ;
1865 Bool even = False ;
1866 if ( (nchan/2)*2 == nchan ) even = True ;
1867 ROScalarQuantColumn<Double> tmpQuantCol( spwtab, "TOTAL_BANDWIDTH" ) ;
1868 Double totbw = tmpQuantCol( spwId ).getValue( "Hz" ) ;
1869 if ( nchan != 4 )
1870 bandwidth = max( bandwidth, totbw ) ;
1871 if ( freqref == "" && nchan != 4)
1872 //sdh.freqref = MFrequency::showType( freqRef ) ;
1873 freqref = "LSRK" ;
1874 if ( reffreq == -1.0 && nchan != 4 ) {
1875 tmpQuantCol.attach( spwtab, "REF_FREQUENCY" ) ;
1876 Quantum<Double> qreffreq = tmpQuantCol( spwId ) ;
1877 if ( freqRef == MFrequency::LSRK ) {
1878 reffreq = qreffreq.getValue("Hz") ;
1879 }
1880 else {
1881 MFrequency::Convert tolsr( freqRef, MFrequency::Ref( MFrequency::LSRK, mf ) ) ;
1882 reffreq = tolsr( qreffreq ).get("Hz").getValue() ;
1883 }
1884 }
1885 Int refchan = nchan / 2 ;
1886 IPosition refip( 1, refchan ) ;
1887 refpix = 0.5*(nchan-1) ;
1888 refval = 0.0 ;
1889 ROArrayQuantColumn<Double> sharedQDArrCol( spwtab, "CHAN_WIDTH" ) ;
1890 increment = sharedQDArrCol( spwId )( refip ).getValue( "Hz" ) ;
1891 // os_ << "nchan = " << nchan << " refchan = " << refchan << "(even=" << even << ") refpix = " << refpix << LogIO::POST ;
1892 sharedQDArrCol.attach( spwtab, "CHAN_FREQ" ) ;
1893 Vector< Quantum<Double> > chanFreqs = sharedQDArrCol( spwId ) ;
1894 if ( nchan > 1 && chanFreqs[0].getValue("Hz") > chanFreqs[1].getValue("Hz") )
1895 increment *= -1.0 ;
1896 if ( freqRef == MFrequency::LSRK ) {
1897 if ( even ) {
1898 IPosition refip0( 1, refchan-1 ) ;
1899 Double refval0 = chanFreqs(refip0).getValue("Hz") ;
1900 Double refval1 = chanFreqs(refip).getValue("Hz") ;
1901 refval = 0.5 * ( refval0 + refval1 ) ;
1902 }
1903 else {
1904 refval = chanFreqs(refip).getValue("Hz") ;
1905 }
1906 }
1907 else {
1908 MFrequency::Convert tolsr( freqRef, MFrequency::Ref( MFrequency::LSRK, mf ) ) ;
1909 if ( even ) {
1910 IPosition refip0( 1, refchan-1 ) ;
1911 Double refval0 = chanFreqs(refip0).getValue("Hz") ;
1912 Double refval1 = chanFreqs(refip).getValue("Hz") ;
1913 refval = 0.5 * ( refval0 + refval1 ) ;
1914 refval = tolsr( refval ).get("Hz").getValue() ;
1915 }
1916 else {
1917 refval = tolsr( chanFreqs(refip) ).get("Hz").getValue() ;
1918 }
1919 }
1920
1921 //double endSec = gettimeofday_sec() ;
1922 //os_ << "end MSFiller::spectralSetup() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
1923}
1924
1925} ;
1926
Note: See TracBrowser for help on using the repository browser.