PlotLoop.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2006 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/Plotting/Pulsar/PlotLoop.h
 
#ifndef __Pulsar_PlotLoop_h
#define __Pulsar_PlotLoop_h
 
#include "TextIndex.h"
#include <stack>
 
namespace Pulsar {
 
  class Archive;
  class Plot;
 
  class PlotLoop : public Reference::Able {
 
  public:
 
    PlotLoop ();
 
    void add_Plot (Plot*);
 
    void configure (const std::vector<std::string>& options);
 
    void finalize ();
 
    void set_Archive (Archive*);
 
    void set_overlay (bool);
    bool get_overlay () const;
 
    void set_stack (bool);
    bool get_stack () const;
 
    void set_colour_indeces (const std::vector<unsigned>& ci);
 
    void set_preprocess (bool);
 
    bool get_preprocess () const;
 
    void add_index (TextIndex*);
 
    // execute the plot for each index in the stack
    void plot ();
 
    // set the colour for the plot
    void set_colour_index (Plot*, unsigned colour_index);
 
  protected:
 
    std::vector< Reference::To<Plot> > plots;
 
    std::vector< Reference::To<Archive> > archives;
 
    bool preprocess;
    bool overlay;
    bool stack;
 
    std::stack< Reference::To<TextIndex> > index_stack;
 
    std::vector<unsigned> colour_indeces;
    unsigned current_colour_index;
 
    void plot ( std::stack< Reference::To<TextIndex> >& indeces );
 
  };
 
}
 
#endif