Tracking/TrkEvent/TrkTrack/TrkTrack/TrackInfo.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef TRKTRACK_TRACKINFO_H
#define TRKTRACK_TRACKINFO_H
 
#include <bitset>
#include <iostream>
#include <string>
#include <vector>
 
#include "TrkEventPrimitives/ParticleHypothesis.h"
 
class MsgStream;
class TrackCnv_p4;
class TrackParticleBaseCnv_p1;
 
namespace Trk {

 
class TrackInfo
{
public:
  enum TrackFitter
  {
    Unknown = 0,

    iPatTrackFitter = 1,

    xKalman = 2,

    KalmanFitter = 3,

    GaussianSumFilter = 4,

    GlobalChi2Fitter = 5,

    DistributedKalmanFilter = 6,

    DeterministicAnnealingFilter = 7,

    KalmanDNAFitter = 8,

    MuonboyFitter = 9,

    NumberOfTrackFitters = 10
  };
 
  enum TrackProperties
  {

    BremFit = 1,

    BremFitSuccessful = 2,

    StraightTrack = 3,

    SlimmedTrack = 4,

    HardScatterOrKink = 5,

    LowPtTrack = 6,

    NumberOfTrackProperties = 7
  };
 
  enum TrackPatternRecoInfo
  {
    SiSPSeededFinder = 0,

    SiCTBTracking = 1,

    InDetAmbiguitySolver = 2,

    InDetExtensionProcessor = 3,

    TRTSeededTrackFinder = 4,

    Muonboy = 5,

    MuGirlUnrefitted = 6,

    STACO = 7,

    StacoLowPt = 8,

    MuTag = 9,

    MooreToTrackTool = 10,

    TrigIDSCAN = 11,

    TrigSiTrack = 12,

    TrigTRTxK = 13,

    TrigTRTLUT = 14,

    Fatras = 15,

    iPatLegacyCnv = 16,

    xKalmanLegacyCnv = 17,

    SimpleAmbiguityProcessorTool = 18,

    InDetAmbiTrackSelectionTool = 19,

    TRTStandalone = 20,

    MuidStandAlone = 21,

    TRTSeededSingleSpTrackFinder = 22,

    MooreLegacyCnv = 23,

    MuidComb = 24,

    Moore = 25,

    MuidCombined = 26,

    MuidVertexAssociator = 27,

    MuGirl = 28,

    iPatRec = 29,

    MuGirlLowBeta = 30,

    FatrasSimulation = 31,

    ReverseOrderedTrack = 32,

    MuonNotHittingTileVolume = 33,

    SiSpacePointsSeedMaker_Cosmic = 34,
    SiSpacePointsSeedMaker_HeavyIon = 35,
    SiSpacePointsSeedMaker_LowMomentum = 36,
    SiSpacePointsSeedMaker_BeamGas = 37,
    SiSpacePointsSeedMaker_VeryLowMomentum = 38,

    MuidMuonRecoveryTool = 39,

    MuidStandaloneRefit = 40,

    TrackInCaloROI = 41,

    SiSpacePointsSeedMaker_ForwardTracks = 42,

    strategyA = 43,
    strategyB = 44,
    strategyC = 45,

    FTK = 46,

    FastTrackFinderSeed = 47,

    SiSPSeededFinderSimple = 48,

    SiSpacePointsSeedMaker_LargeD0 = 49,

    SiSpacePointsSeedMaker_ITkConversionTracks = 50,

    Pseudotracking = 51,

    NumberOfTrackRecoInfo = 52
  };

  TrackInfo(const TrackFitter fitter, const ParticleHypothesis partHyp);
  TrackInfo(const TrackFitter fitter,
            const ParticleHypothesis partHyp,
            const std::bitset<NumberOfTrackProperties>& properties,
            const std::bitset<NumberOfTrackRecoInfo>& patternRecognition);

  TrackInfo() = default;
  TrackInfo(const TrackInfo&) = default;
  TrackInfo(TrackInfo&&) = default;
  TrackInfo& operator=(const TrackInfo&) = default;
  TrackInfo& operator=(TrackInfo&&) = default;
  ~TrackInfo() = default;
 
  TrackInfo* clone() const;
  // Various set methods

  void setTrackProperties(const TrackProperties& properties);

  void setPatternRecognitionInfo(const TrackPatternRecoInfo& patternReco);

  void setTrackFitter(const TrackFitter& fitter);

  void setParticleHypothesis(const ParticleHypothesis& hypothesis);

  const TrackFitter& trackFitter() const;

  bool trackProperties(const TrackProperties& property) const;

  bool patternRecoInfo(const TrackPatternRecoInfo& pattern) const;

  void addPatternRecoAndProperties(const TrackInfo&);

  void addPatternReco(const TrackInfo&);

  std::string dumpInfo() const;
 
  std::string dumpBitsProperties() const;
 
  std::string dumpBitsPatternRecoInfo() const;

  ParticleHypothesis particleHypothesis() const;
 
  const std::bitset<NumberOfTrackProperties>& properties() const;
  const std::bitset<NumberOfTrackRecoInfo>& patternRecognition() const;
 
private:
  friend class ::TrackCnv_p4;
  friend class ::TrackParticleBaseCnv_p1;

  TrackFitter m_fitter = Unknown;

  ParticleHypothesis m_particleHypo = undefined;

  std::bitset<NumberOfTrackProperties> m_properties;

  std::bitset<NumberOfTrackRecoInfo> m_patternRecognition;
 
}; // end of class definitions

MsgStream&
operator<<(MsgStream& sl, const TrackInfo& track);

std::ostream&
operator<<(std::ostream& sl, const TrackInfo& track);
 
} // end of namespace Trk
 
#include "TrkTrack/TrackInfo.icc"
#endif