TauPID.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef TAUEVENT_TAUPID_H
#define TAUEVENT_TAUPID_H
 
//TauEvent includes
#include "tauEvent/TauJetParameters.h"
//C/C++ includes
#include <vector>
#include <bitset>
#include <utility>
 
namespace Analysis
{
    class TauPID
    {
        public:
            TauPID();

            virtual ~TauPID();
 
            TauPID (std::vector<std::pair<TauJetParameters::TauID, double> >&& params,
                    const std::bitset<32>& isTauFlags,
                    const std::bitset<32>& vetoFlags);
 

            TauPID(const TauPID& rhs);

            TauPID &operator = ( const TauPID &rhs );
 
            TauPID &operator = ( TauPID &&rhs );

            bool operator==( const TauPID& rhs ) const;

            bool operator!=( const TauPID& rhs ) const;
            
            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 );

            bool flag(
                    TauJetParameters::VetoFlags flag ) const;
 
            const std::vector<std::pair<TauJetParameters::TauID, double> >& params() const;
            const std::bitset<32>& isTauFlags() const;
            const std::bitset<32>& vetoFlags() const;

        private:
            std::vector<std::pair<TauJetParameters::TauID, double> > m_params;
            std::bitset<32> m_isTauFlags;
            std::bitset<32> m_vetoFlags;
    };
}
 
#include "TauDetails.h"
#include "TauPID.icc"
 
#endif // TAUEVENT_TAUPID_H