source: branches/hpc34/src/MSWriter.cpp@ 2620

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

Use Block instead of Matrix to store processed pair of
(feedId,spwId) and (polId,spwId) for addFeed and addDataDescription.


File size: 76.7 KB
Line 
1//
2// C++ Interface: MSWriter
3//
4// Description:
5//
6// This class is specific writer 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#include <assert.h>
14
15#include <casa/OS/File.h>
16#include <casa/OS/RegularFile.h>
17#include <casa/OS/Directory.h>
18#include <casa/OS/SymLink.h>
19#include <casa/BasicSL/String.h>
20#include <casa/Arrays/Cube.h>
21
22#include <tables/Tables/ExprNode.h>
23#include <tables/Tables/TableDesc.h>
24#include <tables/Tables/SetupNewTab.h>
25#include <tables/Tables/TableIter.h>
26#include <tables/Tables/RefRows.h>
27#include <tables/Tables/TableRow.h>
28
29#include <ms/MeasurementSets/MeasurementSet.h>
30#include <ms/MeasurementSets/MSColumns.h>
31#include <ms/MeasurementSets/MSPolIndex.h>
32#include <ms/MeasurementSets/MSDataDescIndex.h>
33#include <ms/MeasurementSets/MSSourceIndex.h>
34
35#include "MSWriter.h"
36#include "STHeader.h"
37#include "STFrequencies.h"
38#include "STMolecules.h"
39#include "STTcal.h"
40#include "MathUtils.h"
41#include "TableTraverse.h"
42
43using namespace casa ;
44using namespace std ;
45
46namespace asap {
47
48class CorrTypeHandler {
49public:
50 CorrTypeHandler()
51 {}
52 virtual ~CorrTypeHandler() {}
53 virtual Vector<Stokes::StokesTypes> corrType() = 0 ;
54 virtual void reset()
55 {
56 npol = 0 ;
57 }
58 void append( uInt polno )
59 {
60 polnos[npol] = polno ;
61 npol++ ;
62 }
63 uInt nPol() { return npol ; }
64protected:
65 Vector<Stokes::StokesTypes> polmap ;
66 uInt polnos[4] ;
67 uInt npol ;
68};
69
70class LinearHandler : public CorrTypeHandler {
71public:
72 LinearHandler()
73 : CorrTypeHandler()
74 {
75 initMap() ;
76 }
77 virtual ~LinearHandler() {}
78 virtual Vector<Stokes::StokesTypes> corrType()
79 {
80 Vector<Stokes::StokesTypes> ret( npol, Stokes::Undefined ) ;
81 if ( npol < 4 ) {
82 for ( uInt ipol = 0 ; ipol < npol ; ipol++ )
83 ret[ipol] = polmap[polnos[ipol]] ;
84 }
85 else if ( npol == 4 ) {
86 ret[0] = polmap[0] ;
87 ret[1] = polmap[2] ;
88 ret[2] = polmap[3] ;
89 ret[4] = polmap[1] ;
90 }
91 else {
92 throw( AipsError("npol > 4") ) ;
93 }
94 return ret ;
95 }
96protected:
97 void initMap()
98 {
99 polmap.resize( 4 ) ;
100 polmap[0] = Stokes::XX ;
101 polmap[1] = Stokes::YY ;
102 polmap[2] = Stokes::XY ;
103 polmap[3] = Stokes::YX ;
104 }
105};
106class CircularHandler : public CorrTypeHandler {
107public:
108 CircularHandler()
109 : CorrTypeHandler()
110 {
111 initMap() ;
112 }
113 virtual ~CircularHandler() {}
114 virtual Vector<Stokes::StokesTypes> corrType()
115 {
116 Vector<Stokes::StokesTypes> ret( npol, Stokes::Undefined ) ;
117 if ( npol < 4 ) {
118 for ( uInt ipol = 0 ; ipol < npol ; ipol++ )
119 ret[ipol] = polmap[polnos[ipol]] ;
120 }
121 else if ( npol == 4 ) {
122 ret[0] = polmap[0] ;
123 ret[1] = polmap[2] ;
124 ret[2] = polmap[3] ;
125 ret[3] = polmap[1] ;
126 }
127 else {
128 throw( AipsError("npol > 4") ) ;
129 }
130 return ret ;
131 }
132private:
133 void initMap()
134 {
135 polmap.resize( 4 ) ;
136 polmap[0] = Stokes::RR ;
137 polmap[1] = Stokes::LL ;
138 polmap[2] = Stokes::RL ;
139 polmap[3] = Stokes::LR ;
140 }
141};
142class StokesHandler : public CorrTypeHandler {
143public:
144 StokesHandler()
145 : CorrTypeHandler()
146 {
147 initMap() ;
148 }
149 virtual ~StokesHandler() {}
150 virtual Vector<Stokes::StokesTypes> corrType()
151 {
152 Vector<Stokes::StokesTypes> ret( npol, Stokes::Undefined ) ;
153 if ( npol <= 4 ) {
154 for ( uInt ipol = 0 ; ipol < npol ; ipol++ )
155 ret[ipol] = polmap[polnos[ipol]] ;
156 }
157 else {
158 throw( AipsError("npol > 4") ) ;
159 }
160 return ret ;
161 }
162private:
163 void initMap()
164 {
165 polmap.resize( 4 ) ;
166 polmap[0] = Stokes::I ;
167 polmap[1] = Stokes::Q ;
168 polmap[2] = Stokes::U ;
169 polmap[3] = Stokes::V ;
170 }
171};
172class LinPolHandler : public CorrTypeHandler {
173public:
174 LinPolHandler()
175 : CorrTypeHandler()
176 {
177 initMap() ;
178 }
179 virtual ~LinPolHandler() {}
180 virtual Vector<Stokes::StokesTypes> corrType()
181 {
182 Vector<Stokes::StokesTypes> ret( npol, Stokes::Undefined ) ;
183 if ( npol <= 2 ) {
184 for ( uInt ipol = 0 ; ipol < npol ; ipol++ )
185 ret[ipol] = polmap[polnos[ipol]] ;
186 }
187 else {
188 throw( AipsError("npol > 4") ) ;
189 }
190 return ret ;
191 }
192private:
193 void initMap()
194 {
195 polmap.resize( 2 ) ;
196 polmap[0] = Stokes::Plinear ;
197 polmap[1] = Stokes::Pangle ;
198 }
199};
200
201class DataHolder {
202public:
203 DataHolder( TableRow &tableRow, String polType )
204 : row( tableRow )
205 {
206 nchan = 0 ;
207 npol = 0 ;
208 makeCorrTypeHandler( polType ) ;
209 attach() ;
210 flagRow.resize( 4 ) ;
211 reset() ;
212 sigmaTemplate.resize( 4 ) ;
213 sigmaTemplate = 1.0 ;
214 }
215 virtual ~DataHolder() {}
216 virtual void post() = 0 ;
217 virtual void reset()
218 {
219 corr->reset() ;
220 flagRow = False ;
221 npol = 0 ;
222 }
223 virtual void accumulate( uInt id, Vector<Float> &sp, Vector<Bool> &fl, Bool &flr )
224 {
225 accumulateCorrType( id ) ;
226 accumulateData( id, sp ) ;
227 accumulateFlag( id, fl ) ;
228 accumulateFlagRow( id, flr ) ;
229 }
230 uInt nPol() { return npol ; }
231 uInt nChan() { return nchan ; }
232 Vector<Int> corrTypeInt()
233 {
234 Vector<Int> v( npol ) ;
235 convertArray( v, corr->corrType() ) ;
236 return v ;
237 }
238 Vector<Stokes::StokesTypes> corrType() { return corr->corrType() ; }
239 void setNchan( uInt num )
240 {
241 nchan = num ;
242 resize() ;
243 }
244protected:
245 void postAuxiliary()
246 {
247 Vector<Float> w = sigmaTemplate( IPosition(1,0), IPosition(1,npol-1) ) ;
248 sigmaRF.define( w ) ;
249 weightRF.define( w ) ;
250 Cube<Bool> c( npol, nchan, 1, False ) ;
251 flagCategoryRF.define( c ) ;
252 }
253 inline void accumulateCorrType( uInt &id )
254 {
255 corr->append( id ) ;
256 npol = corr->nPol() ;
257 }
258 inline void accumulateFlagRow( uInt &id, Bool &flr )
259 {
260 flagRow[id] = flr ;
261 }
262 void postFlagRow()
263 {
264 *flagRowRF = anyEQ( flagRow, True ) ;
265 }
266 inline void accumulateFlag( uInt &id, Vector<Bool> &fl )
267 {
268 flag.row( id ) = fl ;
269 }
270 virtual void postFlag() = 0 ;
271 inline void accumulateData( uInt &id, Vector<Float> &sp )
272 {
273 data.row( id ) = sp ;
274 }
275 virtual void postData() = 0 ;
276 TableRow &row ;
277 uInt nchan ;
278 uInt npol ;
279 CountedPtr<CorrTypeHandler> corr;
280 RecordFieldPtr< Vector<Float> > sigmaRF ;
281 RecordFieldPtr< Vector<Float> > weightRF ;
282 RecordFieldPtr< Array<Bool> > flagRF ;
283 RecordFieldPtr<Bool> flagRowRF ;
284 RecordFieldPtr< Cube<Bool> > flagCategoryRF ;
285 Vector<Bool> flagRow ;
286 Matrix<Bool> flag ;
287 Matrix<Float> data ;
288 Vector<Float> sigmaTemplate ;
289private:
290 void makeCorrTypeHandler( String &polType )
291 {
292 if ( polType == "linear" )
293 corr = new LinearHandler() ;
294 else if ( polType == "circular" )
295 corr = new CircularHandler() ;
296 else if ( polType == "stokes" )
297 corr = new StokesHandler() ;
298 else if ( polType == "linpol" )
299 corr = new LinPolHandler() ;
300 else
301 throw( AipsError("Invalid polarization type") ) ;
302 }
303 void attach()
304 {
305 TableRecord &rec = row.record() ;
306 sigmaRF.attachToRecord( rec, "SIGMA" ) ;
307 weightRF.attachToRecord( rec, "WEIGHT" ) ;
308 flagRF.attachToRecord( rec, "FLAG" ) ;
309 flagRowRF.attachToRecord( rec, "FLAG_ROW" ) ;
310 flagCategoryRF.attachToRecord( rec, "FLAG_CATEGORY" ) ;
311 }
312 void resize()
313 {
314 flag.resize( 4, nchan ) ;
315 data.resize( 4, nchan ) ;
316 }
317};
318
319class FloatDataHolder : public DataHolder {
320public:
321 FloatDataHolder( TableRow &tableRow, String polType )
322 : DataHolder( tableRow, polType )
323 {
324 attachData() ;
325 }
326 virtual ~FloatDataHolder() {}
327 virtual void post()
328 {
329 postData() ;
330 postFlag() ;
331 postFlagRow() ;
332 postAuxiliary() ;
333 }
334protected:
335 virtual void postFlag()
336 {
337 flagRF.define( flag( IPosition( 2, 0, 0 ), IPosition( 2, npol-1, nchan-1 ) ) ) ;
338 }
339 virtual void postData()
340 {
341 dataRF.define( data( IPosition( 2, 0, 0 ), IPosition( 2, npol-1, nchan-1 ) ) ) ;
342 }
343private:
344 void attachData()
345 {
346 dataRF.attachToRecord( row.record(), "FLOAT_DATA" ) ;
347 }
348 RecordFieldPtr< Matrix<Float> > dataRF;
349};
350
351class ComplexDataHolder : public DataHolder {
352public:
353 ComplexDataHolder( TableRow &tableRow, String polType )
354 : DataHolder( tableRow, polType )
355 {
356 attachData() ;
357 }
358 virtual ~ComplexDataHolder() {}
359 virtual void accumulate( uInt id, Vector<Float> &sp, Vector<Bool> &fl, Bool &flr )
360 {
361 DataHolder::accumulate( id, sp, fl, flr ) ;
362 isFilled[id] = True ;
363 }
364 virtual void post()
365 {
366 postData() ;
367 postFlag() ;
368 postFlagRow() ;
369 postAuxiliary() ;
370 }
371 virtual void reset()
372 {
373 DataHolder::reset() ;
374 for ( uInt i = 0 ; i < 4 ; i++ )
375 isFilled[i] = False ;
376 }
377protected:
378 virtual void postFlag()
379 {
380 if ( npol == 4 ) {
381 Vector<Bool> tmp = flag.row( 3 ) ;
382 flag.row( 3 ) = flag.row( 1 ) ;
383 flag.row( 2 ) = flag.row( 2 ) || tmp ;
384 flag.row( 1 ) = flag.row( 2 ) ;
385 flagRF.define( flag ) ;
386 }
387 else {
388 flagRF.define( flag( IPosition( 2, 0, 0 ), IPosition( 2, npol-1, nchan-1 ) ) ) ;
389 }
390 }
391 virtual void postData()
392 {
393 Matrix<Float> tmp( 2, nchan, 0.0 ) ;
394 Matrix<Complex> v( npol, nchan ) ;
395 if ( isFilled[0] ) {
396 tmp.row( 0 ) = data.row( 0 ) ;
397 v.row( 0 ) = RealToComplex( tmp ) ;
398 }
399 if ( isFilled[1] ) {
400 tmp.row( 0 ) = data.row( 1 ) ;
401 v.row( npol-1 ) = RealToComplex( tmp ) ;
402 }
403 if ( isFilled[2] && isFilled[3] ) {
404 tmp.row( 0 ) = data.row( 2 ) ;
405 tmp.row( 1 ) = data.row( 3 ) ;
406 v.row( 1 ) = RealToComplex( tmp ) ;
407 v.row( 2 ) = conj( v.row( 1 ) ) ;
408 }
409 dataRF.define( v ) ;
410 }
411private:
412 void attachData()
413 {
414 dataRF.attachToRecord( row.record(), "DATA" ) ;
415 }
416 RecordFieldPtr< Matrix<Complex> > dataRF;
417 Bool isFilled[4] ;
418};
419
420class BaseMSWriterVisitor: public TableVisitor {
421 const String *lastFieldName;
422 uInt lastRecordNo;
423 uInt lastBeamNo, lastScanNo, lastIfNo, lastPolNo;
424 Int lastSrcType;
425 uInt lastCycleNo;
426 Double lastTime;
427protected:
428 const Table &table;
429 uInt count;
430public:
431 BaseMSWriterVisitor(const Table &table)
432 : table(table)
433 {
434 static const String dummy;
435 lastFieldName = &dummy;
436 count = 0;
437 }
438
439 virtual void enterFieldName(const uInt recordNo, const String &columnValue) {
440 }
441 virtual void leaveFieldName(const uInt recordNo, const String &columnValue) {
442 }
443 virtual void enterBeamNo(const uInt recordNo, uInt columnValue) { }
444 virtual void leaveBeamNo(const uInt recordNo, uInt columnValue) { }
445 virtual void enterScanNo(const uInt recordNo, uInt columnValue) { }
446 virtual void leaveScanNo(const uInt recordNo, uInt columnValue) { }
447 virtual void enterIfNo(const uInt recordNo, uInt columnValue) { }
448 virtual void leaveIfNo(const uInt recordNo, uInt columnValue) { }
449 virtual void enterSrcType(const uInt recordNo, Int columnValue) { }
450 virtual void leaveSrcType(const uInt recordNo, Int columnValue) { }
451 virtual void enterCycleNo(const uInt recordNo, uInt columnValue) { }
452 virtual void leaveCycleNo(const uInt recordNo, uInt columnValue) { }
453 virtual void enterTime(const uInt recordNo, Double columnValue) { }
454 virtual void leaveTime(const uInt recordNo, Double columnValue) { }
455 virtual void enterPolNo(const uInt recordNo, uInt columnValue) { }
456 virtual void leavePolNo(const uInt recordNo, uInt columnValue) { }
457
458 virtual Bool visitRecord(const uInt recordNo,
459 const String &fieldName,
460 const uInt beamNo,
461 const uInt scanNo,
462 const uInt ifNo,
463 const Int srcType,
464 const uInt cycleNo,
465 const Double time,
466 const uInt polNo) { return True ;}
467
468 virtual Bool visit(Bool isFirst, const uInt recordNo,
469 const uInt nCols, void const *const colValues[]) {
470 const String *fieldName = NULL;
471 uInt beamNo, scanNo, ifNo;
472 Int srcType;
473 uInt cycleNo;
474 Double time;
475 uInt polNo;
476 { // prologue
477 uInt i = 0;
478 {
479 const String *col = (const String*)colValues[i++];
480 fieldName = &col[recordNo];
481 }
482 {
483 const uInt *col = (const uInt *)colValues[i++];
484 beamNo = col[recordNo];
485 }
486 {
487 const uInt *col = (const uInt *)colValues[i++];
488 scanNo = col[recordNo];
489 }
490 {
491 const uInt *col = (const uInt *)colValues[i++];
492 ifNo = col[recordNo];
493 }
494 {
495 const Int *col = (const Int *)colValues[i++];
496 srcType = col[recordNo];
497 }
498 {
499 const uInt *col = (const uInt *)colValues[i++];
500 cycleNo = col[recordNo];
501 }
502 {
503 const Double *col = (const Double *)colValues[i++];
504 time = col[recordNo];
505 }
506 {
507 const Int *col = (const Int *)colValues[i++];
508 polNo = col[recordNo];
509 }
510 assert(nCols == i);
511 }
512
513 if (isFirst) {
514 enterFieldName(recordNo, *fieldName);
515 enterBeamNo(recordNo, beamNo);
516 enterScanNo(recordNo, scanNo);
517 enterIfNo(recordNo, ifNo);
518 enterSrcType(recordNo, srcType);
519 enterCycleNo(recordNo, cycleNo);
520 enterTime(recordNo, time);
521 enterPolNo(recordNo, polNo);
522 } else {
523 if (lastFieldName->compare(*fieldName) != 0) {
524 leavePolNo(lastRecordNo, lastPolNo);
525 leaveTime(lastRecordNo, lastTime);
526 leaveCycleNo(lastRecordNo, lastCycleNo);
527 leaveSrcType(lastRecordNo, lastSrcType);
528 leaveIfNo(lastRecordNo, lastIfNo);
529 leaveScanNo(lastRecordNo, lastScanNo);
530 leaveBeamNo(lastRecordNo, lastBeamNo);
531 leaveFieldName(lastRecordNo, *lastFieldName);
532
533 enterFieldName(recordNo, *fieldName);
534 enterBeamNo(recordNo, beamNo);
535 enterScanNo(recordNo, scanNo);
536 enterIfNo(recordNo, ifNo);
537 enterSrcType(recordNo, srcType);
538 enterCycleNo(recordNo, cycleNo);
539 enterTime(recordNo, time);
540 enterPolNo(recordNo, polNo);
541 } else if (lastBeamNo != beamNo) {
542 leavePolNo(lastRecordNo, lastPolNo);
543 leaveTime(lastRecordNo, lastTime);
544 leaveCycleNo(lastRecordNo, lastCycleNo);
545 leaveSrcType(lastRecordNo, lastSrcType);
546 leaveIfNo(lastRecordNo, lastIfNo);
547 leaveScanNo(lastRecordNo, lastScanNo);
548 leaveBeamNo(lastRecordNo, lastBeamNo);
549
550 enterBeamNo(recordNo, beamNo);
551 enterScanNo(recordNo, scanNo);
552 enterIfNo(recordNo, ifNo);
553 enterSrcType(recordNo, srcType);
554 enterCycleNo(recordNo, cycleNo);
555 enterTime(recordNo, time);
556 enterPolNo(recordNo, polNo);
557 } else if (lastScanNo != scanNo) {
558 leavePolNo(lastRecordNo, lastPolNo);
559 leaveTime(lastRecordNo, lastTime);
560 leaveCycleNo(lastRecordNo, lastCycleNo);
561 leaveSrcType(lastRecordNo, lastSrcType);
562 leaveIfNo(lastRecordNo, lastIfNo);
563 leaveScanNo(lastRecordNo, lastScanNo);
564
565 enterScanNo(recordNo, scanNo);
566 enterIfNo(recordNo, ifNo);
567 enterSrcType(recordNo, srcType);
568 enterCycleNo(recordNo, cycleNo);
569 enterTime(recordNo, time);
570 enterPolNo(recordNo, polNo);
571 } else if (lastIfNo != ifNo) {
572 leavePolNo(lastRecordNo, lastPolNo);
573 leaveTime(lastRecordNo, lastTime);
574 leaveCycleNo(lastRecordNo, lastCycleNo);
575 leaveSrcType(lastRecordNo, lastSrcType);
576 leaveIfNo(lastRecordNo, lastIfNo);
577
578 enterIfNo(recordNo, ifNo);
579 enterSrcType(recordNo, srcType);
580 enterCycleNo(recordNo, cycleNo);
581 enterTime(recordNo, time);
582 enterPolNo(recordNo, polNo);
583 } else if (lastSrcType != srcType) {
584 leavePolNo(lastRecordNo, lastPolNo);
585 leaveTime(lastRecordNo, lastTime);
586 leaveCycleNo(lastRecordNo, lastCycleNo);
587 leaveSrcType(lastRecordNo, lastSrcType);
588
589 enterSrcType(recordNo, srcType);
590 enterCycleNo(recordNo, cycleNo);
591 enterTime(recordNo, time);
592 enterPolNo(recordNo, polNo);
593 } else if (lastCycleNo != cycleNo) {
594 leavePolNo(lastRecordNo, lastPolNo);
595 leaveTime(lastRecordNo, lastTime);
596 leaveCycleNo(lastRecordNo, lastCycleNo);
597
598 enterCycleNo(recordNo, cycleNo);
599 enterTime(recordNo, time);
600 enterPolNo(recordNo, polNo);
601 } else if (lastTime != time) {
602 leavePolNo(lastRecordNo, lastPolNo);
603 leaveTime(lastRecordNo, lastTime);
604
605 enterTime(recordNo, time);
606 enterPolNo(recordNo, polNo);
607 } else if (lastPolNo != polNo) {
608 leavePolNo(lastRecordNo, lastPolNo);
609 enterPolNo(recordNo, polNo);
610 }
611 }
612 count++;
613 Bool result = visitRecord(recordNo, *fieldName, beamNo, scanNo, ifNo, srcType,
614 cycleNo, time, polNo);
615
616 { // epilogue
617 lastRecordNo = recordNo;
618
619 lastFieldName = fieldName;
620 lastBeamNo = beamNo;
621 lastScanNo = scanNo;
622 lastIfNo = ifNo;
623 lastSrcType = srcType;
624 lastCycleNo = cycleNo;
625 lastTime = time;
626 lastPolNo = polNo;
627 }
628 return result ;
629 }
630
631 virtual void finish() {
632 if (count > 0) {
633 leavePolNo(lastRecordNo, lastPolNo);
634 leaveTime(lastRecordNo, lastTime);
635 leaveCycleNo(lastRecordNo, lastCycleNo);
636 leaveSrcType(lastRecordNo, lastSrcType);
637 leaveIfNo(lastRecordNo, lastIfNo);
638 leaveScanNo(lastRecordNo, lastScanNo);
639 leaveBeamNo(lastRecordNo, lastBeamNo);
640 leaveFieldName(lastRecordNo, *lastFieldName);
641 }
642 }
643};
644
645class MSWriterVisitor: public BaseMSWriterVisitor, public MSWriterUtils {
646public:
647 MSWriterVisitor(const Table &table, Table &mstable)
648 : BaseMSWriterVisitor(table),
649 ms(mstable)
650 {
651 rowidx = 0 ;
652 fieldName = "" ;
653 defaultFieldId = 0 ;
654 spwId = -1 ;
655 subscan = 1 ;
656 ptName = "" ;
657 srcId = 0 ;
658
659 row = TableRow( ms ) ;
660
661 initPolarization() ;
662 initFrequencies() ;
663
664 //
665 // add rows to MS
666 //
667 uInt addrow = table.nrow() ;
668 ms.addRow( addrow ) ;
669
670 // attach to Scantable columns
671 spectraCol.attach( table, "SPECTRA" ) ;
672 flagtraCol.attach( table, "FLAGTRA" ) ;
673 flagRowCol.attach( table, "FLAGROW" ) ;
674 tcalIdCol.attach( table, "TCAL_ID" ) ;
675 intervalCol.attach( table, "INTERVAL" ) ;
676 directionCol.attach( table, "DIRECTION" ) ;
677 scanRateCol.attach( table, "SCANRATE" ) ;
678 timeCol.attach( table, "TIME" ) ;
679 freqIdCol.attach( table, "FREQ_ID" ) ;
680 sourceNameCol.attach( table, "SRCNAME" ) ;
681 sourceDirectionCol.attach( table, "SRCDIRECTION" ) ;
682 fieldNameCol.attach( table, "FIELDNAME" ) ;
683
684 // MS subtables
685 attachSubtables() ;
686
687 // attach to MS columns
688 attachMain() ;
689 attachPointing() ;
690 }
691
692 virtual void enterFieldName(const uInt recordNo, const String &columnValue) {
693 //printf("%u: FieldName: %s\n", recordNo, columnValue.c_str());
694 fieldName = fieldNameCol.asString( recordNo ) ;
695 String::size_type pos = fieldName.find( "__" ) ;
696 if ( pos != String::npos ) {
697 fieldId = String::toInt( fieldName.substr( pos+2 ) ) ;
698 fieldName = fieldName.substr( 0, pos ) ;
699 }
700 else {
701 fieldId = defaultFieldId ;
702 defaultFieldId++ ;
703 }
704 Double tSec = timeCol.asdouble( recordNo ) * 86400.0 ;
705 Vector<Double> srcDir = sourceDirectionCol( recordNo ) ;
706 Vector<Double> srate = scanRateCol( recordNo ) ;
707 String srcName = sourceNameCol.asString( recordNo ) ;
708
709 addField( fieldId, fieldName, srcName, srcDir, srate, tSec ) ;
710
711 // put value
712 *fieldIdRF = fieldId ;
713 }
714 virtual void leaveFieldName(const uInt recordNo, const String &columnValue) {
715 }
716 virtual void enterBeamNo(const uInt recordNo, uInt columnValue) {
717 //printf("%u: BeamNo: %u\n", recordNo, columnValue);
718
719 feedId = (Int)columnValue ;
720
721 // put value
722 *feed1RF = feedId ;
723 *feed2RF = feedId ;
724 }
725 virtual void leaveBeamNo(const uInt recordNo, uInt columnValue) {
726 }
727 virtual void enterScanNo(const uInt recordNo, uInt columnValue) {
728 //printf("%u: ScanNo: %u\n", recordNo, columnValue);
729
730 // put value
731 // SCAN_NUMBER is 0-based in Scantable while 1-based in MS
732 *scanNumberRF = (Int)columnValue + 1 ;
733 }
734 virtual void leaveScanNo(const uInt recordNo, uInt columnValue) {
735 subscan = 1 ;
736 }
737 virtual void enterIfNo(const uInt recordNo, uInt columnValue) {
738 //printf("%u: IfNo: %u\n", recordNo, columnValue);
739
740 spwId = (Int)columnValue ;
741 uInt freqId = freqIdCol.asuInt( recordNo ) ;
742
743 Vector<Float> sp = spectraCol( recordNo ) ;
744 uInt nchan = sp.nelements() ;
745 holder->setNchan( nchan ) ;
746
747 addSpectralWindow( spwId, freqId ) ;
748
749 addFeed( feedId, spwId ) ;
750 }
751 virtual void leaveIfNo(const uInt recordNo, uInt columnValue) {
752 }
753 virtual void enterSrcType(const uInt recordNo, Int columnValue) {
754 //printf("%u: SrcType: %d\n", recordNo, columnValue);
755
756 Int stateId = addState( columnValue ) ;
757
758 // put value
759 *stateIdRF = stateId ;
760 }
761 virtual void leaveSrcType(const uInt recordNo, Int columnValue) {
762 }
763 virtual void enterCycleNo(const uInt recordNo, uInt columnValue) {
764 //printf("%u: CycleNo: %u\n", recordNo, columnValue);
765 }
766 virtual void leaveCycleNo(const uInt recordNo, uInt columnValue) {
767 }
768 virtual void enterTime(const uInt recordNo, Double columnValue) {
769 //printf("%u: Time: %f\n", recordNo, columnValue);
770
771 Double timeSec = columnValue * 86400.0 ;
772 Double interval = intervalCol.asdouble( recordNo ) ;
773
774 if ( ptName.empty() ) {
775 Vector<Double> dir = directionCol( recordNo ) ;
776 Vector<Double> rate = scanRateCol( recordNo ) ;
777 if ( anyNE( rate, 0.0 ) ) {
778 Matrix<Double> msdir( 2, 2 ) ;
779 msdir.column( 0 ) = dir ;
780 msdir.column( 1 ) = rate ;
781 addPointing( timeSec, interval, msdir ) ;
782 }
783 else {
784 Matrix<Double> msdir( 2, 1 ) ;
785 msdir.column( 0 ) = dir ;
786 addPointing( timeSec, interval, msdir ) ;
787 }
788 }
789
790 // put value
791 *timeRF = timeSec ;
792 *timeCentroidRF = timeSec ;
793 *intervalRF = interval ;
794 *exposureRF = interval ;
795 }
796 virtual void leaveTime(const uInt recordNo, Double columnValue) {
797 if ( holder->nPol() > 0 ) {
798 Int polId = addPolarization() ;
799 Int ddId = addDataDescription( polId, spwId ) ;
800
801 // put field
802 *dataDescIdRF = ddId ;
803 holder->post() ;
804
805 // commit row
806 row.put( rowidx ) ;
807 rowidx++ ;
808
809 // reset holder
810 holder->reset() ;
811 }
812 }
813 virtual void enterPolNo(const uInt recordNo, uInt columnValue) {
814 //printf("%u: PolNo: %d\n", recordNo, columnValue);
815 }
816 virtual void leavePolNo(const uInt recordNo, uInt columnValue) {
817 }
818
819 virtual Bool visitRecord(const uInt recordNo,
820 const String &fieldName,
821 const uInt beamNo,
822 const uInt scanNo,
823 const uInt ifNo,
824 const Int srcType,
825 const uInt cycleNo,
826 const Double time,
827 const uInt polNo) {
828 //printf("%u: %s, %u, %u, %u, %d, %u, %f, %d\n", recordNo,
829 // fieldName.c_str(), beamNo, scanNo, ifNo, srcType, cycleNo, time, polNo);
830
831 Vector<Float> sp = spectraCol( recordNo ) ;
832 Vector<uChar> tmp = flagtraCol( recordNo ) ;
833 Vector<Bool> fl( tmp.shape() ) ;
834 convertArray( fl, tmp ) ;
835 Bool flr = (Bool)flagRowCol.asuInt( recordNo ) ;
836 holder->accumulate( polNo, sp, fl, flr ) ;
837
838 return True ;
839 }
840
841 virtual void finish() {
842 BaseMSWriterVisitor::finish();
843 //printf("Total: %u\n", count);
844
845 // remove rows
846 if ( ms.nrow() > rowidx ) {
847 uInt numRemove = ms.nrow() - rowidx ;
848 //cout << "numRemove = " << numRemove << endl ;
849 Vector<uInt> rows( numRemove ) ;
850 indgen( rows, rowidx ) ;
851 ms.removeRow( rows ) ;
852 }
853
854 // fill empty SPECTRAL_WINDOW rows
855 infillSpectralWindow() ;
856 }
857
858 void dataColumnName( String name )
859 {
860 if ( name == "DATA" )
861 holder = new ComplexDataHolder( row, poltype ) ;
862 else if ( name == "FLOAT_DATA" )
863 holder = new FloatDataHolder( row, poltype ) ;
864 }
865 void pointingTableName( String name ) {
866 ptName = name ;
867 }
868 void setSourceRecord( Record &r ) {
869 srcRec = r ;
870 }
871private:
872 void addField( Int &fid, String &fname, String &srcName,
873 Vector<Double> &sdir, Vector<Double> &srate,
874 Double &tSec )
875 {
876 uInt nrow = fieldtab.nrow() ;
877 while( (Int)nrow <= fid ) {
878 fieldtab.addRow( 1, True ) ;
879 nrow++ ;
880 }
881
882 Matrix<Double> dir ;
883 Int numPoly = 0 ;
884 if ( anyNE( srate, 0.0 ) ) {
885 dir.resize( 2, 2 ) ;
886 dir.column( 0 ) = sdir ;
887 dir.column( 1 ) = srate ;
888 numPoly = 1 ;
889 }
890 else {
891 dir.resize( 2, 1 ) ;
892 dir.column( 0 ) = sdir ;
893 }
894 srcId = srcRec.asInt( srcName ) ;
895
896 TableRow tr( fieldtab ) ;
897 TableRecord &r = tr.record() ;
898 putField( "NAME", r, fname ) ;
899 putField( "NUM_POLY", r, numPoly ) ;
900 putField( "TIME", r, tSec ) ;
901 putField( "SOURCE_ID", r, srcId ) ;
902 defineField( "DELAY_DIR", r, dir ) ;
903 defineField( "REFERENCE_DIR", r, dir ) ;
904 defineField( "PHASE_DIR", r, dir ) ;
905 tr.put( fid ) ;
906
907 // for POINTING table
908 *poNameRF = fname ;
909 }
910 Int addState( Int &id )
911 {
912 String obsMode ;
913 Bool isSignal ;
914 Double tnoise ;
915 Double tload ;
916 queryType( id, obsMode, isSignal, tnoise, tload ) ;
917
918 String key = obsMode+"_"+String::toString( subscan ) ;
919 Int idx = -1 ;
920 uInt nEntry = stateEntry.nelements() ;
921 for ( uInt i = 0 ; i < nEntry ; i++ ) {
922 if ( stateEntry[i] == key ) {
923 idx = i ;
924 break ;
925 }
926 }
927 if ( idx == -1 ) {
928 uInt nrow = statetab.nrow() ;
929 statetab.addRow( 1, True ) ;
930 TableRow tr( statetab ) ;
931 TableRecord &r = tr.record() ;
932 putField( "OBS_MODE", r, obsMode ) ;
933 putField( "SIG", r, isSignal ) ;
934 isSignal = !isSignal ;
935 putField( "REF", r, isSignal ) ;
936 putField( "CAL", r, tnoise ) ;
937 putField( "LOAD", r, tload ) ;
938 tr.put( nrow ) ;
939 idx = nrow ;
940
941 stateEntry.resize( nEntry+1, True ) ;
942 stateEntry[nEntry] = key ;
943 }
944 subscan++ ;
945
946 return idx ;
947 }
948 void addPointing( Double &tSec, Double &interval, Matrix<Double> &dir )
949 {
950 uInt nrow = potab.nrow() ;
951 potab.addRow() ;
952
953 *poNumPolyRF = dir.ncolumn() - 1 ;
954 *poTimeRF = tSec ;
955 *poTimeOriginRF = tSec ;
956 *poIntervalRF = interval ;
957 poDirectionRF.define( dir ) ;
958 poTargetRF.define( dir ) ;
959 porow.put( nrow ) ;
960 }
961 Int addPolarization()
962 {
963 Int idx = -1 ;
964 Vector<Int> corrType = holder->corrTypeInt() ;
965 uInt nEntry = polEntry.size() ;
966 for ( uInt i = 0 ; i < nEntry ; i++ ) {
967 if ( polEntry[i].conform( corrType ) && allEQ( polEntry[i], corrType ) ) {
968 idx = i ;
969 break ;
970 }
971 }
972
973 Int numCorr = holder->nPol() ;
974 Matrix<Int> corrProduct = corrProductTemplate[numCorr] ;
975
976 if ( idx == -1 ) {
977 uInt nrow = poltab.nrow() ;
978 poltab.addRow( 1, True ) ;
979 TableRow tr( poltab ) ;
980 TableRecord &r = tr.record() ;
981 putField( "NUM_CORR", r, numCorr ) ;
982 defineField( "CORR_TYPE", r, corrType ) ;
983 defineField( "CORR_PRODUCT", r, corrProduct ) ;
984 tr.put( nrow ) ;
985 idx = nrow ;
986
987 polEntry.resize( nEntry+1 ) ;
988 polEntry[nEntry] = corrType ;
989 }
990
991 return idx ;
992 }
993 Int addDataDescription( Int pid, Int sid )
994 {
995 Int idx = -1 ;
996 uInt nItem = 2 ;
997 uInt len = ddEntry.nelements() ;
998 uInt nEntry = len / nItem ;
999 const Int *dd_p = ddEntry.storage() ;
1000 for ( uInt i = 0 ; i < nEntry ; i++ ) {
1001 Int pol = *dd_p ;
1002 dd_p++ ;
1003 Int spw = *dd_p ;
1004 dd_p++ ;
1005 if ( pid == pol && sid == spw ) {
1006 idx = i ;
1007 break ;
1008 }
1009 }
1010
1011 if ( idx == -1 ) {
1012 uInt nrow = ddtab.nrow() ;
1013 ddtab.addRow( 1, True ) ;
1014 TableRow tr( ddtab ) ;
1015 TableRecord &r = tr.record() ;
1016 putField( "POLARIZATION_ID", r, pid ) ;
1017 putField( "SPECTRAL_WINDOW_ID", r, sid ) ;
1018 tr.put( nrow ) ;
1019 idx = nrow ;
1020
1021 ddEntry.resize( len+nItem ) ;
1022 ddEntry[len] = pid ;
1023 ddEntry[len+1] = sid ;
1024 }
1025
1026 return idx ;
1027 }
1028 void infillSpectralWindow()
1029 {
1030 ROScalarColumn<Int> nchanCol( spwtab, "NUM_CHAN" ) ;
1031 Vector<Int> nchan = nchanCol.getColumn() ;
1032 TableRow tr( spwtab ) ;
1033 TableRecord &r = tr.record() ;
1034 Int mfr = 1 ;
1035 Vector<Double> dummy( 1, 0.0 ) ;
1036 putField( "MEAS_FREQ_REF", r, mfr ) ;
1037 defineField( "CHAN_FREQ", r, dummy ) ;
1038 defineField( "CHAN_WIDTH", r, dummy ) ;
1039 defineField( "EFFECTIVE_BW", r, dummy ) ;
1040 defineField( "RESOLUTION", r, dummy ) ;
1041
1042 for ( uInt i = 0 ; i < spwtab.nrow() ; i++ ) {
1043 if ( nchan[i] == 0 )
1044 tr.put( i ) ;
1045 }
1046 }
1047 void addSpectralWindow( Int sid, uInt fid )
1048 {
1049 if ( !processedFreqId[fid] ) {
1050 uInt nrow = spwtab.nrow() ;
1051 while( (Int)nrow <= sid ) {
1052 spwtab.addRow( 1, True ) ;
1053 nrow++ ;
1054 }
1055 processedFreqId[fid] = True ;
1056 }
1057 else {
1058 return ;
1059 }
1060
1061 Double rp = refpix[fid] ;
1062 Double rv = refval[fid] ;
1063 Double ic = increment[fid] ;
1064
1065 Int mfrInt = (Int)freqframe ;
1066 Int nchan = holder->nChan() ;
1067 Double bw = nchan * abs( ic ) ;
1068 Double reffreq = rv - rp * ic ;
1069 Int netsb = 0 ; // USB->0, LSB->1
1070 if ( ic < 0 )
1071 netsb = 1 ;
1072 Vector<Double> res( nchan, abs(ic) ) ;
1073 Vector<Double> cw( nchan, ic ) ;
1074 Vector<Double> chanf( nchan ) ;
1075 indgen( chanf, reffreq, ic ) ;
1076
1077 TableRow tr( spwtab ) ;
1078 TableRecord &r = tr.record() ;
1079 putField( "MEAS_FREQ_REF", r, mfrInt ) ;
1080 putField( "NUM_CHAN", r, nchan ) ;
1081 putField( "TOTAL_BANDWIDTH", r, bw ) ;
1082 putField( "REF_FREQUENCY", r, reffreq ) ;
1083 putField( "NET_SIDEBAND", r, netsb ) ;
1084 defineField( "RESOLUTION", r, res ) ;
1085// defineField( "CHAN_WIDTH", r, res ) ;
1086 defineField( "CHAN_WIDTH", r, cw ) ;
1087 defineField( "EFFECTIVE_BW", r, res ) ;
1088 defineField( "CHAN_FREQ", r, chanf ) ;
1089 tr.put( sid ) ;
1090 }
1091 void addFeed( Int fid, Int sid )
1092 {
1093 Int idx = -1 ;
1094 uInt nItem = 2 ;
1095 uInt len = feedEntry.nelements() ;
1096 uInt nEntry = len / nItem ;
1097 const Int *fe_p = feedEntry.storage() ;
1098 for ( uInt i = 0 ; i < nEntry ; i++ ) {
1099 Int feed = *fe_p ;
1100 fe_p++ ;
1101 Int spw = *fe_p ;
1102 fe_p++ ;
1103 if ( fid == feed && sid == spw ) {
1104 idx = i ;
1105 break ;
1106 }
1107 }
1108
1109
1110 if ( idx == -1 ) {
1111 uInt nrow = feedtab.nrow() ;
1112 feedtab.addRow( 1, True ) ;
1113 Int numReceptors = 2 ;
1114 Vector<String> polType( numReceptors ) ;
1115 Matrix<Double> beamOffset( 2, numReceptors, 0.0 ) ;
1116 Vector<Double> receptorAngle( numReceptors, 0.0 ) ;
1117 if ( poltype == "linear" ) {
1118 polType[0] = "X" ;
1119 polType[1] = "Y" ;
1120 }
1121 else if ( poltype == "circular" ) {
1122 polType[0] = "R" ;
1123 polType[1] = "L" ;
1124 }
1125 else {
1126 polType[0] = "X" ;
1127 polType[1] = "Y" ;
1128 }
1129 Matrix<Complex> polResponse( numReceptors, numReceptors, 0.0 ) ;
1130
1131 TableRow tr( feedtab ) ;
1132 TableRecord &r = tr.record() ;
1133 putField( "FEED_ID", r, fid ) ;
1134 putField( "BEAM_ID", r, fid ) ;
1135 Int tmp = 0 ;
1136 putField( "ANTENNA_ID", r, tmp ) ;
1137 putField( "SPECTRAL_WINDOW_ID", r, sid ) ;
1138 putField( "NUM_RECEPTORS", r, numReceptors ) ;
1139 defineField( "POLARIZATION_TYPE", r, polType ) ;
1140 defineField( "BEAM_OFFSET", r, beamOffset ) ;
1141 defineField( "RECEPTOR_ANGLE", r, receptorAngle ) ;
1142 defineField( "POL_RESPONSE", r, polResponse ) ;
1143 tr.put( nrow ) ;
1144
1145 feedEntry.resize( len+nItem ) ;
1146 feedEntry[len] = fid ;
1147 feedEntry[len+1] = sid ;
1148 }
1149 }
1150 void initPolarization()
1151 {
1152 const TableRecord &keys = table.keywordSet() ;
1153 poltype = keys.asString( "POLTYPE" ) ;
1154
1155 initCorrProductTemplate() ;
1156 }
1157 void initFrequencies()
1158 {
1159 const TableRecord &keys = table.keywordSet() ;
1160 Table tab = keys.asTable( "FREQUENCIES" ) ;
1161 ROScalarColumn<uInt> idcol( tab, "ID" ) ;
1162 ROScalarColumn<Double> rpcol( tab, "REFPIX" ) ;
1163 ROScalarColumn<Double> rvcol( tab, "REFVAL" ) ;
1164 ROScalarColumn<Double> iccol( tab, "INCREMENT" ) ;
1165 Vector<uInt> id = idcol.getColumn() ;
1166 Vector<Double> rp = rpcol.getColumn() ;
1167 Vector<Double> rv = rvcol.getColumn() ;
1168 Vector<Double> ic = iccol.getColumn() ;
1169 for ( uInt i = 0 ; i < id.nelements() ; i++ ) {
1170 processedFreqId.insert( pair<uInt,Bool>( id[i], False ) ) ;
1171 refpix.insert( pair<uInt,Double>( id[i], rp[i] ) ) ;
1172 refval.insert( pair<uInt,Double>( id[i], rv[i] ) ) ;
1173 increment.insert( pair<uInt,Double>( id[i], ic[i] ) ) ;
1174 }
1175 String frameStr = tab.keywordSet().asString( "BASEFRAME" ) ;
1176 MFrequency::getType( freqframe, frameStr ) ;
1177 }
1178 void attachSubtables()
1179 {
1180 const TableRecord &keys = table.keywordSet() ;
1181 TableRecord &mskeys = ms.rwKeywordSet() ;
1182
1183 // FIELD table
1184 fieldtab = mskeys.asTable( "FIELD" ) ;
1185
1186 // SPECTRAL_WINDOW table
1187 spwtab = mskeys.asTable( "SPECTRAL_WINDOW" ) ;
1188
1189 // POINTING table
1190 potab = mskeys.asTable( "POINTING" ) ;
1191
1192 // POLARIZATION table
1193 poltab = mskeys.asTable( "POLARIZATION" ) ;
1194
1195 // DATA_DESCRIPTION table
1196 ddtab = mskeys.asTable( "DATA_DESCRIPTION" ) ;
1197
1198 // STATE table
1199 statetab = mskeys.asTable( "STATE" ) ;
1200
1201 // FEED table
1202 feedtab = mskeys.asTable( "FEED" ) ;
1203 }
1204 void attachMain()
1205 {
1206 TableRecord &r = row.record() ;
1207 dataDescIdRF.attachToRecord( r, "DATA_DESC_ID" ) ;
1208 timeRF.attachToRecord( r, "TIME" ) ;
1209 timeCentroidRF.attachToRecord( r, "TIME_CENTROID" ) ;
1210 intervalRF.attachToRecord( r, "INTERVAL" ) ;
1211 exposureRF.attachToRecord( r, "EXPOSURE" ) ;
1212 fieldIdRF.attachToRecord( r, "FIELD_ID" ) ;
1213 feed1RF.attachToRecord( r, "FEED1" ) ;
1214 feed2RF.attachToRecord( r, "FEED2" ) ;
1215 scanNumberRF.attachToRecord( r, "SCAN_NUMBER" ) ;
1216 stateIdRF.attachToRecord( r, "STATE_ID" ) ;
1217
1218 // constant values
1219 Int id = 0 ;
1220 RecordFieldPtr<Int> intRF( r, "OBSERVATION_ID" ) ;
1221 *intRF = 0 ;
1222 intRF.attachToRecord( r, "ANTENNA1" ) ;
1223 *intRF = 0 ;
1224 intRF.attachToRecord( r, "ANTENNA2" ) ;
1225 *intRF = 0 ;
1226 intRF.attachToRecord( r, "ARRAY_ID" ) ;
1227 *intRF = 0 ;
1228 intRF.attachToRecord( r, "PROCESSOR_ID" ) ;
1229 *intRF = 0 ;
1230 RecordFieldPtr< Vector<Double> > arrayRF( r, "UVW" ) ;
1231 arrayRF.define( Vector<Double>( 3, 0.0 ) ) ;
1232 }
1233 void attachPointing()
1234 {
1235 porow = TableRow( potab ) ;
1236 TableRecord &r = porow.record() ;
1237 poNumPolyRF.attachToRecord( r, "NUM_POLY" ) ;
1238 poTimeRF.attachToRecord( r, "TIME" ) ;
1239 poTimeOriginRF.attachToRecord( r, "TIME_ORIGIN" ) ;
1240 poIntervalRF.attachToRecord( r, "INTERVAL" ) ;
1241 poNameRF.attachToRecord( r, "NAME" ) ;
1242 poDirectionRF.attachToRecord( r, "DIRECTION" ) ;
1243 poTargetRF.attachToRecord( r, "TARGET" ) ;
1244
1245 // constant values
1246 RecordFieldPtr<Int> antIdRF( r, "ANTENNA_ID" ) ;
1247 *antIdRF = 0 ;
1248 RecordFieldPtr<Bool> trackingRF( r, "TRACKING" ) ;
1249 *trackingRF = True ;
1250 }
1251 void queryType( Int type, String &stype, Bool &b, Double &t, Double &l )
1252 {
1253 t = 0.0 ;
1254 l = 0.0 ;
1255
1256 String sep1="#" ;
1257 String sep2="," ;
1258 String target="OBSERVE_TARGET" ;
1259 String atmcal="CALIBRATE_TEMPERATURE" ;
1260 String onstr="ON_SOURCE" ;
1261 String offstr="OFF_SOURCE" ;
1262 String pswitch="POSITION_SWITCH" ;
1263 String nod="NOD" ;
1264 String fswitch="FREQUENCY_SWITCH" ;
1265 String sigstr="SIG" ;
1266 String refstr="REF" ;
1267 String unspecified="UNSPECIFIED" ;
1268 String ftlow="LOWER" ;
1269 String fthigh="HIGHER" ;
1270 switch ( type ) {
1271 case SrcType::PSON:
1272 stype = target+sep1+onstr+sep2+pswitch ;
1273 b = True ;
1274 break ;
1275 case SrcType::PSOFF:
1276 stype = target+sep1+offstr+sep2+pswitch ;
1277 b = False ;
1278 break ;
1279 case SrcType::NOD:
1280 stype = target+sep1+onstr+sep2+nod ;
1281 b = True ;
1282 break ;
1283 case SrcType::FSON:
1284 stype = target+sep1+onstr+sep2+fswitch+sep1+sigstr ;
1285 b = True ;
1286 break ;
1287 case SrcType::FSOFF:
1288 stype = target+sep1+onstr+sep2+fswitch+sep1+refstr ;
1289 b = False ;
1290 break ;
1291 case SrcType::SKY:
1292 stype = atmcal+sep1+offstr+sep2+unspecified ;
1293 b = False ;
1294 break ;
1295 case SrcType::HOT:
1296 stype = atmcal+sep1+offstr+sep2+unspecified ;
1297 b = False ;
1298 break ;
1299 case SrcType::WARM:
1300 stype = atmcal+sep1+offstr+sep2+unspecified ;
1301 b = False ;
1302 break ;
1303 case SrcType::COLD:
1304 stype = atmcal+sep1+offstr+sep2+unspecified ;
1305 b = False ;
1306 break ;
1307 case SrcType::PONCAL:
1308 stype = atmcal+sep1+onstr+sep2+pswitch ;
1309 b = True ;
1310 break ;
1311 case SrcType::POFFCAL:
1312 stype = atmcal+sep1+offstr+sep2+pswitch ;
1313 b = False ;
1314 break ;
1315 case SrcType::NODCAL:
1316 stype = atmcal+sep1+onstr+sep2+nod ;
1317 b = True ;
1318 break ;
1319 case SrcType::FONCAL:
1320 stype = atmcal+sep1+onstr+sep2+fswitch+sep1+sigstr ;
1321 b = True ;
1322 break ;
1323 case SrcType::FOFFCAL:
1324 stype = atmcal+sep1+offstr+sep2+fswitch+sep1+refstr ;
1325 b = False ;
1326 break ;
1327 case SrcType::FSLO:
1328 stype = target+sep1+onstr+sep2+fswitch+sep1+ftlow ;
1329 b = True ;
1330 break ;
1331 case SrcType::FLOOFF:
1332 stype = target+sep1+offstr+sep2+fswitch+sep1+ftlow ;
1333 b = False ;
1334 break ;
1335 case SrcType::FLOSKY:
1336 stype = atmcal+sep1+offstr+sep2+fswitch+sep1+ftlow ;
1337 b = False ;
1338 break ;
1339 case SrcType::FLOHOT:
1340 stype = atmcal+sep1+offstr+sep2+fswitch+sep1+ftlow ;
1341 b = False ;
1342 break ;
1343 case SrcType::FLOWARM:
1344 stype = atmcal+sep1+offstr+sep2+fswitch+sep1+ftlow ;
1345 b = False ;
1346 break ;
1347 case SrcType::FLOCOLD:
1348 stype = atmcal+sep1+offstr+sep2+fswitch+sep1+ftlow ;
1349 b = False ;
1350 break ;
1351 case SrcType::FSHI:
1352 stype = target+sep1+onstr+sep2+fswitch+sep1+fthigh ;
1353 b = True ;
1354 break ;
1355 case SrcType::FHIOFF:
1356 stype = target+sep1+offstr+sep2+fswitch+sep1+fthigh ;
1357 b = False ;
1358 break ;
1359 case SrcType::FHISKY:
1360 stype = atmcal+sep1+offstr+sep2+fswitch+sep1+fthigh ;
1361 b = False ;
1362 break ;
1363 case SrcType::FHIHOT:
1364 stype = atmcal+sep1+offstr+sep2+fswitch+sep1+fthigh ;
1365 b = False ;
1366 break ;
1367 case SrcType::FHIWARM:
1368 stype = atmcal+sep1+offstr+sep2+fswitch+sep1+fthigh ;
1369 b = False ;
1370 break ;
1371 case SrcType::FHICOLD:
1372 stype = atmcal+sep1+offstr+sep2+fswitch+sep1+fthigh ;
1373 b = False ;
1374 break ;
1375 case SrcType::SIG:
1376 stype = target+sep1+onstr+sep2+unspecified ;
1377 b = True ;
1378 break ;
1379 case SrcType::REF:
1380 stype = target+sep1+offstr+sep2+unspecified ;
1381 b = False ;
1382 break ;
1383 default:
1384 stype = unspecified ;
1385 b = True ;
1386 break ;
1387 }
1388 }
1389 void initCorrProductTemplate()
1390 {
1391 Int n = 1 ;
1392 {
1393 Matrix<Int> c( 2, n, 0 ) ;
1394 corrProductTemplate[n] = c ;
1395 }
1396 n = 2 ;
1397 {
1398 Matrix<Int> c( 2, n, 0 ) ;
1399 c.column( 1 ) = 1 ;
1400 corrProductTemplate[n] = c ;
1401 }
1402 n = 4 ;
1403 {
1404 Matrix<Int> c( 2, n, 0 ) ;
1405 c( 0, 2 ) = 1 ;
1406 c( 0, 3 ) = 1 ;
1407 c( 1, 1 ) = 1 ;
1408 c( 1, 3 ) = 1 ;
1409 corrProductTemplate[n] = c ;
1410 }
1411 }
1412
1413 Table &ms;
1414 TableRow row;
1415 uInt rowidx;
1416 String fieldName;
1417 Int fieldId;
1418 Int srcId;
1419 Int defaultFieldId;
1420 Int spwId;
1421 Int feedId;
1422 Int subscan;
1423 CountedPtr<DataHolder> holder;
1424 String ptName;
1425 Bool useFloat;
1426 String poltype;
1427
1428 // MS subtables
1429 Table spwtab;
1430 Table statetab;
1431 Table ddtab;
1432 Table poltab;
1433 Table fieldtab;
1434 Table feedtab;
1435 Table potab;
1436
1437 // Scantable MAIN columns
1438 ROArrayColumn<Float> spectraCol;
1439 ROArrayColumn<Double> directionCol,scanRateCol,sourceDirectionCol;
1440 ROArrayColumn<uChar> flagtraCol;
1441 ROTableColumn tcalIdCol,intervalCol,flagRowCol,timeCol,freqIdCol,
1442 sourceNameCol,fieldNameCol;
1443
1444 // MS MAIN columns
1445 RecordFieldPtr<Int> dataDescIdRF,fieldIdRF,feed1RF,feed2RF,
1446 scanNumberRF,stateIdRF;
1447 RecordFieldPtr<Double> timeRF,timeCentroidRF,intervalRF,exposureRF;
1448
1449 // MS POINTING columns
1450 TableRow porow;
1451 RecordFieldPtr<Int> poNumPolyRF ;
1452 RecordFieldPtr<Double> poTimeRF,
1453 poTimeOriginRF,
1454 poIntervalRF ;
1455 RecordFieldPtr<String> poNameRF ;
1456 RecordFieldPtr< Matrix<Double> > poDirectionRF,
1457 poTargetRF ;
1458
1459 Vector<String> stateEntry;
1460 Block<Int> ddEntry;
1461 Block<Int> feedEntry;
1462 vector< Vector<Int> > polEntry;
1463 map<uInt,Bool> processedFreqId;
1464 map<uInt,Double> refpix;
1465 map<uInt,Double> refval;
1466 map<uInt,Double> increment;
1467 MFrequency::Types freqframe;
1468 Record srcRec;
1469 map< Int, Matrix<Int> > corrProductTemplate;
1470};
1471
1472class BaseMSSysCalVisitor: public TableVisitor {
1473 uInt lastRecordNo;
1474 uInt lastBeamNo, lastIfNo, lastPolNo;
1475 Double lastTime;
1476protected:
1477 const Table &table;
1478 uInt count;
1479public:
1480 BaseMSSysCalVisitor(const Table &table)
1481 : table(table)
1482 {
1483 count = 0;
1484 }
1485
1486 virtual void enterBeamNo(const uInt recordNo, uInt columnValue) { }
1487 virtual void leaveBeamNo(const uInt recordNo, uInt columnValue) { }
1488 virtual void enterIfNo(const uInt recordNo, uInt columnValue) { }
1489 virtual void leaveIfNo(const uInt recordNo, uInt columnValue) { }
1490 virtual void enterPolNo(const uInt recordNo, uInt columnValue) { }
1491 virtual void leavePolNo(const uInt recordNo, uInt columnValue) { }
1492 virtual void enterTime(const uInt recordNo, Double columnValue) { }
1493 virtual void leaveTime(const uInt recordNo, Double columnValue) { }
1494
1495 virtual Bool visitRecord(const uInt recordNo,
1496 const uInt beamNo,
1497 const uInt ifNo,
1498 const uInt polNo,
1499 const Double time) { return True ;}
1500
1501 virtual Bool visit(Bool isFirst, const uInt recordNo,
1502 const uInt nCols, void const *const colValues[]) {
1503 uInt beamNo, ifNo, polNo;
1504 Double time;
1505 { // prologue
1506 uInt i = 0;
1507 {
1508 const uInt *col = (const uInt *)colValues[i++];
1509 beamNo = col[recordNo];
1510 }
1511 {
1512 const uInt *col = (const uInt *)colValues[i++];
1513 ifNo = col[recordNo];
1514 }
1515 {
1516 const Double *col = (const Double *)colValues[i++];
1517 time = col[recordNo];
1518 }
1519 {
1520 const uInt *col = (const uInt *)colValues[i++];
1521 polNo = col[recordNo];
1522 }
1523 assert(nCols == i);
1524 }
1525
1526 if (isFirst) {
1527 enterBeamNo(recordNo, beamNo);
1528 enterIfNo(recordNo, ifNo);
1529 enterTime(recordNo, time);
1530 enterPolNo(recordNo, polNo);
1531 } else {
1532 if (lastBeamNo != beamNo) {
1533 leavePolNo(lastRecordNo, lastPolNo);
1534 leaveTime(lastRecordNo, lastTime);
1535 leaveIfNo(lastRecordNo, lastIfNo);
1536 leaveBeamNo(lastRecordNo, lastBeamNo);
1537
1538 enterBeamNo(recordNo, beamNo);
1539 enterIfNo(recordNo, ifNo);
1540 enterTime(recordNo, time);
1541 enterPolNo(recordNo, polNo);
1542 } else if (lastIfNo != ifNo) {
1543 leavePolNo(lastRecordNo, lastPolNo);
1544 leaveTime(lastRecordNo, lastTime);
1545 leaveIfNo(lastRecordNo, lastIfNo);
1546
1547 enterIfNo(recordNo, ifNo);
1548 enterTime(recordNo, time);
1549 enterPolNo(recordNo, polNo);
1550 } else if (lastTime != time) {
1551 leavePolNo(lastRecordNo, lastPolNo);
1552 leaveTime(lastRecordNo, lastTime);
1553
1554 enterTime(recordNo, time);
1555 enterPolNo(recordNo, polNo);
1556 } else if (lastPolNo != polNo) {
1557 leavePolNo(lastRecordNo, lastPolNo);
1558 enterPolNo(recordNo, polNo);
1559 }
1560 }
1561 count++;
1562 Bool result = visitRecord(recordNo, beamNo, ifNo, polNo, time);
1563
1564 { // epilogue
1565 lastRecordNo = recordNo;
1566
1567 lastBeamNo = beamNo;
1568 lastIfNo = ifNo;
1569 lastPolNo = polNo;
1570 lastTime = time;
1571 }
1572 return result ;
1573 }
1574
1575 virtual void finish() {
1576 if (count > 0) {
1577 leavePolNo(lastRecordNo, lastPolNo);
1578 leaveTime(lastRecordNo, lastTime);
1579 leaveIfNo(lastRecordNo, lastIfNo);
1580 leaveBeamNo(lastRecordNo, lastBeamNo);
1581 }
1582 }
1583};
1584
1585class BaseTsysHolder
1586{
1587public:
1588 BaseTsysHolder( ROArrayColumn<Float> &tsysCol )
1589 : col( tsysCol ),
1590 nchan(0)
1591 {
1592 reset() ;
1593 }
1594 virtual ~BaseTsysHolder() {}
1595 virtual Array<Float> getTsys() = 0 ;
1596 void setNchan( uInt n ) { nchan = n ; }
1597 void appendTsys( uInt row )
1598 {
1599 Vector<Float> v = col( row ) ;
1600 uInt len = tsys.nrow() ;
1601 tsys.resize( len+1, nchan, True ) ;
1602 if ( v.nelements() == nchan )
1603 tsys.row( len ) = v ;
1604 else
1605 tsys.row( len ) = v[0] ;
1606 }
1607 void setTsys( uInt row, uInt idx )
1608 {
1609 if ( idx >= nrow() )
1610 appendTsys( row ) ;
1611 else {
1612 Vector<Float> v = col( row ) ;
1613 if ( v.nelements() == nchan )
1614 tsys.row( idx ) = v ;
1615 else
1616 tsys.row( idx ) = v[0] ;
1617 }
1618 }
1619 void reset()
1620 {
1621 tsys.resize() ;
1622 }
1623 uInt nrow() { return tsys.nrow() ; }
1624 Bool isEffective()
1625 {
1626 return ( !(tsys.empty()) && anyNE( tsys, (Float)1.0 ) ) ;
1627 }
1628 BaseTsysHolder &operator= ( const BaseTsysHolder &v )
1629 {
1630 if ( this != &v )
1631 tsys.assign( v.tsys ) ;
1632 return *this ;
1633 }
1634protected:
1635 ROArrayColumn<Float> col ;
1636 Matrix<Float> tsys ;
1637 uInt nchan ;
1638};
1639
1640class TsysHolder : public BaseTsysHolder
1641{
1642public:
1643 TsysHolder( ROArrayColumn<Float> &tsysCol )
1644 : BaseTsysHolder( tsysCol )
1645 {}
1646 virtual ~TsysHolder() {}
1647 virtual Array<Float> getTsys()
1648 {
1649 return tsys.column( 0 ) ;
1650 }
1651};
1652
1653class TsysSpectrumHolder : public BaseTsysHolder
1654{
1655public:
1656 TsysSpectrumHolder( ROArrayColumn<Float> &tsysCol )
1657 : BaseTsysHolder( tsysCol )
1658 {}
1659 virtual ~TsysSpectrumHolder() {}
1660 virtual Array<Float> getTsys()
1661 {
1662 return tsys ;
1663 }
1664};
1665
1666class BaseTcalProcessor
1667{
1668public:
1669 BaseTcalProcessor( ROArrayColumn<Float> &tcalCol )
1670 : col( tcalCol )
1671 {}
1672 virtual ~BaseTcalProcessor() {}
1673 void setTcalId( Vector<uInt> &tcalId ) { id.assign( tcalId ) ; }
1674 virtual Array<Float> getTcal() = 0 ;
1675protected:
1676 ROArrayColumn<Float> col ;
1677 Vector<uInt> id ;
1678};
1679
1680class TcalProcessor : public BaseTcalProcessor
1681{
1682public:
1683 TcalProcessor( ROArrayColumn<Float> &tcalCol )
1684 : BaseTcalProcessor( tcalCol )
1685 {}
1686 virtual ~TcalProcessor() {}
1687 virtual Array<Float> getTcal()
1688 {
1689 uInt npol = id.nelements() ;
1690 Vector<Float> tcal( npol ) ;
1691 for ( uInt ipol = 0 ; ipol < npol ; ipol++ )
1692 tcal[ipol] = col( id[ipol] ).data()[0] ;
1693 //cout << "TcalProcessor: tcal = " << tcal << endl ;
1694 return tcal ;
1695 }
1696};
1697
1698class TcalSpectrumProcessor : public BaseTcalProcessor
1699{
1700public:
1701 TcalSpectrumProcessor( ROArrayColumn<Float> &tcalCol )
1702 : BaseTcalProcessor( tcalCol )
1703 {}
1704 virtual ~TcalSpectrumProcessor() {}
1705 virtual Array<Float> getTcal()
1706 {
1707 uInt npol = id.nelements() ;
1708 Vector<Float> tcal0 = col( 0 ) ;
1709 uInt nchan = tcal0.nelements() ;
1710 Matrix<Float> tcal( npol, nchan ) ;
1711 tcal.row( 0 ) = tcal0 ;
1712 for ( uInt ipol = 1 ; ipol < npol ; ipol++ )
1713 tcal.row( ipol ) = col( id[ipol] ) ;
1714 return tcal ;
1715 }
1716};
1717
1718class MSSysCalVisitor : public BaseMSSysCalVisitor
1719{
1720public:
1721 MSSysCalVisitor( const Table &from, Table &to )
1722 : BaseMSSysCalVisitor( from ),
1723 sctab( to ),
1724 rowidx( 0 )
1725 {
1726 scrow = TableRow( sctab ) ;
1727
1728 lastTcalId.resize() ;
1729 theTcalId.resize() ;
1730 startTime = 0.0 ;
1731 endTime = 0.0 ;
1732
1733 const TableRecord &keys = table.keywordSet() ;
1734 Table tcalTable = keys.asTable( "TCAL" ) ;
1735 tcalCol.attach( tcalTable, "TCAL" ) ;
1736 tsysCol.attach( table, "TSYS" ) ;
1737 tcalIdCol.attach( table, "TCAL_ID" ) ;
1738 intervalCol.attach( table, "INTERVAL" ) ;
1739 effectiveTcal.resize( tcalTable.nrow() ) ;
1740 for ( uInt irow = 0 ; irow < tcalTable.nrow() ; irow++ ) {
1741 if ( allEQ( tcalCol( irow ), (Float)1.0 ) )
1742 effectiveTcal[irow] = False ;
1743 else
1744 effectiveTcal[irow] = True ;
1745 }
1746
1747 TableRecord &r = scrow.record() ;
1748 RecordFieldPtr<Int> antennaIdRF( r, "ANTENNA_ID" ) ;
1749 *antennaIdRF = 0 ;
1750 feedIdRF.attachToRecord( r, "FEED_ID" ) ;
1751 specWinIdRF.attachToRecord( r, "SPECTRAL_WINDOW_ID" ) ;
1752 timeRF.attachToRecord( r, "TIME" ) ;
1753 intervalRF.attachToRecord( r, "INTERVAL" ) ;
1754 if ( r.isDefined( "TCAL" ) ) {
1755 tcalRF.attachToRecord( r, "TCAL" ) ;
1756 tcalProcessor = new TcalProcessor( tcalCol ) ;
1757 }
1758 else if ( r.isDefined( "TCAL_SPECTRUM" ) ) {
1759 tcalRF.attachToRecord( r, "TCAL_SPECTRUM" ) ;
1760 tcalProcessor = new TcalSpectrumProcessor( tcalCol ) ;
1761 }
1762 if ( r.isDefined( "TSYS" ) ) {
1763 tsysRF.attachToRecord( r, "TSYS" ) ;
1764 theTsys = new TsysHolder( tsysCol ) ;
1765 lastTsys = new TsysHolder( tsysCol ) ;
1766 }
1767 else {
1768 tsysRF.attachToRecord( r, "TSYS_SPECTRUM" ) ;
1769 theTsys = new TsysSpectrumHolder( tsysCol ) ;
1770 lastTsys = new TsysSpectrumHolder( tsysCol ) ;
1771 }
1772
1773 }
1774
1775 virtual void enterBeamNo(const uInt recordNo, uInt columnValue)
1776 {
1777 *feedIdRF = (Int)columnValue ;
1778 }
1779 virtual void leaveBeamNo(const uInt recordNo, uInt columnValue)
1780 {
1781 }
1782 virtual void enterIfNo(const uInt recordNo, uInt columnValue)
1783 {
1784 //cout << "enterIfNo" << endl ;
1785 ROArrayColumn<Float> sp( table, "SPECTRA" ) ;
1786 uInt nchan = sp( recordNo ).nelements() ;
1787 theTsys->setNchan( nchan ) ;
1788 lastTsys->setNchan( nchan ) ;
1789
1790 *specWinIdRF = (Int)columnValue ;
1791 }
1792 virtual void leaveIfNo(const uInt recordNo, uInt columnValue)
1793 {
1794 //cout << "leaveIfNo" << endl ;
1795 post() ;
1796 lastTsys->reset() ;
1797 lastTcalId.resize() ;
1798 theTsys->reset() ;
1799 theTcalId.resize() ;
1800 startTime = 0.0 ;
1801 endTime = 0.0 ;
1802 }
1803 virtual void enterTime(const uInt recordNo, Double columnValue)
1804 {
1805 //cout << "enterTime" << endl ;
1806 interval = intervalCol.asdouble( recordNo ) ;
1807 // start time and end time
1808 if ( startTime == 0.0 ) {
1809 startTime = columnValue * 86400.0 - 0.5 * interval ;
1810 endTime = columnValue * 86400.0 + 0.5 * interval ;
1811 }
1812 }
1813 virtual void leaveTime(const uInt recordNo, Double columnValue)
1814 {
1815 //cout << "leaveTime" << endl ;
1816 if ( isUpdated() ) {
1817 post() ;
1818 *lastTsys = *theTsys ;
1819 lastTcalId = theTcalId ;
1820 theTsys->reset() ;
1821 theTcalId.resize() ;
1822 startTime = columnValue * 86400.0 - 0.5 * interval ;
1823 endTime = columnValue * 86400.0 + 0.5 * interval ;
1824 }
1825 else {
1826 endTime = columnValue * 86400.0 + 0.5 * interval ;
1827 }
1828 }
1829 virtual void enterPolNo(const uInt recordNo, uInt columnValue)
1830 {
1831 //cout << "enterPolNo" << endl ;
1832 Vector<Float> tsys = tsysCol( recordNo ) ;
1833 uInt tcalId = tcalIdCol.asuInt( recordNo ) ;
1834 // lastTsys.nrow() must be npol
1835 if ( lastTsys->nrow() == columnValue )
1836 lastTsys->appendTsys( recordNo ) ;
1837 // lastTcalId.nelements() must be npol
1838 if ( lastTcalId.nelements() == columnValue )
1839 appendTcalId( lastTcalId, tcalId, columnValue ) ;
1840 // theTsys.nrow() must be npol
1841 if ( theTsys->nrow() == columnValue )
1842 theTsys->appendTsys( recordNo ) ;
1843 else {
1844 theTsys->setTsys( recordNo, columnValue ) ;
1845 }
1846 if ( theTcalId.nelements() == columnValue )
1847 appendTcalId( theTcalId, tcalId, columnValue ) ;
1848 else
1849 setTcalId( theTcalId, tcalId, columnValue ) ;
1850 }
1851 virtual void leavePolNo( const uInt recordNo, uInt columnValue )
1852 {
1853 }
1854
1855private:
1856 void appendTcalId( Vector<uInt> &v, uInt &elem, uInt &polId )
1857 {
1858 v.resize( polId+1, True ) ;
1859 v[polId] = elem ;
1860 }
1861 void setTcalId( Vector<uInt> &v, uInt &elem, uInt &polId )
1862 {
1863 v[polId] = elem ;
1864 }
1865 void post()
1866 {
1867 // check if given Tcal and Tsys is effective
1868 Bool isEffective = False ;
1869 for ( uInt ipol = 0 ; ipol < lastTcalId.nelements() ; ipol++ ) {
1870 if ( effectiveTcal[lastTcalId[ipol]] ) {
1871 isEffective = True ;
1872 break ;
1873 }
1874 }
1875 if ( !isEffective ) {
1876 if ( !(lastTsys->isEffective()) )
1877 return ;
1878 }
1879
1880 //cout << " interval: " << (endTime-startTime) << " lastTcalId = " << lastTcalId << endl ;
1881 Double midTime = 0.5 * ( startTime + endTime ) ;
1882 Double interval = endTime - startTime ;
1883 *timeRF = midTime ;
1884 *intervalRF = interval ;
1885 tcalProcessor->setTcalId( lastTcalId ) ;
1886 Array<Float> tcal = tcalProcessor->getTcal() ;
1887 tcalRF.define( tcal ) ;
1888 tsysRF.define( lastTsys->getTsys() ) ;
1889 sctab.addRow( 1, True ) ;
1890 scrow.put( rowidx ) ;
1891 rowidx++ ;
1892 }
1893
1894 Bool isUpdated()
1895 {
1896 Bool ret = False ;
1897 ret = anyNE( theTcalId, lastTcalId ) ;
1898 if ( !ret )
1899 ret = anyNE( theTsys->getTsys(), lastTsys->getTsys() ) ;
1900 return ret ;
1901 }
1902
1903 Table &sctab;
1904 TableRow scrow;
1905 uInt rowidx;
1906
1907 Double startTime,endTime,interval;
1908
1909 CountedPtr<BaseTsysHolder> lastTsys,theTsys;
1910 Vector<uInt> lastTcalId,theTcalId;
1911 CountedPtr<BaseTcalProcessor> tcalProcessor ;
1912 Vector<Bool> effectiveTcal;
1913
1914 RecordFieldPtr<Int> feedIdRF,specWinIdRF;
1915 RecordFieldPtr<Double> timeRF,intervalRF;
1916 RecordFieldPtr< Array<Float> > tcalRF,tsysRF;
1917
1918 ROArrayColumn<Float> tsysCol,tcalCol;
1919 ROTableColumn tcalIdCol,intervalCol;
1920};
1921
1922MSWriter::MSWriter(CountedPtr<Scantable> stable)
1923 : table_(stable),
1924 isWeather_(False),
1925 tcalSpec_(False),
1926 tsysSpec_(False),
1927 ptTabName_("")
1928{
1929 os_ = LogIO() ;
1930 os_.origin( LogOrigin( "MSWriter", "MSWriter()", WHERE ) ) ;
1931// os_ << "MSWriter::MSWriter()" << LogIO::POST ;
1932
1933 // initialize writer
1934 init() ;
1935}
1936
1937MSWriter::~MSWriter()
1938{
1939 os_.origin( LogOrigin( "MSWriter", "~MSWriter()", WHERE ) ) ;
1940// os_ << "MSWriter::~MSWriter()" << LogIO::POST ;
1941
1942 if ( mstable_ != 0 )
1943 delete mstable_ ;
1944}
1945
1946bool MSWriter::write(const string& filename, const Record& rec)
1947{
1948 os_.origin( LogOrigin( "MSWriter", "write()", WHERE ) ) ;
1949 //double startSec = mathutil::gettimeofday_sec() ;
1950 //os_ << "start MSWriter::write() startSec=" << startSec << LogIO::POST ;
1951
1952 filename_ = filename ;
1953
1954 // parsing MS options
1955 Bool overwrite = False ;
1956 if ( rec.isDefined( "ms" ) ) {
1957 Record msrec = rec.asRecord( "ms" ) ;
1958 if ( msrec.isDefined( "overwrite" ) ) {
1959 overwrite = msrec.asBool( "overwrite" ) ;
1960 }
1961 }
1962
1963 os_ << "Parsing MS options" << endl ;
1964 os_ << " overwrite = " << overwrite << LogIO::POST ;
1965
1966 File file( filename_ ) ;
1967 if ( file.exists() ) {
1968 if ( overwrite ) {
1969 os_ << filename_ << " exists. Overwrite existing data... " << LogIO::POST ;
1970 if ( file.isRegular() ) RegularFile(file).remove() ;
1971 else if ( file.isDirectory() ) Directory(file).removeRecursive() ;
1972 else SymLink(file).remove() ;
1973 }
1974 else {
1975 os_ << LogIO::SEVERE << "ERROR: " << filename_ << " exists..." << LogIO::POST ;
1976 return False ;
1977 }
1978 }
1979
1980 // set up MS
1981 setupMS() ;
1982
1983 // subtables
1984 // OBSERVATION
1985 fillObservation() ;
1986
1987 // ANTENNA
1988 fillAntenna() ;
1989
1990 // PROCESSOR
1991 fillProcessor() ;
1992
1993 // SOURCE
1994 fillSource() ;
1995
1996 // WEATHER
1997 if ( isWeather_ )
1998 fillWeather() ;
1999
2000 // SYSCAL
2001 fillSysCal() ;
2002
2003 /***
2004 * Start iteration using TableVisitor
2005 ***/
2006 {
2007 static const char *cols[] = {
2008 "FIELDNAME", "BEAMNO", "SCANNO", "IFNO", "SRCTYPE", "CYCLENO", "TIME",
2009 "POLNO",
2010 NULL
2011 };
2012 static const TypeManagerImpl<uInt> tmUInt;
2013 static const TypeManagerImpl<Int> tmInt;
2014 static const TypeManagerImpl<Double> tmDouble;
2015 static const TypeManagerImpl<String> tmString;
2016 static const TypeManager *const tms[] = {
2017 &tmString, &tmUInt, &tmUInt, &tmUInt, &tmInt, &tmUInt, &tmDouble, &tmUInt, NULL
2018 };
2019 //double t0 = mathutil::gettimeofday_sec() ;
2020 MSWriterVisitor myVisitor(table_->table(),*mstable_);
2021 //double t1 = mathutil::gettimeofday_sec() ;
2022 //cout << "MSWriterVisitor(): elapsed time " << t1-t0 << " sec" << endl ;
2023 String dataColName = "FLOAT_DATA" ;
2024 if ( useData_ )
2025 dataColName = "DATA" ;
2026 myVisitor.dataColumnName( dataColName ) ;
2027 myVisitor.pointingTableName( ptTabName_ ) ;
2028 myVisitor.setSourceRecord( srcRec_ ) ;
2029 //double t2 = mathutil::gettimeofday_sec() ;
2030 traverseTable(table_->table(), cols, tms, &myVisitor);
2031 //double t3 = mathutil::gettimeofday_sec() ;
2032 //cout << "traverseTable(): elapsed time " << t3-t2 << " sec" << endl ;
2033 }
2034 /***
2035 * End iteration using TableVisitor
2036 ***/
2037
2038 // ASDM tables
2039 const TableRecord &stKeys = table_->table().keywordSet() ;
2040 TableRecord &msKeys = mstable_->rwKeywordSet() ;
2041 uInt nfields = stKeys.nfields() ;
2042 for ( uInt ifield = 0 ; ifield < nfields ; ifield++ ) {
2043 String kname = stKeys.name( ifield ) ;
2044 if ( kname.find( "ASDM" ) != String::npos ) {
2045 String asdmpath = stKeys.asString( ifield ) ;
2046 os_ << "found ASDM table: " << asdmpath << LogIO::POST ;
2047 if ( Table::isReadable( asdmpath ) ) {
2048 Table newAsdmTab( asdmpath, Table::Old ) ;
2049 newAsdmTab.copy( filename_+"/"+kname, Table::New ) ;
2050 os_ << "add subtable: " << kname << LogIO::POST ;
2051 msKeys.defineTable( kname, Table( filename_+"/"+kname, Table::Old ) ) ;
2052 }
2053 }
2054 }
2055
2056 // replace POINTING table with original one if exists
2057 if ( ptTabName_ != "" ) {
2058 delete mstable_ ;
2059 mstable_ = 0 ;
2060 Table newPtTab( ptTabName_, Table::Old ) ;
2061 newPtTab.copy( filename_+"/POINTING", Table::New ) ;
2062 }
2063
2064 //double endSec = mathutil::gettimeofday_sec() ;
2065 //os_ << "end MSWriter::write() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
2066
2067 os_ << "Exported data as MS" << LogIO::POST ;
2068
2069 return True ;
2070}
2071
2072void MSWriter::init()
2073{
2074// os_.origin( LogOrigin( "MSWriter", "init()", WHERE ) ) ;
2075// double startSec = mathutil::gettimeofday_sec() ;
2076// os_ << "start MSWriter::init() startSec=" << startSec << LogIO::POST ;
2077
2078 // access to scantable
2079 header_ = table_->getHeader() ;
2080
2081 // FLOAT_DATA? or DATA?
2082 if ( header_.npol > 2 ) {
2083 useFloatData_ = False ;
2084 useData_ = True ;
2085 }
2086 else {
2087 useFloatData_ = True ;
2088 useData_ = False ;
2089 }
2090
2091 // polarization type
2092 polType_ = header_.poltype ;
2093 if ( polType_ == "" )
2094 polType_ = "stokes" ;
2095 else if ( polType_.find( "linear" ) != String::npos )
2096 polType_ = "linear" ;
2097 else if ( polType_.find( "circular" ) != String::npos )
2098 polType_ = "circular" ;
2099 else if ( polType_.find( "stokes" ) != String::npos )
2100 polType_ = "stokes" ;
2101 else if ( polType_.find( "linpol" ) != String::npos )
2102 polType_ = "linpol" ;
2103 else
2104 polType_ = "notype" ;
2105
2106 // Check if some subtables are exists
2107 Bool isTcal = False ;
2108 if ( table_->tcal().table().nrow() != 0 ) {
2109 ROTableColumn col( table_->tcal().table(), "TCAL" ) ;
2110 if ( col.isDefined( 0 ) ) {
2111 os_ << "TCAL table exists: nrow=" << table_->tcal().table().nrow() << LogIO::POST ;
2112 isTcal = True ;
2113 }
2114 else {
2115 os_ << "No TCAL rows" << LogIO::POST ;
2116 }
2117 }
2118 else {
2119 os_ << "No TCAL rows" << LogIO::POST ;
2120 }
2121 if ( table_->weather().table().nrow() != 0 ) {
2122 ROTableColumn col( table_->weather().table(), "TEMPERATURE" ) ;
2123 if ( col.isDefined( 0 ) ) {
2124 os_ << "WEATHER table exists: nrow=" << table_->weather().table().nrow() << LogIO::POST ;
2125 isWeather_ =True ;
2126 }
2127 else {
2128 os_ << "No WEATHER rows" << LogIO::POST ;
2129 }
2130 }
2131 else {
2132 os_ << "No WEATHER rows" << LogIO::POST ;
2133 }
2134
2135 // Are TCAL_SPECTRUM and TSYS_SPECTRUM necessary?
2136 if ( header_.nchan != 1 ) {
2137 if ( isTcal ) {
2138 // examine TCAL subtable
2139 Table tcaltab = table_->tcal().table() ;
2140 ROArrayColumn<Float> tcalCol( tcaltab, "TCAL" ) ;
2141 for ( uInt irow = 0 ; irow < tcaltab.nrow() ; irow++ ) {
2142 if ( tcalCol( irow ).size() != 1 )
2143 tcalSpec_ = True ;
2144 }
2145 }
2146 // examine spectral data
2147 TableIterator iter0( table_->table(), "IFNO" ) ;
2148 while( !iter0.pastEnd() ) {
2149 Table t0( iter0.table() ) ;
2150 ROArrayColumn<Float> sharedFloatArrCol( t0, "SPECTRA" ) ;
2151 uInt len = sharedFloatArrCol( 0 ).size() ;
2152 if ( len != 1 ) {
2153 sharedFloatArrCol.attach( t0, "TSYS" ) ;
2154 if ( sharedFloatArrCol( 0 ).size() != 1 )
2155 tsysSpec_ = True ;
2156 }
2157 iter0.next() ;
2158 }
2159 }
2160
2161 // check if reference for POINTING table exists
2162 const TableRecord &rec = table_->table().keywordSet() ;
2163 if ( rec.isDefined( "POINTING" ) ) {
2164 ptTabName_ = rec.asString( "POINTING" ) ;
2165 if ( !Table::isReadable( ptTabName_ ) ) {
2166 ptTabName_ = "" ;
2167 }
2168 }
2169
2170// double endSec = mathutil::gettimeofday_sec() ;
2171// os_ << "end MSWriter::init() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
2172}
2173
2174void MSWriter::setupMS()
2175{
2176// os_.origin( LogOrigin( "MSWriter", "setupMS()", WHERE ) ) ;
2177// double startSec = mathutil::gettimeofday_sec() ;
2178// os_ << "start MSWriter::setupMS() startSec=" << startSec << LogIO::POST ;
2179
2180 String dunit = table_->getHeader().fluxunit ;
2181
2182 TableDesc msDesc = MeasurementSet::requiredTableDesc() ;
2183 if ( useFloatData_ )
2184 MeasurementSet::addColumnToDesc( msDesc, MSMainEnums::FLOAT_DATA, 2 ) ;
2185 else if ( useData_ )
2186 MeasurementSet::addColumnToDesc( msDesc, MSMainEnums::DATA, 2 ) ;
2187
2188 SetupNewTable newtab( filename_, msDesc, Table::New ) ;
2189
2190 mstable_ = new MeasurementSet( newtab ) ;
2191
2192 TableColumn col ;
2193 if ( useFloatData_ )
2194 col.attach( *mstable_, "FLOAT_DATA" ) ;
2195 else if ( useData_ )
2196 col.attach( *mstable_, "DATA" ) ;
2197 col.rwKeywordSet().define( "UNIT", dunit ) ;
2198
2199 // create subtables
2200 TableDesc antennaDesc = MSAntenna::requiredTableDesc() ;
2201 SetupNewTable antennaTab( mstable_->antennaTableName(), antennaDesc, Table::New ) ;
2202 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::ANTENNA ), Table( antennaTab ) ) ;
2203
2204 TableDesc dataDescDesc = MSDataDescription::requiredTableDesc() ;
2205 SetupNewTable dataDescTab( mstable_->dataDescriptionTableName(), dataDescDesc, Table::New ) ;
2206 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::DATA_DESCRIPTION ), Table( dataDescTab ) ) ;
2207
2208 TableDesc dopplerDesc = MSDoppler::requiredTableDesc() ;
2209 SetupNewTable dopplerTab( mstable_->dopplerTableName(), dopplerDesc, Table::New ) ;
2210 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::DOPPLER ), Table( dopplerTab ) ) ;
2211
2212 TableDesc feedDesc = MSFeed::requiredTableDesc() ;
2213 SetupNewTable feedTab( mstable_->feedTableName(), feedDesc, Table::New ) ;
2214 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::FEED ), Table( feedTab ) ) ;
2215
2216 TableDesc fieldDesc = MSField::requiredTableDesc() ;
2217 SetupNewTable fieldTab( mstable_->fieldTableName(), fieldDesc, Table::New ) ;
2218 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::FIELD ), Table( fieldTab ) ) ;
2219
2220 TableDesc flagCmdDesc = MSFlagCmd::requiredTableDesc() ;
2221 SetupNewTable flagCmdTab( mstable_->flagCmdTableName(), flagCmdDesc, Table::New ) ;
2222 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::FLAG_CMD ), Table( flagCmdTab ) ) ;
2223
2224 TableDesc freqOffsetDesc = MSFreqOffset::requiredTableDesc() ;
2225 SetupNewTable freqOffsetTab( mstable_->freqOffsetTableName(), freqOffsetDesc, Table::New ) ;
2226 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::FREQ_OFFSET ), Table( freqOffsetTab ) ) ;
2227
2228 TableDesc historyDesc = MSHistory::requiredTableDesc() ;
2229 SetupNewTable historyTab( mstable_->historyTableName(), historyDesc, Table::New ) ;
2230 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::HISTORY ), Table( historyTab ) ) ;
2231
2232 TableDesc observationDesc = MSObservation::requiredTableDesc() ;
2233 SetupNewTable observationTab( mstable_->observationTableName(), observationDesc, Table::New ) ;
2234 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::OBSERVATION ), Table( observationTab ) ) ;
2235
2236 TableDesc pointingDesc = MSPointing::requiredTableDesc() ;
2237 SetupNewTable pointingTab( mstable_->pointingTableName(), pointingDesc, Table::New ) ;
2238 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::POINTING ), Table( pointingTab ) ) ;
2239
2240 TableDesc polarizationDesc = MSPolarization::requiredTableDesc() ;
2241 SetupNewTable polarizationTab( mstable_->polarizationTableName(), polarizationDesc, Table::New ) ;
2242 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::POLARIZATION ), Table( polarizationTab ) ) ;
2243
2244 TableDesc processorDesc = MSProcessor::requiredTableDesc() ;
2245 SetupNewTable processorTab( mstable_->processorTableName(), processorDesc, Table::New ) ;
2246 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::PROCESSOR ), Table( processorTab ) ) ;
2247
2248 TableDesc sourceDesc = MSSource::requiredTableDesc() ;
2249 MSSource::addColumnToDesc( sourceDesc, MSSourceEnums::TRANSITION, 1 ) ;
2250 MSSource::addColumnToDesc( sourceDesc, MSSourceEnums::REST_FREQUENCY, 1 ) ;
2251 MSSource::addColumnToDesc( sourceDesc, MSSourceEnums::SYSVEL, 1 ) ;
2252 SetupNewTable sourceTab( mstable_->sourceTableName(), sourceDesc, Table::New ) ;
2253 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::SOURCE ), Table( sourceTab ) ) ;
2254
2255 TableDesc spwDesc = MSSpectralWindow::requiredTableDesc() ;
2256 SetupNewTable spwTab( mstable_->spectralWindowTableName(), spwDesc, Table::New ) ;
2257 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::SPECTRAL_WINDOW ), Table( spwTab ) ) ;
2258
2259 TableDesc stateDesc = MSState::requiredTableDesc() ;
2260 SetupNewTable stateTab( mstable_->stateTableName(), stateDesc, Table::New ) ;
2261 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::STATE ), Table( stateTab ) ) ;
2262
2263 TableDesc sysCalDesc = MSSysCal::requiredTableDesc() ;
2264 if ( tcalSpec_ )
2265 MSSysCal::addColumnToDesc( sysCalDesc, MSSysCalEnums::TCAL_SPECTRUM, 2 ) ;
2266 else
2267 MSSysCal::addColumnToDesc( sysCalDesc, MSSysCalEnums::TCAL, 1 ) ;
2268 if ( tsysSpec_ )
2269 MSSysCal::addColumnToDesc( sysCalDesc, MSSysCalEnums::TSYS_SPECTRUM, 2 ) ;
2270 else
2271 MSSysCal::addColumnToDesc( sysCalDesc, MSSysCalEnums::TSYS, 1 ) ;
2272 SetupNewTable sysCalTab( mstable_->sysCalTableName(), sysCalDesc, Table::New ) ;
2273 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::SYSCAL ), Table( sysCalTab ) ) ;
2274
2275 TableDesc weatherDesc = MSWeather::requiredTableDesc() ;
2276 MSWeather::addColumnToDesc( weatherDesc, MSWeatherEnums::TEMPERATURE ) ;
2277 MSWeather::addColumnToDesc( weatherDesc, MSWeatherEnums::PRESSURE ) ;
2278 MSWeather::addColumnToDesc( weatherDesc, MSWeatherEnums::REL_HUMIDITY ) ;
2279 MSWeather::addColumnToDesc( weatherDesc, MSWeatherEnums::WIND_SPEED ) ;
2280 MSWeather::addColumnToDesc( weatherDesc, MSWeatherEnums::WIND_DIRECTION ) ;
2281 SetupNewTable weatherTab( mstable_->weatherTableName(), weatherDesc, Table::New ) ;
2282 mstable_->rwKeywordSet().defineTable( MeasurementSet::keywordName( MeasurementSet::WEATHER ), Table( weatherTab ) ) ;
2283
2284 mstable_->initRefs() ;
2285
2286// double endSec = mathutil::gettimeofday_sec() ;
2287// os_ << "end MSWriter::setupMS() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
2288}
2289
2290void MSWriter::fillObservation()
2291{
2292 //double startSec = mathutil::gettimeofday_sec() ;
2293 //os_ << "start MSWriter::fillObservation() startSec=" << startSec << LogIO::POST ;
2294
2295 // only 1 row
2296 mstable_->observation().addRow( 1, True ) ;
2297 MSObservationColumns msObsCols( mstable_->observation() ) ;
2298 msObsCols.observer().put( 0, header_.observer ) ;
2299 // tentatively put antennaname (from ANTENNA subtable)
2300 String hAntennaName = header_.antennaname ;
2301 String::size_type pos = hAntennaName.find( "//" ) ;
2302 String telescopeName ;
2303 if ( pos != String::npos ) {
2304 telescopeName = hAntennaName.substr( 0, pos ) ;
2305 }
2306 else {
2307 pos = hAntennaName.find( "@" ) ;
2308 telescopeName = hAntennaName.substr( 0, pos ) ;
2309 }
2310// os_ << "telescopeName = " << telescopeName << LogIO::POST ;
2311 msObsCols.telescopeName().put( 0, telescopeName ) ;
2312 msObsCols.project().put( 0, header_.project ) ;
2313 //ScalarMeasColumn<MEpoch> timeCol( table_->table().sort("TIME"), "TIME" ) ;
2314 Table sortedtable = table_->table().sort("TIME") ;
2315 ScalarMeasColumn<MEpoch> timeCol( sortedtable, "TIME" ) ;
2316 Vector<MEpoch> trange( 2 ) ;
2317 trange[0] = timeCol( 0 ) ;
2318 trange[1] = timeCol( table_->nrow()-1 ) ;
2319 msObsCols.timeRangeMeas().put( 0, trange ) ;
2320
2321 //double endSec = mathutil::gettimeofday_sec() ;
2322 //os_ << "end MSWriter::fillObservation() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
2323}
2324
2325void MSWriter::antennaProperty( String &name, String &m, String &t, Double &d )
2326{
2327 name.upcase() ;
2328
2329 m = "ALT-AZ" ;
2330 t = "GROUND-BASED" ;
2331 if ( name.matches( Regex( "DV[0-9]+$" ) )
2332 || name.matches( Regex( "DA[0-9]+$" ) )
2333 || name.matches( Regex( "PM[0-9]+$" ) ) )
2334 d = 12.0 ;
2335 else if ( name.matches( Regex( "CM[0-9]+$" ) ) )
2336 d = 7.0 ;
2337 else if ( name.contains( "GBT" ) )
2338 d = 104.9 ;
2339 else if ( name.contains( "MOPRA" ) )
2340 d = 22.0 ;
2341 else if ( name.contains( "PKS" ) || name.contains( "PARKS" ) )
2342 d = 64.0 ;
2343 else if ( name.contains( "TIDBINBILLA" ) )
2344 d = 70.0 ;
2345 else if ( name.contains( "CEDUNA" ) )
2346 d = 30.0 ;
2347 else if ( name.contains( "HOBART" ) )
2348 d = 26.0 ;
2349 else if ( name.contains( "APEX" ) )
2350 d = 12.0 ;
2351 else if ( name.contains( "ASTE" ) )
2352 d = 10.0 ;
2353 else if ( name.contains( "NRO" ) )
2354 d = 45.0 ;
2355 else
2356 d = 1.0 ;
2357}
2358
2359void MSWriter::fillAntenna()
2360{
2361 //double startSec = mathutil::gettimeofday_sec() ;
2362 //os_ << "start MSWriter::fillAntenna() startSec=" << startSec << LogIO::POST ;
2363
2364 // only 1 row
2365 Table anttab = mstable_->antenna() ;
2366 anttab.addRow( 1, True ) ;
2367
2368 Table &table = table_->table() ;
2369 const TableRecord &keys = table.keywordSet() ;
2370 String hAntName = keys.asString( "AntennaName" ) ;
2371 String::size_type pos = hAntName.find( "//" ) ;
2372 String antennaName ;
2373 String stationName ;
2374 if ( pos != String::npos ) {
2375 stationName = hAntName.substr( 0, pos ) ;
2376 hAntName = hAntName.substr( pos+2 ) ;
2377 }
2378 pos = hAntName.find( "@" ) ;
2379 if ( pos != String::npos ) {
2380 antennaName = hAntName.substr( 0, pos ) ;
2381 stationName = hAntName.substr( pos+1 ) ;
2382 }
2383 else {
2384 antennaName = hAntName ;
2385 }
2386 Vector<Double> antpos = keys.asArrayDouble( "AntennaPosition" ) ;
2387
2388 String mount, atype ;
2389 Double diameter ;
2390 antennaProperty( antennaName, mount, atype, diameter ) ;
2391
2392 TableRow tr( anttab ) ;
2393 TableRecord &r = tr.record() ;
2394 RecordFieldPtr<String> nameRF( r, "NAME" ) ;
2395 RecordFieldPtr<String> stationRF( r, "STATION" ) ;
2396 RecordFieldPtr<String> mountRF( r, "MOUNT" ) ;
2397 RecordFieldPtr<String> typeRF( r, "TYPE" ) ;
2398 RecordFieldPtr<Double> dishDiameterRF( r, "DISH_DIAMETER" ) ;
2399 RecordFieldPtr< Vector<Double> > positionRF( r, "POSITION" ) ;
2400 *nameRF = antennaName ;
2401 *mountRF = mount ;
2402 *typeRF = atype ;
2403 *dishDiameterRF = diameter ;
2404 *positionRF = antpos ;
2405 *stationRF = stationName ;
2406
2407 tr.put( 0 ) ;
2408
2409 //double endSec = mathutil::gettimeofday_sec() ;
2410 //os_ << "end MSWriter::fillAntenna() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
2411}
2412
2413void MSWriter::fillProcessor()
2414{
2415// double startSec = mathutil::gettimeofday_sec() ;
2416// os_ << "start MSWriter::fillProcessor() startSec=" << startSec << LogIO::POST ;
2417
2418 // only add empty 1 row
2419 MSProcessor msProc = mstable_->processor() ;
2420 msProc.addRow( 1, True ) ;
2421
2422// double endSec = mathutil::gettimeofday_sec() ;
2423// os_ << "end MSWriter::fillProcessor() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
2424}
2425
2426void MSWriter::fillSource()
2427{
2428// double startSec = mathutil::gettimeofday_sec() ;
2429// os_ << "start MSWriter::fillSource() startSec=" << startSec << LogIO::POST ;
2430
2431 // access to MS SOURCE subtable
2432 MSSource msSrc = mstable_->source() ;
2433
2434 // access to MOLECULE subtable
2435 STMolecules stm = table_->molecules() ;
2436
2437 Int srcId = 0 ;
2438 Vector<Double> restFreq ;
2439 Vector<String> molName ;
2440 Vector<String> fMolName ;
2441
2442 // row based
2443 TableRow row( msSrc ) ;
2444 TableRecord &rec = row.record() ;
2445 RecordFieldPtr<Int> srcidRF( rec, "SOURCE_ID" ) ;
2446 RecordFieldPtr<String> nameRF( rec, "NAME" ) ;
2447 RecordFieldPtr< Array<Double> > srcpmRF( rec, "PROPER_MOTION" ) ;
2448 RecordFieldPtr< Array<Double> > srcdirRF( rec, "DIRECTION" ) ;
2449 RecordFieldPtr<Int> numlineRF( rec, "NUM_LINES" ) ;
2450 RecordFieldPtr< Array<Double> > restfreqRF( rec, "REST_FREQUENCY" ) ;
2451 RecordFieldPtr< Array<Double> > sysvelRF( rec, "SYSVEL" ) ;
2452 RecordFieldPtr< Array<String> > transitionRF( rec, "TRANSITION" ) ;
2453 RecordFieldPtr<Double> timeRF( rec, "TIME" ) ;
2454 RecordFieldPtr<Double> intervalRF( rec, "INTERVAL" ) ;
2455 RecordFieldPtr<Int> spwidRF( rec, "SPECTRAL_WINDOW_ID" ) ;
2456
2457 //
2458 // ITERATION: SRCNAME
2459 //
2460 TableIterator iter0( table_->table(), "SRCNAME" ) ;
2461 while( !iter0.pastEnd() ) {
2462 //Table t0( iter0.table() ) ;
2463 Table t0 = iter0.table() ;
2464
2465 // get necessary information
2466 ROScalarColumn<String> srcNameCol( t0, "SRCNAME" ) ;
2467 String srcName = srcNameCol( 0 ) ;
2468 ROArrayColumn<Double> sharedDArrRCol( t0, "SRCPROPERMOTION" ) ;
2469 Vector<Double> srcPM = sharedDArrRCol( 0 ) ;
2470 sharedDArrRCol.attach( t0, "SRCDIRECTION" ) ;
2471 Vector<Double> srcDir = sharedDArrRCol( 0 ) ;
2472 ROScalarColumn<Double> srcVelCol( t0, "SRCVELOCITY" ) ;
2473 Double srcVel = srcVelCol( 0 ) ;
2474 srcRec_.define( srcName, srcId ) ;
2475
2476 // NAME
2477 *nameRF = srcName ;
2478
2479 // SOURCE_ID
2480 *srcidRF = srcId ;
2481
2482 // PROPER_MOTION
2483 *srcpmRF = srcPM ;
2484
2485 // DIRECTION
2486 *srcdirRF = srcDir ;
2487
2488 //
2489 // ITERATION: MOLECULE_ID
2490 //
2491 TableIterator iter1( t0, "MOLECULE_ID" ) ;
2492 while( !iter1.pastEnd() ) {
2493 //Table t1( iter1.table() ) ;
2494 Table t1 = iter1.table() ;
2495
2496 // get necessary information
2497 ROScalarColumn<uInt> molIdCol( t1, "MOLECULE_ID" ) ;
2498 uInt molId = molIdCol( 0 ) ;
2499 stm.getEntry( restFreq, molName, fMolName, molId ) ;
2500
2501 uInt numFreq = restFreq.size() ;
2502
2503 // NUM_LINES
2504 *numlineRF = numFreq ;
2505
2506 // REST_FREQUENCY
2507 *restfreqRF = restFreq ;
2508
2509 // TRANSITION
2510 //*transitionRF = fMolName ;
2511 Vector<String> transition ;
2512 if ( fMolName.size() != 0 ) {
2513 transition = fMolName ;
2514 }
2515 else if ( molName.size() != 0 ) {
2516 transition = molName ;
2517 }
2518 else {
2519 transition.resize( numFreq ) ;
2520 transition = "" ;
2521 }
2522 *transitionRF = transition ;
2523
2524 // SYSVEL
2525 Vector<Double> sysvelArr( numFreq, srcVel ) ;
2526 *sysvelRF = sysvelArr ;
2527
2528 //
2529 // ITERATION: IFNO
2530 //
2531 TableIterator iter2( t1, "IFNO" ) ;
2532 while( !iter2.pastEnd() ) {
2533 //Table t2( iter2.table() ) ;
2534 Table t2 = iter2.table() ;
2535 uInt nrow = msSrc.nrow() ;
2536
2537 // get necessary information
2538 ROScalarColumn<uInt> ifNoCol( t2, "IFNO" ) ;
2539 uInt ifno = ifNoCol( 0 ) ; // IFNO = SPECTRAL_WINDOW_ID
2540 Double midTime ;
2541 Double interval ;
2542 getValidTimeRange( midTime, interval, t2 ) ;
2543
2544 // fill SPECTRAL_WINDOW_ID
2545 *spwidRF = ifno ;
2546
2547 // fill TIME, INTERVAL
2548 *timeRF = midTime ;
2549 *intervalRF = interval ;
2550
2551 // add row
2552 msSrc.addRow( 1, True ) ;
2553 row.put( nrow ) ;
2554
2555 iter2.next() ;
2556 }
2557
2558 iter1.next() ;
2559 }
2560
2561 // increment srcId if SRCNAME changed
2562 srcId++ ;
2563
2564 iter0.next() ;
2565 }
2566
2567// double endSec = mathutil::gettimeofday_sec() ;
2568// os_ << "end MSWriter::fillSource() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
2569}
2570
2571void MSWriter::fillWeather()
2572{
2573// double startSec = mathutil::gettimeofday_sec() ;
2574// os_ << "start MSWriter::fillWeather() startSec=" << startSec << LogIO::POST ;
2575
2576 // access to MS WEATHER subtable
2577 MSWeather msw = mstable_->weather() ;
2578
2579 // access to WEATHER subtable
2580 Table stw = table_->weather().table() ;
2581 uInt nrow = stw.nrow() ;
2582
2583 if ( nrow == 0 )
2584 return ;
2585
2586 msw.addRow( nrow, True ) ;
2587 MSWeatherColumns mswCols( msw ) ;
2588
2589 // ANTENNA_ID is always 0
2590 Vector<Int> antIdArr( nrow, 0 ) ;
2591 mswCols.antennaId().putColumn( antIdArr ) ;
2592
2593 // fill weather status
2594 ROScalarColumn<Float> sharedFloatCol( stw, "TEMPERATURE" ) ;
2595 mswCols.temperature().putColumn( sharedFloatCol ) ;
2596 sharedFloatCol.attach( stw, "PRESSURE" ) ;
2597 mswCols.pressure().putColumn( sharedFloatCol ) ;
2598 sharedFloatCol.attach( stw, "HUMIDITY" ) ;
2599 mswCols.relHumidity().putColumn( sharedFloatCol ) ;
2600 sharedFloatCol.attach( stw, "WINDSPEED" ) ;
2601 mswCols.windSpeed().putColumn( sharedFloatCol ) ;
2602 sharedFloatCol.attach( stw, "WINDAZ" ) ;
2603 mswCols.windDirection().putColumn( sharedFloatCol ) ;
2604
2605 // fill TIME and INTERVAL
2606 Double midTime ;
2607 Double interval ;
2608 Vector<Double> intervalArr( nrow, 0.0 ) ;
2609 TableIterator iter( table_->table(), "WEATHER_ID" ) ;
2610 while( !iter.pastEnd() ) {
2611 //Table tab( iter.table() ) ;
2612 Table tab = iter.table() ;
2613
2614 ROScalarColumn<uInt> widCol( tab, "WEATHER_ID" ) ;
2615 uInt wid = widCol( 0 ) ;
2616
2617 getValidTimeRange( midTime, interval, tab ) ;
2618 mswCols.time().put( wid, midTime ) ;
2619 intervalArr[wid] = interval ;
2620
2621 iter.next() ;
2622 }
2623 mswCols.interval().putColumn( intervalArr ) ;
2624
2625// double endSec = mathutil::gettimeofday_sec() ;
2626// os_ << "end MSWriter::fillWeather() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
2627}
2628
2629void MSWriter::fillSysCal()
2630{
2631 Table mssc = mstable_->sysCal() ;
2632
2633 {
2634 static const char *cols[] = {
2635 "BEAMNO", "IFNO", "TIME", "POLNO",
2636 NULL
2637 };
2638 static const TypeManagerImpl<uInt> tmUInt;
2639 static const TypeManagerImpl<Double> tmDouble;
2640 static const TypeManager *const tms[] = {
2641 &tmUInt, &tmUInt, &tmDouble, &tmUInt, NULL
2642 };
2643 //double t0 = mathutil::gettimeofday_sec() ;
2644 MSSysCalVisitor myVisitor(table_->table(),mssc);
2645 //double t1 = mathutil::gettimeofday_sec() ;
2646 //cout << "MSWriterVisitor(): elapsed time " << t1-t0 << " sec" << endl ;
2647 traverseTable(table_->table(), cols, tms, &myVisitor);
2648 //double t3 = mathutil::gettimeofday_sec() ;
2649 //cout << "traverseTable(): elapsed time " << t3-t2 << " sec" << endl ;
2650 }
2651
2652}
2653
2654void MSWriter::getValidTimeRange( Double &me, Double &interval, Table &tab )
2655{
2656// double startSec = mathutil::gettimeofday_sec() ;
2657// os_ << "start MSWriter::getVaridTimeRange() startSec=" << startSec << LogIO::POST ;
2658
2659 // sort table
2660 //Table stab = tab.sort( "TIME" ) ;
2661
2662 ROScalarColumn<Double> timeCol( tab, "TIME" ) ;
2663 Vector<Double> timeArr = timeCol.getColumn() ;
2664 Double minTime ;
2665 Double maxTime ;
2666 minMax( minTime, maxTime, timeArr ) ;
2667 Double midTime = 0.5 * ( minTime + maxTime ) * 86400.0 ;
2668 // unit for TIME
2669 // Scantable: "d"
2670 // MS: "s"
2671 me = midTime ;
2672 interval = ( maxTime - minTime ) * 86400.0 ;
2673
2674// double endSec = mathutil::gettimeofday_sec() ;
2675// os_ << "end MSWriter::getValidTimeRange() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
2676}
2677
2678void MSWriter::getValidTimeRange( Double &me, Double &interval, Vector<Double> &atime, Vector<Double> &ainterval )
2679{
2680// double startSec = mathutil::gettimeofday_sec() ;
2681// os_ << "start MSWriter::getVaridTimeRange() startSec=" << startSec << LogIO::POST ;
2682
2683 // sort table
2684 //Table stab = tab.sort( "TIME" ) ;
2685
2686 Double minTime ;
2687 Double maxTime ;
2688 minMax( minTime, maxTime, atime ) ;
2689 Double midTime = 0.5 * ( minTime + maxTime ) * 86400.0 ;
2690 // unit for TIME
2691 // Scantable: "d"
2692 // MS: "s"
2693 me = midTime ;
2694 interval = ( maxTime - minTime ) * 86400.0 + mean( ainterval ) ;
2695
2696// double endSec = mathutil::gettimeofday_sec() ;
2697// os_ << "end MSWriter::getValidTimeRange() endSec=" << endSec << " (" << endSec-startSec << "sec)" << LogIO::POST ;
2698}
2699
2700}
Note: See TracBrowser for help on using the repository browser.