TauConstituent.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PANTAUALGS_TAUCONSTITUENT_H
#define PANTAUALGS_TAUCONSTITUENT_H
 
#include <string>
#include <vector>
 
#include "xAODBase/IParticle.h"
#include "xAODPFlow/PFO.h"
#include <TLorentzVector.h>
 
 
namespace PanTau {
 
  class TauConstituent : public xAOD::IParticle {
 
  public:
    
    enum Type {
        t_NoType    = 0,
        t_Charged   = 1,
        t_Neutral   = 2,
        t_Pi0Neut   = 3,
        t_OutChrg   = 4,
        t_OutNeut   = 5,
        t_NeutLowA  = 6,
        t_NeutLowB  = 7,
        t_nTypes    = 8
    };
    
    static double       DefaultBDTValue()       {return -4711.1337;}
    static double       DefaultCharge()         {return -47111337;}
    static std::string  AllConstituentsName()   {return "All";}
    
    static std::string  getTypeName(PanTau::TauConstituent::Type aType);
    static bool         isNeutralType(int tauConstituentType);
    static bool         isCoreType(int tauConstituentType);    
 
      
    TauConstituent();
    
    TauConstituent(const TLorentzVector&   itsMomentum,
                   int              itsCharge,
                   const std::vector<int>& itsType,
                   double           itsBDTValue,
                   const xAOD::PFO* itsPFO);
    
    
    virtual ~TauConstituent();
    
    
    TauConstituent(const TauConstituent& tauConst);
 
    TauConstituent& operator=(const TauConstituent& tauConst);
 
     
     virtual double           pt() const;
 
     virtual double           eta() const;
 
     virtual double           phi() const;
 
     virtual double           m() const;
 
     virtual double           e() const;
 
     virtual double           rapidity() const;
     
     typedef IParticle::FourMom_t FourMom_t;
 
     virtual FourMom_t p4() const;
 
     virtual xAOD::Type::ObjectType type() const;
 
 
     void setP4(float pt, float eta, float phi, float m);
 
     void setPt(float pt);
 
     void setEta(float eta);
 
     void setPhi(float phi);
 
     void setM(float m);
 
 
    //no setter functions needed as all properties are set in constructor
    void                                    removeTypeFlag(TauConstituent::Type aType);
    
    //Getter functions
    std::vector<std::string>                getTypeName() const;
    std::string                             getTypeNameString() const;
    std::vector<int>                        getTypeFlags() const;
    bool                                    isOfType(TauConstituent::Type aType) const;
    double                                  getBDTValue() const;
    int                                     getCharge() const;
    const xAOD::PFO*                        getPFO() const;
    
    void                                    addShot(TauConstituent* shot);
    std::vector<TauConstituent*>            getShots() const;
    unsigned int                            getNShots() const;
    
    void                                    setNPhotonsInShot(int nPhotons);
    int                                     getNPhotonsInShot() const;
 
  private:
 
     FourMom_t m_p4;
 
 
  protected:
    
    std::vector<int>                m_TypeFlags;
    double                          m_BDTValue;
    int                             m_Charge;
    
    const xAOD::PFO*                m_PFO;
    
    // the constituents in there are owned by this! delete them!
    std::vector<TauConstituent*>    m_Shots;
    int                             m_nPhotonsInShot;    
 
  };
 
 
} //end namespace
#endif // PANTAUALGS_TAUCONSTITUENT_H