MuonStationNtupleHelperTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#include "MuonStationNtupleHelperTool.h"
 
#include "TrkEventUtils/IdentifierExtractor.h"
#include "TrkEventPrimitives/TrackStateDefs.h"
 
#include "TTree.h"
 
//================ Constructor =================================================
 
Muon::MuonStationNtupleHelperTool::MuonStationNtupleHelperTool(const std::string& t,
              const std::string& n,
              const IInterface*  p )
  :
  AthAlgTool(t,n,p),
  m_mdtSectorIx(nullptr),
  m_mdtStationIx(nullptr),
  m_rpcSectorIx(nullptr),
  m_rpcStationIx(nullptr),
  m_rpcMeasuresPhi(nullptr),
  m_tgcStationIx(nullptr),
  m_tgcMeasuresPhi(nullptr)
{
  declareInterface<Trk::IValidationNtupleHelperTool>(this);
}
 
//================ Initialisation =================================================
 
StatusCode Muon::MuonStationNtupleHelperTool::initialize()
{
  ATH_CHECK(m_idHelperSvc.retrieve());
 
  m_mdtSectorIx      = new std::vector<int>();
  m_mdtStationIx     = new std::vector<int>();
  m_rpcSectorIx      = new std::vector<int>();
  m_rpcStationIx     = new std::vector<int>();
  m_rpcMeasuresPhi   = new std::vector<int>();
  m_tgcStationIx     = new std::vector<int>();
  m_tgcMeasuresPhi   = new std::vector<int>();
 
  ATH_MSG_INFO ("initialize() successful in " << name());
  return StatusCode::SUCCESS;
}
 
//================ Finalisation =================================================
 
StatusCode Muon::MuonStationNtupleHelperTool::finalize()
{
  delete m_mdtSectorIx; m_mdtSectorIx=nullptr;
  delete m_mdtStationIx; m_mdtStationIx=nullptr;
  delete m_rpcSectorIx; m_rpcSectorIx=nullptr;
  delete m_rpcStationIx; m_rpcStationIx=nullptr;
  delete m_rpcMeasuresPhi; m_rpcMeasuresPhi=nullptr;
  delete m_tgcStationIx; m_tgcStationIx=nullptr;
  delete m_tgcMeasuresPhi; m_tgcMeasuresPhi=nullptr;
  return StatusCode::SUCCESS;
}
 
StatusCode Muon::MuonStationNtupleHelperTool::addNtupleItems (
    TTree* tree,
    const int& detectorType) {
 
    // add items to the ntuple
    if (detectorType==Trk::TrackState::MDT) {
 
      tree->Branch("MdtSectorIndex",      &m_mdtSectorIx);
      tree->Branch("MdtStationIndex",     &m_mdtStationIx);
    }
    if (detectorType==Trk::TrackState::RPC) {
      tree->Branch("RpcSectorIndex",      &m_rpcSectorIx);
      tree->Branch("RpcStationIndex",     &m_rpcStationIx);
      tree->Branch("RpcMeasuresPhi",      &m_rpcMeasuresPhi);
    }
    if (detectorType==Trk::TrackState::TGC) {
      // TGC sectors are not defined, will cause warning if tried to compute from helper
      tree->Branch("TgcStationIndex",     &m_tgcStationIx);
      tree->Branch("TgcMeasuresPhi",      &m_tgcMeasuresPhi);
    }
 
    ATH_MSG_VERBOSE ("added items to ntuple.");
    return StatusCode::SUCCESS;
}
 
StatusCode Muon::MuonStationNtupleHelperTool::fillMeasurementData (
    const Trk::MeasurementBase* hit,
    const Trk::TrackParameters* ,
    const int& detectorType,
    const bool& /*isOutlier*/) {
 
    // todo is this actually called for outliers?
 
    // identify the detector type:
    Identifier id = Trk::IdentifierExtractor::extract(hit);
    using namespace Muon::MuonStationIndex;
    const int stIdx = toInt(m_idHelperSvc->stationIndex(id));
    if (detectorType==Trk::TrackState::MDT) {
      m_mdtSectorIx->push_back(m_idHelperSvc->sector(id));
      m_mdtStationIx->push_back(stIdx);
    }
    if (detectorType==Trk::TrackState::RPC) {
      m_rpcSectorIx->push_back(m_idHelperSvc->sector(id));
      m_rpcStationIx->push_back(stIdx);
      m_rpcMeasuresPhi->push_back(m_idHelperSvc->measuresPhi(id));
    }
    if (detectorType==Trk::TrackState::TGC) {
      m_tgcStationIx->push_back(stIdx);
      m_tgcMeasuresPhi->push_back(m_idHelperSvc->measuresPhi(id));
    }
 
    return StatusCode::SUCCESS;
}
 
StatusCode Muon::MuonStationNtupleHelperTool::resetVariables (
        const int& detectorType ) {
    if (detectorType==Trk::TrackState::MDT) {
      m_mdtSectorIx->clear();
      m_mdtStationIx->clear();
    }
    if (detectorType==Trk::TrackState::RPC) {
      m_rpcSectorIx->clear();
      m_rpcStationIx->clear();
      m_rpcMeasuresPhi->clear();
    }
    if (detectorType==Trk::TrackState::TGC) {
      m_tgcStationIx->clear();
      m_tgcMeasuresPhi->clear();
    }
    return StatusCode::SUCCESS;
}
 
StatusCode Muon::MuonStationNtupleHelperTool::fillHoleData (
        const Trk::TrackStateOnSurface&,
        const int&) {
    // we do nothing with holes
    return StatusCode::SUCCESS;
}