PerigeeBLPredictionFillerTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#include "PerigeeBLPredictionFillerTool.h"
#include "InDetTestPixelLayer/InDetTestPixelLayerTool.h"
#include "InDetTestPixelLayer/TrackStateOnPixelLayerInfo.h"
#include "AthenaKernel/errorcheck.h"
#include "InDetReadoutGeometry/SiDetectorElement.h"
#include "InDetIdentifier/PixelID.h"
 
namespace D3PD {
 
PerigeeBLPredictionFillerTool::PerigeeBLPredictionFillerTool
    (const std::string& type,
     const std::string& name,
     const IInterface* parent)
  : BlockFillerTool<Trk::Perigee> (type, name, parent)
{
  book().ignore(); // Avoid coverity warnings.
}
 
 
StatusCode PerigeeBLPredictionFillerTool::initialize()
{
  CHECK( BlockFillerTool<Trk::Perigee>::initialize() );
  CHECK( m_inDetTestPixelLayerTool.retrieve() );
  CHECK( detStore()->retrieve(m_pixId, "PixelID") );
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode PerigeeBLPredictionFillerTool::book()
{
 
  CHECK( addVariable ("x",  m_x, "predicted x global position in the B-layer" ) );
  CHECK( addVariable ("y",  m_y, "predicted y global position in the B-layer" ) );
  CHECK( addVariable ("z",  m_z, "predicted z global position in the B-layer" ) );
  CHECK( addVariable ("locX",  m_locX, "predicted x local position in the B-layer" ) );
  CHECK( addVariable ("locY",  m_locY, "predicted y local position in the B-layer" ) );
  CHECK( addVariable ("err_locX",  m_err_locX, "error on local x prediction in the B-layer" ) );
  CHECK( addVariable ("err_locY",  m_err_locY, "error on local y prediction in the B-layer" ) );
  CHECK( addVariable ("etaDistToEdge",  m_etaDistToEdge, "distance of B-layer prediction to module edge along eta") );
  CHECK( addVariable ("phiDistToEdge",  m_phiDistToEdge, "distance of B-layer prediction to module edge along phi") );
  CHECK( addVariable ("detElementId",  m_detElementId, "detector element identifier of B-layer prediction" ) );
  CHECK( addVariable ("row",  m_row, "" ) );
  CHECK( addVariable ("col",  m_col, "" ) );
  CHECK( addVariable ("type",  m_type, "" ) );
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode
PerigeeBLPredictionFillerTool::fill (const Trk::Perigee& p) {
 
  this->clearData();
 
  std::vector<InDet::TrackStateOnPixelLayerInfo>  trackStateBlayer;
  if(!m_inDetTestPixelLayerTool->getTrackStateOnInnermostPixelLayerInfo(&p, trackStateBlayer)){
    return StatusCode::SUCCESS;
  }
  
  std::vector<InDet::TrackStateOnPixelLayerInfo>::const_iterator itr = trackStateBlayer.begin();
  std::vector<InDet::TrackStateOnPixelLayerInfo>::const_iterator end = trackStateBlayer.end();
  for (; itr!=end; ++itr){
    Identifier id = itr->moduleId();
    
    if(id.is_valid()){
      m_detElementId->push_back(id.get_identifier32().get_compact());
    }else{
      m_detElementId->push_back(0);
    }
    
    m_type->push_back((int)itr->type());
    
    m_etaDistToEdge->push_back(itr->distToModuleEdgeEta());
    m_phiDistToEdge->push_back(itr->distToModuleEdgePhi());
    
    Identifier holeId_c = itr->pixelId();
    if(holeId_c.is_valid()){
      m_col->push_back(m_pixId->eta_index(holeId_c));
      m_row->push_back(m_pixId->phi_index(holeId_c));
    }else{
      m_col->push_back(-1);
      m_row->push_back(-1);
    }
    
    m_x->push_back(itr->globalPosition().x());
    m_y->push_back(itr->globalPosition().y());
    m_z->push_back(itr->globalPosition().z());
    
    m_locX->push_back(itr->localX());
    m_locY->push_back(itr->localY());
    
    m_err_locX->push_back(itr->errLocalX());
    m_err_locY->push_back(itr->errLocalY());
  }
 
  return StatusCode::SUCCESS;
}
 
void PerigeeBLPredictionFillerTool::clearData(){
 
  m_x->clear();
  m_y->clear();
  m_z->clear();
  m_locX->clear();
  m_locY->clear();
  m_err_locX->clear();
  m_err_locY->clear();
  m_etaDistToEdge->clear();
  m_phiDistToEdge->clear();
  m_detElementId->clear();
  m_row->clear();
  m_col->clear();
  m_type->clear();
}
 
} // namespace D3PD