Back to home page

EIC code displayed by LXR

 
 

    


File indexing completed on 2024-06-26 07:05:40

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2022 Whitney Armstrong, Wouter Deconinck, Sylvester Joosten, Dmitry Romanov
0003 
0004 #pragma once
0005 
0006 #include <TRandomGen.h>
0007 #include <algorithms/algorithm.h>
0008 #include <edm4eic/MCRecoTrackerHitAssociationCollection.h>
0009 #include <edm4eic/RawTrackerHitCollection.h>
0010 #include <edm4hep/SimTrackerHitCollection.h>
0011 #include <functional>
0012 #include <string>
0013 #include <string_view>
0014 
0015 #include "SiliconTrackerDigiConfig.h"
0016 #include "algorithms/interfaces/WithPodConfig.h"
0017 
0018 namespace eicrecon {
0019 
0020 using SiliconTrackerDigiAlgorithm =
0021     algorithms::Algorithm<algorithms::Input<edm4hep::SimTrackerHitCollection>,
0022                           algorithms::Output<edm4eic::RawTrackerHitCollection,
0023                                              edm4eic::MCRecoTrackerHitAssociationCollection>>;
0024 
0025 class SiliconTrackerDigi : public SiliconTrackerDigiAlgorithm,
0026                            public WithPodConfig<SiliconTrackerDigiConfig> {
0027 
0028 public:
0029   SiliconTrackerDigi(std::string_view name)
0030       : SiliconTrackerDigiAlgorithm{name,
0031                                     {"inputHitCollection"},
0032                                     {"outputRawHitCollection", "outputHitAssociations"},
0033                                     "Apply threshold, digitize within ADC range, "
0034                                     "convert time with smearing resolution."} {}
0035 
0036   void init() final;
0037   void process(const Input&, const Output&) const final;
0038 
0039 private:
0040   /** Random number generation*/
0041   TRandomMixMax m_random;
0042   std::function<double()> m_gauss;
0043 
0044   // FIXME replace with standard random engine
0045   // std::default_random_engine generator; // TODO: need something more appropriate here
0046   // std::normal_distribution<double> m_normDist; // defaults to mean=0, sigma=1
0047 
0048   // algorithms::Generator m_rng = algorithms::RandomSvc::instance().generator();
0049 };
0050 
0051 } // namespace eicrecon