JetTruthNtupleTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// JetTruthNtupleTool.h
//   Header file for JetTruthNtupleTool
// (c) ATLAS Detector software
// Wolfgang.Liebig@cern.ch
 
#ifndef TRK_JETTRUTHNTUPLETOOL_H
#define TRK_JETTRUTHNTUPLETOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "xAODEventInfo/EventInfo.h"
 
#include "TrkValInterfaces/IJetTruthNtupleTool.h"
#include <vector>
 
class TTree;
 
namespace Trk {
 
 
class JetTruthNtupleTool : virtual public Trk::IJetTruthNtupleTool, public AthAlgTool {
public:
 
    // standard AlgToolmethods
    JetTruthNtupleTool(const std::string&,const std::string&,const IInterface*);
    ~JetTruthNtupleTool();
 
    // standard Athena methods
    StatusCode initialize();
    StatusCode finalize();
 
    virtual StatusCode writeJetTruthData (const std::vector< Trk::GenParticleJet >&,
                                          const int& );
 
    virtual unsigned int getNumberOfTreeRecords() const;
 
private:
 
    // jobOptions
    std::string m_ntupleTreeName;       
    std::string m_ntupleFileName;       
    std::vector<double> m_etaBins;
 
    TTree* m_nt; 
 
    unsigned int m_numberOfTreeEntries;
 
    // ntuple variables
    int         m_runNumber;
    int         m_eventNumber;
 
    // truth jet information
    std::vector<unsigned int>*   m_genParticleLinkIndex;
    std::vector<float>* m_mc_trackToJetAngle;
    float       m_mc_jetMultiplicity;
    float       m_mc_jetSphericity;
    float       m_mc_jetThrust;
    float       m_mc_jetEnergy;
    float       m_mc_jetMomentum;
 
    SG::ReadHandleKey<xAOD::EventInfo>    m_evt  {this, "EvtInfo", "EventInfo", "EventInfo name"};
};
 
 
} // end of namespace
 
#endif // TRK_TRUTHNTUPLETOOL_H