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