#include <GNNR3_DataStorage.h>

Collaboration diagram for TrigFTF_GNNR3_DataStorage:

Public Member Functions
	TrigFTF_GNNR3_DataStorage (const TrigFTF_GNNR3_Geometry &)
	~TrigFTF_GNNR3_DataStorage ()
int	loadPixelGraphNodes (short, const std::vector< TrigFTF_GNNR3_Node > &, bool)
int	loadStripGraphNodes (short, const std::vector< TrigFTF_GNNR3_Node > &)
unsigned int	numberOfNodes () const
void	sortByPhi ()
void	initializeNodes (bool)
void	generatePhiIndexing (float)
TrigFTF_GNNR3_EtaBin &	getEtaBin (int idx)

Protected Attributes
const TrigFTF_GNNR3_Geometry &	m_geo
std::vector< TrigFTF_GNNR3_EtaBin >	m_etaBins

Detailed Description

Definition at line 79 of file GNNR3_DataStorage.h.

Constructor & Destructor Documentation

◆ TrigFTF_GNNR3_DataStorage()

TrigFTF_GNNR3_DataStorage::TrigFTF_GNNR3_DataStorage ( const TrigFTF_GNNR3_Geometry & g )

Definition at line 95 of file GNNR3_DataStorage.cxx.

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

◆ ~TrigFTF_GNNR3_DataStorage()

TrigFTF_GNNR3_DataStorage::~TrigFTF_GNNR3_DataStorage ( )

Definition at line 100 of file GNNR3_DataStorage.cxx.

                                                      {
 
}

Member Function Documentation

◆ generatePhiIndexing()

void TrigFTF_GNNR3_DataStorage::generatePhiIndexing ( float dphi )

Definition at line 226 of file GNNR3_DataStorage.cxx.

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

◆ getEtaBin()

TrigFTF_GNNR3_EtaBin & TrigFTF_GNNR3_DataStorage::getEtaBin ( int idx )

inline

Definition at line 93 of file GNNR3_DataStorage.h.

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

◆ initializeNodes()

void TrigFTF_GNNR3_DataStorage::initializeNodes ( bool useML )

Definition at line 182 of file GNNR3_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_GNNR3_Layer* pL = m_geo.getTrigFTF_GNNR3_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_GNNR3_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++) {
        float cluster_width = B.m_vn[nIdx]->pixelClusterWidth();
    //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;
        
      }
    }
  }
}

◆ loadPixelGraphNodes()

int TrigFTF_GNNR3_DataStorage::loadPixelGraphNodes	(	short	layerIndex,
		const std::vector< TrigFTF_GNNR3_Node > &	coll,
		bool	useML )

Definition at line 104 of file GNNR3_DataStorage.cxx.

                                                                                                                          {
 
  int nLoaded = 0;
 
  const TrigFTF_GNNR3_Layer* pL = m_geo.getTrigFTF_GNNR3_LayerByIndex(layerIndex);
 
  if(pL == nullptr) {
    return -1;
  }
  
  bool isBarrel = (pL->m_layer.m_type == 0);
 
  for(const auto& node : coll) {
 
    int binIndex = pL->getEtaBin(node.z(), node.r());
 
    if(binIndex == -1) {
      continue;
    }
    
    if(isBarrel) {
      m_etaBins.at(binIndex).m_vn.push_back(&node);
    }
    else {
      if (useML) {
    float cluster_width = node.pixelClusterWidth();
    if(cluster_width > 0.2) continue;
      }
      m_etaBins.at(binIndex).m_vn.push_back(&node);
    }
    
    nLoaded++;
    
  }
  
  return nLoaded;
}

◆ loadStripGraphNodes()

int TrigFTF_GNNR3_DataStorage::loadStripGraphNodes	(	short	layerIndex,
		const std::vector< TrigFTF_GNNR3_Node > &	coll )

Definition at line 142 of file GNNR3_DataStorage.cxx.

                                                                                                              {
 
  int nLoaded = 0;
 
  const TrigFTF_GNNR3_Layer* pL = m_geo.getTrigFTF_GNNR3_LayerByIndex(layerIndex);
 
  if(pL == nullptr) {
    return -1;
  }
   
  for(const auto& node : coll) {
 
    int binIndex = pL->getEtaBin(node.z(), node.r());
 
    if(binIndex == -1) {
      continue;
    }
 
    m_etaBins.at(binIndex).m_vn.push_back(&node);
    nLoaded++;
  }
  
  return nLoaded;
}

◆ numberOfNodes()

unsigned int TrigFTF_GNNR3_DataStorage::numberOfNodes ( ) const

Definition at line 167 of file GNNR3_DataStorage.cxx.

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

◆ sortByPhi()

void TrigFTF_GNNR3_DataStorage::sortByPhi ( )

Definition at line 177 of file GNNR3_DataStorage.cxx.

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

Member Data Documentation

◆ m_etaBins

std::vector<TrigFTF_GNNR3_EtaBin> TrigFTF_GNNR3_DataStorage::m_etaBins

protected

Definition at line 102 of file GNNR3_DataStorage.h.

◆ m_geo

const TrigFTF_GNNR3_Geometry& TrigFTF_GNNR3_DataStorage::m_geo