TrigFTF_GNN_DataStorage Class Reference

#include <GNN_DataStorage.h>

Collaboration diagram for TrigFTF_GNN_DataStorage:

Public Member Functions
	TrigFTF_GNN_DataStorage (const TrigFTF_GNN_Geometry &)
	~TrigFTF_GNN_DataStorage ()
int	addSpacePoint (const TrigSiSpacePointBase *, bool)
unsigned int	numberOfNodes () const
void	sortByPhi ()
void	initializeNodes (bool)
void	generatePhiIndexing (float)
TrigFTF_GNN_EtaBin &	getEtaBin (int idx)

Protected Attributes
const TrigFTF_GNN_Geometry &	m_geo
std::vector< TrigFTF_GNN_EtaBin >	m_etaBins

Detailed Description

Definition at line 49 of file GNN_DataStorage.h.

Constructor & Destructor Documentation

TrigFTF_GNN_DataStorage()

TrigFTF_GNN_DataStorage::TrigFTF_GNN_DataStorage

(

const TrigFTF_GNN_Geometry &

g

)

Definition at line 69 of file GNN_DataStorage.cxx.

                                                                              : m_geo(g) {
  m_etaBins.resize(g.num_bins());
}

~TrigFTF_GNN_DataStorage()

TrigFTF_GNN_DataStorage::~TrigFTF_GNN_DataStorage

(

)

Definition at line 73 of file GNN_DataStorage.cxx.

{}

Member Function Documentation

addSpacePoint()

int TrigFTF_GNN_DataStorage::addSpacePoint	(	const TrigSiSpacePointBase *	sp,
		bool	useML = false
	)

Definition at line 75 of file GNN_DataStorage.cxx.

                                                                                             {
 
  const TrigFTF_GNN_Layer* pL = m_geo.getTrigFTF_GNN_LayerByIndex(sp->layer());
 
  if(pL==nullptr) return -1;
 
  int binIndex = pL->getEtaBin(sp->z(), sp->r());
 
  if(binIndex == -1) {
    return -2;
  }
 
  bool isStrip = sp->offlineSpacePoint()->clusterList().second != nullptr;
 
  if(isStrip) {
    m_etaBins.at(binIndex).m_vn.push_back(sp);
    return 0;
  }
 
  bool isBarrel = (pL->m_layer.m_type == 0);
 
  if(isBarrel) {
    m_etaBins.at(binIndex).m_vn.push_back(sp);
  }
  else {
    if (useML) {
      const Trk::SpacePoint* osp = sp->offlineSpacePoint();
      const InDet::PixelCluster* pCL = dynamic_cast<const InDet::PixelCluster*>(osp->clusterList().first);
      float cluster_width = pCL->width().widthPhiRZ().y();
      if(cluster_width > 0.2) return -3;
    }
    m_etaBins.at(binIndex).m_vn.push_back(sp);
    
  }
 
  return 0;
}

generatePhiIndexing()

void TrigFTF_GNN_DataStorage::generatePhiIndexing

(

float

dphi

)

Definition at line 174 of file GNN_DataStorage.cxx.

                                                            {
  for(auto& b : m_etaBins) b.generatePhiIndexing(dphi);
}

getEtaBin()

TrigFTF_GNN_EtaBin& TrigFTF_GNN_DataStorage::getEtaBin ( int  idx )

inline

Definition at line 61 of file GNN_DataStorage.h.

                                         {
    if(idx >= static_cast<int>(m_etaBins.size())) idx = idx-1;
    return m_etaBins.at(idx);
  }

initializeNodes()

void TrigFTF_GNN_DataStorage::initializeNodes

(

bool

useML = false

)

Definition at line 127 of file GNN_DataStorage.cxx.

                                                                {
  for(auto& b : m_etaBins) {
    b.initializeNodes();
  }
  
  if(!useML) return;
 
  unsigned int nL = m_geo.num_layers();
 
  for(unsigned int layerIdx=0;layerIdx<nL;layerIdx++) {
 
    const TrigFTF_GNN_Layer* pL = m_geo.getTrigFTF_GNN_LayerByIndex(layerIdx);
 
    if(pL->m_layer.m_subdet < 20000) {//skip strips volumes: layers in range [1200X-1400X]
      continue;
    }
    
    bool isBarrel = (pL->m_layer.m_type == 0);
 
    if(!isBarrel) continue;
 
    int nBins = pL->m_bins.size();
 
    for(int b=0;b<nBins;b++) {//loop over eta-bins in Layer
 
      TrigFTF_GNN_EtaBin& B = m_etaBins.at(pL->m_bins.at(b));
 
      if(B.empty()) continue;
      
      for(unsigned int nIdx=0;nIdx<B.m_vn.size();nIdx++) {
        
        const Trk::SpacePoint* osp = B.m_vn[nIdx]->offlineSpacePoint();
        const InDet::PixelCluster* pCL = dynamic_cast<const InDet::PixelCluster*>(osp->clusterList().first);
 
        float cluster_width = pCL->width().widthPhiRZ().y();
    //adjusting cuts using fitted boundaries of |cot(theta)| vs. cluster z-width distribution 
        float min_tau = 6.7*(cluster_width - 0.2);//linear fit
        float max_tau = 1.6 + 0.15/(cluster_width + 0.2) + 6.1*(cluster_width - 0.2);//linear fit + correction for short clusters
 
        B.m_params[nIdx][0] = min_tau;
        B.m_params[nIdx][1] = max_tau;
        
      }
    }
  }
}

numberOfNodes()

unsigned int TrigFTF_GNN_DataStorage::numberOfNodes

(

)

const

Definition at line 113 of file GNN_DataStorage.cxx.

                                                          {
 
  unsigned int n=0;
  
  for(auto& b : m_etaBins) {
    n += b.m_vn.size();
  }
  return n;
}

sortByPhi()

void TrigFTF_GNN_DataStorage::sortByPhi

(

)

Definition at line 123 of file GNN_DataStorage.cxx.

                                        {
  for(auto& b : m_etaBins) b.sortByPhi();
}

Member Data Documentation

m_etaBins

std::vector<TrigFTF_GNN_EtaBin> TrigFTF_GNN_DataStorage::m_etaBins

protected

Definition at line 70 of file GNN_DataStorage.h.

m_geo

const TrigFTF_GNN_Geometry& TrigFTF_GNN_DataStorage::m_geo

protected

Definition at line 68 of file GNN_DataStorage.h.

---

The documentation for this class was generated from the following files:

• GNN_DataStorage.h
• GNN_DataStorage.cxx