ProfileAmps.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2002 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/Base/Classes/Pulsar/ProfileAmps.h
 
#ifndef __Pulsar_ProfileAmps_h
#define __Pulsar_ProfileAmps_h
 
#include "Pulsar/Container.h"
 
namespace Pulsar {
 
 
  class ProfileAmps : public Container {
 
  public:
 
    static bool no_amps;
 
    ProfileAmps (unsigned nbin = 0);
 
    virtual ~ProfileAmps ();
    
    ProfileAmps (const ProfileAmps&);
 
    unsigned get_nbin () const { return nbin; }
 
    virtual void resize (unsigned nbin);
 
    const float* get_amps () const;
 
    float* get_amps ();
 
    template <typename T> void set_amps (const T* data);
 
    template <typename T> void set_amps (const std::vector<T>& data);
 
    template <typename T> void get_amps (std::vector<T>& data) const;
 
 
    void remove (const std::vector<unsigned>& indeces);
 
    class Expert;
 
  private:
 
    friend class Expert;
 
    unsigned nbin;
 
    float *amps;
 
    unsigned amps_size;
 
  };
 
}
 
template <typename T>
void Pulsar::ProfileAmps::set_amps (const T* data)
{
  for (unsigned ibin=0; ibin<nbin; ibin++)
    amps[ibin] = static_cast<float>( data[ibin] );
}
 
template <typename T>
void Pulsar::ProfileAmps::set_amps (const std::vector<T>& data)
{
  resize (data.size());
  for (unsigned ibin=0; ibin<nbin; ibin++)
    amps[ibin] = static_cast<float>( data[ibin] );
}
 
template <typename T>
void Pulsar::ProfileAmps::get_amps (std::vector<T>& data) const
{
  data.resize (nbin);
  for (unsigned ibin=0; ibin<nbin; ibin++)
    data[ibin] = static_cast<T>( amps[ibin] );
}
 
#endif // !defined __Pulsar_ProfileAmps_h