InDetSecVtxTruthMatchTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef InDetSecVtxTruthMatchTool_h
#define InDetSecVtxTruthMatchTool_h
 
// Framework include(s):
#include "AsgTools/AsgTool.h"
#include "AsgTools/PropertyWrapper.h"
 
// EDM include(s):
#include "xAODTracking/VertexContainerFwd.h"
#include "xAODTruth/TruthParticleFwd.h"
#include "xAODTruth/TruthVertexFwd.h"
#include "xAODTruth/TruthVertexContainer.h"
#include "InDetSecVtxTruthMatchTool/IInDetSecVtxTruthMatchTool.h"
 
// standard includes
#include<vector>
 
namespace InDetSecVtxTruthMatchUtils {
 
  //Namespace for useful analysis things on the truth matching decorations applied to the VertexContainer
 
  //How the matching info is stored; link to the truth vertex, a float with its contribution to the relative track weight, and a float with its contribution to the track sumpt2 of the truth vertex
  typedef std::tuple<ElementLink<xAOD::TruthVertexContainer>, float, float> VertexTruthMatchInfo;
 
  //type codes for vertex matching on all vertices
  enum VertexMatchType {
    Matched=0, // > threshold (default 50%) from one truth interaction
    Merged,    // not matched
    Split,     // highest weight truth interaction contributes to >1 vtx (vtx with highest fraction of sumpT2 remains matched/merged)
    Fake,      // highest contribution is fake (if pile-up MC info not present those tracks end up as "fakes")
    Other
  };
 
  //type codes for truth vertices
  //NOTE: types are subsets of subsequent types
  enum TruthVertexMatchType {
    Reconstructable=0, // fiducial cuts, >= 2 charged daughters
    Accepted,          // >= 2 reco tracks
    Seeded,            // tracks pass cuts
    Reconstructed,     // matched to reco vtx
    ReconstructedSplit // matched to multiple vtx
  };
 
  inline bool isMatched(int matchInfo) {
    if (matchInfo & (0x1 << Matched)) return true;
    return false;
  }
  inline bool isMerged(int matchInfo) {
    if (matchInfo & (0x1 << Merged)) return true;
    return false;
  }
  inline bool isSplit(int matchInfo) {
    if (matchInfo & (0x1 << Split)) return true;
    return false;
  }
  inline bool isFake(int matchInfo) {
    if (matchInfo & (0x1 << Fake)) return true;
    return false;
  }
  inline bool isOther(int matchInfo) {
    if (matchInfo & (0x1 << Other)) return true;
    return false;
  }
 
  inline bool isReconstructable(int matchInfo) {
    if (matchInfo & (0x1 << Reconstructable)) return true;
    return false;
  }
  inline bool isAccepted(int matchInfo) {
    if (matchInfo & (0x1 << Accepted)) return true;
    return false;
  }
  inline bool isSeeded(int matchInfo) {
    if (matchInfo & (0x1 << Seeded)) return true;
    return false;
  }
  inline bool isReconstructed(int matchInfo) {
    if (matchInfo & (0x1 << Reconstructed)) return true;
    return false;
  }
  inline bool isReconstructedSplit(int matchInfo) {
    if (matchInfo & (0x1 << ReconstructedSplit)) return true;
    return false;
  }
}
 
class InDetSecVtxTruthMatchTool : public virtual IInDetSecVtxTruthMatchTool,
                                     public asg::AsgTool {
 
  ASG_TOOL_CLASS( InDetSecVtxTruthMatchTool, IInDetSecVtxTruthMatchTool )
 
 public:
 
  InDetSecVtxTruthMatchTool( const std::string & name );
 
  virtual StatusCode initialize() override final;
 
  // take collection of vertices, match them, and decorate with matching info
  virtual StatusCode matchVertices( std::vector<const xAOD::Vertex*> recoVerticesToMatch, std::vector<const xAOD::TruthVertex*> truthVerticesToMatch, const xAOD::TrackParticleContainer* trackParticles ) override;
 
 private:
 
  Gaudi::Property<float> m_trkMatchProb{this, "trackMatchProb", 0.5, "Required MC match probability to consider track a good match" };
  Gaudi::Property<float> m_vxMatchWeight{this, "vertexMatchWeight", 0.5, "Relative weight threshold to consider vertex matched"};
  Gaudi::Property<float> m_trkPtCut{this, "trackPtCut", 1000., "pt cut to apply on tracks"};
  Gaudi::Property<std::string> m_selectedTrackFlag{this, "selectedTrackFlag", "is_selected", "Aux decoration on tracks for seeding efficiencies"};
 
  //private methods to check if particles are good to use
  //returns barcode of LLP production truth vertex
  int checkProduction( const xAOD::TruthParticle& truthPart, std::vector<const xAOD::TruthVertex*> truthVerticesToMatch ) const;
  void countReconstructibleDescendentParticles(const xAOD::TruthVertex& signalTruthVertex,
                                               std::vector<const xAOD::TruthParticle*>& set, int counter) const;
  std::vector<int> checkParticle( const xAOD::TruthParticle& part, const xAOD::TrackParticleContainer* tkCont ) const;
};
 
#endif