CrossCovarianceMatrix.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2016 by Aditya Parthasarathy & Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/Base/Extensions/Pulsar/CrossCovarianceMatrix.h
 
#ifndef __CrossCovarianceMatrix_h
#define __CrossCovarianceMatrix_h
 
#include "Pulsar/ArchiveExtension.h"
#include "Pulsar/Profile.h"
 
#include <vector> 
 
namespace Pulsar {
 
  class CrossCovarianceMatrix : public Pulsar::Archive::Extension
  {
 
  public:
 
    CrossCovarianceMatrix ();

    CrossCovarianceMatrix (const CrossCovarianceMatrix& extension);  

    const CrossCovarianceMatrix& operator= (const CrossCovarianceMatrix&);

    ~CrossCovarianceMatrix ();

    CrossCovarianceMatrix* clone () const 
    { return new CrossCovarianceMatrix( *this ); }

    void set_nbin (unsigned);
    
    unsigned get_nbin () const;

    void set_npol (unsigned);
    
    unsigned get_npol () const;

    void set_nlag (unsigned);

    unsigned get_nlag () const;

    unsigned get_ncross (unsigned ilag) const;

    unsigned get_ncross_total () const;
 
    class Stream;

    bool has_stream () const;

    void set_stream (Stream*);

    const Stream* get_stream () const;

    void resize_data ();
    
    std::vector<double>& get_data () { return covariance; }
    
    const std::vector<double>& get_data () const { return covariance; }
    
  protected:
 
    unsigned nbin;
    unsigned npol;
    unsigned nlag;
 
    std::vector<double> covariance;
 
    Reference::To<Stream> stream;
            
  };

  class CrossCovarianceMatrix::Stream : public Reference::Able
  {
    public:
      typedef double Type;
 
      virtual unsigned get_ndat () const = 0;
      virtual void get_data (unsigned off, unsigned n, double*) const = 0;
      virtual void verify_end_of_data () const = 0;
  };
 
}
 
#endif