Define eltype() for PseudoJet and EEjet

Returns the parametrised element type

Author: graeme-a-stewart

src/EEAlgorithm.jl

@@ -275,7 +275,7 @@ function _ee_genkt_algorithm(; particles::Vector{EEjet{T}}, p = 1, R = 4.0,
     R2 = R^2
 
     # Numerical type?
-    ParticleType = typeof(particles[1].E)
+    ParticleType = eltype(particles[1])
 
     # Constant factor for the dij metric and the beam distance function
     if algorithm == JetAlgorithm.Durham

src/EEjet.jl

@@ -27,6 +27,14 @@ mutable struct EEjet{T <: Real} <: FourMomentum
     _cluster_hist_index::Int
 end
 
+
+"""
+    Base.eltype(::Type{EEjet{T}}) where T
+
+Return the element type of the `EEjet` struct.
+"""
+Base.eltype(::Type{EEjet{T}}) where T = T
+
 function EEjet(px::T, py::T, pz::T, E::T, _cluster_hist_index::Integer) where {T <: Real}
     @muladd p2 = px * px + py * py + pz * pz
     inv_p = @fastmath 1.0 / sqrt(p2)

src/PlainAlgo.jl

@@ -287,8 +287,8 @@ function _plain_jet_reconstruct(; particles::Vector{PseudoJet{T}}, p = -1, R = 1
     # Parameters
     R2 = R^2
 
-    # Numerical type?
-    ParticleType = typeof(particles[1].E)
+    # Numerical type for this reconstruction
+    ParticleType = eltype(particles[1])
 
     # Optimised compact arrays for determining the next merge step
     # We make sure these arrays are type stable - have seen issues where, depending on the values

src/Pseudojet.jl

@@ -49,6 +49,13 @@ mutable struct PseudoJet{T <: Real} <: FourMomentum
     _phi::T
 end
 
+"""
+    Base.eltype(::Type{PseudoJet{T}}) where T
+
+Return the element type of the `PseudoJet` struct.
+"""
+Base.eltype(::Type{PseudoJet{T}}) where T = T
+
 """
     PseudoJet(px::T, py::T, pz::T, E::T,
         _cluster_hist_index::Int,

src/TiledAlgoLL.jl

@@ -440,8 +440,8 @@ function _tiled_jet_reconstruct(particles::Vector{PseudoJet{T}}; p::Real = -1, R
     R2::Float64 = R * R
     p = (round(p) == p) ? Int(p) : p # integer p if possible
 
-    # Numerical type?
-    ParticleType = typeof(particles[1].E)
+    # Numerical type for this reconstruction
+    ParticleType = eltype(particles[1])
 
     # This will be used quite deep inside loops, so declare it here so that
     # memory (de)allocation gets done only once