StokesCovariancePlot.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2014 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/Plotting/Pulsar/StokesCovariancePlot.h
 
#ifndef __Pulsar_StokesCovariancePlot_h
#define __Pulsar_StokesCovariancePlot_h
 
#include "Pulsar/FluxPlot.h"
 
namespace Pulsar {

  class StokesCovariancePlot : public FluxPlot {
 
  public:

    StokesCovariancePlot ();

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

    void get_profiles (const Archive* data);
 
    enum What { Variances,
                EigenValues,
                NaturalCovariances,
                NonOrthogonality,
                Theta,
                RegressionCoefficients,
                ModulationIndex,
                DegreeOfPolarization};
 
    void set_what (What w) { what = w; }
    What get_what () const { return what; }

    void set_plot_colours (const std::string& plot) { plot_colours = plot; }
    std::string get_plot_colours () const { return plot_colours; }

    void set_plot_lines (const std::string& plot) { plot_lines = plot; }
    std::string get_plot_lines () const { return plot_lines; }

    void set_plot_widths (const std::string& plot) { plot_widths = plot; }
    std::string get_plot_widths () const { return plot_widths; }

    void set_std_deviations (bool flag) { plot_std_deviations = flag; }
    bool get_std_deviations () const { return plot_std_deviations; }

    void set_remove_eigen_baseline (bool flag) { remove_eigen_baseline = flag; }
    bool get_remove_eigen_baseline () const { return remove_eigen_baseline; }
 
  private:
 
    What what;
 
    std::string plot_colours;
    std::string plot_lines;
 
    std::string plot_widths;
 
    bool plot_std_deviations;
    bool remove_eigen_baseline;
  };
 
  std::ostream& operator << (std::ostream& os, StokesCovariancePlot::What);
 
  std::istream& operator >> (std::istream& is, StokesCovariancePlot::What&);
}
 
#endif