#include <WFSClusterMaker.h>

Inheritance diagram for Gep::WFSClusterMaker:

Collaboration diagram for Gep::WFSClusterMaker:

Public Member Functions
	WFSClusterMaker ()
virtual	~WFSClusterMaker ()=default
std::vector< Gep::Cluster >	makeClusters (const pGepCellMap &) const override
std::string	getName () const override

Private Member Functions
bool	isSeedCell (const Gep::GepCaloCell &cell) const
bool	isInAllowedSampling (int sampling, const std::vector< int > &list_of_samplings) const
bool	isNewCell (unsigned int id, const std::vector< unsigned int > &seenCells) const
std::vector< Gep::GepCaloCell >	clusterFromCells (const Gep::GepCaloCell &seed, const pGepCellMap &) const
Gep::Cluster	getClusterFromListOfCells (const std::vector< Gep::GepCaloCell > &cells) const
double	calculateClusterPhi (double seed_phi, double delta_phi) const
void	orderClustersInEt (std::vector< Gep::Cluster > &v_clusters) const
double	getDeltaPhi (double phi, double seed_phi) const

Private Attributes
float	m_seed_threshold = 4.0
float	m_clustering_threshold = 2.0
int	m_max_shells = 8
std::vector< int >	m_allowed_seed_samplings = {0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20}
std::vector< int >	m_allowed_clustering_samplings = {0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20}

Detailed Description

Definition at line 14 of file WFSClusterMaker.h.

Constructor & Destructor Documentation

◆ WFSClusterMaker()

Gep::WFSClusterMaker::WFSClusterMaker ( )

inline

Definition at line 18 of file WFSClusterMaker.h.

18{}

◆ ~WFSClusterMaker()

virtual Gep::WFSClusterMaker::~WFSClusterMaker ( )

virtualdefault

Member Function Documentation

◆ calculateClusterPhi()

double Gep::WFSClusterMaker::calculateClusterPhi	(	double	seed_phi,
		double	delta_phi ) const

private

Definition at line 163 of file WFSClusterMaker.cxx.

                                                                                      {
  double phi = seed_phi + delta_phi;
  if (phi > TMath::Pi()) phi = -1.0*TMath::Pi() + (phi - TMath::Pi());
  if (phi < (-1.0)*TMath::Pi()) phi = TMath::Pi() + (phi + TMath::Pi());
  return phi;
}

◆ clusterFromCells()

std::vector< Gep::GepCaloCell > Gep::WFSClusterMaker::clusterFromCells	(	const Gep::GepCaloCell &	seed,
		const pGepCellMap &	caloCellsMap ) const

private

Definition at line 62 of file WFSClusterMaker.cxx.

                                                             {
 
  std::vector<Gep::GepCaloCell> v_clusterCells;
 
  std::vector<Gep::GepCaloCell> cellsNextLayer, cellsThisLayer;
  std::vector<unsigned int> seenCells;
 
  // Fill seed into supporting vectors
  v_clusterCells.push_back(seed);
  cellsNextLayer.push_back(seed);
  seenCells.push_back(seed.id);
 
  int i_shell = 1;
 
  while (!cellsNextLayer.empty() && i_shell <= m_max_shells) {
 
        cellsThisLayer = cellsNextLayer;
        cellsNextLayer.clear();
        ++i_shell;
 
        // Loop over all cells in this shell
        for (unsigned int i_cell = 0; i_cell < cellsThisLayer.size(); ++i_cell) {
 
                // Go through list of neighbouring cells and check whether they are part of the cluster
                for (unsigned int i_neighbour = 0; i_neighbour < (cellsThisLayer[i_cell]).neighbours.size(); ++i_neighbour) {
 
            // Check whether this neighbouring cell was sent to the GEP
            auto const& nghbr_itr = caloCellsMap->find((cellsThisLayer[i_cell]).neighbours[i_neighbour]);
            if (nghbr_itr == caloCellsMap->end()) continue;
 
            Gep::GepCaloCell neighbour = nghbr_itr->second;//caloCellsMap->at((cellsThisLayer[i_cell]).neighbours[i_neighbour]);
 
                        // reject if bad cell
                        if (neighbour.isBadCell()) continue;
 
                        // Reject if cell is not above clustering threshold
                        if (std::fabs(neighbour.sigma) < m_clustering_threshold) continue;
 
                        // Reject if cell was already considered
                        if (!isNewCell(neighbour.id, seenCells)) continue;
 
                        // Ignore cells in disallowed samplings
                        if (!isInAllowedSampling(neighbour.sampling, m_allowed_clustering_samplings)) continue;
 
                        seenCells.push_back(neighbour.id);
                        cellsNextLayer.push_back(neighbour);
                        v_clusterCells.push_back(std::move(neighbour));
                }
        }
        cellsThisLayer.clear();
  }
 
  return v_clusterCells;
}

◆ getClusterFromListOfCells()

Gep::Cluster Gep::WFSClusterMaker::getClusterFromListOfCells ( const std::vector< Gep::GepCaloCell > & cells ) const

private

Definition at line 119 of file WFSClusterMaker.cxx.

                                                                                                     {
 
  Gep::Cluster cluster;
 
  std::vector<unsigned int> v_cellIDs;
  double cluster_e = 0.0;
  double etaSum = 0.0;
  double phiSum = 0.0;
  double abs_e = 0.0;
  float weight = 0.0;
 
  double seed_phi = cells[0].phi;
  for (unsigned int i_cell = 0; i_cell < cells.size(); ++i_cell) {
    float cell_e = cells[i_cell].et * TMath::CosH(cells[i_cell].eta);
        cluster_e += cell_e;
        abs_e += std::fabs(cell_e);
        v_cellIDs.push_back(cells[i_cell].id);
        etaSum += std::fabs(cell_e) * cells[i_cell].eta;
        phiSum += std::fabs(cell_e) * getDeltaPhi(cells[i_cell].phi, seed_phi);
        if (std::fabs(cells[i_cell].sigma) > m_seed_threshold) weight += 1.0;
  }
 
  cluster.ncells = cells.size();
  cluster.time = cells[0].time; // Take time of seed cell
  cluster.cell_id = std::move(v_cellIDs);
  if (abs_e == 0.) [[unlikely]] throw std::runtime_error("Gep::WFSClusterMaker::getClusterFromListOfCells: abs_e is zero");
  double cluster_eta = etaSum / abs_e;
  double cluster_phi = calculateClusterPhi(seed_phi, phiSum / abs_e);
  if (weight == 0.) [[unlikely]] throw std::runtime_error("Gep::WFSClusterMaker::getClusterFromListOfCells: weight is zero");
  double cluster_et = (cluster_e * (1.0 / std::cosh(cluster_eta))) / weight;
  cluster.setEtEtaPhi(cluster_et, cluster_eta, cluster_phi);
 
  return cluster;
}

◆ getDeltaPhi()

double Gep::WFSClusterMaker::getDeltaPhi	(	double	phi,
		double	seed_phi ) const

private

Definition at line 155 of file WFSClusterMaker.cxx.

                                                                        {
  double delta_phi = std::fabs(std::fabs( std::fabs( phi - seed_phi ) - TMath::Pi() ) - TMath::Pi());
  if (phi < seed_phi) delta_phi *= -1.00;
  // Taking care of the -pi/pi split
  if ((std::fabs(phi + seed_phi) < TMath::Pi()) && (std::fabs(phi) + std::fabs(seed_phi) > 5.0)) delta_phi *= -1.00;
  return delta_phi;
}

◆ getName()

std::string Gep::WFSClusterMaker::getName ( ) const

overridevirtual

Implements Gep::IClusterMaker.

Definition at line 202 of file WFSClusterMaker.cxx.

                                            {
  return "CaloWFS";
}

◆ isInAllowedSampling()

bool Gep::WFSClusterMaker::isInAllowedSampling	(	int	sampling,
		const std::vector< int > &	list_of_samplings ) const

private

Definition at line 42 of file WFSClusterMaker.cxx.

                                                                                                        {
 
  for (unsigned int i = 0; i < list_of_samplings.size(); ++i) {
    if (list_of_samplings[i] == sampling) return true;
  }
  return false;
}

◆ isNewCell()

bool Gep::WFSClusterMaker::isNewCell	(	unsigned int	id,
		const std::vector< unsigned int > &	seenCells ) const

private

Definition at line 51 of file WFSClusterMaker.cxx.

                                                                                                  {
 
  for (unsigned int i = 0; i < seenCells.size(); ++i) {
        if (id == seenCells[i]) return false;
  }
 
  return true;
}

◆ isSeedCell()

bool Gep::WFSClusterMaker::isSeedCell ( const Gep::GepCaloCell & cell ) const

private

Definition at line 32 of file WFSClusterMaker.cxx.

                                                                     {
 
  if (cell.isBadCell()) return false;
  if (std::fabs(cell.sigma) < m_seed_threshold) return false;
  if (!isInAllowedSampling(cell.sampling, m_allowed_seed_samplings)) return false;
 
  return true;
}

◆ makeClusters()

std::vector< Gep::Cluster > Gep::WFSClusterMaker::makeClusters ( const pGepCellMap & caloCellsMap ) const

overridevirtual

Implements Gep::IClusterMaker.

Definition at line 10 of file WFSClusterMaker.cxx.

                                                                      {
 
  std::vector<Gep::Cluster> clusters;
 
  // Loop over all cells
  for (auto const& cell_itr : *caloCellsMap) {
 
    // Select only seed cells
    if (!isSeedCell(cell_itr.second)) continue;
 
    // Clustering
        std::vector<Gep::GepCaloCell> cluster_cells = clusterFromCells(cell_itr.second, caloCellsMap);
        clusters.push_back(getClusterFromListOfCells(cluster_cells));
  }
 
  // Order topo clusters according to their et
  orderClustersInEt(clusters);
 
  return clusters;
}

◆ orderClustersInEt()

void Gep::WFSClusterMaker::orderClustersInEt ( std::vector< Gep::Cluster > & v_clusters ) const

private

Definition at line 171 of file WFSClusterMaker.cxx.

                                                                                  {
 
  std::vector<Gep::Cluster> v_ordered;
  for (unsigned int i_cluster = 0; i_cluster < v_clusters.size(); ++i_cluster) {
        float et = v_clusters[i_cluster].et();
 
        // Fill first cluster
        if (v_ordered.empty()) {
                v_ordered.push_back(v_clusters[i_cluster]);
                continue;
        }
 
        // Find correct position for filling
        for (unsigned int i = 0; i < v_ordered.size(); ++i) {
                if (v_ordered[i].et() < et) {
                        v_ordered.insert(v_ordered.begin()+i, v_clusters[i_cluster]);
                        break;
                }
        }
 
        // If cluster has smallest et so far, it wasn't filled at all so we need to take care of it
        if (v_ordered.size() != i_cluster+1) v_ordered.push_back(v_clusters[i_cluster]);
  }
 
  v_clusters = std::move(v_ordered);
 
  return;
 
}

Member Data Documentation

◆ m_allowed_clustering_samplings

std::vector<int> Gep::WFSClusterMaker::m_allowed_clustering_samplings = {0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20}

private

Definition at line 32 of file WFSClusterMaker.h.

32{0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20};

◆ m_allowed_seed_samplings

std::vector<int> Gep::WFSClusterMaker::m_allowed_seed_samplings = {0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20}

private

Definition at line 31 of file WFSClusterMaker.h.

31{0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20};

◆ m_clustering_threshold

float Gep::WFSClusterMaker::m_clustering_threshold = 2.0

private

Definition at line 29 of file WFSClusterMaker.h.

◆ m_max_shells

int Gep::WFSClusterMaker::m_max_shells = 8

private

Definition at line 30 of file WFSClusterMaker.h.

◆ m_seed_threshold

float Gep::WFSClusterMaker::m_seed_threshold = 4.0

private

Definition at line 28 of file WFSClusterMaker.h.

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

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ WFSClusterMaker()

◆ ~WFSClusterMaker()

Member Function Documentation

◆ calculateClusterPhi()

◆ clusterFromCells()

◆ getClusterFromListOfCells()

◆ getDeltaPhi()

◆ getName()

◆ isInAllowedSampling()

◆ isNewCell()

◆ isSeedCell()

◆ makeClusters()

◆ orderClustersInEt()

Member Data Documentation

◆ m_allowed_clustering_samplings

◆ m_allowed_seed_samplings

◆ m_clustering_threshold

◆ m_max_shells

◆ m_seed_threshold