Program Listing for File TruthContainer.h

↰ Return to documentation for file (xAODAnaHelpers/TruthContainer.h)

#ifndef xAODAnaHelpers_TruthContainer_H
#define xAODAnaHelpers_TruthContainer_H

#include <TTree.h>
#include <TLorentzVector.h>

#include <vector>
#include <string>

#include "xAODTruth/TruthParticle.h"

#include <xAODAnaHelpers/HelperClasses.h>
#include <xAODAnaHelpers/HelperFunctions.h>

#include <xAODAnaHelpers/TruthPart.h>
#include <xAODAnaHelpers/ParticleContainer.h>

namespace xAH {

    class TruthContainer : public ParticleContainer<TruthPart,HelperClasses::TruthInfoSwitch>
    {
    public:
      TruthContainer(const std::string& name = "truth", const std::string& detailStr="", float units = 1e3);
      virtual ~TruthContainer();

      virtual void setTree    (TTree *tree);
      virtual void setBranches(TTree *tree);
      virtual void clear();
      virtual void FillTruth( const xAOD::TruthParticle* truth );
      virtual void FillTruth( const xAOD::IParticle* particle );
      using ParticleContainer::setTree; // make other overloaded version of execute() to show up in subclass

    protected:

      virtual void updateParticle(uint idx, TruthPart& truth);

    private:


      //
      // Vector branches
      //

      // All
      std::vector<int>* m_pdgId;
      std::vector<int>* m_status;
      std::vector<int>* m_barcode;

      // type
      std::vector<int>* m_is_higgs;
      std::vector<int>* m_is_bhad;

      // bVtx
      std::vector<float>* m_Bdecay_x;
      std::vector<float>* m_Bdecay_y;
      std::vector<float>* m_Bdecay_z;

      // parents
      std::vector<int>* m_nParents;
      std::vector< std::vector<int> >* m_parent_pdgId;
      std::vector< std::vector<int> >* m_parent_barcode;
      std::vector< std::vector<int> >* m_parent_status;

      // children
      std::vector<int>* m_nChildren;
      std::vector< std::vector<int> >* m_child_pdgId;
      std::vector< std::vector<int> >* m_child_barcode;
      std::vector< std::vector<int> >* m_child_status;

      // dressed
      std::vector<float>* m_pt_dressed;
      std::vector<float>* m_eta_dressed;
      std::vector<float>* m_phi_dressed;
      std::vector<float>* m_e_dressed;

      // origin
      std::vector<unsigned int>* m_origin;

      // particle type
      std::vector<unsigned int>* m_particleType;

    };
}



#endif // xAODAnaHelpers_TruthContainer_H