TrackCaloCluster_v1.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
// $Id: TrackCaloCluster_v1.h $
#ifndef XAODPFLOW_VERSIONS_TRACKCALOCLUSTER_V1_H
#define XAODPFLOW_VERSIONS_TRACKCALOCLUSTER_V1_H
 
// Core include(s):
#include "AthContainers/DataVector.h"
#include "AthLinks/ElementLink.h"
 
// xAOD include(s):
#include "xAODBase/IParticle.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
 
namespace xAOD {
  
  class TrackCaloCluster_v1 : public IParticle {
    
  public: 
    
    enum Taste { Charged = 0, Neutral, Combined };
    
    // Default Constructor
    TrackCaloCluster_v1();
    // Destuctor
    ~TrackCaloCluster_v1();
    
    virtual double           pt() const;
    virtual double           eta() const;
    virtual double           phi() const;
    virtual double           m() const;
    virtual double           e() const;
    virtual double           rapidity() const;
    virtual int              taste() const;
    
    typedef IParticle::FourMom_t FourMom_t;
    
    virtual FourMom_t        p4() const;

    typedef ROOT::Math::LorentzVector<ROOT::Math::PtEtaPhiM4D<double> > GenVecFourMom_t;

    GenVecFourMom_t genvecP4() const; 
    
    virtual Type::ObjectType type() const;
    
    void setParameters(float pt, float eta, float phi, float m, TrackCaloCluster_v1::Taste taste, 
               const ElementLink<xAOD::TrackParticleContainer>& particle, 
               const std::vector< ElementLink< xAOD::CaloClusterContainer > >& clusters);       
    
    const ElementLink<xAOD::TrackParticleContainer>& trackParticleLink() const;
    void setTrackParticleLink(const ElementLink< xAOD::TrackParticleContainer >& particle);
    const xAOD::TrackParticle* trackParticle() const;       
    const std::vector< ElementLink<xAOD::CaloClusterContainer> > & caloClusterLinks() const;
    void setCaloClusterLinks(const std::vector< ElementLink< xAOD::CaloClusterContainer > >& caloClusterLinks);
    
  }; 
  
}
 
#endif // XAODPFLOW_VERSIONS_TRACKCALOCLUSTER_V1_H