TrigFTF_GNN_EtaBin Class Reference

#include <GNN_DataStorage.h>

Collaboration diagram for TrigFTF_GNN_EtaBin:

Classes
struct	CompareNodesByPhi

Public Member Functions
	TrigFTF_GNN_EtaBin ()
	~TrigFTF_GNN_EtaBin ()
void	sortByPhi ()
void	initializeNodes ()
bool	empty () const
void	generatePhiIndexing (float)
float	getMinBinRadius () const
float	getMaxBinRadius () const

Public Attributes
std::vector< const TrigFTF_GNN_Node * >	m_vn
std::vector< std::pair< float, unsigned int > >	m_vPhiNodes
std::vector< std::vector< unsigned int > >	m_in
std::vector< std::array< float, 5 > >	m_params
float	m_minRadius
float	m_maxRadius
unsigned int	m_layerKey {0}

Detailed Description

Definition at line 41 of file GNN_DataStorage.h.

Constructor & Destructor Documentation

TrigFTF_GNN_EtaBin()

TrigFTF_GNN_EtaBin::TrigFTF_GNN_EtaBin

(

)

Definition at line 15 of file GNN_DataStorage.cxx.

                                      : m_minRadius(0), m_maxRadius(0) {
 
  m_in.clear();
  m_vn.clear();
  m_params.clear();
  m_vn.reserve(1000);
}

~TrigFTF_GNN_EtaBin()

TrigFTF_GNN_EtaBin::~TrigFTF_GNN_EtaBin

(

)

Definition at line 23 of file GNN_DataStorage.cxx.

                                        {
  m_in.clear();
  m_vn.clear();
  m_params.clear();
}

Member Function Documentation

empty()

bool TrigFTF_GNN_EtaBin::empty ( ) const

inline

Definition at line 57 of file GNN_DataStorage.h.

                     {
    return m_vn.empty();
  }

generatePhiIndexing()

void TrigFTF_GNN_EtaBin::generatePhiIndexing

(

float

dphi

)

Definition at line 72 of file GNN_DataStorage.cxx.

                                                       {
 
  for(unsigned int nIdx=0;nIdx<m_vn.size();nIdx++) {
 
    float phi = m_params[nIdx][2];
    if(phi <= M_PI-dphi) continue;    
    m_vPhiNodes.push_back(std::pair<float, unsigned int>(phi - 2*M_PI, nIdx));
    
  }
 
  for(unsigned int nIdx=0;nIdx<m_vn.size();nIdx++) {
    float phi = m_params[nIdx][2];
    m_vPhiNodes.push_back(std::pair<float, unsigned int>(phi, nIdx));
  }
 
  for(unsigned int nIdx=0;nIdx<m_vn.size();nIdx++) {
 
    float phi = m_params[nIdx][2];
    if(phi >= -M_PI + dphi) break;
    m_vPhiNodes.push_back(std::pair<float, unsigned int>(phi + 2*M_PI, nIdx));
  }
  
}

getMaxBinRadius()

float TrigFTF_GNN_EtaBin::getMaxBinRadius ( ) const

inline

Definition at line 67 of file GNN_DataStorage.h.

                                {
    return m_maxRadius;
  }

getMinBinRadius()

float TrigFTF_GNN_EtaBin::getMinBinRadius ( ) const

inline

Definition at line 63 of file GNN_DataStorage.h.

                                {
    return m_minRadius;
  }

initializeNodes()

void TrigFTF_GNN_EtaBin::initializeNodes

(

)

Definition at line 54 of file GNN_DataStorage.cxx.

                                         {
  
  if(m_vn.empty()) return;
  
  m_params.resize(m_vn.size());
  
  m_in.resize(m_vn.size());
  for(auto& v : m_in) v.reserve(50);//reasonably high number of incoming edges per node
  
  std::transform(m_vn.begin(), m_vn.end(), m_params.begin(),
                   [](const TrigFTF_GNN_Node* pN) { std::array<float,5> a = {-100.0, 100.0, pN->phi(), pN->r(), pN->z()}; return a;});
    
  auto [min_iter, max_iter] = std::minmax_element(m_vn.begin(), m_vn.end(),
                          [](const TrigFTF_GNN_Node* s, const TrigFTF_GNN_Node* s1) { return (s->r() < s1->r()); });
  m_maxRadius = (*max_iter)->r();
  m_minRadius = (*min_iter)->r();
}

sortByPhi()

void TrigFTF_GNN_EtaBin::sortByPhi

(

)

Definition at line 29 of file GNN_DataStorage.cxx.

                                   {
  
  std::vector<std::pair<float, const TrigFTF_GNN_Node*> > phiBuckets[32];
 
  int nBuckets = 31;
 
  for(const auto& n : m_vn) {
    int bIdx = (int)(0.5*nBuckets*(n->phi()/(float)M_PI + 1.0f));
    phiBuckets[bIdx].push_back(std::make_pair(n->phi(), n));
  }
 
  for(auto& b : phiBuckets) {
    std::sort(b.begin(), b.end());
  }
 
  int idx = 0;
  for(const auto& b : phiBuckets) {
    for(const auto& p : b) {
      m_vn[idx++] = p.second;
    }
  }
 
}

Member Data Documentation

m_in

std::vector<std::vector<unsigned int> > TrigFTF_GNN_EtaBin::m_in

Definition at line 73 of file GNN_DataStorage.h.

m_layerKey

unsigned int TrigFTF_GNN_EtaBin::m_layerKey {0}

Definition at line 77 of file GNN_DataStorage.h.

{0};

m_maxRadius

float TrigFTF_GNN_EtaBin::m_maxRadius

Definition at line 76 of file GNN_DataStorage.h.

m_minRadius

float TrigFTF_GNN_EtaBin::m_minRadius

Definition at line 76 of file GNN_DataStorage.h.

m_params

std::vector<std::array<float,5> > TrigFTF_GNN_EtaBin::m_params

Definition at line 74 of file GNN_DataStorage.h.

m_vn

std::vector<const TrigFTF_GNN_Node*> TrigFTF_GNN_EtaBin::m_vn

Definition at line 71 of file GNN_DataStorage.h.

m_vPhiNodes

std::vector<std::pair<float, unsigned int> > TrigFTF_GNN_EtaBin::m_vPhiNodes

Definition at line 72 of file GNN_DataStorage.h.

---

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

• GNN_DataStorage.h
• GNN_DataStorage.cxx