#include <METTriggerAugmentationTool.h>

Inheritance diagram for DerivationFramework::METTriggerAugmentationTool:

Collaboration diagram for DerivationFramework::METTriggerAugmentationTool:

Public Member Functions
	METTriggerAugmentationTool (const std::string &t, const std::string &n, const IInterface *p)

virtual StatusCode	initialize () override final

virtual StatusCode	addBranches (const EventContext &ctx) const override final

Private Attributes
SG::ReadHandleKey< xAOD::EnergySumRoI >	m_L1METName {this, "L1METName", "LVL1EnergySumRoI"}

SG::ReadHandleKey< xAOD::JetRoIContainer >	m_L1JetName {this, "L1JetName", "LVL1JetRoIs"}

SG::WriteHandleKey< xAOD::EnergySumRoI >	m_outputName {this, "OutputName", "LVL1EnergySumRoI_KF"}

Gaudi::Property< std::string >	m_LUTFileName {this, "LUTFile", "LUT_data15.root"}

std::unique_ptr< TH2 >	m_LUT {}

Detailed Description

Definition at line 19 of file METTriggerAugmentationTool.h.

Constructor & Destructor Documentation

◆ METTriggerAugmentationTool()

DerivationFramework::METTriggerAugmentationTool::METTriggerAugmentationTool	(	const std::string &	t,
		const std::string &	n,
		const IInterface *	p
	)

Definition at line 16 of file METTriggerAugmentationTool.cxx.

                                                                               :
     base_class(t,n,p)
   {
   }

Member Function Documentation

◆ addBranches()

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

finaloverridevirtual

Definition at line 51 of file METTriggerAugmentationTool.cxx.

   {
     ATH_MSG_DEBUG(" In L1KF_METMaker::makeKFMET()" );
     // if the output has already been written we don't need to do anything
     if (evtStore()->contains<xAOD::EnergySumRoI>(m_outputName.key()) ) return StatusCode::SUCCESS; // FIXME  tool should not have been configured in this case
  
     SG::ReadHandle<xAOD::EnergySumRoI> originalL1{m_L1METName, ctx};
     if (!originalL1.isValid()){
       ATH_MSG_ERROR("Unable to retrieve EnergySumRoI: " << m_L1METName << "!");
       return StatusCode::FAILURE;
     }
  
     SG::ReadHandle<xAOD::JetRoIContainer> l1Jets{m_L1JetName, ctx};
     if (!originalL1.isValid()){
       ATH_MSG_ERROR("Unable to retrieve JetRoIContainer: " << m_L1JetName << "!");
       return StatusCode::FAILURE;
     }
  
     SG::WriteHandle<xAOD::EnergySumRoI> l1_kf{m_outputName, ctx};
     if (!originalL1.isValid()){
       ATH_MSG_ERROR("Invalid WriteHandle for EnergySumRoI: " << m_outputName << "!");
       return StatusCode::FAILURE;
     }
     ATH_CHECK(l1_kf.record(std::make_unique<xAOD::EnergySumRoI>(), std::make_unique<xAOD::EnergySumRoIAuxInfo>()));
  
     ATH_MSG_DEBUG( "Making deep copy" );
     // copy across the info
     *l1_kf = *originalL1;
  
     ATH_MSG_DEBUG( "Building KF MET" );
     float KFMETx = l1_kf->exMiss();
     float KFMETy = l1_kf->eyMiss();
     float KFSumEt = l1_kf->energyT();
  
     for (const auto jet : *l1Jets) {
       int etaBin = m_LUT->GetXaxis()->FindBin( fabs( jet->eta() ) );
       int ptBin = m_LUT->GetYaxis()->FindBin( log2( jet->et8x8() / 1000.) );
       if (ptBin==0) ptBin=1;
       float KFweight = m_LUT->GetBinContent(etaBin, ptBin);
       float jetContribution = jet->et8x8() * KFweight;
  
       KFMETx -= jetContribution * cos( jet->phi() );
       KFMETy -= jetContribution * sin( jet->phi() );
       KFSumEt += jetContribution;
     }
     l1_kf->setEnergyX(-KFMETx);
     l1_kf->setEnergyY(-KFMETy);
     l1_kf->setEnergyT(KFSumEt);
  
     ATH_MSG_DEBUG( "Built KF MET" );
     return StatusCode::SUCCESS;
   }

◆ initialize()

StatusCode DerivationFramework::METTriggerAugmentationTool::initialize ( )

finaloverridevirtual

Definition at line 23 of file METTriggerAugmentationTool.cxx.

   {
     ATH_CHECK(m_L1METName.initialize());
     ATH_CHECK(m_L1JetName.initialize());
     ATH_CHECK(m_outputName.initialize());
     std::string fullLUTFileName = PathResolver::find_file(m_LUTFileName, "DATAPATH");
  
     if (fullLUTFileName.empty() ) {
       ATH_MSG_ERROR( "File " << m_LUTFileName << " not found!" );
       return StatusCode::FAILURE;
     }
  
     TFile* lutFile = TFile::Open(fullLUTFileName.c_str() );
     if (lutFile->IsZombie() ) return StatusCode::FAILURE;
  
     TH2* lutFromFile{};
     lutFile->GetObject("LUT", lutFromFile);
     if (!lutFromFile) {
       ATH_MSG_ERROR( "LUT file doesn't contain a 'LUT' object!" );
       return StatusCode::FAILURE;
     }
  
     m_LUT.reset(dynamic_cast<TH2*>(lutFromFile->Clone() ) );
     m_LUT->SetDirectory(0);
     lutFile->Close();
     return StatusCode::SUCCESS;
   }