#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	getConnectingNodes (std::vector< const TrigFTF_GNN_Node * > &)

void	sortByPhi ()

void	generatePhiIndexing (float)

const TrigFTF_GNN_EtaBin &	getEtaBin (int idx) const

Protected Attributes
const TrigFTF_GNN_Geometry &	m_geo

std::vector< TrigFTF_GNN_EtaBin >	m_etaBins

Detailed Description

Definition at line 83 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 67 of file GNN_DataStorage.cxx.

                                                                               : m_geo(g) {
  
  
   for(int k=0;k<g.num_bins();k++) {
     m_etaBins.push_back(TrigFTF_GNN_EtaBin());
   }
 }

◆ ~TrigFTF_GNN_DataStorage()

TrigFTF_GNN_DataStorage::~TrigFTF_GNN_DataStorage ( )

Definition at line 75 of file GNN_DataStorage.cxx.

                                                   {
  
 }

Member Function Documentation

◆ addSpacePoint()

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

Definition at line 79 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 isBarrel = (pL->m_layer.m_type == 0);
  
   if(isBarrel) {
     float min_tau = -100.0;
     float max_tau =  100.0;
     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();
       min_tau = 6.7*(cluster_width - 0.2);
       max_tau = 1.6 + 0.15/(cluster_width + 0.2) + 6.1*(cluster_width - 0.2);
     }
     m_etaBins.at(binIndex).m_vn.push_back(new TrigFTF_GNN_Node(sp, min_tau, max_tau));
   }
   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(new TrigFTF_GNN_Node(sp));
   }
  
   return 0;
 }

◆ generatePhiIndexing()

void TrigFTF_GNN_DataStorage::generatePhiIndexing ( float dphi )

Definition at line 131 of file GNN_DataStorage.cxx.

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

◆ getConnectingNodes()

void TrigFTF_GNN_DataStorage::getConnectingNodes ( std::vector< const TrigFTF_GNN_Node * > & vn )

Definition at line 136 of file GNN_DataStorage.cxx.

                                                                                        {
   
   vn.clear();
   vn.reserve(numberOfNodes());
   
   for(const auto& b : m_etaBins) {
     for(std::vector<TrigFTF_GNN_Node*>::const_iterator nIt = b.m_vn.begin();nIt!=b.m_vn.end();++nIt) {
       if((*nIt)->m_in.empty()) continue;
       if((*nIt)->m_out.empty()) continue;
       vn.push_back(*nIt);
     }
   }
 }

◆ getEtaBin()

const TrigFTF_GNN_EtaBin& TrigFTF_GNN_DataStorage::getEtaBin ( int idx ) const

inline

Definition at line 95 of file GNN_DataStorage.h.

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

◆ numberOfNodes()

unsigned int TrigFTF_GNN_DataStorage::numberOfNodes ( ) const

Definition at line 117 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 127 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 104 of file GNN_DataStorage.h.

◆ m_geo

const TrigFTF_GNN_Geometry& TrigFTF_GNN_DataStorage::m_geo