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