#include <HIGlobalAugmentationTool.h>

Inheritance diagram for DerivationFramework::HIGlobalAugmentationTool:

Collaboration diagram for DerivationFramework::HIGlobalAugmentationTool:

Public Member Functions
virtual StatusCode	initialize () override final
virtual StatusCode	finalize () override final
virtual StatusCode	addBranches (const EventContext &ctx) const override final

Private Attributes
Gaudi::Property< int >	m_nHarmonic {this, "nHarmonic", 1, "Flow harmonic starting from v2"}
Gaudi::Property< bool >	m_doTopoClusDec {this, "doTopoClusDec", false, "Decorate with CaloTopoCluster FCal cut, non-HI mode only"}
SG::ReadHandleKey< xAOD::EventInfo >	m_eventInfoKey { this, "EventInfoKey", "EventInfo", "" }
SG::ReadHandleKey< xAOD::TrackParticleContainer >	m_TP_key {this, "InDetTrackParticlesKey", "InDetTrackParticles"}
SG::ReadHandleKey< xAOD::HIEventShapeContainer >	m_eventShapeKey {this, "HIEventShapeKey", "HIEventShape", ""}
SG::ReadHandleKey< xAOD::CaloClusterContainer >	m_caloClusterKey {this, "CaloClusterKey", "", "Only needed if doTopoClusDec is true"}
PublicToolHandleArray< InDet::IInDetTrackSelectionTool >	m_trkSelTools {this, "TrackSelectionTools", {}, "Track selection tools (optional)"}
	track selection tool which can be optionally used for N_trk and sum pt cuts
Gaudi::Property< std::vector< std::string > >	m_cutLevels {this, "cutLevels", {}, "Cut levels"}
std::vector< SG::AuxElement::Decorator< float > >	m_decFCalEtA_Qnx
std::vector< SG::AuxElement::Decorator< float > >	m_decFCalEtA_Qny
std::vector< SG::AuxElement::Decorator< float > >	m_decFCalEtC_Qnx
std::vector< SG::AuxElement::Decorator< float > >	m_decFCalEtC_Qny
std::vector< SG::AuxElement::Decorator< float > >	m_decHalfFCalEtA_Qnx
std::vector< SG::AuxElement::Decorator< float > >	m_decHalfFCalEtA_Qny
std::vector< SG::AuxElement::Decorator< float > >	m_decHalfFCalEtC_Qnx
std::vector< SG::AuxElement::Decorator< float > >	m_decHalfFCalEtC_Qny
std::vector< SG::AuxElement::Decorator< int > >	m_decTrack_count

Detailed Description

Definition at line 28 of file HIGlobalAugmentationTool.h.

Member Function Documentation

◆ addBranches()

StatusCode DerivationFramework::HIGlobalAugmentationTool::addBranches ( const EventContext & ctx ) const

finaloverridevirtual

Definition at line 48 of file HIGlobalAugmentationTool.cxx.

  {
    SG::ReadHandle<xAOD::EventInfo> eventInfo (m_eventInfoKey, ctx);
 
    //Load track particle container
    SG::ReadHandle<xAOD::TrackParticleContainer> tracks(m_TP_key, ctx);
    if (!tracks.isValid()) {
      ATH_MSG_ERROR ("Couldn't retrieve TrackParticleContainer with key " << m_TP_key);
      return StatusCode::FAILURE;
    }
 
    //Count tracks
    unsigned int nTrackSels  = m_trkSelTools.size();
    std::vector <int> TrackParticleMultiplicity(nTrackSels,0);
    for (const auto* tp : *tracks) {
      if (tp) {
        for (unsigned int its = 0; its < nTrackSels; ++its) {
          if (m_trkSelTools[its]->accept(*tp)) ++TrackParticleMultiplicity[its];
        }
      }
    }
    //decorate eventInfo with track multiplicity
    for (unsigned int its = 0; its < nTrackSels; ++its) {
      (m_decTrack_count[its])(*eventInfo) = TrackParticleMultiplicity[its];
    }
 
    float FCalEtA = 0;
    float FCalEtC = 0;
 
    std::vector<float> FCalEtA_Qnx(m_nHarmonic,0);
    std::vector<float> FCalEtA_Qny(m_nHarmonic,0);
    std::vector<float> FCalEtC_Qnx(m_nHarmonic,0);
    std::vector<float> FCalEtC_Qny(m_nHarmonic,0);
 
    float HalfFCalEtA = 0;
    float HalfFCalEtC = 0;
 
    std::vector<float> HalfFCalEtA_Qnx(m_nHarmonic,0);
    std::vector<float> HalfFCalEtA_Qny(m_nHarmonic,0);
    std::vector<float> HalfFCalEtC_Qnx(m_nHarmonic,0);
    std::vector<float> HalfFCalEtC_Qny(m_nHarmonic,0);
 
    // Set up the decorators for FCal Et
    SG::AuxElement::Decorator< float > decFCalEtA("FCalEtA");
    SG::AuxElement::Decorator< float > decFCalEtC("FCalEtC");
 
    SG::AuxElement::Decorator< float > decHalfFCalEtA("HalfFCalEtA");
    SG::AuxElement::Decorator< float > decHalfFCalEtC("HalfFCalEtC");
 
    //Retrieve HIEventShape
    SG::ReadHandle<xAOD::HIEventShapeContainer> eventShape(m_eventShapeKey, ctx);
    if (!eventShape.isValid()){
      ATH_MSG_ERROR ("Cannot retrieve HIEventShape");
      return StatusCode::FAILURE;
    }
 
    //Calculate FCal A and FCal C sums
    for(const auto* ptrEvtShp : *eventShape){
      if(ptrEvtShp->layer()!=21 && ptrEvtShp->layer()!=22 &&
        ptrEvtShp->layer()!=23) continue;
 
      float eta = ptrEvtShp->etaMin();
      const std::vector<float>&c1 = ptrEvtShp->etCos();
      const std::vector<float>&s1 = ptrEvtShp->etSin();
 
      if (eta > 0) {
        FCalEtA += ptrEvtShp->et();
        for (int vn = 1; vn <= m_nHarmonic; ++vn){
            FCalEtA_Qnx.at(vn-1) += c1.at(vn);
            FCalEtA_Qny.at(vn-1) += s1.at(vn);
        }
        if(eta > 4.0) {
          HalfFCalEtA += ptrEvtShp->et();
          for (int vn = 1; vn <= m_nHarmonic; ++vn){
            HalfFCalEtA_Qnx.at(vn-1) += c1.at(vn);
            HalfFCalEtA_Qny.at(vn-1) += s1.at(vn);
          }
        }
      } else if (eta < 0) {
          FCalEtC += ptrEvtShp->et();
          for (int vn = 1; vn <= m_nHarmonic; ++vn){
              FCalEtC_Qnx.at(vn-1) += c1.at(vn);
              FCalEtC_Qny.at(vn-1) += s1.at(vn);
          }
          if(eta < -4.0){
            HalfFCalEtC += ptrEvtShp->et();
            for (int vn = 1; vn <= m_nHarmonic; ++vn){
              HalfFCalEtC_Qnx.at(vn-1) += c1.at(vn);
              HalfFCalEtC_Qny.at(vn-1) += s1.at(vn);
            }
                }
        }
    }
 
    if (m_doTopoClusDec) {
        // Setup the decorator for TopoCaloCluster cut
        // If this is true, the event is not compatible with UPC topologies in the FCal
        SG::AuxElement::Decorator< bool > decTopoClusterFCalCut("passUPCTopoCaloCut");
        //Default decoration set to false
        decTopoClusterFCalCut(*eventInfo) = false;
 
        //access topoClusters
        SG::ReadHandle<xAOD::CaloClusterContainer> topos(m_caloClusterKey, ctx);
        bool hasTowerA{false};
        bool hasTowerC{false};
        for (const auto topo : *topos) {
          float topo_eta = topo->eta();
          if (abs(topo_eta) > 3.2 && abs(topo_eta) < 4.9) {
            float topo_pt = topo->pt() * 1e-3;
            if (topo_pt > 0.4) {
              if (topo_eta > 0) {
                hasTowerA = true;
              } else {
                hasTowerC = true;
              }
            }
          }
        }
        //decorate
        decTopoClusterFCalCut(*eventInfo) = (hasTowerA && hasTowerC);
    }
 
 
    //decorate
    decFCalEtA(*eventInfo) = FCalEtA;
    decFCalEtC(*eventInfo) = FCalEtC;
    decHalfFCalEtA(*eventInfo) = HalfFCalEtA;
    decHalfFCalEtC(*eventInfo) = HalfFCalEtC;
 
    for (int vn = 0; vn < m_nHarmonic; ++vn){
      (m_decFCalEtA_Qnx[vn])(*eventInfo) = FCalEtA_Qnx.at(vn);
      (m_decFCalEtA_Qny[vn])(*eventInfo) = FCalEtA_Qny.at(vn);
      (m_decFCalEtC_Qnx[vn])(*eventInfo) = FCalEtC_Qnx.at(vn);
      (m_decFCalEtC_Qny[vn])(*eventInfo) = FCalEtC_Qny.at(vn);
      (m_decHalfFCalEtA_Qnx[vn])(*eventInfo) = HalfFCalEtA_Qnx.at(vn);
      (m_decHalfFCalEtA_Qny[vn])(*eventInfo) = HalfFCalEtA_Qny.at(vn);
      (m_decHalfFCalEtC_Qnx[vn])(*eventInfo) = HalfFCalEtC_Qnx.at(vn);
      (m_decHalfFCalEtC_Qny[vn])(*eventInfo) = HalfFCalEtC_Qny.at(vn);
    }
 
    return StatusCode::SUCCESS;
  }

◆ finalize()

StatusCode DerivationFramework::HIGlobalAugmentationTool::finalize ( )

finaloverridevirtual

Definition at line 39 of file HIGlobalAugmentationTool.cxx.

  {
    for (auto trkSelTool : m_trkSelTools ) {
      ATH_CHECK(trkSelTool->finalize());
    }
 
    return StatusCode::SUCCESS;
  }

◆ initialize()

StatusCode DerivationFramework::HIGlobalAugmentationTool::initialize ( )

finaloverridevirtual

Definition at line 11 of file HIGlobalAugmentationTool.cxx.

  {
    ATH_CHECK( m_eventInfoKey.initialize() );
    ATH_CHECK( m_TP_key.initialize() );
    ATH_CHECK( m_eventShapeKey.initialize() );
    ATH_CHECK( m_caloClusterKey.initialize(SG::AllowEmpty) ); // Only needed if m_doTopoClusDec = True
    unsigned int nTrackSels  = m_trkSelTools.size();
    for (unsigned int its = 0; its < nTrackSels; ++its) {
      ATH_CHECK(m_trkSelTools[its].retrieve());
      m_decTrack_count.emplace_back(SG::AuxElement::Decorator<int>("TrackParticleMultiplicity_" + m_cutLevels[its] ));
    }
    //initilize decorators for flow vectors
    for (int vn = 1; vn <= m_nHarmonic; ++vn) {
      m_decFCalEtA_Qnx.emplace_back(SG::AuxElement::Decorator<float>(std::format("FCalEtA_Q{}x", vn + 1)));
      m_decFCalEtA_Qny.emplace_back(SG::AuxElement::Decorator<float>(std::format("FCalEtA_Q{}y", vn + 1)));
      m_decFCalEtC_Qnx.emplace_back(SG::AuxElement::Decorator<float>(std::format("FCalEtC_Q{}x", vn + 1)));
      m_decFCalEtC_Qny.emplace_back(SG::AuxElement::Decorator<float>(std::format("FCalEtC_Q{}y", vn + 1)));
 
      // half FCal is for FCal with eta > 4.0 only
      m_decHalfFCalEtA_Qnx.emplace_back(SG::AuxElement::Decorator<float>(std::format("HalfFCalEtA_Q{}x", vn + 1)));
      m_decHalfFCalEtA_Qny.emplace_back(SG::AuxElement::Decorator<float>(std::format("HalfFCalEtA_Q{}y", vn + 1)));
      m_decHalfFCalEtC_Qnx.emplace_back(SG::AuxElement::Decorator<float>(std::format("HalfFCalEtC_Q{}x", vn + 1)));
      m_decHalfFCalEtC_Qny.emplace_back(SG::AuxElement::Decorator<float>(std::format("HalfFCalEtC_Q{}y", vn + 1)));
    }
 
 
    return StatusCode::SUCCESS;
  }

Member Data Documentation

◆ m_caloClusterKey

SG::ReadHandleKey<xAOD::CaloClusterContainer> DerivationFramework::HIGlobalAugmentationTool::m_caloClusterKey {this, "CaloClusterKey", "", "Only needed if doTopoClusDec is true"}

private

Definition at line 46 of file HIGlobalAugmentationTool.h.

46{this, "CaloClusterKey", "", "Only needed if doTopoClusDec is true"};

◆ m_cutLevels

Gaudi::Property<std::vector<std::string> > DerivationFramework::HIGlobalAugmentationTool::m_cutLevels {this, "cutLevels", {}, "Cut levels"}

private

Definition at line 48 of file HIGlobalAugmentationTool.h.

48{this, "cutLevels", {}, "Cut levels"};

◆ m_decFCalEtA_Qnx

std::vector< SG::AuxElement::Decorator< float > > DerivationFramework::HIGlobalAugmentationTool::m_decFCalEtA_Qnx

private

Definition at line 51 of file HIGlobalAugmentationTool.h.

◆ m_decFCalEtA_Qny

std::vector< SG::AuxElement::Decorator< float > > DerivationFramework::HIGlobalAugmentationTool::m_decFCalEtA_Qny

private

Definition at line 52 of file HIGlobalAugmentationTool.h.

◆ m_decFCalEtC_Qnx

std::vector< SG::AuxElement::Decorator< float > > DerivationFramework::HIGlobalAugmentationTool::m_decFCalEtC_Qnx

private

Definition at line 53 of file HIGlobalAugmentationTool.h.

◆ m_decFCalEtC_Qny

std::vector< SG::AuxElement::Decorator< float > > DerivationFramework::HIGlobalAugmentationTool::m_decFCalEtC_Qny

private

Definition at line 54 of file HIGlobalAugmentationTool.h.

◆ m_decHalfFCalEtA_Qnx

std::vector< SG::AuxElement::Decorator< float > > DerivationFramework::HIGlobalAugmentationTool::m_decHalfFCalEtA_Qnx

private

Definition at line 56 of file HIGlobalAugmentationTool.h.

◆ m_decHalfFCalEtA_Qny

std::vector< SG::AuxElement::Decorator< float > > DerivationFramework::HIGlobalAugmentationTool::m_decHalfFCalEtA_Qny

private

Definition at line 57 of file HIGlobalAugmentationTool.h.

◆ m_decHalfFCalEtC_Qnx

std::vector< SG::AuxElement::Decorator< float > > DerivationFramework::HIGlobalAugmentationTool::m_decHalfFCalEtC_Qnx

private

Definition at line 58 of file HIGlobalAugmentationTool.h.

◆ m_decHalfFCalEtC_Qny

std::vector< SG::AuxElement::Decorator< float > > DerivationFramework::HIGlobalAugmentationTool::m_decHalfFCalEtC_Qny

private

Definition at line 59 of file HIGlobalAugmentationTool.h.

◆ m_decTrack_count

std::vector< SG::AuxElement::Decorator< int > > DerivationFramework::HIGlobalAugmentationTool::m_decTrack_count

private

Definition at line 61 of file HIGlobalAugmentationTool.h.

◆ m_doTopoClusDec

Gaudi::Property<bool> DerivationFramework::HIGlobalAugmentationTool::m_doTopoClusDec {this, "doTopoClusDec", false, "Decorate with CaloTopoCluster FCal cut, non-HI mode only"}

private

Definition at line 42 of file HIGlobalAugmentationTool.h.

42{this, "doTopoClusDec", false, "Decorate with CaloTopoCluster FCal cut, non-HI mode only"};

◆ m_eventInfoKey

SG::ReadHandleKey<xAOD::EventInfo> DerivationFramework::HIGlobalAugmentationTool::m_eventInfoKey { this, "EventInfoKey", "EventInfo", "" }

private

Definition at line 43 of file HIGlobalAugmentationTool.h.

43{ this, "EventInfoKey", "EventInfo", "" };

◆ m_eventShapeKey

SG::ReadHandleKey<xAOD::HIEventShapeContainer> DerivationFramework::HIGlobalAugmentationTool::m_eventShapeKey {this, "HIEventShapeKey", "HIEventShape", ""}

private

Definition at line 45 of file HIGlobalAugmentationTool.h.

45{this, "HIEventShapeKey", "HIEventShape", ""};

◆ m_nHarmonic

Gaudi::Property<int> DerivationFramework::HIGlobalAugmentationTool::m_nHarmonic {this, "nHarmonic", 1, "Flow harmonic starting from v2"}

private

Definition at line 41 of file HIGlobalAugmentationTool.h.

41{this, "nHarmonic", 1, "Flow harmonic starting from v2"};

◆ m_TP_key

SG::ReadHandleKey<xAOD::TrackParticleContainer> DerivationFramework::HIGlobalAugmentationTool::m_TP_key {this, "InDetTrackParticlesKey", "InDetTrackParticles"}

private

Definition at line 44 of file HIGlobalAugmentationTool.h.

44{this, "InDetTrackParticlesKey", "InDetTrackParticles"};

◆ m_trkSelTools

PublicToolHandleArray< InDet::IInDetTrackSelectionTool > DerivationFramework::HIGlobalAugmentationTool::m_trkSelTools {this, "TrackSelectionTools", {}, "Track selection tools (optional)"}

private

track selection tool which can be optionally used for N_trk and sum pt cuts

Definition at line 47 of file HIGlobalAugmentationTool.h.

47{this, "TrackSelectionTools", {}, "Track selection tools (optional)"};

The documentation for this class was generated from the following files:

Public Member Functions

Private Attributes

Detailed Description

Member Function Documentation

◆ addBranches()

◆ finalize()

◆ initialize()

Member Data Documentation

◆ m_caloClusterKey

◆ m_cutLevels

◆ m_decFCalEtA_Qnx

◆ m_decFCalEtA_Qny

◆ m_decFCalEtC_Qnx

◆ m_decFCalEtC_Qny

◆ m_decHalfFCalEtA_Qnx

◆ m_decHalfFCalEtA_Qny

◆ m_decHalfFCalEtC_Qnx

◆ m_decHalfFCalEtC_Qny

◆ m_decTrack_count

◆ m_doTopoClusDec

◆ m_eventInfoKey

◆ m_eventShapeKey

◆ m_nHarmonic

◆ m_TP_key

◆ m_trkSelTools