TauJet_v3.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
 
/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
// $Id: TauJet_v3.h 725228 2016-02-19 22:59:42Z griffith $
#ifndef XAODTAU_VERSIONS_TAUJET_V3_H
#define XAODTAU_VERSIONS_TAUJET_V3_H
 
 
// Core include(s):
#include "AthLinks/ElementLink.h"
 
 
// xAOD include(s):
#include "xAODBase/IParticle.h"
#include "xAODBase/IParticleContainer.h"
#include "xAODTau/TauDefs.h"
#include "xAODTau/TauTrackContainer.h"
#include "xAODTau/TauTrack.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/VertexContainer.h"
#include "xAODJet/JetContainer.h"
#include "xAODPFlow/PFOContainer.h"
#include "xAODCaloEvent/CaloVertexedTopoCluster.h"
 
// ROOT include(s):
#include "Math/Vector4D.h"
 
namespace xAOD {
 
 
 
  class TauJet_v3 : public IParticle {
 
  public:
    TauJet_v3();
    
    
    virtual double           pt() const;
    virtual double           eta() const;
    virtual double           phi() const;
    virtual double           e() const;
    
    virtual double           m() const;
    virtual double           rapidity() const;
    
    virtual FourMom_t p4() const;
    virtual FourMom_t p4(const TauJetParameters::TauCalibType calib) const;
 
    typedef ROOT::Math::LorentzVector<ROOT::Math::PtEtaPhiM4D<double> > GenVecFourMom_t;
 
    GenVecFourMom_t genvecP4() const; 
    GenVecFourMom_t genvecP4(const TauJetParameters::TauCalibType calib) const; 
 
    void setP4(double pt, double eta, double phi, double m);
    bool setP4(const TauJetParameters::TauCalibType calib, double pt, double eta, double phi, double m);
 
    //setters and getters for the individual 4-vector components at different calibration levels
    double ptJetSeed() const;
    double etaJetSeed() const;
    double phiJetSeed() const;
    double mJetSeed() const;
 
    double ptDetectorAxis() const;
    double etaDetectorAxis() const;
    double phiDetectorAxis() const;
    double mDetectorAxis() const;
 
    double ptIntermediateAxis() const;
    double etaIntermediateAxis() const;
    double phiIntermediateAxis() const;
    double mIntermediateAxis() const;
 
    double ptTauEnergyScale() const;
    double etaTauEnergyScale() const;
    double phiTauEnergyScale() const;
    double mTauEnergyScale() const;
 
    double ptTauEtaCalib() const;
    double etaTauEtaCalib() const;
    double phiTauEtaCalib() const;
    double mTauEtaCalib() const;
 
    double ptPanTauCellBasedProto() const;
    double etaPanTauCellBasedProto() const;
    double phiPanTauCellBasedProto() const;
    double mPanTauCellBasedProto() const;
 
    double ptPanTauCellBased() const;
    double etaPanTauCellBased() const;
    double phiPanTauCellBased() const;
    double mPanTauCellBased() const;
 
    double ptTrigCaloOnly() const;
    double etaTrigCaloOnly() const;
    double phiTrigCaloOnly() const;
    double mTrigCaloOnly() const;
 
    double ptFinalCalib() const;
    double etaFinalCalib() const;
    double phiFinalCalib() const;
    double mFinalCalib() const;
 
    virtual Type::ObjectType type() const;
    
 
    // the summed up charge of all tau tracks
    float charge() const;
 
    // set the charge
    void setCharge(float);
 
    unsigned int ROIWord() const;
 
    // set the ROIWord, needed by TauTrigger 
    void setROIWord(unsigned int);
 
    bool hasDiscriminant( TauJetParameters::TauID discID ) const;
    
    double discriminant( TauJetParameters::TauID discID ) const;
 
    void setDiscriminant( TauJetParameters::TauID discID, double disc );
 
    bool isTau( TauJetParameters::IsTauFlag flag ) const;
 
    void setIsTau( TauJetParameters::IsTauFlag flag, bool value );
 
    // void setFlag( TauJetParameters::VetoFlags flag, bool value );//r21 cleanup
 
    // bool flag( TauJetParameters::VetoFlags flag ) const;//r21 cleanup
 
 
    bool detail( TauJetParameters::Detail detail, int& value ) const;
    bool detail( TauJetParameters::Detail detail, float& value ) const;
    bool detail( TauJetParameters::Detail detail, const IParticle* &value ) const;
    void setDetail( TauJetParameters::Detail detail, int value );
    void setDetail( TauJetParameters::Detail detail, float value );
    void setDetail( TauJetParameters::Detail detail, const IParticle* value, const IParticleContainer* cont=0 );
 
    // getter for specialized types. Compilation error if used with other types.
    template <class T>
    typename xAODTAU_return_type<T>::type detail( TauJetParameters::Detail detail ) const
    {
        T temp;
        if (this->detail(detail,temp)) return temp;
        else return 0;
    }
 
    bool panTauDetail( TauJetParameters::PanTauDetails panTauDetail, int& value ) const;
    bool panTauDetail( TauJetParameters::PanTauDetails panTauDetail, float& value ) const;
    void setPanTauDetail( TauJetParameters::PanTauDetails panTauDetail, int value );
    void setPanTauDetail( TauJetParameters::PanTauDetails panTauDetail, float value );
 
    // getter for specialized types. Compilation error if used with other types.
    template <class T>
    typename xAODTAU_return_type<T>::type panTauDetail( TauJetParameters::PanTauDetails panTauDetail ) const
    {
        T temp;
        if (this->panTauDetail(panTauDetail,temp)) return temp;
        else return 0;
    }
 
    //seed jet
    typedef ElementLink< xAOD::JetContainer > JetLink_t;
 
    const JetLink_t& jetLink() const;
    
    void setJetLink( const JetLink_t& jetLink ); 
 
    void setJet( const xAOD::JetContainer *cont, const xAOD::Jet *jet ); 
 
    const Jet* jet() const;
 
  
    //vertex that the tau comes from
    typedef ElementLink< xAOD::VertexContainer > VertexLink_t;
 
    const VertexLink_t& vertexLink() const;
    
    void setVertexLink( const VertexLink_t& vertexLink ); 
 
    void setVertex( const xAOD::VertexContainer *cont, const xAOD::Vertex *vertex ); 
 
    const Vertex* vertex() const;
 
 
    //secondary vertex that the tau decays in
 
    const VertexLink_t& secondaryVertexLink() const;
    
    void setSecondaryVertexLink( const VertexLink_t& vertexLink ); 
 
    void setSecondaryVertex( const xAOD::VertexContainer *cont, const xAOD::Vertex *vertex ); 
 
    const Vertex* secondaryVertex() const;
  
 
    //tauTracks
    typedef std::vector< ElementLink< xAOD::TauTrackContainer > >  TauTrackLinks_t;
 
    const TauTrackLinks_t tauTrackLinks(TauJetParameters::TauTrackFlag=TauJetParameters::TauTrackFlag::classifiedCharged) const;
    const TauTrackLinks_t tauTrackLinksWithMask(unsigned int) const;
    TauTrackLinks_t& allTauTrackLinksNonConst();
    
 
    //retrieve all TauJetAuxContainer::tauTrackLinks, regardless of flags
    const TauTrackLinks_t& allTauTrackLinks() const;
    //set all TauJetAuxContainer::tauTrackLinks
    void setAllTauTrackLinks( const TauTrackLinks_t& tauTracks );
    const TauTrack* track( size_t i, TauJetParameters::TauTrackFlag flag=TauJetParameters::TauTrackFlag::classifiedCharged, int* container_index=0 ) const;
    const TauTrack* trackWithMask( size_t i, unsigned int mask, int* container_index=0 ) const;
#ifndef XAOD_STANDALONE
    //see TauxAODHelpers for a solution in XAOD_STANDALONE
    TauTrack* trackNonConst( size_t i, TauJetParameters::TauTrackFlag flag=TauJetParameters::TauTrackFlag::classifiedCharged );
#endif
    std::vector<const TauTrack*> tracks(TauJetParameters::TauTrackFlag flag=TauJetParameters::TauTrackFlag::classifiedCharged ) const;
    std::vector<const TauTrack*> tracksWithMask(unsigned int mask ) const;
    std::vector<const TauTrack*> allTracks() const;
    //number of tracks with a given classification
    size_t nTracks(TauJetParameters::TauTrackFlag flag=TauJetParameters::TauTrackFlag::classifiedCharged) const;
    //number of tracks classified as classifiedCharged
    size_t nTracksCharged() const;
    //number of tracks classified as classifiedIsolation
    size_t nTracksIsolation() const;
    //number of tracks with a given classification
    size_t nTracksWithMask(unsigned int classification) const;
    //all tracks regardless of classification or lack thereof
    size_t nAllTracks() const;
    void addTauTrackLink( const ElementLink< TauTrackContainer >& tr);
    void clearTauTrackLinks();
    
    //clusters
    typedef std::vector< ElementLink< xAOD::IParticleContainer > >  IParticleLinks_t;
 
    const IParticleLinks_t& clusterLinks() const;
 
    void setClusterLinks( const IParticleLinks_t& clusters );
    const IParticle* cluster( size_t i) const;
    
    //* @brief Get the clusters */
    std::vector<const IParticle*> clusters() const;
    
    FourMom_t calibratedCluster( size_t i, xAOD::CaloCluster::State state=xAOD::CaloCluster::State::CALIBRATED) const;
    //number of cluster with associated to tau
    size_t nClusters() const;
    void addClusterLink( const ElementLink< IParticleContainer >& tr);
    void clearClusterLinks();
 
    //* @brief Get the clusters corrected to poinst at tau vertex*/
    std::vector<xAOD::CaloVertexedTopoCluster> vertexedClusters() const;
 
    const IParticleLinks_t& pi0Links() const;
 
    void setPi0Links( const IParticleLinks_t& pi0s );
    const IParticle* pi0( size_t i) const;
    //number of pi0 with associated to tau
    size_t nPi0s() const;
    void addPi0Link( const ElementLink< IParticleContainer >& tr);
    void clearPi0Links();
 
  
    //name accessors for trackFilterProngs
    int trackFilterProngs() const;
    //name accessors for trackFilterProngs
    void setTrackFilterProngs(int);
 
    //name accessors for trackFilterQuality
    int trackFilterQuality() const;
    //name accessors for trackFilterQuality
    void setTrackFilterQuality(int);
 
    //name accessors for pi0ConeDR
    float pi0ConeDR() const;
    //name accessors for pi0ConeDR
    void setPi0ConeDR(float);
 
 
    typedef std::vector< ElementLink< xAOD::PFOContainer > >  PFOLinks_t;
 
 
    //photon shot PFOs
    const PFOLinks_t& hadronicPFOLinks() const;
 
    void setHadronicPFOLinks( const PFOLinks_t& hadronicPFOs );
    const PFO* hadronicPFO( size_t i ) const;
    size_t nHadronicPFOs() const;
    void addHadronicPFOLink( const ElementLink< PFOContainer >& pfo);
    void clearHadronicPFOLinks();
 
 
    //photon shot PFOs
    const PFOLinks_t& shotPFOLinks() const;
 
    void setShotPFOLinks( const PFOLinks_t& shotPFOs );
    const PFO* shotPFO( size_t i ) const;
    size_t nShotPFOs() const;
    void addShotPFOLink( const ElementLink< PFOContainer >& pfo);
    void clearShotPFOLinks();
 
 
    //
    //charged PFOs
    const PFOLinks_t& chargedPFOLinks() const;
 
    void setChargedPFOLinks( const PFOLinks_t& chargedPFOs );
    const PFO* chargedPFO( size_t i ) const;
    size_t nChargedPFOs() const;
    void addChargedPFOLink( const ElementLink< PFOContainer >& pfo);
    void clearChargedPFOLinks();
 
 
    //neutral PFOs
    const PFOLinks_t& neutralPFOLinks() const;
 
    void setNeutralPFOLinks( const PFOLinks_t& neutralPFOs );
    const PFO* neutralPFO( size_t i ) const;
    size_t nNeutralPFOs() const;
    void addNeutralPFOLink( const ElementLink< PFOContainer >& pfo);
    void clearNeutralPFOLinks();
 
 
    //pi0 PFOs
    const PFOLinks_t& pi0PFOLinks() const;
 
    void setPi0PFOLinks( const PFOLinks_t& pi0PFOs );
    const PFO* pi0PFO( size_t i ) const;
    size_t nPi0PFOs() const;
    void addPi0PFOLink( const ElementLink< PFOContainer >& pfo);
    void clearPi0PFOLinks();
 
 
 
 
    //
    //cellbased_charged PFOs
    const PFOLinks_t& protoChargedPFOLinks() const;
 
    void setProtoChargedPFOLinks( const PFOLinks_t& protoChargedPFOs );
    const PFO* protoChargedPFO( size_t i ) const;
    size_t nProtoChargedPFOs() const;
    void addProtoChargedPFOLink( const ElementLink< PFOContainer >& pfo);
    void clearProtoChargedPFOLinks();
 
 
    //cellbased_neutral PFOs
    const PFOLinks_t& protoNeutralPFOLinks() const;
 
    void setProtoNeutralPFOLinks( const PFOLinks_t& protoNeutralPFOs );
    const PFO* protoNeutralPFO( size_t i ) const;
    size_t nProtoNeutralPFOs() const;
    void addProtoNeutralPFOLink( const ElementLink< PFOContainer >& pfo);
    void clearProtoNeutralPFOLinks();
 
 
    //pi0 PFOs
    const PFOLinks_t& protoPi0PFOLinks() const;
 
    void setProtoPi0PFOLinks( const PFOLinks_t& protoPi0PFOs );
    const PFO* protoPi0PFO( size_t i ) const;
    size_t nProtoPi0PFOs() const;
    void addProtoPi0PFOLink( const ElementLink< PFOContainer >& pfo);
    void clearProtoPi0PFOLinks();
 
 
  
    
 
    void dump() const;
 
 
  }; // class TauJet
 
} // namespace xAOD
 
 
// Declare IParticle as a base class of TauJet_v3:
#include "AthContainers/DataVector.h"
DATAVECTOR_BASE( xAOD::TauJet_v3, xAOD::IParticle );
 
#endif // XAODTAU_VERSIONS_TAUJET_V3_H
 
//  LocalWords:  newROIWord