StokesCrossCovariance.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2009 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/Polarimetry/Pulsar/StokesCrossCovariance.h
 
#ifndef __Pulsar_StokesCrossCovariance_h
#define __Pulsar_StokesCrossCovariance_h
 
#include "Pulsar/Container.h"
 
#include "Stokes.h"
#include "Jones.h"
#include "Matrix.h"
 
namespace Pulsar {
 
  class CrossCovarianceMatrix;
 
 
  class StokesCrossCovariance : public Container
  {
    
  public:
 
    StokesCrossCovariance (const CrossCovarianceMatrix* = 0);
 
    void load (const CrossCovarianceMatrix*);
 
    void unload (CrossCovarianceMatrix*);
 
    class Stream;
    
    virtual StokesCrossCovariance* clone () const;
 
    unsigned get_nbin () const;
    void set_nbin (unsigned);
 
    unsigned get_nlag () const;
    void set_nlag (unsigned);
 
    void resize ();
 
    void set_all (double val = 0.0);
 
    Matrix<4,4,double> get_cross_covariance (unsigned ibin,
                                             unsigned jbin,
                                             unsigned ilag = 0) const;
 
    Matrix<4,4,double>& get_cross_covariance (unsigned ibin,
                                              unsigned jbin,
                                              unsigned ilag = 0);
 
    void set_cross_covariance (unsigned ibin, unsigned jbin,
                               const Matrix<4,4,double>&, bool lock = false);
 
    void set_cross_covariance (unsigned ibin, unsigned jbin, unsigned ilag,
                               const Matrix<4,4,double>&, bool lock = false);
 
    unsigned get_icross (unsigned ibin, unsigned jbin, unsigned ilag=0) const;
 
    unsigned get_ncross (unsigned ilag) const;
 
    unsigned get_ncross_total () const;
 
  protected:
 
    unsigned nbin;
    unsigned nlag;
 
    unsigned get_icross_check (unsigned ibin, unsigned jbin, unsigned ilag,
                               const char*) const;
    
    std::vector< Matrix<4,4,double> > cross_covariance;
    std::vector< bool > locked;
  };
 
}
 
 
#endif