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