.. _program_listing_file_xAODAnaHelpers_VertexContainer.h:

Program Listing for File VertexContainer.h
==========================================

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

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

.. code-block:: cpp

   #ifndef xAODAnaHelpers_VertexContainer_H
   #define xAODAnaHelpers_VertexContainer_H
   
   #include <TTree.h>
   #include <TLorentzVector.h>
   
   #include <vector>
   #include <string>
   
   #include <xAODAnaHelpers/HelperClasses.h>
   #include <xAODAnaHelpers/HelperFunctions.h>
   
   #include "xAODTracking/VertexContainer.h"
   #include "xAODTruth/TruthVertexContainer.h"
   
   namespace xAH {
   
       class VertexContainer
       {
       public:
         VertexContainer(const std::string& detailStr, const std::string& name = "vertex");
         virtual ~VertexContainer();
   
         virtual void setTree    (TTree *tree);
         virtual void setBranches(TTree *tree);
         virtual void clear();
         virtual void FillVertices( const xAOD::VertexContainer* vertices);
         virtual void FillTruthVertices( const xAOD::TruthVertexContainer* truthVertices);
   
         std::string m_name;
   
         std::string branchName(const std::string& varName)
         {
           std::string name = m_name + "_" + varName;
           return name;
         }
   
         template <typename T_BR> void connectBranch(TTree *tree, const std::string& branch, std::vector<T_BR> **variable)
         {
           std::string name = branchName(branch);
           if(*variable) { delete (*variable); (*variable)=0; }
           if(tree->GetBranch(name.c_str()))
             {
               (*variable)=new std::vector<T_BR>();
               tree->SetBranchStatus  (name.c_str()  , 1);
               tree->SetBranchAddress (name.c_str()  , variable);
             }
         }
   
         template<typename T> void setBranch(TTree* tree, std::string varName, std::vector<T>* localVectorPtr){
           std::string name = branchName(varName);
           tree->Branch(name.c_str(),        localVectorPtr);
         }
   
       private:
         // Vector branches
         std::vector<float>* m_vertex_x;
         std::vector<float>* m_vertex_y;
         std::vector<float>* m_vertex_z;
         std::string         m_detailStr;
       };
   
   }
   
   #endif // xAODAnaHelpers_VertexContainer_H