00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028 #ifndef MSVIS_VISIBILITYITERATORIMPL2_H
00029 #define MSVIS_VISIBILITYITERATORIMPL2_H
00030
00031 #include <casa/aips.h>
00032 #include <ms/MSOper/MSDerivedValues.h>
00033 #include <msvis/MSVis/ViImplementation2.h>
00034 #include <msvis/MSVis/ViColumns2.h>
00035 #include <tables/Tables/ArrayColumn.h>
00036 #include <tables/Tables/ScalarColumn.h>
00037 #include <msvis/MSVis/VisImagingWeight.h>
00038 #include <msvis/MSVis/VisBuffer2.h>
00039
00040 #include <tuple>
00041 #include <map>
00042 #include <vector>
00043
00044 namespace casa {
00045
00046 template <typename T> class ArrayColumn;
00047 template <typename T> class CountedPtr;
00048 class VisModelDataI;
00049 class MsIter;
00050 template <typename T> class ROArrayColumn;
00051 template <typename T, Int N> class RigidVector;
00052 template <typename T, Int N> class SquareMatrix;
00053
00054 namespace vi {
00055
00056
00057
00058 class VisBuffer2;
00059
00060 class ChannelSelector;
00061 class ChannelSelectorCache;
00062 class SpectralWindowChannelsCache;
00063 class SpectralWindowChannels;
00064 class SubtableColumns;
00065
00066
00067
00068
00069
00070
00071
00072
00073
00074
00075
00076
00077
00078
00079
00080
00081
00082
00083
00084
00085
00086
00087
00088
00089
00090
00091
00092
00093
00094
00095
00096
00097
00098
00099
00100
00101
00102
00103
00104
00105
00106
00107
00108
00109
00110
00111
00112
00113
00114
00115
00116 class VisibilityIteratorImpl2 : public ViImplementation2 {
00117
00118 friend class AsyncEnabler;
00119 friend class ViImplAsync2;
00120 friend class VisibilityIterator2;
00121 friend class VLAT;
00122 friend class VisBuffer2Adapter;
00123
00124
00125 public:
00126
00127
00128
00129
00130
00131
00132
00133
00134
00135
00136
00137
00138
00139
00140
00141
00142
00143
00144
00145 typedef VisibilityIterator2::DataColumn DataColumn;
00146
00147
00148
00149
00150
00151
00152
00153
00154
00155
00156
00157
00158
00159
00160
00161
00163
00164
00165
00166
00167
00168
00169
00170
00171
00172
00173
00174
00175
00176
00177
00178
00179
00180
00181
00182
00183
00184
00185
00186
00187
00188
00189
00190
00191
00192
00193 VisibilityIteratorImpl2 (const Block<const MeasurementSet *> & mss,
00194 const SortColumns & sortColumns,
00195 Double timeInterval,
00196 VisBufferType vbType,
00197 Bool isWritable);
00198
00199
00200
00201
00202
00203
00204
00205 virtual ~VisibilityIteratorImpl2 ();
00206
00207
00208
00209
00210
00211 virtual String ViiType() const { return String("DiskIO()"); };
00212
00213 virtual Bool isWritable () const;
00214
00215
00216 virtual void origin ();
00217
00218 virtual void originChunks ();
00219
00220
00221
00222
00223 virtual Double getInterval() const;
00224
00225
00226
00227
00228 virtual void setInterval (Double timeInterval);
00229
00230
00231
00232
00233
00234
00235 virtual void setRowBlocking (Int nRows = 0);
00236
00237 virtual Bool existsColumn (VisBufferComponent2 id) const;
00238
00239
00240 virtual Bool more () const;
00241
00242 virtual Subchunk getSubchunkId () const;
00243
00244 virtual const SortColumns & getSortColumns() const;
00245
00246 virtual void setFrequencySelections (const FrequencySelections & selection);
00247
00248
00249 virtual Bool moreChunks () const;
00250
00251
00252
00253 virtual Bool isNewArrayId () const;
00254 virtual Bool isNewFieldId () const;
00255 virtual Bool isNewMs () const;
00256 virtual Bool isNewSpectralWindow () const;
00257
00258 virtual Int msId () const;
00259 virtual Int getNMs () const;
00260
00261 virtual VisBuffer2 * getVisBuffer ();
00262 virtual VisBuffer2 * getVisBuffer (const VisibilityIterator2 *);
00263
00264
00265 virtual const MeasurementSet & ms () const;
00266
00267
00268 virtual String msName() const;
00269
00270
00271
00272 virtual void next ();
00273
00274
00275 virtual void nextChunk ();
00276
00277
00278 virtual void antenna1 (Vector<Int> & ant1) const;
00279
00280
00281 virtual void antenna2 (Vector<Int> & ant2) const;
00282
00283
00284 virtual void feed1 (Vector<Int> & fd1) const;
00285
00286
00287 virtual void feed2 (Vector<Int> & fd2) const;
00288
00289
00290 void jonesC (Vector<SquareMatrix<Complex, 2> > & cjones) const;
00291
00292
00293
00294
00295
00296
00297
00298 virtual const Cube<Double> & receptorAngles () const;
00299
00300
00301 virtual const Vector<String> & antennaMounts () const;
00302
00303
00304
00305
00306
00307 virtual const Cube<RigidVector<Double, 2> > & getBeamOffsets () const;
00308
00309
00310 virtual Bool allBeamOffsetsZero () const;
00311
00312
00313 virtual const Vector<Float> & feed_pa (Double time) const;
00314
00315
00316
00317
00318 virtual const Float & parang0 (Double time) const;
00319
00320
00321 virtual const Vector<Float> & parang (Double time) const;
00322
00323
00324 virtual MDirection azel0 (Double time) const;
00325
00326 virtual const Vector<MDirection> & azel (Double time) const;
00327
00328
00329 virtual Double hourang (Double time) const;
00330
00331
00332 virtual void fieldIds (Vector<Int>&) const;
00333
00334
00335 virtual void arrayIds (Vector<Int>&) const;
00336
00337
00338 virtual String fieldName () const;
00339
00340
00341 virtual String sourceName () const;
00342
00343
00344 virtual void flag (Cube<Bool> & flags) const;
00345
00346
00347 virtual void flag (Matrix<Bool> & flags) const;
00348
00349
00350 Bool flagCategoryExists () const;
00351
00352
00353 virtual void flagCategory (Array<Bool> & flagCategories) const;
00354
00355
00356 virtual void flagRow (Vector<Bool> & rowflags) const;
00357
00358
00359 virtual void scan (Vector<Int> & scans) const;
00360
00361
00362 virtual void observationId (Vector<Int> & obsids) const;
00363
00364
00365 virtual void processorId (Vector<Int> & procids) const;
00366
00367
00368 virtual void stateId (Vector<Int> & stateids) const;
00369
00370
00371 virtual const MDirection & phaseCenter () const;
00372
00373
00374 virtual Int polFrame () const;
00375
00376
00377 virtual void corrType (Vector<Int> & corrTypes) const;
00378
00379
00380 virtual void sigma (Matrix<Float> & sig) const;
00381
00382
00383 virtual Int spectralWindow () const;
00384
00385 virtual void spectralWindows (Vector<Int> & spws) const;
00386
00387
00388 virtual Int polarizationId () const;
00389
00390
00391 virtual Int dataDescriptionId () const;
00392
00393 virtual void dataDescriptionIds (Vector<Int> & ddis) const;
00394
00395
00396 virtual void time (Vector<Double> & t) const;
00397
00398
00399 virtual void timeCentroid (Vector<Double> & t) const;
00400
00401
00402 virtual void timeInterval (Vector<Double> & ti) const;
00403
00404
00405 virtual void exposure (Vector<Double> & expo) const;
00406
00407
00408 virtual void visibilityCorrected (Cube<Complex> & vis) const;
00409 virtual void visibilityModel (Cube<Complex> & vis) const;
00410 virtual void visibilityObserved (Cube<Complex> & vis) const;
00411
00412
00413 virtual void floatData (Cube<Float> & fcube) const;
00414
00415
00416
00417
00418
00419
00420
00421
00422
00423 virtual IPosition visibilityShape () const;
00424
00425
00426
00427 virtual void uvw (Matrix<Double> & uvwmat) const;
00428
00429
00430 virtual void weight (Matrix<Float> & wt) const;
00431
00432
00433 Bool weightSpectrumExists () const;
00434
00435
00436 Bool sigmaSpectrumExists () const;
00437
00438
00439 virtual void weightSpectrum (Cube<Float> & wtsp) const;
00440
00441
00442 virtual void sigmaSpectrum (Cube<Float> & sigsp) const;
00443
00444
00445 virtual void setWeightScaling (CountedPtr<WeightScaling> weightscaling);
00446 virtual CountedPtr<WeightScaling> getWeightScaling () const;
00447 virtual Bool hasWeightScaling () const;
00448
00449
00450
00451 const VisImagingWeight & getImagingWeightGenerator () const;
00452
00453
00454 virtual Bool newFieldId () const;
00455
00456
00457 virtual Bool newArrayId () const;
00458
00459
00460 virtual Bool newSpectralWindow () const;
00461
00462
00463 virtual Int nRows () const;
00464
00465
00466
00467
00468 virtual void getRowIds (Vector<uInt> & rowids) const;
00469
00470
00471
00472 virtual Int nRowsInChunk () const;
00473
00474
00475
00476
00477
00478
00479
00480
00481
00482
00483
00484
00485
00486
00487 virtual void slurp () const;
00488
00489
00490
00491
00492
00493
00494
00495
00496
00497
00498
00499
00500
00501
00502
00503
00504
00505
00506 virtual const vi::SubtableColumns & subtableColumns () const;
00507
00508
00509
00510
00511 virtual const SpectralWindowChannels & getSpectralWindowChannels (Int msId, Int spectralWindowId) const;
00512
00513
00514 virtual void useImagingWeight (const VisImagingWeight & imWgt);
00515
00516 virtual Int nAntennas () const;
00517
00518 virtual Int nRowsViWillSweep () const;
00519
00520
00521
00522 virtual Int nSpectralWindows () const;
00523 virtual Int nPolarizationIds () const;
00524 virtual Int nDataDescriptionIds () const;
00525
00526 virtual MEpoch getEpoch () const;
00527 MFrequency::Types getObservatoryFrequencyType () const;
00528 MPosition getObservatoryPosition () const;
00529 Vector<Float> getReceptor0Angle ();
00530
00531
00532
00533
00534
00535
00536
00537
00538
00539
00540
00541 virtual void writeFlag (const Cube<Bool> & flag);
00542
00543
00544 virtual void writeFlagRow (const Vector<Bool> & rowflags);
00545
00546 void writeFlagCategory(const Array<Bool>& fc);
00547
00548
00549
00550
00551
00552
00553
00554
00555
00556
00557
00558
00559
00560 virtual void writeVisCorrected (const Cube<Complex> & vis);
00561 virtual void writeVisModel (const Cube<Complex> & vis);
00562 virtual void writeVisObserved (const Cube<Complex> & vis);
00563
00564
00565 virtual void writeWeight (const Matrix<Float> & wt);
00566
00567
00568
00569
00570
00571 virtual void writeWeightSpectrum (const Cube<Float> & wtsp);
00572
00573
00574 virtual void initWeightSpectrum (const Cube<Float> & weightSpectrum);
00575
00576
00577 virtual void writeSigmaSpectrum (const Cube<Float> & sigsp);
00578
00579
00580 virtual void initSigmaSpectrum (const Cube<Float> & sigmaSpectrum);
00581
00582
00583 virtual void writeSigma (const Matrix<Float> & sig);
00584
00585
00586
00587
00588 virtual void writeModel(const RecordInterface& rec, Bool iscomponentlist=True,
00589 Bool incremental=False);
00590
00591 virtual void writeBackChanges (VisBuffer2 *);
00592
00593 protected:
00594
00595 typedef std::tuple <Vector<Int>, Vector<Int>, Vector<Int>, Vector<Int> > ChannelInfo;
00596
00597 void addDataSelection (const MeasurementSet & ms);
00598
00599 void allSpectralWindowsSelected (Vector<Int> & spectralWindows,
00600 Vector<Int> & nChannels) const;
00601
00602 void attachColumnsSafe (const Table & t);
00603
00604
00605
00606 virtual void attachColumns (const Table & t);
00607
00608
00609
00610
00611 virtual const Table attachTable () const;
00612
00613 virtual void applyPendingChanges ();
00614
00615
00616
00617 virtual void configureNewChunk ();
00618
00619
00620
00621 virtual void configureNewSubchunk ();
00622
00623 const ChannelSelector *
00624 createDefaultChannelSelector (Double time, Int msId, Int spectralWindowId);
00625
00626 virtual const vi::ChannelSelector * determineChannelSelection (Double time,
00627 Int spectralWindowId = -1,
00628 Int polarizationId = -1,
00629 Int msId = -1) const;
00630
00631 bool fillFromVirtualModel (Cube <Complex> & value) const;
00632
00633 Slice findChannelsInRange (Double lowerFrequency, Double upperFrequency,
00634 const vi::SpectralWindowChannels & spectralWindowChannels) const;
00635
00636 ChannelInfo
00637 getChannelInformation(Bool now) const;
00638
00639 ChannelInfo
00640 getChannelInformationUsingFrequency (Bool now) const;
00641
00642
00643
00644
00645 template <typename T>
00646 void getColumnRows (const ROArrayColumn<T> & column, Array<T> & array) const;
00647
00648 template <typename T>
00649 void
00650 getColumnRowsMatrix (const ROArrayColumn<T> & column, Matrix<T> & array,
00651 Bool correlationSlicing) const;
00652
00653 template <typename T>
00654 void getColumnRows (const ROScalarColumn<T> & column, Vector<T> & array) const;
00655
00656 Vector<Double> getFrequencies (Double time, Int frameOfReference,
00657 Int spectralWindowId, Int msId) const;
00658
00659 Vector<Int> getChannels (Double time, Int frameOfReference,
00660 Int spectralWindowId, Int msId) const;
00661 Vector<Int> getCorrelations () const;
00662 Vector<Stokes::StokesTypes> getCorrelationTypesDefined () const;
00663 Vector<Stokes::StokesTypes> getCorrelationTypesSelected () const;
00664
00665 int getMeasurementFrame (int spectralWindowId) const;
00666
00667 Int getPolarizationId (Int spectralWindowId, Int msId) const;
00668
00669 Int getReportingFrameOfReference () const;
00670
00671
00672
00673 Block <MeasurementSet> getMeasurementSets () const;
00674
00675
00676
00677 const MSDerivedValues & getMsd () const;
00678
00679
00680
00681 VisibilityIterator2 * getViP () const;
00682
00683
00684
00685
00686
00687
00688 virtual void initialize (const Block<const MeasurementSet *> & mss);
00689
00690
00691
00692
00693 Bool isInASelectedSpectralWindow () const;
00694
00695
00696
00697
00698 vi::ChannelSelector *
00699 makeChannelSelectorC (const FrequencySelection & selection,
00700 Double time, Int msId, Int spectralWindowId,
00701 Int polarizationId) const;
00702
00703 vi::ChannelSelector *
00704 makeChannelSelectorF (const FrequencySelection & selection,
00705 Double time, Int msId, Int spectralWindowId,
00706 Int polarizationId) const;
00707
00708 MFrequency::Convert makeFrequencyConverter (Double time,
00709 int spectralWindowId,
00710 Int otherFrameOfReference,
00711 Bool toObservedFrame, Unit) const;
00712
00713
00714
00715 const ROMSColumns * msColumnsKluge () const;
00716
00717
00718
00719
00720
00721 virtual void originChunks (Bool forceRewind);
00722
00723
00724
00725
00726 void positionMsIterToASelectedSpectralWindow ();
00727
00728
00729
00730
00731
00732 void setReportingFrameOfReference (Int);
00733
00734
00735
00736
00737 virtual void setTileCache ();
00738 void setMsCacheSizes (const MeasurementSet & ms,
00739 vector<MSMainEnums::PredefinedColumns> columnIds);
00740 void setMsColumnCacheSizes (const MeasurementSet&, const string &);
00741
00742
00743
00744
00745
00746 virtual void throwIfPendingChanges ();
00747
00748
00749
00750 Bool usesTiledDataManager (const String & columnName, const MeasurementSet & ms) const;
00751
00752
00753
00754
00755
00756
00757
00758
00759
00760
00761 class BackWriter {
00762
00763 public:
00764
00765 virtual ~BackWriter () {}
00766
00767 virtual void operator () (VisibilityIteratorImpl2 * vi, VisBuffer2 * vb) = 0;
00768
00769 };
00770
00771
00772 template <typename Setter, typename Getter>
00773 class BackWriterImpl : public BackWriter {
00774 public:
00775
00776 BackWriterImpl (Setter setter, Getter getter) : getter_p (getter), setter_p (setter) {}
00777 void operator () (VisibilityIteratorImpl2 * vi, VisBuffer2 * vb) {
00778 (vi ->* setter_p) ((vb ->* getter_p) ());
00779 }
00780
00781 private:
00782
00783 Getter getter_p;
00784 Setter setter_p;
00785 };
00786
00787
00788
00789
00790 template <typename Setter, typename Getter>
00791 class BackWriterImpl2 : public BackWriter {
00792 public:
00793
00794 typedef VisibilityIteratorImpl2::DataColumn DataColumn;
00795
00796 BackWriterImpl2 (Setter setter, Getter getter, DataColumn dc)
00797 : dataColumn_p (dc), getter_p (getter), setter_p (setter)
00798 {}
00799 void operator () (VisibilityIteratorImpl2 * vi, VisBuffer2 * vb) {
00800 (vi ->* setter_p) ((vb ->* getter_p) (), dataColumn_p);
00801 }
00802
00803 private:
00804
00805 DataColumn dataColumn_p;
00806 Getter getter_p;
00807 Setter setter_p;
00808 };
00809
00810
00811
00812
00813 template <typename Ret>
00814 static
00815 BackWriter *
00816 makeBackWriter (void (VisibilityIteratorImpl2::* setter) (Ret), Ret (VisBuffer2::* getter) () const) {
00817 return new BackWriterImpl <void (VisibilityIteratorImpl2:: *) (Ret),
00818 Ret (VisBuffer2:: *) () const >
00819 (setter, getter);
00820 }
00821
00822 template <typename Ret>
00823 static
00824 BackWriter *
00825 makeBackWriter2 (void (VisibilityIteratorImpl2::* setter) (Ret, VisibilityIteratorImpl2::DataColumn),
00826 Ret (VisBuffer2::* getter) () const,
00827 VisibilityIterator2::DataColumn dc) {
00828
00829
00830
00831 typedef void (VisibilityIteratorImpl2::* Setter) (Ret, VisibilityIteratorImpl2::DataColumn);
00832 typedef Ret (VisBuffer2::* Getter) () const;
00833
00834 return new BackWriterImpl2 < Setter, Getter> (setter, getter, dc);
00835 }
00836
00837 void initializeBackWriters ();
00838
00839 template <typename T>
00840 void putColumnRows (ArrayColumn<T> & column, const Array<T> & array);
00841
00842 template <typename T>
00843 void putColumnRows (ArrayColumn<T> & column, const Matrix<T> & array);
00844
00845 template <typename T>
00846 void putColumnRows (ScalarColumn<T> & column, const Vector <T> & array);
00847
00848
00849
00850
00851 class Cache {
00852
00853 public:
00854
00855 Cache ();
00856
00857 Cache & operator= (const Cache & other);
00858
00859 void flush ();
00860
00861 MDirection azel0_p;
00862 Double azel0Time_p;
00863 Vector<MDirection> azel_p;
00864 Double azelTime_p;
00865 Vector<uInt> chunkRowIds_p;
00866
00867 Vector<Float> feedpa_p;
00868 Double feedpaTime_p;
00869 Double hourang_p;
00870 Double hourangTime_p;
00871 Matrix<Float> imagingWeight_p;
00872 Bool msHasFlagCategory_p;
00873 Bool msHasWeightSpectrum_p;
00874 Bool msHasSigmaSpectrum_p;
00875 Float parang0_p;
00876 Double parang0Time_p;
00877 Vector<Float> parang_p;
00878 Double parangTime_p;
00879 };
00880
00881 class PendingChanges {
00882
00883 public:
00884
00885 PendingChanges ();
00886 ~PendingChanges ();
00887
00888 PendingChanges * clone () const;
00889 Bool empty () const;
00890
00891 pair<Bool, FrequencySelections *> popFrequencySelections ();
00892 pair<Bool, Double> popInterval ();
00893 pair<Bool, Int> popNRowBlocking ();
00894
00895 void setFrequencySelections (FrequencySelections *);
00896 void setInterval (Double);
00897 void setNRowBlocking (Int);
00898
00899 private:
00900
00901 enum {Empty = -1};
00902
00903 FrequencySelections * frequencySelections_p;
00904 Bool frequencySelectionsPending_p;
00905 Double interval_p;
00906 Int nRowBlocking_p;
00907 };
00908
00909 typedef Block <MeasurementSet> MeasurementSets;
00910
00911 class RowBounds {
00912 public:
00913
00914 RowBounds () : chunkNRows_p (-1), subchunkBegin_p (-1),
00915 subchunkEnd_p (-1), subchunkNRows_p (-1),
00916 subchunkRows_p (0, 0),
00917 timeMax_p (-1), timeMin_p (-1)
00918 {}
00919
00920 Int chunkNRows_p;
00921 Int subchunkBegin_p;
00922 Int subchunkEnd_p;
00923 Int subchunkNRows_p;
00924 RefRows subchunkRows_p;
00925 Vector<Double> times_p;
00926 Double timeMax_p;
00927 Double timeMin_p;
00928
00929 };
00930
00931 Bool autoTileCacheSizing_p;
00932 std::map <VisBufferComponent2, BackWriter *> backWriters_p;
00933 mutable Cache cache_p;
00934 const ChannelSelector * channelSelector_p;
00935 ChannelSelectorCache * channelSelectorCache_p;
00936 ViColumns2 columns_p;
00937 Bool floatDataFound_p;
00938 FrequencySelections * frequencySelections_p;
00939 VisImagingWeight imwgt_p;
00940 mutable Int measurementFrame_p;
00941 MeasurementSets measurementSets_p;
00942 VisModelDataI * modelDataGenerator_p;
00943 Bool more_p;
00944 Int msIndex_p;
00945 Bool msIterAtOrigin_p;
00946 CountedPtr<MSIter> msIter_p;
00947 mutable MSDerivedValues msd_p;
00948 Int nCorrelations_p;
00949 Int nRowBlocking_p;
00950 PendingChanges pendingChanges_p;
00951 Int reportingFrame_p;
00952
00953 RowBounds rowBounds_p;
00954 SortColumns sortColumns_p;
00955 mutable SpectralWindowChannelsCache * spectralWindowChannelsCache_p;
00956 Subchunk subchunk_p;
00957 SubtableColumns * subtableColumns_p;
00958 Vector<Bool> tileCacheIsSet_p;
00959 MeasRef<MEpoch> timeFrameOfReference_p;
00960 Double timeInterval_p;
00961 VisBuffer2 * vb_p;
00962 CountedPtr<WeightScaling> weightScaling_p;
00963 Bool writable_p;
00964 };
00965
00966 }
00967
00968 }
00969
00970 #endif
00971