LambdaReconstruction.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/ReconstructedParticleCollection.h>
#include <spdlog/logger.h>
#include <gsl/pointers>
#include <memory>
#include <string>      // for basic_string
#include <string_view> // for string_view
 
#include "algorithms/interfaces/WithPodConfig.h"
#include "services/particle/ParticleSvc.h"
#include "algorithms/reco/LambdaReconstructionConfig.h"
 
namespace eicrecon {
 
using LambdaReconstructionAlgorithm = algorithms::Algorithm<
    algorithms::Input<const edm4eic::ReconstructedParticleCollection,
                      const edm4eic::ReconstructedParticleCollection,
                      const edm4eic::ReconstructedParticleCollection,
                      const edm4eic::ReconstructedParticleCollection>,
    /*output collections contain the lambda candidates and their decay products in the CM frame*/
    algorithms::Output<edm4eic::ReconstructedParticleCollection,
                       edm4eic::ReconstructedParticleCollection>>;
class LambdaReconstruction : public LambdaReconstructionAlgorithm,
                             public WithPodConfig<LambdaReconstructionConfig> {
public:
  LambdaReconstruction(std::string_view name)
      : LambdaReconstructionAlgorithm{
            name,
 
            {"inputNeutralsHcal", "inputNeutralsB0", "inputNeutralsEcalEndCapP",
             "inputNeutralsLFHCAL"},
 
            {"outputLambdas", "outputLambdaDecayProductsCM"},
 
            "Reconstructs lambda candidates and their decay products (in the CM frame) from the "
            "reconstructed neutrons and photons"} {}
 
  void init() final;
  void process(const Input&, const Output&) const final;
 
private:
  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_zMax{0};
 
  bool reconstruct_from_triplet(const edm4eic::ReconstructedParticle& n_in,
                                const edm4eic::ReconstructedParticle& g1_in,
                                const edm4eic::ReconstructedParticle& g2_in,
                                edm4eic::ReconstructedParticleCollection* out_lambdas,
                                edm4eic::ReconstructedParticleCollection* out_decay_products) const;
};
} // namespace eicrecon