algorithms/digi/SiliconPulseDiscretization.h

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2025 Chun Yuen Tsang
0003 //
0004 // Creates a discrete pulse from a continuous pulse
0005 //
0006
0007 #pragma once
0008
0009 #include <TGraph.h>
0010 #include <algorithms/algorithm.h>
0011 #include <edm4hep/RawTimeSeriesCollection.h>
0012 #include <edm4eic/EDM4eicVersion.h>
0013 #if EDM4EIC_VERSION_MAJOR > 8 || (EDM4EIC_VERSION_MAJOR == 8 && EDM4EIC_VERSION_MINOR >= 1)
0014 #include <edm4eic/SimPulseCollection.h>
0015 #else
0016 #include <edm4hep/TimeSeriesCollection.h>
0017 #endif
0018 #include <string>
0019 #include <string_view>
0020
0021 #include "algorithms/digi/SiliconPulseDiscretizationConfig.h"
0022 #include "algorithms/interfaces/WithPodConfig.h"
0023
0024 namespace eicrecon {
0025
0026 #if EDM4EIC_VERSION_MAJOR > 8 || (EDM4EIC_VERSION_MAJOR == 8 && EDM4EIC_VERSION_MINOR >= 1)
0027 using PulseType = edm4eic::SimPulse;
0028 #else
0029 using PulseType = edm4hep::TimeSeries;
0030 #endif
0031
0032 using SiliconPulseDiscretizationAlgorithm =
0033     algorithms::Algorithm<algorithms::Input<PulseType::collection_type>,
0034                           algorithms::Output<edm4hep::RawTimeSeriesCollection>>;
0035
0036 class SiliconPulseDiscretization : public SiliconPulseDiscretizationAlgorithm,
0037                                    public WithPodConfig<SiliconPulseDiscretizationConfig> {
0038
0039 public:
0040   SiliconPulseDiscretization(std::string_view name)
0041       : SiliconPulseDiscretizationAlgorithm{name, {"OutputPulses"}, {"DiscretePulses"}, {}} {}
0042   virtual void init() final;
0043   void process(const Input&, const Output&) const;
0044
0045 private:
0046   double _interpolateOrZero(const TGraph& graph, double t, double tMin, double tMax) const;
0047 };
0048
0049 } // namespace eicrecon