/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/InDetTrackSystematicsTools/InDetTrackSystematicsTools/InDetTrackTruthOriginTool.h

#pragma once

#include "IInDetTrackTruthOriginTool.h"
#include "AsgTools/AsgTool.h"
#include "xAODTracking/TrackParticle.h"
#include "xAODTruth/TruthParticle.h"
#include <string>

namespace InDet {

  class InDetTrackTruthOriginTool : public virtual IInDetTrackTruthOriginTool, public asg::AsgTool {

    ASG_TOOL_CLASS2( InDetTrackTruthOriginTool,
             asg::IAsgTool,
             InDet::IInDetTrackTruthOriginTool )

  public:

  InDetTrackTruthOriginTool (const std::string& name);
  virtual ~InDetTrackTruthOriginTool();

  virtual StatusCode initialize() override;
  virtual void prepare() override {};
  virtual StatusCode finalize() override;

  virtual int getTrackOrigin(const xAOD::TrackParticle* track) const override;
  virtual int getTrackOrigin(const xAOD::TrackParticle* track, int &parentID) const override;

private:

  bool isFrom(const xAOD::TruthParticle* part, int flav) const;
  int getParentID(const xAOD::TruthParticle* part) const;

  float m_matchingProbabilityCut = 0.75;
  float m_barcodeG4 = 2e5;

  std::string m_truthParticleLinkName = "truthParticleLink";
  std::string m_truthMatchProbabilityAuxName = "truthMatchProbability";

}; // class InDetTrackTruthOriginTool

} // namespace InDet

---