Transposer.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2004 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/General/Pulsar/Transposer.h
 
#ifndef __Pulsar_Transposer_h
#define __Pulsar_Transposer_h
 
#include "Pulsar/Algorithm.h"
 
#include "Reference.h"
#include "Types.h"
 
namespace Pulsar {
  
  class Archive;
  class Integration;
  class Profile;
  class Dimensions;

  class Transposer : public Algorithm {
 
  public:

    Transposer (const Archive* archive = 0);

    ~Transposer ();

    void set_Archive (const Pulsar::Archive* archive);

    void set_dim (unsigned idim, Signal::Dimension dim);

    unsigned get_ndim (unsigned idim);

    Dimensions get_stride (const Dimensions& dimensions) const;

    Dimensions get_stride () const;

    void get_amps (std::vector<float>& amps) const;

    void get_amps (const Integration*, float*, const Dimensions& stride) const;

    void get_amps (const Profile*, float*, unsigned jbin) const;

    bool verbose;
 
  protected:
 
    void range_check (unsigned idim, const char* method) const;
 
    Reference::To<const Archive> archive;
    Signal::Dimension dim[4];
 
  };

  class Dimensions {
 
  public:

    Dimensions (const Pulsar::Archive* archive = 0);
    
    unsigned get_ndim (Signal::Dimension axis) const;
 
  protected:
 
    friend class Transposer;

    void set_if_zero (unsigned ndim);

    void set_ndim (Signal::Dimension axis, unsigned ndim);
 
    unsigned nsub;
    unsigned npol;
    unsigned nchan;
    unsigned nbin;
 
  };
 
}
 
#endif