QChannelCovariance.hh

Go to the documentation of this file.

#ifndef _Q_CHANNELCOVARIANCE_HH_
#define _Q_CHANNELCOVARIANCE_HH_
 
#include "QObject.hh"
#include "QMatrixC.hh"
#include "QMatrix.hh"
#include "QVectorC.hh"
#include <vector>
#include <map>
 
class QChannelCovariance : public Diana::QObject {
 
public:
  QChannelCovariance();  
 
  QChannelCovariance(size_t nfreq, const std::vector<int> &channels);
  
  void Clear() 
  {
    // mandatory method: reset members to default
    fCovariance.clear();
    fChannels.clear();
    fNevents.Clear();
  }
 
  void SetNFreq(size_t nfreq);
  void SetChannels(const std::vector<int> &channels);
  void SetCovariance(int chI, int chJ,const Diana::QVectorC& cov);
  void SetNEvents(int chI, int chJ, int nev);
 
  inline size_t GetNFreq() const {return fCovariance.size();};
  std::vector<int> GetChannels() const;
  inline size_t GetNChan() const {return fChannels.size();};
  Diana::QVectorC GetCovariance(int chI, int chJ) const;
  Diana::QVectorC GetCovarianceByIndex(int i, int j) const;
  int GetNEvents(int chI, int chJ) const;
 
  QChannelCovariance Inverse() const;
  QChannelCovariance GetSubMatrix(const std::vector<int> &channels) const;
  Diana::QVectorC GetCorrelation(int chI, int chJ) const;
  double GetMeanCorrelation(int chI,int chJ, int fmin = -1, int fmax = -1) const;
 
  std::vector<int> GetChannelsByCorrelation(int chI, int fmin = -1, int fmax = -1) const;
 
 
  void DumpMeanCorrelationMatrix(std::ostream& o,const int fmin =-1, const int fmax = -1) const;
 
  void Dump(std::ostream& o) const;
  
 
private:
 
  void Channels2Indices(const int chI, const int chJ, int& i, int& j) const;
  /* @brief vector of QMatrixC. 
   * The  index of the vector is the frequency and the indexes of the
   * matrix are the channel indices
   */
  std::vector<Diana::QMatrixC> fCovariance;
 
  std::map<int,int> fChannels;
 
  Diana::QMatrix fNevents;
  
 
  QObjectDef(QChannelCovariance,2);
  
};
 
 
 
#endif