TrackCollHandle_TrkTrack.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 

//                                                            //
//  Implementation of class TrackCollHandle_TrkTrack          //
//                                                            //
//  Author: Thomas H. Kittelmann (Thomas.Kittelmann@cern.ch)  //
//  Initial version: February 2008                            //
//                                                            //
 
#include "VP1TrackSystems/TrackCollHandle_TrkTrack.h"
#include "VP1TrackSystems/TrackHandle_TrkTrack.h"
#include "VP1Base/IVP1System.h"
#include "VP1TrackSystems/TrackSystemController.h"
#include "VP1Base/VP1QtUtils.h"
#include "VP1Utils/VP1JobConfigInfo.h"
#include "VP1Utils/VP1SGContentsHelper.h"
#include "VP1Utils/VP1SGAccessHelper.h"
 
#include "TrkTrack/Track.h"
#include "TrkTrack/TrackCollection.h"
 
#include <QRegExp>
 
//____________________________________________________________________
class TrackCollHandle_TrkTrack::Imp {
public:
  static bool isFatrasTruthTrack(const QString& key);
  static bool alwaysShowAllTrackColls;
  TrackCollHandle_TrkTrack * theclass;
  TrackCommonFlags::DETAILLEVEL detailLevel;
 
};
 
bool TrackCollHandle_TrkTrack::Imp::alwaysShowAllTrackColls
  = VP1QtUtils::environmentVariableIsOn("VP1_ALWAYS_SHOW_ALL_TRACK_COLLECTIONS");
 
//____________________________________________________________________
QStringList TrackCollHandle_TrkTrack::availableCollections(IVP1System*sys)
{
  //There must be at least one of the trackers available for this to make sense.
  if ( ! ( VP1JobConfigInfo::hasPixelGeometry()||VP1JobConfigInfo::hasSCTGeometry()
       ||VP1JobConfigInfo::hasTRTGeometry()||VP1JobConfigInfo::hasMuonGeometry() ) ) {
    sys->messageDebug("TrackCollHandle_TrkTrack::availableCollections: Neither"
              " Pixel, SCT, TRT or Muon geometry available. Won't"
              " look in event store for Trk::Track collections");
    return {};
  }
 
  QStringList keysInSG;
  for(const QString& key : VP1SGContentsHelper(sys).getKeys<TrackCollection>()) {
    // if ( key=="CombinedInDetTracks") // Useful for debugging to limit to one collection
      keysInSG<<key;
  }
 
  if (keysInSG.isEmpty()||Imp::alwaysShowAllTrackColls)
    return keysInSG;
 
  QRegExp needsMuonsPattern("Converted.*|.*MBoy.*|.*Moore.*|.*Mu.*|.*Staco.*");
  QRegExp needsTRTPattern(".*trt.*|.*TRT.*|.*Trt.*");
  QRegExp needsAllIDPattern("ExtendedTracks|ResolvedTracks|Tracks|.*Id.*|.*ID.*|.*InDet.*|.*Inner.*");
 
  const bool jobcfgAllID
    = VP1JobConfigInfo::hasPixelGeometry()
    &&VP1JobConfigInfo::hasSCTGeometry()
    &&VP1JobConfigInfo::hasTRTGeometry();
 
  QStringList outkeys;
 
  for (const QString& key : keysInSG) {
    if (!VP1JobConfigInfo::hasMuonGeometry() && needsMuonsPattern.exactMatch(key)) {
      sys->messageDebug("TrackCollHandle_TrkTrack::availableCollections: Ignoring key '"
               +key+"' since muon geometry is not present in job.");
      continue;
    }
    if (!jobcfgAllID && needsAllIDPattern.exactMatch(key)) {
      sys->messageDebug("TrackCollHandle_TrkTrack::availableCollections: Ignoring key '"
            +key+"' since pixel, sct and trt geometry are not all present in job.");
      continue;
    }
    if (!VP1JobConfigInfo::hasTRTGeometry()&&needsTRTPattern.exactMatch(key)) {
      sys->messageDebug("TrackCollHandle_TrkTrack::availableCollections: Ignoring key '"
            +key+"' since trt geometry is not present in job.");
      continue;
    }
 
    outkeys<<key;
  }
  return outkeys;
 
}
 
//____________________________________________________________________
TrackCollHandle_TrkTrack::TrackCollHandle_TrkTrack( TrackSysCommonData * cd,
                            const QString& name,
                            TrackType::Type type )
  : TrackCollHandleBase(cd,name,type), m_d(new Imp)
{
  m_d->theclass = this;
  m_d->detailLevel = TrackCommonFlags::AUTO;
}
 
//____________________________________________________________________
TrackCollHandle_TrkTrack::~TrackCollHandle_TrkTrack()
{
  delete m_d;
}
 
//____________________________________________________________________
bool TrackCollHandle_TrkTrack::Imp::isFatrasTruthTrack(const QString& key)
{
  QString env = VP1QtUtils::environmentVariableValue("VP1_FATRAS_TRACKCOLLS_PATTERN");
  return QRegExp((env.isEmpty()?"*fatras*":env),Qt::CaseInsensitive,QRegExp::Wildcard).exactMatch(key);
}
 
//____________________________________________________________________
bool TrackCollHandle_TrkTrack::load()
{
  //Get collection:
  const TrackCollection * trackColl(nullptr);
  if (!VP1SGAccessHelper(systemBase()).retrieve(trackColl, name())) {
    message("Error: Could not retrieve track collection with key="+name());
    return false;
  }
 
  //Make appropriate trk::track handles:
  hintNumberOfTracksInEvent(trackColl->size());
  int i(0);
  TrackCollection::const_iterator trackItr, trackItrEnd = trackColl->end();
  for ( trackItr = trackColl->begin() ; trackItr != trackItrEnd; ++trackItr) {
    if (*trackItr)
      addTrackHandle(new TrackHandle_TrkTrack(this,*trackItr));
    else
      messageDebug("WARNING: Ignoring null Trk::Track pointer.");
    if (!(i++%100))
      systemBase()->updateGUI();
  }
 
  //Maybe we need to show measurements, etc.:
  //  updateVisibleAssociatedObjects();
 
  return true;
}
 
//____________________________________________________________________
bool TrackCollHandle_TrkTrack::cut(TrackHandleBase* handle)
{
  if (!TrackCollHandleBase::cut(handle))
    return false;
 
  //fixme: more?
  return true;
}
 
//____________________________________________________________________
void TrackCollHandle_TrkTrack::visibleStateUpdated(TrackHandle_TrkTrack* trkhandle)
{
  common()->visTrkTracksToMaterialHelper()->setState(trkhandle->trkTrackPointer(), (trkhandle->visible() ? trkhandle->currentMaterial() : nullptr ));
}