cellsPrefixSum. More...

#include <EventInfoDefinitions.h>

Inheritance diagram for CaloRecGPU::ClusterInfoArr:

Collaboration diagram for CaloRecGPU::ClusterInfoArr:

Classes
struct	ClusterMomentsArr

Public Member Functions
template<class F, class ... Args>
constexpr void	for_all_moments (F &&f, Args &&... args) const
	F receives the array associated with the moment and any additional arguments.
template<class F, class ... Args>
constexpr void	for_all_moments (F &&f, Args &&... args)
	F receives the array associated with the moment and any additional arguments.
constexpr const tag_type &	get_extra_cell_info (const int idx) const
	Extra representation for the cell tags, to save time for calculating the isolation moment after building the list of cells per cluster.
constexpr tag_type &	get_extra_cell_info (const int idx)
template<class NewT = tag_type>
constexpr const NewT *	secondary_tag_array () const
template<class NewT = tag_type>
constexpr NewT *	secondary_tag_array ()
template<class NewT = tag_type>
constexpr const NewT &	secondary_tag_array (const int idx) const
template<class NewT = tag_type>
constexpr NewT &	secondary_tag_array (const int idx)
template<class NewT = tag_type>
constexpr const NewT *	tertiary_tag_array () const
template<class NewT = tag_type>
constexpr NewT *	tertiary_tag_array ()
template<class NewT = tag_type>
constexpr const NewT &	tertiary_tag_array (const int idx) const
template<class NewT = tag_type>
constexpr NewT &	tertiary_tag_array (const int idx)
constexpr bool	has_cells_per_cluster () const
constexpr bool	has_basic_info (const ClusterInformationState this_state) const
constexpr bool	has_basic_info () const
constexpr bool	has_moments () const

Static Public Member Functions
static constexpr bool	has_cells_per_cluster (const ClusterInformationState this_state)
static constexpr bool	has_moments (const ClusterInformationState this_state)

Public Attributes
float	clusterEnergy [NMaxClusters]
float	clusterEt [NMaxClusters]
float	clusterEta [NMaxClusters]
float	clusterPhi [NMaxClusters]
int	seedCellIndex [NMaxClusters]
	Invalid(ated) clusters have seedCellIndex < 0.
int	cellsPrefixSum [NMaxClusters+1]
union {
int indices [2 *NCaloCells]
tag_type tags [NCaloCells]
}	cells
float	cellWeights [2 *NCaloCells]
int	clusterIndices [2 *NCaloCells]
struct CaloRecGPU::ClusterInfoArr::ClusterMomentsArr	moments
ClusterInformationState	state
bool	has_deleted_clusters
int	number
int	number_cells

Detailed Description

cellsPrefixSum.

Contains the per-event information of the clusters, including both the kinematic variables and the moments.

In the latter case, we store cell index, cell weight and cluster index so that it is easy to iterate through all cells inside clusters.

Also includes an easy look-up for the index of the seed cell, with clusters with seed cells less than zero being invalid; this is useful for e. g. applying cuts on the GPU without requiring re-sorting everything. Cluster assignment depends on the ClusterInformationState as reported by the ClusterBaseInfo: we may have an array of tags (which assumes primary_weight = 1 - secondary_weight) or an actual jagged vector expressed by the

Definition at line 481 of file EventInfoDefinitions.h.

Member Function Documentation

◆ for_all_moments() [1/2]

template<class F, class ... Args>

void CaloRecGPU::ClusterInfoArr::for_all_moments	(	F &&	f,
		Args &&...	args )

inlineconstexpr

F receives the array associated with the moment and any additional arguments.

Given that the moments are of different types, it must be a functor or lambda ready to deal with that!

Definition at line 719 of file EventInfoDefinitions.h.

    {
      CALORECGPU_FORALLMOMENTS_INSTANTIATE(CALORECGPU_FORALLMOMENTS_HELPER)
    }

◆ for_all_moments() [2/2]

template<class F, class ... Args>

void CaloRecGPU::ClusterInfoArr::for_all_moments	(	F &&	f,
		Args &&...	args ) const

inlineconstexpr

F receives the array associated with the moment and any additional arguments.

Given that the moments are of different types, it must be a functor or lambda ready to deal with that!

Definition at line 708 of file EventInfoDefinitions.h.

    {
      // call f as an lvalue every time
      CALORECGPU_FORALLMOMENTS_INSTANTIATE(CALORECGPU_FORALLMOMENTS_HELPER)
    }

◆ get_extra_cell_info() [1/2]

tag_type & CaloRecGPU::ClusterInfoArr::get_extra_cell_info ( const int idx )

inlineconstexpr

Definition at line 754 of file EventInfoDefinitions.h.

    {
      constexpr auto max_size = NMaxClusters * sizeof(float) / sizeof(tag_type);
      
      const int outer_idx = idx / max_size;
      const int inner_idx = idx % max_size;
      
      using PtrType = tag_type *;
      
      auto get_laundered_pointer = [&](auto * ptr) -> PtrType
      {
        return std::launder(static_cast<PtrType>(static_cast<void *>(ptr)));
      };
 
      PtrType arr[6] = { get_laundered_pointer(this->moments.engCalibTot),
                         get_laundered_pointer(this->moments.engCalibOutL),
                         get_laundered_pointer(this->moments.engCalibOutM),
                         get_laundered_pointer(this->moments.engCalibOutT),
                         get_laundered_pointer(this->moments.engCalibDeadL),
                         get_laundered_pointer(this->moments.engCalibDeadM)  };
 
      return *(arr[outer_idx] + inner_idx);
    }

◆ get_extra_cell_info() [2/2]

const tag_type & CaloRecGPU::ClusterInfoArr::get_extra_cell_info ( const int idx ) const

inlineconstexpr

Extra representation for the cell tags, to save time for calculating the isolation moment after building the list of cells per cluster.

Can (and should) also be used if any other processing (before calculating the relevant moments...) needs some sort of cell -> cluster mapping.

Definition at line 730 of file EventInfoDefinitions.h.

    {
      constexpr auto max_size = NMaxClusters * sizeof(float) / sizeof(tag_type);
      
      const int outer_idx = idx / max_size;
      const int inner_idx = idx % max_size;
 
      using PtrType = const tag_type *;
      
      auto get_laundered_pointer = [&](auto * ptr) -> PtrType
      {
        return std::launder(static_cast<PtrType>(static_cast<const void *>(ptr)));
      };
 
      PtrType arr[6] = { get_laundered_pointer(this->moments.engCalibTot),
                         get_laundered_pointer(this->moments.engCalibOutL),
                         get_laundered_pointer(this->moments.engCalibOutM),
                         get_laundered_pointer(this->moments.engCalibOutT),
                         get_laundered_pointer(this->moments.engCalibDeadL),
                         get_laundered_pointer(this->moments.engCalibDeadM)  };
 
      return *(arr[outer_idx] + inner_idx);
    }

◆ has_basic_info() [1/2]

bool CaloRecGPU::ClusterBaseInfo::has_basic_info ( ) const

inlineconstexprinherited

Definition at line 442 of file EventInfoDefinitions.h.

    {
      return has_basic_info(state);
    }

◆ has_basic_info() [2/2]

bool CaloRecGPU::ClusterBaseInfo::has_basic_info ( const ClusterInformationState this_state ) const

inlineconstexprinherited

Definition at line 425 of file EventInfoDefinitions.h.

    {
      switch(this_state)
      {
        case ClusterInformationState::TagsWithBasicInfo:
          return true;
        case ClusterInformationState::WithBasicInfo:
          return true;
        case ClusterInformationState::WithMoments:
          return true;
        case ClusterInformationState::WithExtraMoments:
          return true;
        default:
          return false;
      }
    }

◆ has_cells_per_cluster() [1/2]

bool CaloRecGPU::ClusterBaseInfo::has_cells_per_cluster ( ) const

inlineconstexprinherited

Definition at line 420 of file EventInfoDefinitions.h.

    {
      return has_cells_per_cluster(state);
    }

◆ has_cells_per_cluster() [2/2]

constexpr bool CaloRecGPU::ClusterBaseInfo::has_cells_per_cluster ( const ClusterInformationState this_state )

inlinestaticconstexprinherited

Definition at line 403 of file EventInfoDefinitions.h.

    {
      switch(this_state)
      {
        case ClusterInformationState::Full:
          return true;
        case ClusterInformationState::WithBasicInfo:
          return true;
        case ClusterInformationState::WithMoments:
          return true;
        case ClusterInformationState::WithExtraMoments:
          return true;
        default:
          return false;
      }
    }

◆ has_moments() [1/2]

bool CaloRecGPU::ClusterBaseInfo::has_moments ( ) const

inlineconstexprinherited

Definition at line 460 of file EventInfoDefinitions.h.

    {
      return has_moments(state);
    }

◆ has_moments() [2/2]

constexpr bool CaloRecGPU::ClusterBaseInfo::has_moments ( const ClusterInformationState this_state )

inlinestaticconstexprinherited

Definition at line 447 of file EventInfoDefinitions.h.

    {
      switch(this_state)
      {
        case ClusterInformationState::WithMoments:
          return true;
        case ClusterInformationState::WithExtraMoments:
          return true;
        default:
          return false;
      }
    }