JetOriginHelpers.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#include "JetUtils/JetOriginHelpers.h"
 
#include "xAODCaloEvent/CaloVertexedTopoCluster.h"
#include "xAODTracking/Vertex.h"
 
namespace jet {
 
  xAOD::JetFourMom_t clusterOriginCorrection(const xAOD::Jet& jet, const xAOD::Vertex & vx, const xAOD::CaloCluster::State state){
    
 
    xAOD::CaloCluster::State constitScale = state;
    if(constitScale==xAOD::CaloCluster::UNKNOWN) constitScale = (xAOD::CaloCluster::State) jet.getConstituentsSignalState();
 
    int numC = jet.numConstituents();
    const Amg::Vector3D& vxpos= vx.position();
 
    xAOD::IParticle::FourMom_t corrP4(0,0,0,0);
 
    for(int i=0; i<numC;i++){
      const xAOD::CaloCluster* cl = dynamic_cast<const xAOD::CaloCluster*>(jet.rawConstituent(i) );
      if( ! cl ) continue;
 
      xAOD::CaloVertexedTopoCluster corrCl(*cl, constitScale, vxpos );
      xAOD::IParticle::FourMom_t p4CorrCl=corrCl.p4();
      if(cl->e(constitScale) < 0.) p4CorrCl*=-1.;
      corrP4 += p4CorrCl;   
    }
    
    xAOD::JetFourMom_t corrJet;
    corrJet.SetPxPyPzE(corrP4.Px(), corrP4.Py(), corrP4.Pz(), corrP4.E() );
    return corrJet;
    
  }
 
}