CalorimeterClusterShape.h

Go to the documentation of this file.

// SPDX-License-Identifier: LGPL-3.0-or-later
// Copyright (C) 2025 Chao Peng, Dhevan Gangadharan, Sebouh Paul, Derek Anderson
 
#pragma once
 
#include <algorithms/algorithm.h>
#include <edm4eic/ClusterCollection.h>
#include <edm4eic/EDM4eicVersion.h>
#include <edm4eic/MCRecoClusterParticleAssociationCollection.h>
#include <algorithm>
#include <cmath>
#include <functional>
#include <map>
#include <optional>
#include <string>
#include <string_view>
#include <utility>
 
#include "CalorimeterClusterShapeConfig.h"
#include "algorithms/interfaces/WithPodConfig.h"
 
#if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
#include <edm4eic/MCRecoClusterParticleLinkCollection.h>
#endif
 
namespace eicrecon {
 
// --------------------------------------------------------------------------
// --------------------------------------------------------------------------
using CalorimeterClusterShapeAlgorithm = algorithms::Algorithm<
    algorithms::Input<edm4eic::ClusterCollection,
                      std::optional<edm4eic::MCRecoClusterParticleAssociationCollection>>,
    algorithms::Output<edm4eic::ClusterCollection,
#if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
                       std::optional<edm4eic::MCRecoClusterParticleLinkCollection>,
#endif
                       std::optional<edm4eic::MCRecoClusterParticleAssociationCollection>>>;
 
// --------------------------------------------------------------------------
// --------------------------------------------------------------------------
class CalorimeterClusterShape : public CalorimeterClusterShapeAlgorithm,
                                public WithPodConfig<CalorimeterClusterShapeConfig> {
 
public:
  // ctor
  CalorimeterClusterShape(std::string_view name)
      : CalorimeterClusterShapeAlgorithm{name,
                                         {"inputClusters", "inputMCClusterAssociations"},
                                         {"outputClusters",
#if EDM4EIC_BUILD_VERSION >= EDM4EIC_VERSION(8, 7, 0)
                                          "outputMCClusterLinks",
#endif
                                          "outputMCClusterAssociations"},
                                         "Computes cluster shape parameters"} {
  }
 
  // public methods
  void init() final;
  void process(const Input&, const Output&) const final;
 
private:
  static double constWeight(double /*E*/, double /*tE*/, double /*p*/, int /*type*/) { return 1.0; }
 
  static double linearWeight(double E, double /*tE*/, double /*p*/, int /*type*/) { return E; }
 
  static double logWeight(double E, double tE, double base, int /*type*/) {
    return std::max(0., base + std::log(E / tE));
  }
 
  const std::map<std::string, std::function<double(double, double, double, int)>> m_weightMethods =
      {{"none", constWeight}, {"linear", linearWeight}, {"log", logWeight}};
 
  std::function<double(double, double, double, int)> m_weightFunc;
};
 
} // namespace eicrecon