.. _program_listing_file_xAODAnaHelpers_TreeAlgo.h:

Program Listing for File TreeAlgo.h
===================================

|exhale_lsh| :ref:`Return to documentation for file <file_xAODAnaHelpers_TreeAlgo.h>` (``xAODAnaHelpers/TreeAlgo.h``)

.. |exhale_lsh| unicode:: U+021B0 .. UPWARDS ARROW WITH TIP LEFTWARDS

.. code-block:: cpp

   #ifndef xAODAnaHelpers_TreeAlgo_H
   #define xAODAnaHelpers_TreeAlgo_H
   
   #include "TTree.h"
   
   #include <xAODAnaHelpers/HelpTreeBase.h>
   
   // algorithm wrapper
   #include "xAODAnaHelpers/Algorithm.h"
   
   class TreeAlgo : public xAH::Algorithm
   {
     // put your configuration variables here as public variables.
     // that way they can be set directly from CINT and python.
   public:
     // choose whether the tree gets saved in the same directory as output histograms
     bool m_outHistDir = false;
     std::string m_treeStreamName = "tree";
   
     // holds bools that control which branches are filled
     std::string m_evtDetailStr = "";
     std::string m_trigDetailStr = "";
     std::string m_muDetailStr = "";
     std::string m_elDetailStr = "";
     std::string m_jetDetailStr = "";
     std::string m_trigJetDetailStr = "";
     std::string m_truthJetDetailStr = "";
     std::string m_fatJetDetailStr = "";
     std::string m_truthFatJetDetailStr = "";
     std::string m_tauDetailStr = "";
     std::string m_METDetailStr = "";
     std::string m_METReferenceDetailStr = "";
     std::string m_photonDetailStr = "";
     std::string m_clusterDetailStr = "";
     std::string m_truthParticlesDetailStr = "";
     std::string m_trackParticlesDetailStr = "";
     std::string m_vertexDetailStr = "";
   
     std::string m_evtContainerName = "";
     std::string m_muContainerName = "";
     std::string m_elContainerName = "";
     std::string m_jetContainerName = "";
     std::string m_jetBranchName = "jet";
     std::string m_truthJetContainerName = "";
     std::string m_truthJetBranchName = "truthJet";
     std::string m_trigJetContainerName = "";
     std::string m_trigJetBranchName = "trigJet";
     std::string m_fatJetContainerName = "";
     std::string m_fatJetBranchName = ""; // default is to use container name
     std::string m_truthFatJetContainerName = "";
     std::string m_truthFatJetBranchName = "truth_fatjet";
     std::string m_tauContainerName = "";
     std::string m_METContainerName = "";
     std::string m_METReferenceContainerName = "";
     std::string m_photonContainerName = "";
     std::string m_clusterContainerName = "";
     std::string m_clusterBranchName = "CaloCalTopoClusters";
     std::string m_truthParticlesContainerName = "";
     std::string m_truthParticlesBranchName = "xAH_truth";
     std::string m_trackParticlesContainerName = "";
     std::string m_l1JetContainerName = "";
     std::string m_l1JetBranchName    = "L1Jet";
     std::string m_vertexBranchName   = "vertex";
     bool m_sortL1Jets = false;
     bool m_retrievePV = true;
   
     // if these are set, assume systematics are being processed over
     std::string m_muSystsVec = "";
     std::string m_elSystsVec = "";
     std::string m_tauSystsVec = "";
     std::string m_jetSystsVec = "";
     std::string m_photonSystsVec = "";
     std::string m_fatJetSystsVec = "";
     std::string m_metSystsVec = "";
   
     float m_units = 1e3;
   
     int m_autoFlush = 0;
   
   protected:
     std::vector<std::string> m_jetDetails; 
     std::vector<std::string> m_trigJetDetails; 
     std::vector<std::string> m_fatJetDetails; 
   
     std::vector<std::string> m_jetContainers; 
     std::vector<std::string> m_truthJetContainers; 
     std::vector<std::string> m_trigJetContainers; 
     std::vector<std::string> m_fatJetContainers; 
     std::vector<std::string> m_l1JetContainers; 
     std::vector<std::string> m_vertexContainers; 
     std::vector<std::string> m_truthParticlesContainers; 
   
     std::vector<std::string> m_jetBranches; 
     std::vector<std::string> m_truthJetBranches; 
     std::vector<std::string> m_trigJetBranches; 
     std::vector<std::string> m_fatJetBranches; 
     std::vector<std::string> m_l1JetBranches; 
     std::vector<std::string> m_vertexBranches; 
     std::vector<std::string> m_truthParticlesBranches; 
   
     std::vector<std::string> m_clusterDetails; 
     std::vector<std::string> m_clusterContainers; 
     std::vector<std::string> m_clusterBranches; 
   
     std::vector<std::string> m_vertexDetails; 
   
     std::map<std::string, HelpTreeBase*> m_trees;            
   
   public:
   
     // this is a standard constructor
     TreeAlgo ();
   
     // these are the functions inherited from Algorithm
     virtual EL::StatusCode setupJob (EL::Job& job);           
     virtual EL::StatusCode fileExecute ();                    
     virtual EL::StatusCode histInitialize ();                 
     virtual EL::StatusCode changeInput (bool firstFile);      
     virtual EL::StatusCode initialize ();                     
     virtual EL::StatusCode execute ();                        
     virtual EL::StatusCode postExecute ();                    
     virtual EL::StatusCode finalize ();                       
     virtual EL::StatusCode histFinalize ();                   
   
     // Help tree creator function
     virtual HelpTreeBase* createTree(xAOD::TEvent *event, TTree* tree, TFile* file, const float units, bool debug, xAOD::TStore* store); 
   
     // this is needed to distribute the algorithm to the workers
     ClassDef(TreeAlgo, 1);                                 
   };
   
   #endif