TrigFTF_GNNR3_EtaBin Class Reference

#include <GNNR3_DataStorage.h>

Collaboration diagram for TrigFTF_GNNR3_EtaBin:

Classes
struct	CompareNodesByPhi

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

Public Attributes
std::vector< const TrigFTF_GNNR3_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

Detailed Description

Definition at line 40 of file GNNR3_DataStorage.h.

Constructor & Destructor Documentation

TrigFTF_GNNR3_EtaBin()

TrigFTF_GNNR3_EtaBin::TrigFTF_GNNR3_EtaBin

(

)

Definition at line 14 of file GNNR3_DataStorage.cxx.

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

~TrigFTF_GNNR3_EtaBin()

TrigFTF_GNNR3_EtaBin::~TrigFTF_GNNR3_EtaBin

(

)

Definition at line 22 of file GNNR3_DataStorage.cxx.

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

Member Function Documentation

empty()

bool TrigFTF_GNNR3_EtaBin::empty ( ) const

inline

Definition at line 56 of file GNNR3_DataStorage.h.

                     {
    return m_vn.empty();
  }

generatePhiIndexing()

void TrigFTF_GNNR3_EtaBin::generatePhiIndexing

(

float

dphi

)

Definition at line 71 of file GNNR3_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_GNNR3_EtaBin::getMaxBinRadius ( ) const

inline

Definition at line 66 of file GNNR3_DataStorage.h.

                                {
    return m_maxRadius;
  }

getMinBinRadius()

float TrigFTF_GNNR3_EtaBin::getMinBinRadius ( ) const

inline

Definition at line 62 of file GNNR3_DataStorage.h.

                                {
    return m_minRadius;
  }

initializeNodes()

void TrigFTF_GNNR3_EtaBin::initializeNodes

(

)

Definition at line 53 of file GNNR3_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_GNNR3_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_GNNR3_Node* s, const TrigFTF_GNNR3_Node* s1) { return (s->r() < s1->r()); });
  m_maxRadius = (*max_iter)->r();
  m_minRadius = (*min_iter)->r();
}

sortByPhi()

void TrigFTF_GNNR3_EtaBin::sortByPhi

(

)

Definition at line 28 of file GNNR3_DataStorage.cxx.

                                     {
  
  std::vector<std::pair<float, const TrigFTF_GNNR3_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_GNNR3_EtaBin::m_in

Definition at line 72 of file GNNR3_DataStorage.h.

m_maxRadius

float TrigFTF_GNNR3_EtaBin::m_maxRadius

Definition at line 75 of file GNNR3_DataStorage.h.

m_minRadius

float TrigFTF_GNNR3_EtaBin::m_minRadius

Definition at line 75 of file GNNR3_DataStorage.h.

m_params

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

Definition at line 73 of file GNNR3_DataStorage.h.

m_vn

std::vector<const TrigFTF_GNNR3_Node*> TrigFTF_GNNR3_EtaBin::m_vn

Definition at line 70 of file GNNR3_DataStorage.h.

m_vPhiNodes

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

Definition at line 71 of file GNNR3_DataStorage.h.

---

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

• GNNR3_DataStorage.h
• GNNR3_DataStorage.cxx