TrigR3Mon.cxx

Go to the documentation of this file.

 
#include "TrigInDetAnalysis/Filter_AcceptAll.h"
 
#include "TrigInDetAnalysisUtils/Filters.h"
#include "TrigInDetAnalysisUtils/Filter_Track.h"
#include "TrigInDetAnalysisUtils/TagNProbe.h"
 
#include "TrigInDetAnalysisExample/ChainString.h"
#include "TrigInDetAnalysisExample/TIDAHistogram.h"
 
#include "TrigR3Mon.h"
 
#include "TrigInDetAnalysisExample/AnalysisConfigMT_Ntuple.h"
 
#include <mutex>
 
TrigR3Mon::TrigR3Mon( const std::string & name, ISvcLocator* pSvcLocator) 
  :  AthMonitorAlgorithm(name, pSvcLocator), 
     m_tdt("Trig::TrigDecisionTool/TrigDecisionTool"),
     m_buildNtuple(false),
     m_initialisePerRun(true),
     m_firstRun(true),
     m_keepAllEvents(false),
     m_fileopen(false),
     m_useHighestPT(false),
     m_vtxIndex(-1),
     m_runPurity(false),
     m_shifter(false),
     m_shifterChains(1),
     m_sliceTag(""),
     m_containTracks(false),
     m_legacy(true), 
     m_fiducial_radius(32),
     m_requireDecision(false),
     m_filter_on_roi(false)
{
# ifdef USE_DATE_TIME_STAMP   
  msg(MSG::INFO) << "TrigR3Mon::TrigR3Mon() compiled: " << __DATE__ << " " << __TIME__ << endmsg;
# endif
  
  declareProperty( "SliceTag",   m_sliceTag = "" );
 
  declareProperty( "pTCut",   m_pTCut   = 0 );
  declareProperty( "etaCut",  m_etaCut  = 5 );
  declareProperty( "d0Cut",   m_d0Cut   = 1000 );
  declareProperty( "z0Cut",   m_z0Cut   = 2000 );
  declareProperty( "siHits",  m_siHits  = -1 );
  m_pixHits=0;
  m_sctHits=0;
 
  declareProperty( "trtHits",   m_trtHits   = -2 );
  declareProperty( "strawHits", m_strawHits = -2 );
 
  declareProperty( "tauEtCutOffline",   m_tauEtCutOffline = 0 );
  declareProperty( "doTauThreeProng",   m_doTauThreeProng = false);
 
  declareProperty( "pTCutOffline",      m_pTCutOffline      = 1000 );
  declareProperty( "etaCutOffline",     m_etaCutOffline     = 2.5 );
  declareProperty( "d0CutOffline",      m_d0CutOffline      = 1000 );
  declareProperty( "mind0CutOffline",   m_mind0CutOffline   = 0 );
  declareProperty( "z0CutOffline",      m_z0CutOffline      = 2000 );
  declareProperty( "pixHitsOffline",    m_pixHitsOffline    =  2 ); // 1 <- old value ( 2 degrees of freedom = 1 cluster ) 
  declareProperty( "sctHitsOffline",    m_sctHitsOffline    =  6 ); // 6 <- old value ( 6 clusters = 3 spacepoints )
  declareProperty( "siHitsOffline",     m_siHitsOffline     =  8 );
  declareProperty( "blayerHitsOffline", m_blayerHitsOffline = -1 ); // no requirement - in case IBL is off
 
  declareProperty( "pixHolesOffline",   m_pixHolesOffline   =  20 ); // essentially no limit
  declareProperty( "sctHolesOffline",   m_sctHolesOffline   =  20 ); // essentially no limit
  declareProperty( "siHolesOffline",    m_siHolesOffline    =  2 );  // npix holes + nsi holes <= 2 ( not degrees of freedom ! )   
 
  declareProperty( "ContainTracks",     m_containTracks     = false );  // use only basic track containment
  declareProperty( "FiducialRadius",    m_fiducial_radius = 32 );
 
  declareProperty( "trtHitsOffline",    m_trtHitsOffline    = -2 );
  declareProperty( "strawHitsOffline",  m_strawHitsOffline  = -2 );
 
  declareProperty( "matchR",   m_matchR   = 0.1 );
  declareProperty( "matchPhi", m_matchPhi = 0.1 );
 
  declareProperty( "ntupleChainNames",  m_ntupleChainNames );
  declareProperty( "releaseMetaData",   m_releaseMetaData );
 
  declareProperty( "buildNtuple",   m_buildNtuple = false );
  declareProperty( "mcTruth",       m_mcTruthIn = false );
 
  declareProperty( "AnalysisConfig", m_analysis_config = "Ntuple");
 
  declareProperty( "SelectTruthPdgId",       m_selectTruthPdgId = 0 );
  declareProperty( "SelectParentTruthPdgId", m_selectParentTruthPdgId = 0 );
 
  declareProperty( "InitialisePerRun", m_initialisePerRun = false );
  declareProperty( "KeepAllEvents",    m_keepAllEvents = false );
  declareProperty( "UseHighestPT",     m_useHighestPT = false );
  declareProperty( "VtxIndex",         m_vtxIndex = -1 );
 
  declareProperty( "RunPurity",        m_runPurity = false );
  declareProperty( "Shifter",          m_shifter = false );
 
  declareProperty( "ShifterChains",    m_shifterChains = 1 );
 
  declareProperty( "GenericFlag",      m_genericFlag = true );
 
  declareProperty( "Legacy",           m_legacy = true );
 
  declareProperty( "outputFileName",   m_outputFileName = "TrkNtuple.root" );
 
  declareProperty( "RequireDecision", m_requireDecision = false );
 
  declareProperty( "FilterOnRoi",     m_filter_on_roi = false );
  
  msg(MSG::INFO) << "TrigR3Mon::TrigR3Mon() exiting " << gDirectory->GetName() << endmsg;
 
}
 
 
 
TrigR3Mon::~TrigR3Mon() {
 
  if ( m_fileopen ) { 
    for ( unsigned i=0 ; i<m_sequences.size() ; i++ ) m_sequences[i]->finalize();
    for ( unsigned i=0 ; i<m_sequences.size() ; i++ ) delete m_sequences[i];
    m_fileopen = false;
  }
 
  // msg(MSG::INFO) << "TrigR3Mon::~TrigR3Mon()" << endmsg;
  for ( unsigned i=m_filters.size()     ; i-- ; ) delete m_filters[i];
  for ( unsigned i=m_associators.size() ; i-- ; ) delete m_associators[i];
 
}
 
 
 
StatusCode TrigR3Mon::initialize() {
 
  msg(MSG::DEBUG) << " ----- enter initialize() ----- " << endmsg;
 
  msg(MSG::INFO) << "TrigR3Mon::initialize() " << gDirectory->GetName() << " " << m_sliceTag << endmsg;

 
#if 0
  std::cout << "TrigR3Mon::name              = " << name()     << std::endl;
  std::cout << "TrigR3Mon::SliceTag          = " << m_sliceTag << std::endl;
  std::cout << "TrigR3Mon::AnalysisConfig    = " << m_analysis_config << std::endl;
  std::cout << "TrigR3Mon::Legacy            = " << m_legacy   << std::endl;
#endif
 
  ATH_CHECK(m_monTools.retrieve());  
 
  ATH_CHECK(bookHistograms());

 
  ATH_CHECK (m_trackdummykeys.initialize());
  ATH_CHECK (m_vtxdummykeys.initialize());
 
  ATH_MSG_DEBUG( " -----  exit init() ----- " );
 
  return AthMonitorAlgorithm::initialize();
 
}
 
 
StatusCode TrigR3Mon::bookHistograms() {
 
  ATH_MSG_DEBUG( "           ----- enter book() ----- " );
 
  ATH_MSG_INFO( "            TrigR3Mon::book() " << gDirectory->GetName() );

  // "^[[91;1m"
  // "^[[m"
 
  ATH_MSG_INFO( "^[[91;1m" << name() << "\t:AnalysisConfig " << m_analysis_config << "^[[m" );
  
  ATH_MSG_INFO( "configuring chains: " << m_ntupleChainNames.size() << "\tmonTools: " << m_monTools.size() );
  
  std::string lastvtx = "";

 
  bool mcTruth = m_mcTruthIn;
 
  if ( m_buildNtuple ) { 
  
        if ( m_tdt->getNavigationFormat() == "TriggerElement" ) { 
      m_sequences.push_back( new AnalysisConfig_Ntuple( m_ntupleChainNames, 
                                m_outputFileName, m_tauEtCutOffline, m_selectTruthPdgId, 
                                m_keepAllEvents ) );
    }
    else {
      AnalysisConfigMT_Ntuple* ac = new AnalysisConfigMT_Ntuple( m_ntupleChainNames, 
                                     m_outputFileName, m_tauEtCutOffline, m_selectTruthPdgId, 
                                     m_keepAllEvents, m_selectParentTruthPdgId );
 
      ac->set_fiducial_radius( m_fiducial_radius );
      ac->set_ptmin( m_pTCutOffline );
      m_sequences.push_back( ac );
    }
      
    m_sequences.back()->releaseData(m_releaseMetaData);
    if ( m_requireDecision ) m_sequences.back()->setRequireDecision(true);
    if ( m_mcTruthIn )       m_sequences.back()->setMCTruth(mcTruth);
    m_sequences.back()->setFilterOnRoi( m_filter_on_roi );
  }
  else if ( m_analysis_config=="Tier0" ) {

 
    // track filters
    // reference tracks (offline or truth) ...
    TrackFilter* filterRef = new Filter_Track( m_etaCutOffline,    m_d0CutOffline,   m_mind0CutOffline, m_z0CutOffline,  m_pTCutOffline,
                           m_pixHitsOffline,   m_sctHitsOffline, m_siHitsOffline, m_blayerHitsOffline,  
                           m_strawHitsOffline, m_trtHitsOffline, 0, 
                           m_pixHolesOffline, m_sctHolesOffline, m_siHolesOffline );
    
    // test (trigger) tracks..
    //  TrackFilter* filterTest = new Filter_Track( m_etaCut, m_d0Cut, m_z0Cut, m_pTCut, -1, -1, -1, -1,  -2, -2 );
    TrackFilter* filterTest = new Filter_AcceptAll();
    
    // keep track of the filters so they can be cleaned up at the end
    m_filters.push_back(filterRef);
    m_filters.push_back(filterTest);
    // m_filters.push_back(filterTest_TRT);
    
    
    // track associators
    TrackAssociator*  dR_matcher = new Associator_BestDeltaRMatcher("EBdeltaR", m_matchR); // this needs to be set correctly
    
    // keep track of the associators so they can be cleaned up at the end
    m_associators.push_back(dR_matcher);
    
    
    //    TrackAssociator* phi_matcher = new Associator_BestDeltaPhiMatcher("EBdeltaPhi", m_matchPhi); // this needs to be set correctly
    //    m_associators.push_back(phi_matcher);
        
    std::vector<std::string> chains;
    
    chains.reserve( m_monTools.size() );
    
 
    std::vector<ToolHandle<GenericMonitoringTool>*> monTools;
    monTools.reserve(m_monTools.size());
 
    ToolHandleArray<GenericMonitoringTool>::iterator toolitr = m_monTools.begin();;
    
    while ( toolitr!=m_monTools.end() ) {
         
      ChainString chainName = (toolitr->name());
      
      ATH_MSG_DEBUG( "configuring chain: " << chainName.head() << "\t: " << chainName.tail() );

            
      if ( chainName.head() == "" ) { 
    
    std::string selectChain = chainName.raw();

    //     toolitr->setPath( m_sliceTag+"/"+chainName );
 
    if (std::find(chains.begin(), chains.end(), selectChain) == chains.end()) { // deduplicate
      chains.push_back( selectChain );
      monTools.push_back( &(*toolitr) );
    }
    
      }
      else { 
    
    
    if ( chainName.head().find("HLT_")==std::string::npos && chainName.head().find("EF_")==std::string::npos ) {
      ++toolitr;
      continue;
    }
    
    
    std::string selectChain = chainName.head();
    
 
    ATH_MSG_DEBUG( "checking chain: " << chainName.head() );
    
    if ( selectChain=="" ) { 
      msg(MSG::WARNING) << "^[[91;1m" << "No chain matched\tchain input " << chainName.head() << "  :  " << chainName.tail() << "^[[m"<< endmsg;
      ++toolitr;
      continue;
    }
      
    std::string mchain = selectChain; //chainName.head();
 
    if ( chainName.tail()!="" )     mchain += "/"+chainName.tail();
    if ( chainName.roi()!="" )      mchain += "_"+chainName.roi();
    if ( chainName.vtx()!="" )      mchain += "_"+chainName.vtx();
    if ( chainName.element()!="" )  mchain += "_"+chainName.element();
    if ( chainName.extra()!="" )    mchain += "_"+chainName.extra();
    if ( !chainName.passed() )      mchain += "_DTE";
    
    selectChain = chainName.subs( selectChain );
 
#if 0
    std::cout << "\nTrigR3Mon::chain specification: " << chainName << "\t" << chainName.raw() << std::endl;
    std::cout << "\tchain: " << chainName.head()    << std::endl;
    std::cout << "\tkey:   " << chainName.tail()    << std::endl;
    std::cout << "\troi:   " << chainName.roi()     << std::endl;
    std::cout << "\tvtx:   " << chainName.vtx()     << std::endl;
    std::cout << "\tte:    " << chainName.element() << std::endl;
    std::cout << "\textra: " << chainName.extra()   << std::endl;
#endif
    
    
#if 0
    int shifter_efid      = 0;
    int shifter_ftf       = 0;
    int shifter_efid_run1 = 0;
    
    int shifterChains = m_shifterChains;
    if ( chainName.vtx()=="" ) shifterChains = ( m_shifterChains>1 ? 1 : m_shifterChains );
 
    if ( m_sliceTag.find("Shifter")!=std::string::npos || m_shifter ) { 
      if ( chainName.tail().find("_FTF")!=std::string::npos ) { 
        if (   shifter_ftf>=shifterChains || 
           ( shifter_ftf<shifterChains && chainName.vtx()!="" && chainName.vtx()==lastvtx ) ) {  
          msg(MSG::DEBUG) << "^[[91;1m" << "Matching chain " << selectChain << " excluded - Shifter chain already definied^[[m" << endmsg;
          ++toolitr;
          continue;
        }
        ++shifter_ftf;
        lastvtx = chainName.vtx();
      }
      else if ( chainName.tail().find("_IDTrig")!=std::string::npos || chainName.tail().find("CosmicsN_EFID")!=std::string::npos ) { 
        ++shifter_efid;
        if ( shifter_efid>shifterChains ) {
          msg(MSG::DEBUG) << "^[[91;1m" << "Matching chain " << selectChain << " excluded - Shifter chain already definied^[[m" << endmsg;
          ++toolitr;
          continue;
        }
      }
      else if ( chainName.tail().find("_EFID")!=std::string::npos ) { 
        shifter_efid_run1++;
        if ( shifter_efid_run1>shifterChains ) {
          msg(MSG::DEBUG) << "^[[91;1m" << "Matching chain " << selectChain << " excluded - Shifter chain already definied^[[m" << endmsg;
          ++toolitr;
          continue;
        }
      }
    }
#endif  
 
    if (  std::find(chains.begin(), chains.end(), selectChain) == chains.end() ) { // deduplicate
      chains.push_back( selectChain );
      monTools.push_back( &(*toolitr) );
    }
    
      }
     
      ++toolitr;
    }
    
    m_chainNames = chains;
      
    // tag and probe object creation                                                                                                                                                                            
    std::vector<std::string>& allchains = m_ntupleChainNames;
 
    std::vector<ChainString> allcs;
 
    allcs.reserve(allchains.size());
 

    std::vector<std::string> mtypes;
 
    for ( size_t i=0 ; i<allchains.size() ; i++ ) {
      allcs.emplace_back( ChainString( allchains[i] ) );
      if ( allcs.back().head().find("HLT_")==0 ) continue;
      if ( allcs.back().head().find("EF_")==0  ) continue;
      if ( allcs.back().head()=="Offline" ) { 
    mcTruth   = false;
    if ( allcs.back().tail()!="" ) {
      if ( allcs.back().tail().find("+")==0 ) { 
        mtypes.push_back( allcs.back().tail().substr(1) );
        ATH_MSG_INFO( "Adding Offline reference collection: " << mtypes.back() );
      }
      else { 
        if ( mtypes.size()<1 ) ATH_MSG_WARNING( "Too Many reference collections : " << allcs.back().tail() );
        else { 
          mtypes.push_back( allcs.back().tail() );
          ATH_MSG_INFO( "Offline reference collection: " << mtypes.back() );
        }
      }
    }
    else { 
      mtypes.push_back( "" );
    }
      }
    }
 
    for ( size_t i=0 ; i<m_chainNames.size() ; i++ ) {
 
      ChainString probe = m_chainNames[i] ;
 
      TagNProbe* tnp = 0;
 
      if ( probe.extra().find("probe")!=std::string::npos ) {
 
       std::string probe_key = std::string(probe.extra()).erase( probe.extra().find("probe"), 5 ) ;
 
    for ( size_t j=0 ; j<allchains.size() ; j++ ) {
 
      ChainString tag = allchains[j];
 
      // tag chain must be the same as probe chain but with te=0 and extra=*_tag                                                                                                                            
      if ( tag.head() != probe.head() ) continue;
      if ( tag.element() == probe.element() ) continue;
      if ( tag.extra().find("tag")==std::string::npos ) continue;
 
      std::string tag_key = std::string(tag.extra()).erase( tag.extra().find("tag"), 3 ) ;
 
      if ( tag_key != probe_key ) continue;
      
      double massMin = 40;
      double massMax = 150;
 
      if ( tag.head().find("Jpsi")!=std::string::npos ) { 
        massMin = 0;
        massMax = 20;
      }
        
      if ( mcTruth ) tnp = new TagNProbe( "Truth",   massMin, massMax );
      else           tnp = new TagNProbe( "Offline", massMin, massMax );
 
      tnp->tag(tag.raw()) ;
      tnp->probe(probe.raw()) ;
      
      ATH_MSG_DEBUG( "Tag and probe pair found: " << tag << " : " << probe );
      
      break ;
       
    }
      }
 

 
      { 
 
        ATH_MSG_INFO( "configure analysis: " << m_chainNames[i] );
 
    AnalysisConfigR3_Tier0* analysis = new AnalysisConfigR3_Tier0( m_sliceTag, // m_chainNames[i],
                                       m_chainNames[i], "", "",
                                       m_chainNames[i], "", "",
                                       filterTest, filterRef,
                                       dR_matcher,
                                       new AnalysisR3_Tier0( m_chainNames[i], m_pTCut, m_etaCut, m_d0Cut, m_z0Cut ),
                                       tnp );
 
    analysis->setRunPurity(m_runPurity);
    // analysis->setShifter(m_shifter);
    analysis->containTracks(m_containTracks);
 
    analysis->set_monTool( monTools[i] );
    if ( mtypes.size()>0 ) analysis->setTypes( mtypes );
 
    analysis->initialise();
 
    if ( mcTruth ) {
        analysis->setPdgID( m_selectTruthPdgId );
        analysis->setParentPdgID( m_selectParentTruthPdgId );       
 
        analysis->setMCTruthRef( true );
        analysis->setOfflineRef( false );
    }
 
    m_sequences.push_back( analysis );
 
        std::string highestPT_str = "";
    std::string vtxindex_str  = "";
    
    if ( m_useHighestPT ) { 
      highestPT_str = ": using highest PT only";
      m_sequences.back()->setUseHighestPT(true);
    }
    else m_sequences.back()->setUseHighestPT(false); 
    
    if ( !(m_vtxIndex<0) )  {
      vtxindex_str = ": searching for vertex index ";
      m_sequences.back()->setVtxIndex(m_vtxIndex);
    }   
    
    m_sequences.back()->releaseData(m_releaseMetaData);
    
    if ( m_chainNames[i].find("cosmic")!=std::string::npos || 
         m_chainNames[i].find("Cosmic")!=std::string::npos )  m_sequences.back()->setFilterOnRoi(false);
    
    ATH_MSG_DEBUG( " ----- creating analysis " << m_sequences.back()->name() << " : " << m_chainNames[i] << " " << highestPT_str << " " << vtxindex_str << " -----" );
    
      }
    }
  }
  
  if ( !m_fileopen && ( m_initialisePerRun || m_firstRun ) ) {
    m_fileopen = true;
    
    for ( unsigned i=0 ; i<m_sequences.size() ; i++ ) {
      ATH_MSG_VERBOSE( " ----- booking for analysis " << m_sequences[i]->name() << " -----" );
      m_sequences[i]->initialize( this, &m_tdt );
      m_sequences[i]->setGenericFlag(m_genericFlag);
      m_sequences[i]->book();
    }
    m_firstRun = false;
  }
  
  ATH_MSG_DEBUG( " configured " << m_sequences.size() << " sequences" );
  
  ATH_MSG_DEBUG(  " ----- exit book() ----- " );
 
  return StatusCode::SUCCESS;
  
}
 
 
 
 
 
StatusCode TrigR3Mon::fillHistograms(const EventContext &/*context*/) const {
 
  ATH_MSG_DEBUG( " ----- enter fill() ----- " );
 
  const Trig::ChainGroup* chainGroup = m_tdt->getChainGroup( "(HLT|EF)_e.*" );
  const std::vector<std::string> selectChains = chainGroup->getListOfTriggers();

  static std::once_flag flag;
  std::call_once(flag, [&]() {
    for ( unsigned i=0 ; i<selectChains.size() ; i++ ) {
      ATH_MSG_DEBUG( "\tchain " << selectChains[i] << " from TDT" );
    }
    
    for ( size_t i=selectChains.size() ; i-- ; ) {
      if ( i>5 ) i=5;
      ATH_MSG_INFO( "^[[91;1m" << "configured chain " << selectChains[i] << "^[[m" );
    }
  });
  
  if (msgLvl(MSG::DEBUG)) {
    const std::vector<bool> isPassed = chainGroup->isPassedForEach();
    for ( unsigned i=0 ; i<selectChains.size() ; i++ ) {
      if ( isPassed[i] ) {
        ATH_MSG_DEBUG( "chain " << selectChains[i] << "\tpass: " << isPassed[i] << "\tprescale: " << m_tdt->getPrescale(selectChains[i]) );
      }
    }
  }
  
  for ( unsigned i=0 ; i<m_sequences.size() ; i++ ) { 
    m_sequences[i]->execute();
  }
 
  ATH_MSG_DEBUG(" ----- exit fill() ----- ");
 
  return StatusCode::SUCCESS;
}
 
 
 
 
 
// bool newEventsBlock, bool newLumiBlock, bool newRun are protected varibales
// correctly set before this is called
StatusCode TrigR3Mon::finalize() {
 
  ATH_MSG_DEBUG( " ====== enter proc() ====== " );
  if ( m_initialisePerRun ) {
    for ( unsigned i=0 ; i<m_sequences.size() ; i++ ) m_sequences[i]->finalize();
    m_fileopen = false;
  }
  ATH_MSG_DEBUG( " ====== exit proc() ====== " );
 
  return StatusCode::SUCCESS;
}