CorrelatorDataFrameManager.h

Go to the documentation of this file.

#ifndef SZA_UTIL_CORRELATORDATAFRAMEMANAGER_H
#define SZA_UTIL_CORRELATORDATAFRAMEMANAGER_H

#include <map>

#include "carma/szautil/Directives.h"

#if DIR_USE_CORR
#include "carma/correlator/lib/CorrelatorData.h"
#endif

// C header files from the array control code

#include "carma/szaarrayutils/szaregs.h"

#include "carma/szautil/CorrelatorBand.h"
#include "carma/szautil/RegMapDataFrameManager.h"
#include "carma/szautil/TimeVal.h"


#include <complex>

namespace cobra {
  class CorrelatorBand;
  class CorrelatorData;
  class AutoSpectra;
  class CrossSpectra;
}

namespace sza {
  namespace util {
    
    class DataFrame;

    class CorrelatorDataFrameManager : public RegMapDataFrameManager {
    public:
      
      CorrelatorDataFrameManager(CorrelatorDataFrameManager& fm);

      CorrelatorDataFrameManager(bool archivedOnly=false, DataFrame* frame=0);

#if DIR_USE_CORR

      CorrelatorDataFrameManager(bool archivedOnly, 
                                 carma::correlator::lib::
                                 CorrelatorData& corrData);

      void setTo(carma::correlator::lib::CorrelatorData& corrData);

      void setTo(cobra::CorrelatorBand& band, unsigned char antMask=0xff);

#endif

      virtual ~CorrelatorDataFrameManager();
      
      void initialize(bool archivedOnly=false, sza::util::DataFrame* frame = 0);

      void reinitialize();
      
      std::map<unsigned int, std::map<unsigned int, unsigned int> > 
        crossBaselineIndex_;

      void setAntennaBasedGain(Complex<float> gain, 
                               unsigned iAnt, unsigned iBand, unsigned iChan);

      void printAntennaBasedGains();

      void setBandReceived(CorrelatorBand::Id id, bool received);

      void setBandReceived(CorrelatorBand& iBand, bool received);

      void setCrossBaselineReceived(unsigned int iBase, bool received);

      void initializeSources();

      bool isNoise();

      bool isRf();

    private:      

      // Statistics about correlator sources

      bool isRf_;
      bool isNoise_;

      // Arrays in which visibilities will be stored until ready to write

      std::vector<Complex<float>::Data> usbAvg_;
      std::vector<Complex<float>::Data> lsbAvg_;
      std::vector<Complex<float>::Data> usbVar_;
      std::vector<Complex<float>::Data> lsbVar_;

      std::vector<float> usbAmplitude_;
      std::vector<float> lsbAmplitude_;

      std::vector<float> usbAvgAmplitude_;
      std::vector<float> lsbAvgAmplitude_;

      std::vector<Complex<float>::Data> usb_;
      std::vector<Complex<float>::Data> lsb_;

      std::vector<float> autoAvg_;
      std::vector<float> autoVar_;
      std::vector<float> auto_;

      std::vector<unsigned char> baselineReceived_;

      // A static array of antennaBasedComplexGains

      static std::vector<std::vector<std::vector<Complex<float> > > > 
        antennaBasedGains_;

      static bool gainsAreInitialized_;

#if DIR_USE_CORR

      void packCorrData(carma::correlator::lib::CorrelatorData& corrData);

      void packBand(carma::correlator::lib::CorrelatorBand& corrBand);

      void packBaseline(carma::correlator::lib::CorrelatorBaseline& baseline);
      
      void packSideBand(carma::correlator::lib::CorrelatorSideband* sideband,
                        unsigned int iAnt1, unsigned int iAnt2);

      std::string bandName(carma::correlator::lib::CorrelatorBand& corrBand);

#endif

#if DIR_USE_CORR

      void packCorrData(cobra::CorrelatorBand& corrBand, unsigned char antMask=0xff);

      void packSpectrum(cobra::CorrelatorData* spectrum, bool reverseFreqOrder=false, unsigned char antMask=0xff);

      void packAutoSpectrum(cobra::AutoSpectra* spectrum, bool reverseFreqOrder=false);

      void packCrossSpectrum(cobra::CrossSpectra* spectrum, bool reverseFreqOrder=false, unsigned char antMask=0xff);

      void packAuto(unsigned iAnt, float* data, bool reverseFreqOrder=false);

      void packLsb(unsigned iBase, std::complex<float>* data, bool reverseFreqOrder=false);

      void packUsb(unsigned iBase, std::complex<float>* data, bool reverseFreqOrder=false);

      void packCross(unsigned iBase, std::complex<float>* data, 
                     Complex<float>::Data* ptr, 
                     Complex<float>::Data* avgPtr, 
                     Complex<float>::Data* varPtr,
                     float* ampPtr,
                     float* avgAmpPtr,
                     bool reverseFreqOrder=false);

#endif

      std::string bandName(unsigned iBand);

      void writeRegisters(std::string& bName, bool swapSidebands);

      void addSource(std::string& source);

    }; // End class CorrelatorDataFrameManager
    
  } // End namespace util
} // End namespace sza



#endif // End #ifndef SZA_UTIL_CORRELATORDATAFRAMEMANAGER_H