#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 ()

virtual StatusCode	finalize ()

virtual StatusCode	addBranches () const

Private Attributes
std::string	m_outputName

std::string	m_LUTFileName

std::string	m_L1METName

std::string	m_L1JetName

std::unique_ptr< TH2 >	m_LUT

Detailed Description

Definition at line 18 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 18 of file METTriggerAugmentationTool.cxx.

                            : 
     base_class(t,n,p)
   {
     declareProperty("OutputName", m_outputName = "LVL1EnergySumRoI_KF");
     declareProperty("LUTFile", m_LUTFileName = "LUT_data15.root");
     declareProperty("L1METName", m_L1METName = "LVL1EnergySumRoI");
     declareProperty("L1JetName", m_L1JetName = "LVL1JetRoIs");
   }

Member Function Documentation

◆ addBranches()

StatusCode DerivationFramework::METTriggerAugmentationTool::addBranches ( ) const

virtual

Definition at line 61 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) ) return StatusCode::SUCCESS;
  
     const xAOD::EnergySumRoI* originalL1(0);
     ATH_CHECK( evtStore()->retrieve(originalL1, m_L1METName) );
  
     const xAOD::JetRoIContainer* l1Jets(0);
     ATH_CHECK( evtStore()->retrieve(l1Jets, m_L1JetName) );
  
     xAOD::EnergySumRoI* l1_kf = new xAOD::EnergySumRoI();
     xAOD::EnergySumRoIAuxInfo* l1_kfAux = new xAOD::EnergySumRoIAuxInfo();
  
     ATH_MSG_DEBUG( "Setting the store" );
     // set the store
     l1_kf->setStore(l1_kfAux);
  
     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" );
     ATH_CHECK( evtStore()->record(l1_kf, m_outputName) );
     ATH_CHECK( evtStore()->record(l1_kfAux, m_outputName+"Aux.") );
     return StatusCode::SUCCESS;
   }

◆ finalize()

StatusCode DerivationFramework::METTriggerAugmentationTool::finalize ( )

virtual

Definition at line 54 of file METTriggerAugmentationTool.cxx.

   {
     return StatusCode::SUCCESS;
   }

◆ initialize()

StatusCode DerivationFramework::METTriggerAugmentationTool::initialize ( )

virtual

Definition at line 29 of file METTriggerAugmentationTool.cxx.

   {
     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(0);
     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;
   }