ATLAS Offline Software
Classes | Public Member Functions | Public Attributes | List of all members
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
 

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.

15  : m_minRadius(0), m_maxRadius(0) {
16 
17  m_in.clear();
18  m_vn.clear();
19  m_params.clear();
20  m_vn.reserve(1000);
21 }

◆ ~TrigFTF_GNN_EtaBin()

TrigFTF_GNN_EtaBin::~TrigFTF_GNN_EtaBin ( )

Definition at line 23 of file GNN_DataStorage.cxx.

23  {
24  m_in.clear();
25  m_vn.clear();
26  m_params.clear();
27 }

Member Function Documentation

◆ empty()

bool TrigFTF_GNN_EtaBin::empty ( ) const
inline

Definition at line 57 of file GNN_DataStorage.h.

57  {
58  return m_vn.empty();
59  }

◆ generatePhiIndexing()

void TrigFTF_GNN_EtaBin::generatePhiIndexing ( float  dphi)

Definition at line 72 of file GNN_DataStorage.cxx.

72  {
73 
74  for(unsigned int nIdx=0;nIdx<m_vn.size();nIdx++) {
75 
76  float phi = m_params[nIdx][2];
77  if(phi <= M_PI-dphi) continue;
78  m_vPhiNodes.push_back(std::pair<float, unsigned int>(phi - 2*M_PI, nIdx));
79 
80  }
81 
82  for(unsigned int nIdx=0;nIdx<m_vn.size();nIdx++) {
83  float phi = m_params[nIdx][2];
84  m_vPhiNodes.push_back(std::pair<float, unsigned int>(phi, nIdx));
85  }
86 
87  for(unsigned int nIdx=0;nIdx<m_vn.size();nIdx++) {
88 
89  float phi = m_params[nIdx][2];
90  if(phi >= -M_PI + dphi) break;
91  m_vPhiNodes.push_back(std::pair<float, unsigned int>(phi + 2*M_PI, nIdx));
92  }
93 
94 }

◆ getMaxBinRadius()

float TrigFTF_GNN_EtaBin::getMaxBinRadius ( ) const
inline

Definition at line 67 of file GNN_DataStorage.h.

67  {
68  return m_maxRadius;
69  }

◆ getMinBinRadius()

float TrigFTF_GNN_EtaBin::getMinBinRadius ( ) const
inline

Definition at line 63 of file GNN_DataStorage.h.

63  {
64  return m_minRadius;
65  }

◆ initializeNodes()

void TrigFTF_GNN_EtaBin::initializeNodes ( )

Definition at line 54 of file GNN_DataStorage.cxx.

54  {
55 
56  if(m_vn.empty()) return;
57 
58  m_params.resize(m_vn.size());
59 
60  m_in.resize(m_vn.size());
61  for(auto& v : m_in) v.reserve(50);//reasonably high number of incoming edges per node
62 
63  std::transform(m_vn.begin(), m_vn.end(), m_params.begin(),
64  [](const TrigFTF_GNN_Node* pN) { std::array<float,5> a = {-100.0, 100.0, pN->phi(), pN->r(), pN->z()}; return a;});
65 
66  auto [min_iter, max_iter] = std::minmax_element(m_vn.begin(), m_vn.end(),
67  [](const TrigFTF_GNN_Node* s, const TrigFTF_GNN_Node* s1) { return (s->r() < s1->r()); });
68  m_maxRadius = (*max_iter)->r();
69  m_minRadius = (*min_iter)->r();
70 }

◆ sortByPhi()

void TrigFTF_GNN_EtaBin::sortByPhi ( )

Definition at line 29 of file GNN_DataStorage.cxx.

29  {
30 
31  std::vector<std::pair<float, const TrigFTF_GNN_Node*> > phiBuckets[32];
32 
33  int nBuckets = 31;
34 
35  for(const auto& n : m_vn) {
36  int bIdx = (int)(0.5*nBuckets*(n->phi()/(float)M_PI + 1.0f));
37  phiBuckets[bIdx].push_back(std::make_pair(n->phi(), n));
38  }
39 
40  for(auto& b : phiBuckets) {
41  std::sort(b.begin(), b.end());
42  }
43 
44  int idx = 0;
45  for(const auto& b : phiBuckets) {
46  for(const auto& p : b) {
47  m_vn[idx++] = p.second;
48  }
49  }
50 
51 }

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_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:
ReadCellNoiseFromCoolCompare.s1
s1
Definition: ReadCellNoiseFromCoolCompare.py:378
TrigFTF_GNN_EtaBin::m_in
std::vector< std::vector< unsigned int > > m_in
Definition: GNN_DataStorage.h:73
phi
Scalar phi() const
phi method
Definition: AmgMatrixBasePlugin.h:67
M_PI
#define M_PI
Definition: ActiveFraction.h:11
TrigFTF_GNN_EtaBin::m_minRadius
float m_minRadius
Definition: GNN_DataStorage.h:76
TrigFTF_GNN_Node
Definition: GNN_DataStorage.h:18
python.utils.AtlRunQueryDQUtils.p
p
Definition: AtlRunQueryDQUtils.py:209
TrigFTF_GNN_EtaBin::m_vPhiNodes
std::vector< std::pair< float, unsigned int > > m_vPhiNodes
Definition: GNN_DataStorage.h:72
beamspotman.n
n
Definition: beamspotman.py:727
Amg::transform
Amg::Vector3D transform(Amg::Vector3D &v, Amg::Transform3D &tr)
Transform a point from a Trasformation3D.
Definition: GeoPrimitivesHelpers.h:156
TrigFTF_GNN_EtaBin::m_maxRadius
float m_maxRadius
Definition: GNN_DataStorage.h:76
plotBeamSpotMon.b
b
Definition: plotBeamSpotMon.py:76
python.PyAthena.v
v
Definition: PyAthena.py:154
a
TList * a
Definition: liststreamerinfos.cxx:10
python.CaloAddPedShiftConfig.int
int
Definition: CaloAddPedShiftConfig.py:45
std::sort
void sort(typename std::reverse_iterator< DataModel_detail::iterator< DVL > > beg, typename std::reverse_iterator< DataModel_detail::iterator< DVL > > end, const Compare &comp)
Specialization of sort for DataVector/List.
Definition: DVL_algorithms.h:623
LArNewCalib_DelayDump_OFC_Cali.idx
idx
Definition: LArNewCalib_DelayDump_OFC_Cali.py:69
python.SystemOfUnits.s
float s
Definition: SystemOfUnits.py:147
TrigFTF_GNN_EtaBin::m_vn
std::vector< const TrigFTF_GNN_Node * > m_vn
Definition: GNN_DataStorage.h:71
TrigFTF_GNN_EtaBin::m_params
std::vector< std::array< float, 5 > > m_params
Definition: GNN_DataStorage.h:74