ElectronJetDRAssociator.h

Go to the documentation of this file.

// This file's extension implies that it's C, but it's really -*- C++ -*-.
/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#ifndef EGAMMAD3PDMAKER_ELECTRONJETDRASSOCIATOR_H
#define EGAMMAD3PDMAKER_ELECTRONJETDRASSOCIATOR_H
 
 
#include "D3PDMakerInterfaces/ICollectionGetterRegistryTool.h"
#include "D3PDMakerUtils/MultiAssociationTool.h"
#include "xAODEgamma/Electron.h"
#include "xAODJet/Jet.h"
#include "TrkVertexFitterInterfaces/ITrackToVertexIPEstimator.h"
#include "GaudiKernel/ToolHandle.h"
#include <string>
#include <vector>
 
 
 
namespace Trk  { class VxCandidate; }
 
 
namespace D3PD {
 
 
class ICollectionGetterTool;
 
 
class ElectronJetDRAssociator
  : public MultiAssociationTool<xAOD::Electron,xAOD::Jet>
{
public:
  typedef MultiAssociationTool<xAOD::Electron,xAOD::Jet> Base;
 
  ElectronJetDRAssociator (const std::string& type,
                           const std::string& name, 
                           const IInterface* parent);
  
 
  virtual StatusCode initialize() override;
 
 
  virtual StatusCode configureD3PD (IAddVariable* tree,
                                    const std::type_info& ti) override;
 
 
  virtual StatusCode reset (const xAOD::Electron& eg) override;
 
 
  virtual const xAOD::Jet* next() override;
 
 
  virtual StatusCode book() final;
 
 
private:
  StatusCode findReducedPt (const xAOD::Electron* eg,
                const xAOD::Jet* j,
                float& reducedPt);
 
  StatusCode findSignedIPAndPTRelEleJetPair (const xAOD::Electron* eg,
                                             const xAOD::Jet* j0,
                                             float& signedIP, float& PTRel);
 
 
  std::string m_targetLabel;
 
  std::string m_vertexContainerName;
 
  double m_drcut;
 
  ToolHandle< Trk::ITrackToVertexIPEstimator > m_trackToVertexIPEstimator
    { this, "TrackToVertexIPEstimator", "Trk::TrackToVertexIPEstimator", "" };
 
  ToolHandle<ICollectionGetterRegistryTool> m_registry;
 
 
  ICollectionGetterTool* m_target;
 
  // Input electrons; source of the association.
  const xAOD::Electron* m_eg;
 
  float* m_dr;
 
  float* m_signedIP;
 
  float* m_ptrel;
 
  unsigned int* m_jetSize;
 
  float* m_reducedPt;
};
 
 
} // namespace D3PD
 
 
#endif // EGAMMAD3PDMAKER_EGAMMAJETDRASSOCIATOR_H