eFEXNtupleWriter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EFEXTUPLEWRITER_H
#define EFEXTUPLEWRITER_H
#include "AthenaBaseComps/AthAlgorithm.h"
#include "TTree.h"
#include "GaudiKernel/ITHistSvc.h"
#include "xAODTruth/TruthParticle.h"
#include <memory>
#include "L1CaloFEXSim/eFEXOutputCollection.h"
 
namespace LVL1 {
class eFEXNtupleWriter : public AthAlgorithm
{
public:
  eFEXNtupleWriter(const std::string& name, ISvcLocator* pSvcLocator);
 
  ~eFEXNtupleWriter();
 
  StatusCode initialize ();
 
  StatusCode execute ();
 
  StatusCode finalize ();
 
private:
  SG::ReadHandleKey<LVL1::eFEXOutputCollection> m_eFEXOutputCollectionSGKey {this, "MyOutputs", "eFEXOutputCollection", "MyOutputs"};
 
  int m_eg_nTOBs; 
  bool m_load_truth_jet; 
 
  // values from the e-gamma algorithm
  std::vector<float> m_em;
  std::vector<float> m_had;
  std::vector<float> m_truth_e_eta;
  std::vector<float> m_truth_e_phi;
  std::vector<float> m_truth_e_ET;
  std::vector<float> m_truth_jet_eta;
  std::vector<float> m_truth_jet_phi;
  std::vector<float> m_truth_jet_ET;
  std::vector<float> m_truth_tauvisible_eta;
  std::vector<float> m_truth_tauvisible_phi;
  std::vector<float> m_truth_tauvisible_ET;
  std::vector<float> m_eg_haveseed;
  std::vector<float> m_eg_eta;
  std::vector<float> m_eg_phi;
  std::vector<float> m_eg_ET;
  std::vector<float> m_eg_WstotNum;
  std::vector<float> m_eg_WstotDen;
  std::vector<float> m_eg_RetaNum;
  std::vector<float> m_eg_RetaDen;
  std::vector<float> m_eg_RhadNum;
  std::vector<float> m_eg_RhadDen;
 
  // values from the tau algorithm
  std::vector<float> m_tau_realRCore;
  std::vector<float> m_tau_rCoreCore;
  std::vector<float> m_tau_rCoreEnv;
  std::vector<float> m_tau_rCoreWP;
  std::vector<float> m_tau_realRHad;
  std::vector<float> m_tau_rHadCore;
  std::vector<float> m_tau_rHadEnv;
  std::vector<float> m_tau_rHadWP;
  std::vector<float> m_tau_seed;
  std::vector<float> m_tau_und;
  std::vector<float> m_tau_Et;
  std::vector<float> m_tau_Eta;
  std::vector<float> m_tau_Phi;
  std::vector<float> m_tau_floatEta;
  std::vector<float> m_tau_floatPhi;
  std::vector<float> m_tau_isCentralTowerSeed;
  std::vector<float> m_tau_CenterTowerEt;
  std::vector<float> m_tau_OneOffEtaTowerEt;
  std::vector<float> m_tau_OneBelowEtaTowerEt;
  std::vector<float> m_tau_eFEXID;
  std::vector<float> m_tau_FPGAID;
  
  std::vector<int> m_eFex_number; 
  std::vector<uint32_t> m_eg_tob; 
 
  // variables in eg TOB
  std::vector<float> m_eg_TOB_FP;
  std::vector<float> m_eg_TOB_Eta;
  std::vector<float> m_eg_TOB_Phi;
  std::vector<float> m_eg_TOB_ha;
  std::vector<float> m_eg_TOB_f3;
  std::vector<float> m_eg_TOB_Re;
  std::vector<float> m_eg_TOB_Sd;
  std::vector<float> m_eg_TOB_UnD;
  std::vector<float> m_eg_TOB_Max;
  std::vector<float> m_eg_TOB_zeros;
  std::vector<float> m_eg_TOB_energy;
  
  std::string m_jet_container_name = "AntiKt10TruthJets"; 
  TTree *m_myTree;
  
  StatusCode loadegAlgoVariables(SG::ReadHandle<LVL1::eFEXOutputCollection>);
 
  StatusCode loadegAlgoTOBs(SG::ReadHandle<LVL1::eFEXOutputCollection>);
 
  StatusCode loadtauAlgoVariables(SG::ReadHandle<LVL1::eFEXOutputCollection>);
 
  StatusCode loadTruthElectron();
 
  StatusCode loadTruthJets();
 
  StatusCode loadTruthTau();
 
  std::unique_ptr<TLorentzVector> visibleTauP4(const xAOD::TruthParticle*);
 
  std::unique_ptr<TLorentzVector> invisibleTauP4(const xAOD::TruthParticle*);
 
  const xAOD::TruthParticle* getMother(const xAOD::TruthParticle*, int);
 
  uint32_t getbits(uint32_t in, int start, int end);
};
}
#endif