FarForwardNeutralsReconstruction.h

Go to the documentation of this file.

// SPDX-License-Identifier: LGPL-3.0-or-later
// Copyright (C) 2026 Sebouh Paul, Baptiste Fraisse
 
#pragma once
#include <DD4hep/Detector.h>
#include <algorithms/algorithm.h>
#include <algorithms/geo.h>
#include <edm4eic/ClusterCollection.h>
#include <edm4eic/ReconstructedParticleCollection.h>
#include <spdlog/logger.h>
#include <gsl/pointers>
#include <memory>
#include <string>      // for basic_string
#include <string_view> // for string_view
#include <vector>
#include <functional>
#include "algorithms/interfaces/WithPodConfig.h"
#include "services/particle/ParticleSvc.h"
#include "algorithms/reco/FarForwardNeutralsReconstructionConfig.h"
 
namespace eicrecon {
 
using FarForwardNeutralsReconstructionAlgorithm =
 
    algorithms::Algorithm<
        algorithms::Input<const edm4eic::ClusterCollection,  // clusters ZDC-Hcal
                          const edm4eic::ClusterCollection,  // clusters B0-Ecal
                          const edm4eic::ClusterCollection,  // clusters EndcapP-Ecal
                          const edm4eic::ClusterCollection>, // clusters LFHCAL
 
        algorithms::Output<
            edm4eic::ReconstructedParticleCollection,   // neutrons/gamma in ZDC-Hcal
            edm4eic::ReconstructedParticleCollection,   // neutrons/gamma in B0-Ecal
            edm4eic::ReconstructedParticleCollection,   // neutrons/gamma in EndcapP-Ecal
            edm4eic::ReconstructedParticleCollection>>; // neutrons/gamma in LFHCAL
class FarForwardNeutralsReconstruction
    : public FarForwardNeutralsReconstructionAlgorithm,
      public WithPodConfig<FarForwardNeutralsReconstructionConfig> {
public:
  FarForwardNeutralsReconstruction(std::string_view name)
      : FarForwardNeutralsReconstructionAlgorithm{
            name,
 
            {"clustersHcal", "clustersB0", "clustersEcalEndCapP", "clustersLFHCAL"},
 
            {"outputNeutralsHcal", "outputNeutralsB0", "outputNeutralsEcalEndCapP",
             "outputNeutralsLFHCAL"},
 
            "Convert EMCal and HCal clusters into neutron or photon candidates"} {}
 
  void init() final;
  void process(const Input&, const Output&) const final;
 
private:
  bool isGamma(const edm4eic::Cluster& cluster) const;
  std::shared_ptr<spdlog::logger> m_log;
  const algorithms::ParticleSvc& m_particleSvc = algorithms::ParticleSvc::instance();
  const dd4hep::Detector* m_detector{algorithms::GeoSvc::instance().detector()};
  double m_gammaZMax{0};
 
  enum class GammaMode { None, LeaderOnly, AllPassing };
  enum class NeutronMode { None, SumAll, LeaderOnly };
 
  using CorrFunc = std::function<double(double, const std::vector<double>&)>;
 
  static double corrPower(double E, const std::vector<double>& coeffs);
 
  int processNeutralCalo(const edm4eic::ClusterCollection* clusters,
                         edm4eic::ReconstructedParticleCollection* out_neutrals,
                         const std::vector<double>& gammaScaleCoeff,
                         const std::vector<double>& neutronScaleCoeff, bool canDetectGammas,
                         bool canDetectNeutrons, const CorrFunc& gammaCorr,
                         const CorrFunc& neutronCorr, GammaMode gammaMode,
                         double gammaLeaderFracMin, double clusterEmin, NeutronMode neutronMode,
                         bool associateAllClustersToNeutron) const;
};
 
} // namespace eicrecon