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