Shape.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2002 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// dspsr/Signal/General/dsp/Shape.h
 
#ifndef __Shape_h
#define __Shape_h
 
#include "OwnStream.h"
 
#ifdef ACTIVATE_MPI
#include <mpi.h>
#endif
 
namespace dsp {
 
  class Shape : public OwnStream {
 
  public:
    static bool verbose;
 
    Shape ();
    virtual ~Shape ();
    Shape (const Shape&);
    const Shape& operator = (const Shape&);
 
    virtual void resize (unsigned npol, unsigned nchan,
                         unsigned ndat, unsigned ndim);
 
    unsigned get_npol()  const { return npol; }
 
    unsigned get_nchan() const { return nchan; }
 
    unsigned get_ndat()  const { return ndat; }
 
    unsigned get_ndim()  const { return ndim; }
 
    virtual bool matches (const Shape* shape);
 
    void scrunch_to (unsigned ndat);
 
    void rotate (int npt);
 
    void zero ();
 
    void zero_dimensions ();
 
    void borrow (const Shape&, unsigned ichan=0);
 
    const Shape& operator /= (float factor);
 
    const Shape& operator *= (float factor);
 
    const Shape& operator += (const Shape&);
 
    float* get_datptr (unsigned ichan, unsigned ipol) {
      return buffer + offset * ipol + ndat*ndim * ichan;
    }
 
    const float* get_datptr (unsigned ichan, unsigned ipol) const {
      return buffer + offset * ipol + ndat*ndim * ichan;
    }
 
  protected:
 
    float* buffer;
 
    unsigned bufsize;
 
    unsigned offset;
 
    unsigned npol;
 
    unsigned nchan;
 
    unsigned ndat;
 
    unsigned ndim;
 
    bool borrowed;
 
    void init ();
    void size_dataspace ();
    void destroy ();
 
#if PGPLOT
    void plot (float centre, float width, const char* label,
                bool swap=false, int dimension=0, bool one_poln=false);
#endif
 
  };
 
}
 
// these are declared outside of the namespace, to simplify their
// use in the stdmpi templates
#ifdef ACTIVATE_MPI
int mpiPack_size (const dsp::Shape&,
                  MPI_Comm comm, int* size);
int mpiPack      (const dsp::Shape&, void* outbuf,
                  int outcount, int* position, MPI_Comm comm);
int mpiUnpack    (void* inbuf, int insize, int* position,
                  dsp::Shape*, MPI_Comm comm);
#endif
 
#endif // #ifndef __Shape_h