PlotFactory.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2006-2010 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/Plotting/Pulsar/PlotFactory.h
 
#ifndef __Pulsar_PlotFactory_h
#define __Pulsar_PlotFactory_h
 
#include "Pulsar/Plot.h"
 
#include <string>
#include <vector>
 
namespace Pulsar {
 
  class PlotFactory {
 
  public:
 
    PlotFactory ();
 
    Plot* construct (std::string name);
 
    unsigned get_nplot () { return agents.size(); }
 
    std::string get_name (unsigned i);
 
    std::string get_description (unsigned i);
 
    std::string help();
 
  protected:
 
    class Agent;
    
    std::vector<Agent*> agents;
 
  public:
 
    template<class P> class Advocate;
 
    void add (Agent* agent) { agents.push_back(agent); }
 
  };
 
  class PlotFactory::Agent : public Plot::Constructor
  {
  public:
 
    Agent (char c, std::string n, std::string d)
      : shortcut(c), name (n), description (d) {}
 
    std::string get_name () { return name; }
 
    char get_shortcut () { return shortcut; }
 
    std::string get_description () { return description; }
 
  protected:
 
    char shortcut;
    std::string name;
    std::string description;
 
  };
 
  template<class P> class PlotFactory::Advocate : public Agent
  {  
  public:
 
    Advocate (std::string _name, std::string _description)
      : Agent (' ', _name, _description) {}
 
    Advocate (char shortcut, std::string _name, std::string _description)
      : Agent (shortcut, _name, _description) {}
 
    Plot* construct ()
    {
      P* plot = new P;
      plot->set_constructor(this);
      return plot;
    }
 
  };
 
}
 
#endif