Alpaka Multi-depth particle flow clusterizer

RecoParticleFlow/PFClusterProducer/interface/PFMultiDepthClusteringEdgeVarsSoA.h

@@ -0,0 +1,18 @@
+#ifndef RecoParticleFlow_PFRecHitProducer_interface_PFMultiDepthClusteringEdgeVarsSoA_h
+#define RecoParticleFlow_PFRecHitProducer_interface_PFMultiDepthClusteringEdgeVarsSoA_h
+
+#include "DataFormats/SoATemplate/interface/SoACommon.h"
+#include "DataFormats/SoATemplate/interface/SoALayout.h"
+#include "DataFormats/SoATemplate/interface/SoAView.h"
+
+namespace reco {
+
+  GENERATE_SOA_LAYOUT(PFMultiDepthClusteringEdgeVarsSoALayout,
+                        SOA_COLUMN(int, mdpf_adjacencyIndex),  // needs nClusters+1 allocation
+                        SOA_COLUMN(int, mdpf_adjacencyList)    // needs 2*nClusters allocation
+                      )  
+
+  using PFMultiDepthClusteringEdgeVarsSoA = PFMultiDepthClusteringEdgeVarsSoALayout<>;
+}  // namespace reco
+
+#endif

RecoParticleFlow/PFClusterProducer/interface/PFMultiDepthClusteringVarsSoA.h

@@ -0,0 +1,24 @@
+#ifndef RecoParticleFlow_PFClusterProducer_interface_PFMultiDepthClusteringVarsSoA_h
+#define RecoParticleFlow_PFClusterProducer_interface_PFMultiDepthClusteringVarsSoA_h
+
+#include "DataFormats/SoATemplate/interface/SoACommon.h"
+#include "DataFormats/SoATemplate/interface/SoALayout.h"
+#include "DataFormats/SoATemplate/interface/SoAView.h"
+
+namespace reco {
+
+  GENERATE_SOA_LAYOUT(PFMultiDepthClusteringVarsSoALayout,
+                      SOA_COLUMN(float, depth),
+                      SOA_COLUMN(double, etaRMS2),
+                      SOA_COLUMN(double, phiRMS2),
+                      SOA_COLUMN(float, energy),
+                      SOA_COLUMN(float, eta),
+                      SOA_COLUMN(float, phi),
+                      SOA_COLUMN(int, mdpf_topoId),
+                      SOA_SCALAR(int, size)
+                    )
+
+  using PFMultiDepthClusteringVarsSoA = PFMultiDepthClusteringVarsSoALayout<>;
+}  // namespace reco
+
+#endif

RecoParticleFlow/PFClusterProducer/interface/alpaka/PFMultiDepthClusteringEdgeVarsDeviceCollection.h

@@ -0,0 +1,16 @@
+#ifndef RecoParticleFlow_PFRecHitProducer_interface_alpaka_PFMultiDepthClusteringEdgeVarsDeviceCollection_h
+#define RecoParticleFlow_PFRecHitProducer_interface_alpaka_PFMultiDepthClusteringEdgeVarsDeviceCollection_h
+
+#include "DataFormats/Portable/interface/alpaka/PortableCollection.h"
+#include "HeterogeneousCore/AlpakaInterface/interface/config.h"
+
+#include "RecoParticleFlow/PFClusterProducer/interface/PFMultiDepthClusteringEdgeVarsSoA.h"
+
+namespace ALPAKA_ACCELERATOR_NAMESPACE::reco {
+
+  using PFMultiDepthClusteringEdgeVarsDeviceCollection = PortableCollection<::reco::PFMultiDepthClusteringEdgeVarsSoA>;
+  // needs 2*nClusters allocation
+
+}  // namespace ALPAKA_ACCELERATOR_NAMESPACE::reco
+
+#endif

RecoParticleFlow/PFClusterProducer/interface/alpaka/PFMultiDepthClusteringVarsDeviceCollection.h

@@ -0,0 +1,14 @@
+#ifndef RecoParticleFlow_PFRecHitProducer_interface_alpaka_PFMultiDepthClusteringVarsDeviceCollection_h
+#define RecoParticleFlow_PFRecHitProducer_interface_alpaka_PFMultiDepthClusteringVarsDeviceCollection_h
+
+#include "DataFormats/Portable/interface/alpaka/PortableCollection.h"
+#include "HeterogeneousCore/AlpakaInterface/interface/config.h"
+
+#include "RecoParticleFlow/PFClusterProducer/interface/PFMultiDepthClusteringVarsSoA.h"
+
+namespace ALPAKA_ACCELERATOR_NAMESPACE::reco {
+
+  using PFMultiDepthClusteringVarsDeviceCollection = PortableCollection<::reco::PFMultiDepthClusteringVarsSoA>;
+}  // namespace ALPAKA_ACCELERATOR_NAMESPACE::reco
+
+#endif