/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/ParticleJetTools/ParticleJetTools/JetQuarkLabel.h

//Dear emacs, this is -*-c++-*-

// JetQuarkLabel.h
//   Header file for class JetQuarkLabel

#ifndef PARTICLEJETTOOLS_JETQUARKLABEL_H
#define PARTICLEJETTOOLS_JETQUARKLABEL_H

#include "AsgTools/AsgTool.h"
#include "ParticleJetTools/IJetTruthMatching.h"
#include "EventPrimitives/EventPrimitives.h"
#include <string>
#include <map>
#include "xAODJet/Jet.h"
#include "xAODTruth/TruthEventContainer.h"
#include "xAODTruth/TruthParticleContainer.h"
#include "xAODTruth/TruthParticle.h"

//class McEventCollection;
class Jet;

namespace Analysis
{

class JetQuarkLabel : public asg::AsgTool, virtual public IJetTruthMatching {
    ASG_TOOL_CLASS(JetQuarkLabel, IJetTruthMatching)
    public:
        JetQuarkLabel(const std::string& name);
        virtual ~JetQuarkLabel();
        StatusCode initialize();
        StatusCode finalize();

        /* Method to truth tag a jet.
         * NB: for this particular algorithm, matchJet is TRUE if the jet is matched 
         * to a b-quark OR a c-quark. The jetLabel is set accordingly and is no 
         * longer a job option.
         */

        virtual bool matchJet(const xAOD::Jet& myJet);

        virtual void m_printParameterSettings();

        int barcode() const;

        int pdgCode() const;

        inline const int& jetLabel() const { return m_jetLabel; }

        double deltaRMinTo(const std::string&) const;
        inline const std::map<std::string, double>& distanceToQuarks() const { return m_distanceToQuarks; }

        inline const Eigen::Vector3d& BDecVtx() const { return m_BDecVtx; }
        int Bpdg() const { return m_Bpdg; }

        inline int NEventInCollection() const { return m_NEventInCollection; }
        inline void EventSelection(short s) { m_inTime = s; }
        inline short EventSelection() const { return m_inTime; }

    private:
        std::string m_mcEventCollection; 
        double m_deltaRCut; 
        double m_ptCut;     
        bool   m_noDoc;
        short  m_inTime;
        bool m_testJet(const xAOD::Jet&, const xAOD::TruthEventContainer*);
        int m_jetLabel; 
        std::map<std::string, double> m_distanceToQuarks; 
        Eigen::Vector3d m_BDecVtx; 
        int    m_pdg;       
        int    m_barcode;   
        int m_Bpdg;
        int m_NEventInCollection;
};

inline int JetQuarkLabel::barcode()  const { return m_barcode; }

inline int JetQuarkLabel::pdgCode()  const { return m_pdg; }

}
#endif // TRUTHMATCHTOOLS_JETQUARKLABEL_H

---