PowerSpectra.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2006 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/Plotting/Pulsar/PowerSpectra.h
 
#ifndef __Pulsar_PowerSpectra_h
#define __Pulsar_PowerSpectra_h
 
#include "Pulsar/FrequencyPlot.h"
#include "Pulsar/Index.h"
 
namespace Pulsar {
 
  class Profile;

  class PowerSpectra : public FrequencyPlot {
 
  public:

    PowerSpectra ();

    TextInterface::Parser* get_interface ();
 
    // Text interface to the PowerSpectra class
    class Interface : public TextInterface::To<PowerSpectra> {
    public:
      Interface (PowerSpectra* = 0);
    };

    void prepare (const Archive*);

    std::string get_ylabel (const Archive*);

    virtual void get_spectra (const Archive*) = 0;

    void set_subint (const Index& _isubint) { isubint = _isubint; }
    Index get_subint () const { return isubint; }
    
    void set_pol (const Index& _ipol) { ipol = _ipol; }
    Index get_pol () const { return ipol; }

    void set_median_window (unsigned window) { median_window = window; }
    unsigned get_median_window () const { return median_window; }

    void set_draw_lines (bool flag) { draw_lines = flag; }
    bool get_draw_lines () const { return draw_lines; }

    void set_skip_zapped (bool flag) { skip_zapped = flag; }
    bool get_skip_zapped () const { return skip_zapped; }

    void set_logarithmic (bool val) { logarithmic = val; }
    bool get_logarithmic () const { return logarithmic; }

    void draw (const Archive*);

    void draw (const std::vector<float>&) const;
 
  protected:
 
    std::vector< std::vector<float> > spectra;
    std::vector< int > plot_sci;
    std::vector< int > plot_sls;
 
    Index isubint;
    Index ipol;
 
    unsigned median_window;
    bool draw_lines;
    bool skip_zapped;
    bool logarithmic;
 
  private:
 
    std::vector<float> frequencies;
 
  };
 
}
 
#endif