GNN_Geometry.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRIGINDETPATTRECOTOOLS_GNN_GEOMETRY_H
#define TRIGINDETPATTRECOTOOLS_GNN_GEOMETRY_H
 
#include<vector>
#include<map>
#include<algorithm>
#include <memory>
#include "GNN_FasTrackConnector.h"
#include "TrigInDetPattRecoEvent/TrigInDetSiLayer.h"
 
 
class TrigFTF_GNN_Layer {
public:
  TrigFTF_GNN_Layer(const TrigInDetSiLayer&, float, int);
  ~TrigFTF_GNN_Layer();
 
  int getEtaBin(float, float) const;
 
  float getMinBinRadius(int) const;
  float getMaxBinRadius(int) const;
 
  int num_bins() const {return m_bins.size();} 
 
  bool verifyBin(const TrigFTF_GNN_Layer*, int, int, float, float) const;
 
  const TrigInDetSiLayer& m_layer;
  std::vector<int> m_bins;//eta-bin indices
  std::vector<float> m_minRadius;
  std::vector<float> m_maxRadius;
  std::vector<float> m_minBinCoord;
  std::vector<float> m_maxBinCoord;
 
  float m_minEta, m_maxEta, m_etaBin;
 
protected:
 
  float m_etaBinWidth;
 
  float m_r1, m_z1, m_r2, m_z2;
  int m_nBins; 
 
};
 
class TrigFTF_GNN_Geometry {
public:
  TrigFTF_GNN_Geometry(const std::vector<TrigInDetSiLayer>&, const std::unique_ptr<GNN_FasTrackConnector>&);
  ~TrigFTF_GNN_Geometry();
  
  const TrigFTF_GNN_Layer* getTrigFTF_GNN_LayerByKey(unsigned int) const;
  const TrigFTF_GNN_Layer* getTrigFTF_GNN_LayerByIndex(int) const;
  inline unsigned int getTrigFTF_GNN_LayerKeyByIndex(int idx) const {
    return m_layerKeys[idx];
  }
  int num_bins() const {return m_nEtaBins;}
  unsigned int num_layers() const {return m_layArray.size();}
  const std::vector<std::pair<int, std::vector<int> > >& bin_groups() const {return m_binGroups;}
  
protected:
 
  const TrigFTF_GNN_Layer* addNewLayer(const TrigInDetSiLayer&, int);
 
  float m_etaBinWidth;
 
  std::map<unsigned int, TrigFTF_GNN_Layer*> m_layMap;
  std::vector<TrigFTF_GNN_Layer*> m_layArray;
  std::vector<unsigned int> m_layerKeys;
  int m_nEtaBins;
 
  std::vector<std::pair<int, std::vector<int> > > m_binGroups;
  
};
 
 
#endif