ProjectionTreePlotter.hh

// -*- C++ -*-
#include "Rivet/Analysis.hh"
#include "Rivet/AnalysisHandler.hh"
#include "Rivet/ProjectionHandler.hh"
#include "Rivet/Tools/Logging.hh"
 
#ifndef RIVET_RivetProjectionTree_HH
  #define RIVET_RivetProjectionTree_HH
 
 
namespace Rivet {

  class ProjectionTreeGenerator {
  private:


    std::string _path;
 
    //Title of digraph (by default the last bit of the name minus the '.gv')
    std::string _title;

    bool _treeGenerated;
 
    //Tracks how many analyses are involved for the projection tree
    size_t _nAnalyses;



    std::vector<Rivet::ConstProjectionPtr> _projVector;

    std::vector<std::pair<size_t, size_t>> _edgeVector;

    std::vector<std::string> _nameVector; //Should be 1<->1 with _projVector;

 
 
  public:

    ProjectionTreeGenerator(const std::string& path);

    ProjectionTreeGenerator();

    void setPath(const std::string& path);

    void setTitle(const std::string& title);

    int generateProjTree(const std::vector<std::string>& analyses);

    int getProjTree(const AnalysisHandler& ah);

    void writeProjTree() const;

    inline const std::vector<string>& getProjNames() const {
      return _nameVector;
    }

    inline const std::vector<std::pair<size_t, size_t>>& getEdges() const {
      return _edgeVector;
    }

    Log& getLog() const;
  };
}
 
#endif