egammaRec.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EGAMMARECEVENT_EGAMMA_H
#define EGAMMARECEVENT_EGAMMA_H
// INCLUDE HEADER FILES:
 
#include "AthLinks/ElementLink.h"
 
// CaloCluster include
#include "xAODCaloEvent/CaloCluster.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
 
// TrackParticle Includes
#include "xAODTracking/TrackParticle.h"
#include "xAODTracking/TrackParticleContainer.h"
 
// Vertex Includes
#include "xAODTracking/Vertex.h"
#include "xAODTracking/VertexContainer.h"
 
#include <array>
#include <vector>
class egammaRec
{
  // Public typedefs:
public:
  egammaRec() = default;

  egammaRec(const std::vector<ElementLink<xAOD::CaloClusterContainer>> &caloClusterLinks);
  
  egammaRec(const egammaRec&) = default;
  egammaRec(egammaRec&&) = default;
  egammaRec& operator=(const egammaRec&) = default;
  egammaRec& operator=(egammaRec&&) = default;
  ~egammaRec() = default;

  size_t getNumberOfClusters() const;

  const xAOD::CaloCluster* caloCluster(size_t index = 0) const;

  ElementLink<xAOD::CaloClusterContainer> caloClusterElementLink(size_t index = 0) const;

  void setCaloClusters(const std::vector<ElementLink<xAOD::CaloClusterContainer>>& links);

  const std::vector<ElementLink<xAOD::CaloClusterContainer>> & caloClusterElementLinks() const;

  size_t getNumberOfTrackParticles() const;

  const xAOD::TrackParticle* trackParticle(size_t index = 0) const;

  ElementLink<xAOD::TrackParticleContainer> trackParticleElementLink(size_t index = 0) const;

  void setTrackParticles(const std::vector<ElementLink<xAOD::TrackParticleContainer>>& links);
  
  const std::vector<ElementLink<xAOD::TrackParticleContainer>> & trackParticleElementLinks() const;

  size_t getNumberOfVertices() const;

  const xAOD::Vertex* vertex(size_t index = 0) const;

  ElementLink<xAOD::VertexContainer> vertexElementLink(size_t index = 0) const;

  void setVertices(const std::vector<ElementLink<xAOD::VertexContainer>>& links);
  
  const std::vector<ElementLink<xAOD::VertexContainer>> & vertexElementLinks() const;

  void pushBackVertex(const ElementLink<xAOD::VertexContainer>& vertexElementLink);

  void pushFrontVertex(const ElementLink<xAOD::VertexContainer>& vertexElementLink);


  const std::array<double,4>&  deltaEta () const; 

  const std::array<double,4>&  deltaPhi () const;

  const std::array<double,4>&  deltaPhiRescaled () const;

  double deltaPhiLast () const ;
  
  void setDeltaEta(const std::array<double, 4>& deltaEtas);

  void setDeltaPhi(const std::array<double, 4>& deltaPhis);

  void setDeltaPhiRescaled(std::array<double, 4>& deltaPhis);

  void setDeltaPhiLast(double value);


  float deltaEtaVtx() const;

  float deltaPhiVtx() const;

  void setDeltaEtaVtx(float value);

  void setDeltaPhiVtx(float value);
 
private:
  std::vector<ElementLink<xAOD::CaloClusterContainer>> m_caloClusters;
  std::vector<ElementLink<xAOD::TrackParticleContainer>> m_trackParticles;
  std::vector<ElementLink<xAOD::VertexContainer>> m_vertices;
 
  std::array<double, 4> m_deltaEta {-999, -999, -999, -999};
  std::array<double, 4> m_deltaPhi {-999, -999, -999, -999};
  std::array<double, 4> m_deltaPhiRescaled {-999, -999, -999, -999};
 
  double m_deltaPhiLast = -999;
  float m_deltaEtaVtx = -999;
  float m_deltaPhiVtx = -999;
};
#include"egammaRec.icc"
#endif