ProjectionTreePlotter.hh

// -*- C++ -*-
#include "Rivet/Tools/Logging.hh"
#include "Rivet/Analysis.hh"
#include "Rivet/Tools/Logging.hh"
#include "Rivet/AnalysisHandler.hh"
#include "Rivet/ProjectionHandler.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