Tracking/TrkEvent/TrkTrack/src/TrackInfo.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TrkTrack/TrackInfo.h"
#include "GaudiKernel/MsgStream.h"
#include <iostream>
#include <sstream>
 
Trk::TrackInfo::TrackInfo(
  const TrackFitter fitter,
  const ParticleHypothesis partHyp,
  const std::bitset<NumberOfTrackProperties>& properties,
  const std::bitset<NumberOfTrackRecoInfo>& patternRecognition)
  : m_fitter(fitter)
  , m_particleHypo(partHyp)
  , m_properties(properties)
  , m_patternRecognition(patternRecognition)
{
 
  if (m_particleHypo == undefined) {
    switch (m_fitter) // switch on TrackFitter
    {
      case GaussianSumFilter: // GaussianSumFilter as TrackFitter (electron)
        m_particleHypo = electron;
        break;
      case DeterministicAnnealingFilter:
      case DistributedKalmanFilter:
      case GlobalChi2Fitter:
        m_particleHypo = muon;
        break;
      default:
        m_particleHypo = pion;
    } // end of switch operator
  }   // end of particle hypothesis check
}
 
Trk::TrackInfo::TrackInfo(const TrackFitter fitter,
                          const ParticleHypothesis partHyp)
  : TrackInfo(fitter, partHyp, {}, {})
{
}
 
std::string
Trk::TrackInfo::dumpInfo() const
{
  std::stringstream authorInfo;
  authorInfo << "The Fitter of this track is: ";
 
  switch (m_fitter) {
    case Trk::TrackInfo::Unknown:
      authorInfo << "Unknown";
      break;
 
    case Trk::TrackInfo::iPatTrackFitter:
      authorInfo << "iPatTrackFitter";
      break;
 
    case Trk::TrackInfo::xKalman:
      authorInfo << "xKalman";
      break;
 
    case Trk::TrackInfo::KalmanFitter:
      authorInfo << "KalmanFitter";
      break;
 
    case Trk::TrackInfo::GaussianSumFilter:
      authorInfo << "GaussianSumFilter";
      break;
 
    case Trk::TrackInfo::GlobalChi2Fitter:
      authorInfo << "GlobalChi2Fitter";
      break;
 
    case Trk::TrackInfo::DistributedKalmanFilter:
      authorInfo << "DistributedKalmanFilter";
      break;
 
    case Trk::TrackInfo::DeterministicAnnealingFilter:
      authorInfo << "DeterministicAnnealingFilter";
      break;
 
    case Trk::TrackInfo::MuonboyFitter:
      authorInfo << "MuonboyFitter";
      break;
 
    default:
      authorInfo << "Fitter not recognized " << m_fitter;
  }
 
  // adding properties, but checking first
  // that they are there at all
  if (m_properties.any()) {
    authorInfo << (m_properties.test(Trk::TrackInfo::BremFit) ? " + BremFit" : "");
    authorInfo << (m_properties.test(Trk::TrackInfo::BremFitSuccessful) ? " + BremFitSuccess" : "");
    authorInfo << (m_properties.test(Trk::TrackInfo::StraightTrack) ? " + StraightTrack" : "");
    authorInfo << (m_properties.test(Trk::TrackInfo::SlimmedTrack) ? " + SlimmedTrack" : "");
    authorInfo << (m_properties.test(Trk::TrackInfo::HardScatterOrKink) ? " + HardScatterOrKink"
                                                                        : "");
    authorInfo << (m_properties.test(Trk::TrackInfo::LowPtTrack) ? " + LowPtTrack" : "");
  } // end of adding track properties
 
  // adding a pattern recognition structure
  if (m_patternRecognition.any()) {
    //     std::cout<<"calling the pattern reco dump"<<std::endl;
    authorInfo << "; The Pattern Recognition Agorithms used: ";
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::SiSPSeededFinder) ? "SiSPSeedFinder; "
                                                                               : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::SiCTBTracking) ? "SiCTBTracking; "
                                                                            : "");
    //      std::cout<<"3d line pattern reco dump"<<std::endl;
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::InDetAmbiguitySolver)
                     ? "InDetAmbiguitySolver; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::InDetExtensionProcessor)
                     ? "InDetExtensionProcessor; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::TRTSeededTrackFinder)
                     ? "TRTSeededTrackFinder; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::Muonboy) ? "Muonboy; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::MuGirlUnrefitted)
                     ? "MuGirlUnrefitted; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::STACO) ? "STACO; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::StacoLowPt) ? "StacoLowPt; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::MuTag) ? "MuTag; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::MooreToTrackTool)
                     ? "MooreToTrackTool; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::TrigIDSCAN) ? "TrigIDSCAN; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::TrigSiTrack) ? "TrigSiTrack; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::TrigTRTxK) ? "TrigTRTxK; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::TrigTRTLUT) ? "TrigTRTLUT; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::Fatras) ? "Fatras; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::iPatLegacyCnv) ? "iPatLegacyCnv; "
                                                                            : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::xKalmanLegacyCnv)
                     ? "xKalmanLegacyCnv; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::SimpleAmbiguityProcessorTool)
                     ? "SimpleAmbiguityProcessorTool; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::InDetAmbiTrackSelectionTool)
                     ? "InDetAmbiTrackSelectionTool; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::TRTStandalone) ? "TRTStandalone; "
                                                                            : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::MuidStandAlone) ? "MuidStandAlone; "
                                                                             : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::TRTSeededSingleSpTrackFinder)
                     ? "TRTSeededSingleSpTrackFinder; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::MooreLegacyCnv) ? "MooreLegacyCnv; "
                                                                             : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::MuidComb) ? "MuidComb; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::Moore) ? "Moore; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::MuidCombined) ? "MuidCombined; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::MuidVertexAssociator)
                     ? "MuidVertexAssociator; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::MuGirl) ? "MuGirl; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::iPatRec) ? "iPatRec; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::MuGirlLowBeta) ? "MuGirlLowBeta; "
                                                                            : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::FatrasSimulation)
                     ? "FatrasSimulation; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::MuonNotHittingTileVolume)
                     ? "MuonNotHittingTileVolume; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::ReverseOrderedTrack)
                     ? "ReverseOrderedTrack; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::SiSpacePointsSeedMaker_Cosmic)
                     ? "SiSpacePointsSeedMaker_Cosmic; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::SiSpacePointsSeedMaker_HeavyIon)
                     ? "SiSpacePointsSeedMaker_HeavyIon; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::SiSpacePointsSeedMaker_LowMomentum)
                     ? "SiSpacePointsSeedMaker_LowMomentum; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::SiSpacePointsSeedMaker_BeamGas)
                     ? "SiSpacePointsSeedMaker_BeamGas; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::SiSpacePointsSeedMaker_VeryLowMomentum)
                     ? "SiSpacePointsSeedMaker_VeryLowMomentum; "
                     : "");
    authorInfo << (m_patternRecognition.test(
                     Trk::TrackInfo::SiSpacePointsSeedMaker_ITkConversionTracks)
                     ? "SiSpacePointsSeedMaker_ITkConversionTracks; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::SiSpacePointsSeedMaker_LargeD0)
                     ? "SiSpacePointsSeedMaker_LargeD0; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::Pseudotracking) ? "Pseudotracking; "
                                                                             : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::MuidMuonRecoveryTool)
                     ? "MuidMuonRecoveryTool; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::MuidStandaloneRefit)
                     ? "MuidStandaloneRefit; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::TrackInCaloROI) ? "TrackInCaloROI; "
                                                                             : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::SiSpacePointsSeedMaker_ForwardTracks)
                     ? "SiSpacePointsSeedMaker_ForwardTracks; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::strategyA) ? "strategyA; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::strategyB) ? "strategyB; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::strategyC) ? "strategyC; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::FTK) ? "FTK; " : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::FastTrackFinderSeed)
                     ? "FastTrackFinderSeed; "
                     : "");
    authorInfo << (m_patternRecognition.test(Trk::TrackInfo::SiSPSeededFinderSimple)
                     ? "SiSPSeededFinderSimple; "
                     : "");
 
  } // end of adding pattern recognition properties
  return authorInfo.str();
} // end of dump method
 
std::ostream&
Trk::operator<<(std::ostream& sl, const Trk::TrackInfo& info)
{
  sl << info.dumpInfo() << std::endl;
  return sl;
}
 
MsgStream&
Trk::operator<<(MsgStream& sl, const Trk::TrackInfo& info)
{
  sl << info.dumpInfo() << endmsg;
  return sl;
}